feat(main.py): support multiple deployments in 1 completion call

This commit is contained in:
Krrish Dholakia 2023-10-20 13:01:53 -07:00
parent 62d9ce6e66
commit 1f1cf7a11c
3 changed files with 79 additions and 3 deletions

View file

@ -2798,8 +2798,8 @@ def exception_type(
)
exception_mapping_worked = True
raise APIError(
status_code=original_exception.status_code,
message=f"ReplicateException - {original_exception.message}",
status_code=500,
message=f"ReplicateException - {str(original_exception)}",
llm_provider="replicate",
model=model
)