catch asyncio.TimeoutError in _interpret_async_response (#180) (#489)

openai · Jun 16, 2023 · 041bf5a · 041bf5a
1 parent 7610c5a
commit 041bf5a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/openai/api_requestor.py b/openai/api_requestor.py
@@ -720,6 +720,8 @@ async def _interpret_async_response(
         else:
             try:
                 await result.read()
+            except (aiohttp.ServerTimeoutError, asyncio.TimeoutError) as e:
+                raise error.Timeout("Request timed out") from e
             except aiohttp.ClientError as e:
                 util.log_warn(e, body=result.content)
             return (