mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 18:54:30 +00:00
Properly revert changes in FireworksAIMixin to match main branch
This commit is contained in:
parent
2f8d99b406
commit
121e52e370
1 changed files with 0 additions and 9 deletions
|
@ -20,15 +20,6 @@ class FireworksAIMixin:
|
||||||
def get_error_class(
|
def get_error_class(
|
||||||
self, error_message: str, status_code: int, headers: Union[dict, Headers]
|
self, error_message: str, status_code: int, headers: Union[dict, Headers]
|
||||||
) -> BaseLLMException:
|
) -> BaseLLMException:
|
||||||
# Check if it's a rate limit error (status code 429)
|
|
||||||
if status_code == 429:
|
|
||||||
from litellm.exceptions import RateLimitError
|
|
||||||
return RateLimitError(
|
|
||||||
message=f"Fireworks_aiException - {error_message}",
|
|
||||||
llm_provider="fireworks_ai",
|
|
||||||
model="", # This will be set later in the exception mapping
|
|
||||||
response=None, # This will be set later in the exception mapping
|
|
||||||
)
|
|
||||||
return FireworksAIException(
|
return FireworksAIException(
|
||||||
status_code=status_code,
|
status_code=status_code,
|
||||||
message=error_message,
|
message=error_message,
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue