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

adding exception handling for together ai

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-29 12:29:56 -07:00
parent 35ecc91a71
commit 88bd1df3e0
8 changed files with 96 additions and 43 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/exceptions.py
View file

@ -28,6 +28,17 @@ class InvalidRequestError(InvalidRequestError): # type: ignore
self.message, f"{self.model}"
) # Call the base class constructor with the parameters it needs
# sub class of invalid request error - meant to give more granularity for error handling context window exceeded errors
class ContextWindowExceededError(InvalidRequestError): # type: ignore
def __init__(self, message, model, llm_provider):
self.status_code = 400
self.message = message
self.model = model
self.llm_provider = llm_provider
super().__init__(
self.message, self.model, self.llm_provider
) # Call the base class constructor with the parameters it needs
class RateLimitError(RateLimitError): # type: ignore
def __init__(self, message, llm_provider):