phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 18:24:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Provide response to ServiceUnavailableError where needed

Browse source
This commit is contained in:
Jack Collins 2023-11-13 21:20:40 -08:00
parent f499e8bb1d
commit 223ec8933d
1 changed files with 16 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

24
litellm/utils.py
View file

@ -3264,7 +3264,8 @@ def exception_type(
raise ServiceUnavailableError(
message=f"AnthropicException - {original_exception.message}",
llm_provider="anthropic",
model=model
model=model,
response=original_exception.response
)
else:
exception_mapping_worked = True
@ -3346,7 +3347,8 @@ def exception_type(
raise ServiceUnavailableError(
message=f"ReplicateException - {original_exception.message}",
llm_provider="replicate",
model=model
model=model,
response=original_exception.response
)
exception_mapping_worked = True
raise APIError(
@ -3395,7 +3397,8 @@ def exception_type(
raise ServiceUnavailableError(
message=f"BedrockException - {original_exception.message}",
llm_provider="bedrock",
model=model
model=model,
response=original_exception.response
)
elif original_exception.status_code == 401:
exception_mapping_worked = True
@ -3477,7 +3480,8 @@ def exception_type(
raise ServiceUnavailableError(
message=f"CohereException - {original_exception.message}",
llm_provider="cohere",
model=model
model=model,
response=original_exception.response
)
elif (
"CohereConnectionError" in exception_type
@ -3502,7 +3506,8 @@ def exception_type(
raise ServiceUnavailableError(
message=f"CohereException - {original_exception.message}",
llm_provider="cohere",
model=model
model=model,
response=original_exception.response
)
else:
if hasattr(original_exception, "status_code"):
@ -3710,7 +3715,8 @@ def exception_type(
raise ServiceUnavailableError(
message=f"NLPCloudException - {original_exception.message}",
model=model,
llm_provider="nlp_cloud"
llm_provider="nlp_cloud",
response=original_exception.response
)
else:
exception_mapping_worked = True
@ -3847,7 +3853,8 @@ def exception_type(
raise ServiceUnavailableError(
message=f"AlephAlphaException - {original_exception.message}",
llm_provider="aleph_alpha",
model=model
model=model,
response=original_exception.response
)
raise original_exception
raise original_exception
@ -3872,7 +3879,8 @@ def exception_type(
raise ServiceUnavailableError(
message=f"OllamaException: {original_exception}",
llm_provider="ollama",
model=model
model=model,
response=original_exception.response
)
elif "Invalid response object from API" in error_str:
exception_mapping_worked = True