scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-05 08:07:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 136

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
Show stats 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:

4
litellm/llms/predibase.py
View file

@ -487,7 +487,7 @@ class PredibaseChatCompletion(BaseLLM):
headers=headers,
data=json.dumps(data),
stream=stream,
timeout=timeout,
timeout=timeout, # type: ignore
)
_response = CustomStreamWrapper(
response.iter_lines(),
@ -502,7 +502,7 @@ class PredibaseChatCompletion(BaseLLM):
url=completion_url,
headers=headers,
data=json.dumps(data),
timeout=timeout,
timeout=timeout, # type: ignore
)
return self.process_response(
model=model,