fix gemini dynamic model (#470)

This commit is contained in:
Okk Hex
2025-04-02 00:39:23 +05:30
committed by GitHub
parent c5c7916be6
commit a7cc8ce073

View File

@@ -36,7 +36,7 @@ import asyncio
ENDPOINTS = {
"gpt": "https://api.openai.com/v1/chat/completions",
"antr": "https://api.anthropic.com/v1/messages",
"gemini": "https://generativelanguage.googleapis.com/v1/models/gemini-pro:generateContent",
"gemini": "https://generativelanguage.googleapis.com/v1/models/",
"deepseek": "https://api.deepseek.com/chat/completions"
}
@@ -163,6 +163,7 @@ async def get_ai_response(provider, prompt, api_key, stream=False):
continue
elif provider == "gemini":
endpoint = f"{ENDPOINTS[provider]}{model}:generateContent"
params = {"key": api_key}
data = {
"contents": [{
@@ -170,7 +171,7 @@ async def get_ai_response(provider, prompt, api_key, stream=False):
}]
}
response = await async_searcher(
ENDPOINTS[provider],
endpoint,
params=params,
headers=headers,
post=True,