GithubBackup/gemini-balance
1
0
Fork 0
mirror of https://github.com/snailyp/gemini-balance.git synced 2026-09-05 15:36:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'BetterAndBetterII/main'

Browse Source
This commit is contained in:
snaily
2025-03-03 18:45:56 +08:00
parent 7b5b6c7d4c 68ed4da789
commit 5a44a76c48
6 changed files with 20 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/services/chat/api_client.py
+3 -3
View File
@@ -24,13 +24,13 @@ class GeminiApiClient(ApiClient):
self.base_url = base_url
self.timeout = timeout
def generate_content(self, payload: Dict[str, Any], model: str, api_key: str) -> Dict[str, Any]:
async def generate_content(self, payload: Dict[str, Any], model: str, api_key: str) -> Dict[str, Any]:
timeout = httpx.Timeout(self.timeout, read=self.timeout)
if model.endswith("-search"):
model = model[:-7]
with httpx.Client(timeout=timeout) as client:
async with httpx.AsyncClient(timeout=timeout) as client:
url = f"{self.base_url}/models/{model}:generateContent?key={api_key}"
response = client.post(url, json=payload)
response = await client.post(url, json=payload)
if response.status_code != 200:
error_content = response.text
raise Exception(f"API call failed with status code {response.status_code}, {error_content}")