test(test_completion.py): handle anthropic internal server error

This commit is contained in:
Krrish Dholakia 2024-03-05 19:01:40 -08:00
parent 6b6775f68e
commit 10ef40faae

View file

@ -151,6 +151,8 @@ def test_completion_claude_3_function_call():
assert isinstance(
response.choices[0].message.tool_calls[0].function.arguments, str
)
except litellm.ServiceUnavailableError as e:
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")