phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix predibase timeout exceptions

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-30 22:52:08 -07:00
parent 19ab0614c4
commit 82e5ea059b
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/custom_httpx/http_handler.py
View file

@ -110,7 +110,7 @@ class AsyncHTTPHandler:
) )
finally: finally:
await new_client.aclose() await new_client.aclose()
except httpx.ConnectTimeout: except httpx.TimeoutException:
if data is None: if data is None:
data = {} data = {}
raise litellm.Timeout( raise litellm.Timeout(
@ -220,7 +220,7 @@ class HTTPHandler:
) )
response = self.client.send(req, stream=stream) response = self.client.send(req, stream=stream)
return response return response
except httpx.ConnectTimeout: except httpx.TimeoutException:
if data is None: if data is None:
data = {} data = {}
raise litellm.Timeout( raise litellm.Timeout(