fix(utils.py): catch 422-status errors

This commit is contained in:
Krrish Dholakia 2024-06-24 19:41:29 -07:00
parent 95f972ee9f
commit d182ea0f77
2 changed files with 28 additions and 7 deletions

View file

@ -6068,6 +6068,14 @@ def exception_type(
model=model,
llm_provider="replicate",
)
elif original_exception.status_code == 422:
exception_mapping_worked = True
raise UnprocessableEntityError(
message=f"ReplicateException - {original_exception.message}",
llm_provider="replicate",
model=model,
response=original_exception.response,
)
elif original_exception.status_code == 429:
exception_mapping_worked = True
raise RateLimitError(