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 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)