From fcc8eef784142c9a66c4dcf2614ef6b4064bea52 Mon Sep 17 00:00:00 2001 From: harold Date: Wed, 7 May 2025 19:42:59 +0500 Subject: [PATCH] add fix --- gemini_worker/src/gemini_sdk/service.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) 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)