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

fix(utils.py): improve coverage for anthropic exception mapping

Browse source
This commit is contained in:
Krrish Dholakia 2024-06-21 21:15:10 -07:00
parent 24addd8a4f
commit 000d678445
2 changed files with 37 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

22
litellm/tests/test_exceptions.py
View file

@ -670,7 +670,7 @@ def test_litellm_predibase_exception():
# print(f"accuracy_score: {accuracy_score}")
@pytest.mark.parametrize("provider", ["predibase", "vertex_ai_beta"])
@pytest.mark.parametrize("provider", ["predibase", "vertex_ai_beta", "anthropic"])
def test_exception_mapping(provider):
"""
For predibase, run through a set of mock exceptions
@ -712,3 +712,23 @@ def test_exception_mapping(provider):
)
pass
def test_anthropic_tool_calling_exception():
"""
Related - https://github.com/BerriAI/litellm/issues/4348
"""
tools = [
{
"type": "function",
"function": {
"name": "get_current_weather",
"description": "Get the current weather in a given location",
},
}
]
litellm.completion(
model="claude-3.5",
messages=[{"role": "user", "content": "Hey, how's it going?"}],
tools=tools,
)