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

(feat) add openai.NotFoundError

Browse source
This commit is contained in:
ishaan-jaff 2023-12-15 10:18:02 +05:30
parent 4fd59c5847
commit 353433e5ce
3 changed files with 28 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

15
litellm/utils.py
View file

@ -46,6 +46,7 @@ from openai._models import BaseModel as OpenAIObject
from .exceptions import (
AuthenticationError,
BadRequestError,
NotFoundError,
RateLimitError,
ServiceUnavailableError,
OpenAIError,
@ -4169,6 +4170,14 @@ def exception_type(
model=model,
response=original_exception.response
)
elif "invalid_request_error" in error_str and "model_not_found" in error_str:
exception_mapping_worked = True
raise NotFoundError(
message=f"OpenAIException - {original_exception.message}",
llm_provider="openai",
model=model,
response=original_exception.response
)
elif "invalid_request_error" in error_str and "Incorrect API key provided" not in error_str:
exception_mapping_worked = True
raise BadRequestError(
@ -4994,15 +5003,15 @@ def exception_type(
model=model,
response=original_exception.response
)
elif "invalid_request_error" in error_str:
elif "DeploymentNotFound" in error_str:
exception_mapping_worked = True
raise BadRequestError(
raise NotFoundError(
message=f"AzureException - {original_exception.message}",
llm_provider="azure",
model=model,
response=original_exception.response
)
elif "DeploymentNotFound" in error_str:
elif "invalid_request_error" in error_str:
exception_mapping_worked = True
raise BadRequestError(
message=f"AzureException - {original_exception.message}",