overspend1/Ultroid-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix gemini dynamic model (#470)

Browse Source
This commit is contained in:
Okk Hex
2025-04-02 00:39:23 +05:30
committed by GitHub
parent c5c7916be6
commit a7cc8ce073
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
plugins/aiwrapper.py
View File

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