diff --git a/gemini_worker/src/gemini_sdk/service.py b/gemini_worker/src/gemini_sdk/service.py index a125812..0048fd1 100644 --- a/gemini_worker/src/gemini_sdk/service.py +++ b/gemini_worker/src/gemini_sdk/service.py @@ -13,9 +13,9 @@ from src.settings.base import settings class GoogleHelper: def __init__( - self, - api_key: str, - model_name: str, + self, + api_key: str, + model_name: str, ) -> None: self.api_key = api_key self.model = model_name @@ -24,7 +24,7 @@ class GoogleHelper: @staticmethod def _serialize_messages_to_prompt( - chats: MessagesForSendToWorkersSchema, + chats: MessagesForSendToWorkersSchema, ) -> List[dict]: messages_for_request = create_base_message() @@ -41,7 +41,7 @@ class GoogleHelper: @staticmethod def _serialize_response_to_json( - response_text: str, + response_text: str, ) -> ResponseFromGeminiSchema: cleaned_response = response_text.strip().replace('```json\n', '').replace('\n```', '') try: @@ -52,15 +52,25 @@ class GoogleHelper: return ResponseFromGeminiSchema(success=None) def create_request_ai( - self, - messages: MessagesForSendToWorkersSchema, + 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( api_key=settings.GEMINI.API_KEY, model_name=settings.GEMINI.MODEL_NAME, -) \ No newline at end of file +)