mirror of
https://github.com/tiennm99/litellm.git
synced 2026-08-07 10:21:54 +00:00
fix(utils): fixing exception mapping
This commit is contained in:
@@ -451,6 +451,7 @@ class Huggingface(BaseLLM):
|
||||
if self._aclient_session is None:
|
||||
self._aclient_session = self.create_aclient_session()
|
||||
client = self._aclient_session
|
||||
response = None
|
||||
try:
|
||||
response = await client.post(url=api_base, json=data, headers=headers)
|
||||
response_json = response.json()
|
||||
@@ -468,7 +469,7 @@ class Huggingface(BaseLLM):
|
||||
except Exception as e:
|
||||
if isinstance(e,httpx.TimeoutException):
|
||||
raise HuggingfaceError(status_code=500, message="Request Timeout Error")
|
||||
elif response and hasattr(response, "text"):
|
||||
elif response is not None and hasattr(response, "text"):
|
||||
raise HuggingfaceError(status_code=500, message=f"{str(e)}\n\nOriginal Response: {response.text}")
|
||||
else:
|
||||
raise HuggingfaceError(status_code=500, message=f"{str(e)}")
|
||||
|
||||
@@ -3742,8 +3742,6 @@ def exception_type(
|
||||
model=model,
|
||||
request=original_exception.request
|
||||
)
|
||||
exception_mapping_worked = True
|
||||
raise APIError(status_code=500, message=error_str, model=model, llm_provider=custom_llm_provider)
|
||||
elif custom_llm_provider == "ai21":
|
||||
if hasattr(original_exception, "message"):
|
||||
if "Prompt has too many tokens" in original_exception.message:
|
||||
|
||||
Reference in New Issue
Block a user