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

(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

1
litellm/__init__.py
View file

@ -412,6 +412,7 @@ from .exceptions import (
AuthenticationError, AuthenticationError,
InvalidRequestError, InvalidRequestError,
BadRequestError, BadRequestError,
NotFoundError,
RateLimitError, RateLimitError,
ServiceUnavailableError, ServiceUnavailableError,
OpenAIError, OpenAIError,

15
litellm/exceptions.py
View file

@ -12,6 +12,7 @@
from openai import ( from openai import (
AuthenticationError, AuthenticationError,
BadRequestError, BadRequestError,
NotFoundError,
RateLimitError, RateLimitError,
APIStatusError, APIStatusError,
OpenAIError, OpenAIError,
@ -36,6 +37,20 @@ class AuthenticationError(AuthenticationError): # type: ignore
body=None body=None
) # Call the base class constructor with the parameters it needs ) # Call the base class constructor with the parameters it needs
# raise when invalid models passed, example gpt-8
class NotFoundError(NotFoundError): # type: ignore
def __init__(self, message, model, llm_provider, response: httpx.Response):
self.status_code = 404
self.message = message
self.model = model
self.llm_provider = llm_provider
super().__init__(
self.message,
response=response,
body=None
) # Call the base class constructor with the parameters it needs
class BadRequestError(BadRequestError): # type: ignore class BadRequestError(BadRequestError): # type: ignore
def __init__(self, message, model, llm_provider, response: httpx.Response): def __init__(self, message, model, llm_provider, response: httpx.Response):
self.status_code = 400 self.status_code = 400

15
litellm/utils.py
View file

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