This commit is contained in:
harold 2025-05-07 19:30:23 +05:00
parent 16855d1de9
commit dd283bdc23

View File

@ -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(