异步生成完成

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
+2 -2
View File
@@ -79,10 +79,10 @@ class GeminiChatService:
self.key_manager = key_manager
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)
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)
async def stream_generate_content(self, model: str, request: GeminiRequest, api_key: str) -> AsyncGenerator[str, None]: