fix: fix linting errors

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
+2 -6
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:
+2 -2
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,