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

fix: fix linting errors

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-31 07:19:45 -07:00
parent 1b6bf48264
commit c583cbdbcd
2 changed files with 4 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -111,11 +111,9 @@ class AsyncHTTPHandler:
finally:
await new_client.aclose()
except httpx.TimeoutException:
if data is None:
data = {}
raise litellm.Timeout(
message=f"Connection timed out after {timeout} seconds.",
model=data.get("model"),
model="default-model-name",
llm_provider="litellm-httpx-handler",
)
except httpx.HTTPStatusError as e:
@ -221,11 +219,9 @@ class HTTPHandler:
response = self.client.send(req, stream=stream)
return response
except httpx.TimeoutException:
if data is None:
data = {}
raise litellm.Timeout(
message=f"Connection timed out after {timeout} seconds.",
model=data.get("model"),
model="default-model-name",
llm_provider="litellm-httpx-handler",
)
except Exception as e: