advatroniks/Telegram-Scrapper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add fix

Browse Source
This commit is contained in:
harold 2025-05-07 19:42:59 +05:00
parent dd283bdc23
commit fcc8eef784
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
gemini_worker/src/gemini_sdk/service.py
View File

@ -6,7 +6,7 @@ import google.generativeai as genai
from pydantic import ValidationError from pydantic import ValidationError
from src.gemini_sdk.promt import create_base_message 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.schemas import MessagesForSendToWorkersSchema
from src.settings.base import settings from src.settings.base import settings
@ -24,7 +24,7 @@ class GoogleHelper:
@staticmethod @staticmethod
def _serialize_messages_to_prompt( def _serialize_messages_to_prompt(
chats: MessagesForSendToWorkersSchema, chats: FullRequestForGeminiSchema,
) -> List[dict]: ) -> List[dict]:
messages_for_request = create_base_message() messages_for_request = create_base_message()
@ -53,11 +53,10 @@ class GoogleHelper:
def create_request_ai( def create_request_ai(
self, self,
messages: MessagesForSendToWorkersSchema, messages: FullRequestForGeminiSchema,
) -> ResponseFromGeminiSchema: ) -> ResponseFromGeminiSchema:
print(messages.slice_id, "SLICE ID BEFORE REQUEST") for message in messages.chats:
for message in messages.messages: print(message.slice_id, "SLICE ID BEFORE REQUEST")
print(message.id, message.user_id, message.chat_id)
contents = self._serialize_messages_to_prompt(messages) contents = self._serialize_messages_to_prompt(messages)
response = self._model.generate_content(contents=contents) response = self._model.generate_content(contents=contents)
result = self._serialize_response_to_json(response.text) result = self._serialize_response_to_json(response.text)