phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

docs(input.md): add hf_model_name to docs

Browse source
This commit is contained in:
Krrish Dholakia 2023-12-05 16:56:18 -08:00
parent 88845dddb1
commit ff949490de
4 changed files with 39 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/utils.py
View file

@ -4065,6 +4065,14 @@ def exception_type(
llm_provider="sagemaker",
response=original_exception.response
)
elif "Input validation error: `best_of` must be > 0 and <= 2" in error_str:
exception_mapping_worked = True
raise BadRequestError(
message=f"SagemakerException - the value of 'n' must be > 0 and <= 2 for sagemaker endpoints",
model=model,
llm_provider="sagemaker",
response=original_exception.response
)
elif custom_llm_provider == "vertex_ai":
if "Vertex AI API has not been used in project" in error_str or "Unable to find your project" in error_str:
exception_mapping_worked = True