diff --git a/gemini_worker/src/gemini_sdk/service.py b/gemini_worker/src/gemini_sdk/service.py index 0048fd1..dd966ff 100644 --- a/gemini_worker/src/gemini_sdk/service.py +++ b/gemini_worker/src/gemini_sdk/service.py @@ -6,7 +6,7 @@ import google.generativeai as genai from pydantic import ValidationError from src.gemini_sdk.promt import create_base_message -from src.gemini_sdk.schemas import ResponseFromGeminiSchema +from src.gemini_sdk.schemas import ResponseFromGeminiSchema, FullRequestForGeminiSchema from src.schemas import MessagesForSendToWorkersSchema from src.settings.base import settings @@ -24,7 +24,7 @@ class GoogleHelper: @staticmethod def _serialize_messages_to_prompt( - chats: MessagesForSendToWorkersSchema, + chats: FullRequestForGeminiSchema, ) -> List[dict]: messages_for_request = create_base_message() @@ -53,11 +53,10 @@ class GoogleHelper: def create_request_ai( self, - messages: MessagesForSendToWorkersSchema, + messages: FullRequestForGeminiSchema, ) -> ResponseFromGeminiSchema: - print(messages.slice_id, "SLICE ID BEFORE REQUEST") - for message in messages.messages: - print(message.id, message.user_id, message.chat_id) + for message in messages.chats: + print(message.slice_id, "SLICE ID BEFORE REQUEST") contents = self._serialize_messages_to_prompt(messages) response = self._model.generate_content(contents=contents) result = self._serialize_response_to_json(response.text)