GithubBackup/gemini-balance
1
0
Fork 0
mirror of https://github.com/snailyp/gemini-balance.git synced 2026-07-10 23:11:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

异步生成完成

Browse Source
This commit is contained in:
Yuzhong Zhang
2025-03-03 13:41:06 +08:00
parent 88d483c1ef
commit 48d58ef2e8
3 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/api/gemini_routes.py
View File

@@ -63,7 +63,7 @@ async def generate_content(
logger.info(f"Using API key: {api_key}") logger.info(f"Using API key: {api_key}")
try: try:
response = chat_service.generate_content( response = await chat_service.generate_content(
model=model_name, model=model_name,
request=request, request=request,
api_key=api_key api_key=api_key
@@ -122,7 +122,7 @@ async def verify_key(api_key: str):
) )
] ]
) )
response = chat_service.generate_content(settings.TEST_MODEL,gemini_requset, api_key) response = await chat_service.generate_content(settings.TEST_MODEL,gemini_requset, api_key)
if response: if response:
return JSONResponse({"status": "valid"}) return JSONResponse({"status": "valid"})
return JSONResponse({"status": "invalid"}) return JSONResponse({"status": "invalid"})

6
app/services/chat/api_client.py
View File

@@ -24,13 +24,13 @@ class GeminiApiClient(ApiClient):
self.base_url = base_url self.base_url = base_url
self.timeout = timeout 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) timeout = httpx.Timeout(self.timeout, read=self.timeout)
if model.endswith("-search"): if model.endswith("-search"):
model = model[:-7] 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}" 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: if response.status_code != 200:
error_content = response.text error_content = response.text
raise Exception(f"API call failed with status code {response.status_code}, {error_content}") raise Exception(f"API call failed with status code {response.status_code}, {error_content}")

4
app/services/gemini_chat_service.py
View File

@@ -79,10 +79,10 @@ class GeminiChatService:
self.key_manager = key_manager self.key_manager = key_manager
self.response_handler = GeminiResponseHandler() self.response_handler = GeminiResponseHandler()
def generate_content(self, model: str, request: GeminiRequest, api_key: str) -> Dict[str, Any]: async def generate_content(self, model: str, request: GeminiRequest, api_key: str) -> Dict[str, Any]:
"""生成内容""" """生成内容"""
payload = _build_payload(model, request) payload = _build_payload(model, request)
response = self.api_client.generate_content(payload, model, api_key) response = await self.api_client.generate_content(payload, model, api_key)
return self.response_handler.handle_response(response, model, stream=False) return self.response_handler.handle_response(response, model, stream=False)
async def stream_generate_content(self, model: str, request: GeminiRequest, api_key: str) -> AsyncGenerator[str, None]: async def stream_generate_content(self, model: str, request: GeminiRequest, api_key: str) -> AsyncGenerator[str, None]: