add fix
This commit is contained in:
parent
16855d1de9
commit
dd283bdc23
@ -55,9 +55,19 @@ class GoogleHelper:
|
||||
self,
|
||||
messages: MessagesForSendToWorkersSchema,
|
||||
) -> ResponseFromGeminiSchema:
|
||||
print(messages.slice_id, "SLICE ID BEFORE REQUEST")
|
||||
for message in messages.messages:
|
||||
print(message.id, message.user_id, message.chat_id)
|
||||
contents = self._serialize_messages_to_prompt(messages)
|
||||
response = self._model.generate_content(contents=contents)
|
||||
return self._serialize_response_to_json(response.text)
|
||||
result = self._serialize_response_to_json(response.text)
|
||||
|
||||
if result.success:
|
||||
for i in result.success:
|
||||
print(i.slice_id, i.user_id, "SUCCESS")
|
||||
print(i.reason)
|
||||
|
||||
return result
|
||||
|
||||
|
||||
gemini_helper = GoogleHelper(
|
||||
|
Loading…
x
Reference in New Issue
Block a user