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(
|
||||
self, error_message: str, status_code: int, headers: Union[dict, Headers]
|
||||
) -> 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(
|
||||
status_code=status_code,
|
||||
message=error_message,
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue