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

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

Browse source 
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
Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_exceptions.py
View file

@ -724,11 +724,15 @@ def test_anthropic_tool_calling_exception():
"function": {
"name": "get_current_weather",
"description": "Get the current weather in a given location",
"parameters": {},
},
}
]
try:
litellm.completion(
model="claude-3.5",
model="claude-3-5-sonnet-20240620",
messages=[{"role": "user", "content": "Hey, how's it going?"}],
tools=tools,
)
except litellm.BadRequestError:
pass

13
litellm/utils.py
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",
)