Use gemini

This commit is contained in:
2025-02-18 21:58:15 +01:00
parent 8ccd3debce
commit 62956dc5f0

View File

@@ -99,16 +99,21 @@ def get_completion(message: str):
api_key=config.OPENAI_API_KEY, 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}" query = f"Отвечай на русском языке! {message_cleaned}"
completion = client.chat.completions.create( completion = client.chat.completions.create(
extra_body={}, extra_body={},
model="deepseek/deepseek-r1:free", model="google/gemini-2.0-flash-thinking-exp:free",
messages=[ messages=[
{ {
"role": "user", "role": "user",
"content": query "content": [
{
"type": "text",
"text": query
},
]
} }
] ]
) )