fix bug: _achat_completion_stream add request_timeout

geekan · Aug 7, 2024 · e4b8288 · e4b8288
1 parent c036574
commit e4b8288
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metagpt/provider/qianfan_api.py b/metagpt/provider/qianfan_api.py
@@ -117,7 +117,7 @@ async def acompletion(self, messages: list[dict], timeout: int = USE_CONFIG_TIME
         return await self._achat_completion(messages, timeout=self.get_timeout(timeout))
 
     async def _achat_completion_stream(self, messages: list[dict], timeout: int = USE_CONFIG_TIMEOUT) -> str:
-        resp = await self.aclient.ado(**self._const_kwargs(messages=messages, stream=True))
+        resp = await self.aclient.ado(**self._const_kwargs(messages=messages, stream=True),request_timeout = timeout)
         collected_content = []
         usage = {}
         async for chunk in resp: