From 274baac9dfe1280f663dec9f505e4f031c1c181c Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 3 Jul 2025 14:53:30 -0700 Subject: [PATCH] test_mcp_tools_with_responses_api --- .../test_openai_responses_api.py | 8 ++++--- .../test_anthropic_completion.py | 23 ++++++++++++------- 2 files changed, 20 insertions(+), 11 deletions(-) diff --git a/tests/llm_responses_api_testing/test_openai_responses_api.py b/tests/llm_responses_api_testing/test_openai_responses_api.py index d00d8ce5e09..8cb98e2afc0 100644 --- a/tests/llm_responses_api_testing/test_openai_responses_api.py +++ b/tests/llm_responses_api_testing/test_openai_responses_api.py @@ -1117,9 +1117,11 @@ def test_mcp_tools_with_responses_api(): MCP_TOOLS = [ { "type": "mcp", - "server_label": "gitmcp", - "server_url": "https://gitmcp.io/openai/tiktoken", - "allowed_tools": ["search_tiktoken_documentation", "fetch_tiktoken_documentation"], + "server_label": "zapier", + "server_url": "https://mcp.zapier.com/api/mcp/mcp", + "headers": { + "Authorization": f"Bearer {os.getenv('ZAPIER_CI_CD_MCP_TOKEN')}" + } } ] MODEL = "openai/gpt-4.1" diff --git a/tests/llm_translation/test_anthropic_completion.py b/tests/llm_translation/test_anthropic_completion.py index 5877005bfe6..bd49757088b 100644 --- a/tests/llm_translation/test_anthropic_completion.py +++ b/tests/llm_translation/test_anthropic_completion.py @@ -1276,16 +1276,20 @@ def test_anthropic_mcp_server_tool_use(spec: str): tools = [ { "type": "url", - "url": "https://gitmcp.io/openai/tiktoken", - "name": "gitmcp", + "url": "https://mcp.zapier.com/api/mcp/mcp", + "name": "zapier-mcp", + "authorization_token": os.getenv('ZAPIER_CI_CD_MCP_TOKEN') } ] elif spec == "openai": tools = [ { "type": "mcp", - "server_label": "gitmcp", - "server_url": "https://gitmcp.io/openai/tiktoken", + "server_label": "zapier", + "server_url": "https://mcp.zapier.com/api/mcp/mcp", + "headers": { + "Authorization": f"Bearer {os.getenv('ZAPIER_CI_CD_MCP_TOKEN')}" + }, "require_approval": "never", }, ] @@ -1309,19 +1313,22 @@ def test_anthropic_mcp_server_tool_use(spec: str): ) def test_anthropic_mcp_server_responses_api(model: str): from litellm import responses - + litellm._turn_on_debug() tools = [ { "type": "mcp", - "server_label": "gitmcp", - "server_url": "https://gitmcp.io/openai/tiktoken", + "server_label": "zapier", + "server_url": "https://mcp.zapier.com/api/mcp/mcp", "require_approval": "never", + "headers": { + "Authorization": f"Bearer {os.getenv('ZAPIER_CI_CD_MCP_TOKEN')}" + } }, ] response = litellm.responses( model=model, - input="how does tiktoken work?", + input="Who won the World Cup in 2022?", max_output_tokens=100, tools=tools, )