From fd578e26f6480bccdfbefa803eeeccd5ed034e9c Mon Sep 17 00:00:00 2001 From: zzh1996 Date: Sun, 9 Apr 2023 04:37:07 -0700 Subject: [PATCH] make `request_timeout` in Async API behave the same as in non-Async API --- openai/api_requestor.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/openai/api_requestor.py b/openai/api_requestor.py index 827b73b78e..242b44fd38 100644 --- a/openai/api_requestor.py +++ b/openai/api_requestor.py @@ -562,11 +562,12 @@ async def arequest_raw( if isinstance(request_timeout, tuple): timeout = aiohttp.ClientTimeout( connect=request_timeout[0], - total=request_timeout[1], + sock_read=request_timeout[1], ) else: timeout = aiohttp.ClientTimeout( - total=request_timeout if request_timeout else TIMEOUT_SECS + connect=request_timeout if request_timeout else TIMEOUT_SECS, + sock_read=request_timeout if request_timeout else TIMEOUT_SECS, ) if files: