feat use UnsupportedParamsError as litellm error type

This commit is contained in:
Ishaan Jaff 2024-07-24 12:19:10 -07:00
parent d5d2ffffdf
commit 11512c057d
2 changed files with 29 additions and 0 deletions

View file

@ -888,6 +888,7 @@ from .exceptions import (
APIError, APIError,
Timeout, Timeout,
APIConnectionError, APIConnectionError,
UnsupportedParamsError,
APIResponseValidationError, APIResponseValidationError,
UnprocessableEntityError, UnprocessableEntityError,
InternalServerError, InternalServerError,

View file

@ -682,11 +682,39 @@ class JSONSchemaValidationError(APIError):
) )
class UnsupportedParamsError(BadRequestError):
def __init__(
self,
message,
llm_provider: Optional[str] = None,
model: Optional[str] = None,
status_code: int = 400,
response: Optional[httpx.Response] = None,
litellm_debug_info: Optional[str] = None,
max_retries: Optional[int] = None,
num_retries: Optional[int] = None,
):
self.status_code = 400
self.message = "litellm.UnsupportedParamsError: {}".format(message)
self.model = model
self.llm_provider = llm_provider
self.litellm_debug_info = litellm_debug_info
response = response or httpx.Response(
status_code=self.status_code,
request=httpx.Request(
method="GET", url="https://litellm.ai"
), # mock request object
)
self.max_retries = max_retries
self.num_retries = num_retries
LITELLM_EXCEPTION_TYPES = [ LITELLM_EXCEPTION_TYPES = [
AuthenticationError, AuthenticationError,
NotFoundError, NotFoundError,
BadRequestError, BadRequestError,
UnprocessableEntityError, UnprocessableEntityError,
UnsupportedParamsError,
Timeout, Timeout,
PermissionDeniedError, PermissionDeniedError,
RateLimitError, RateLimitError,