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

bump version

Browse source
This commit is contained in:
Krrish Dholakia 2023-09-06 11:05:09 -07:00
parent 0ddda7c035
commit afcd6b28cc
2 changed files with 51 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

67
litellm/utils.py
View file

@ -1572,23 +1572,56 @@ def exception_type(model, original_exception, custom_llm_provider):
llm_provider="replicate", llm_provider="replicate",
model=model model=model
) )
elif ( elif hasattr(original_exception, "status_code"):
exception_type == "ReplicateError" if original_exception.status_code == 401:
): # ReplicateError implies an error on Replicate server side, not user side exception_mapping_worked = True
exception_mapping_worked = True raise AuthenticationError(
raise ServiceUnavailableError( message=f"ReplicateException - {original_exception.message}",
message=f"ReplicateException - {error_str}", llm_provider="replicate",
llm_provider="replicate", model=model
model=model )
) elif original_exception.status_code == 400:
else: exception_mapping_worked = True
exception_mapping_worked = True raise InvalidRequestError(
raise APIError( message=f"ReplicateException - {original_exception.message}",
status_code=original_exception.status_code, model=model,
message=f"ReplicateException - {original_exception.message}", llm_provider="replicate",
llm_provider="replicate", )
model=model elif original_exception.status_code == 408:
) exception_mapping_worked = True
raise Timeout(
message=f"ReplicateException - {original_exception.message}",
model=model,
llm_provider="replicate"
)
elif original_exception.status_code == 413:
exception_mapping_worked = True
raise InvalidRequestError(
message=f"ReplicateException - {original_exception.message}",
model=model,
llm_provider="replicate",
)
elif original_exception.status_code == 429:
exception_mapping_worked = True
raise RateLimitError(
message=f"ReplicateException - {original_exception.message}",
llm_provider="replicate",
model=model
)
elif original_exception.status_code == 500:
exception_mapping_worked = True
raise ServiceUnavailableError(
message=f"ReplicateException - {original_exception.message}",
llm_provider="replicate",
model=model
)
exception_mapping_worked = True
raise APIError(
status_code=original_exception.status_code,
message=f"ReplicateException - {original_exception.message}",
llm_provider="replicate",
model=model
)
elif model in litellm.cohere_models: # Cohere elif model in litellm.cohere_models: # Cohere
if ( if (
"invalid api token" in error_str "invalid api token" in error_str

2
pyproject.toml
View file

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "litellm" name = "litellm"
version = "0.1.540" version = "0.1.541"
description = "Library to easily interface with LLM API providers" description = "Library to easily interface with LLM API providers"
authors = ["BerriAI"] authors = ["BerriAI"]
license = "MIT License" license = "MIT License"