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

This commit is contained in:
Krrish Dholakia 2024-06-24 19:41:29 -07:00 committed by Ishaan Jaff
parent 82c6f31095
commit 438f65666b
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(