From 62956dc5f01f0229352e80fe65ca888b0b7cbdaf Mon Sep 17 00:00:00 2001 From: Bulat Kurbanov Date: Tue, 18 Feb 2025 21:58:15 +0100 Subject: [PATCH] Use gemini --- src/modules/stream_notifications/messages_proc.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/src/modules/stream_notifications/messages_proc.py b/src/modules/stream_notifications/messages_proc.py index a91da9d..06ddedc 100644 --- a/src/modules/stream_notifications/messages_proc.py +++ b/src/modules/stream_notifications/messages_proc.py @@ -99,16 +99,21 @@ def get_completion(message: str): api_key=config.OPENAI_API_KEY, ) - message_cleaned = message.replace("курбез", "").replace("булат", "").replace("kurbezz", "") + message_cleaned = message.replace("курбез", "").replace("булат", "").replace("kurbezz", "").replace("@", "") query = f"Отвечай на русском языке! {message_cleaned}" completion = client.chat.completions.create( extra_body={}, - model="deepseek/deepseek-r1:free", + model="google/gemini-2.0-flash-thinking-exp:free", messages=[ { "role": "user", - "content": query + "content": [ + { + "type": "text", + "text": query + }, + ] } ] )