fix(utils.py): Fix anthropic tool calling exception mapping

Fixes https://github.com/BerriAI/litellm/issues/4348
This commit is contained in:
Krrish Dholakia 2024-06-21 21:20:49 -07:00
parent 44bd9548d9
commit 5e893ed13e
2 changed files with 19 additions and 8 deletions

View file

@ -5922,21 +5922,28 @@ def exception_type(
if "prompt is too long" in error_str or "prompt: length" in error_str:
exception_mapping_worked = True
raise ContextWindowExceededError(
message=error_str,
message="AnthropicError - {}".format(error_str),
model=model,
llm_provider="anthropic",
)
if "Invalid API Key" in error_str:
exception_mapping_worked = True
raise AuthenticationError(
message=error_str,
message="AnthropicError - {}".format(error_str),
model=model,
llm_provider="anthropic",
)
if "content filtering policy" in error_str:
exception_mapping_worked = True
raise ContentPolicyViolationError(
message=error_str,
message="AnthropicError - {}".format(error_str),
model=model,
llm_provider="anthropic",
)
if "Client error '400 Bad Request'" in error_str:
exception_mapping_worked = True
raise BadRequestError(
message="AnthropicError - {}".format(error_str),
model=model,
llm_provider="anthropic",
)