test: handle test

This commit is contained in:
Krrish Dholakia 2025-07-03 14:54:39 -07:00
parent f6a0eef58a
commit e3150075ce

View file

@ -1300,6 +1300,11 @@ def test_anthropic_mcp_server_tool_use(spec: str):
response = litellm.completion(**params)
except litellm.InternalServerError as e:
print(e)
except Exception as e:
if "Connection to MCP server" in str(e):
pass
else:
raise e
assert response is not None