scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test_mcp_tools_with_responses_api

Browse source
This commit is contained in:
Ishaan Jaff 2025-07-03 14:53:30 -07:00
parent adfe63cf3a
commit 274baac9df
2 changed files with 20 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/llm_responses_api_testing/test_openai_responses_api.py
View file

@ -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"

23
tests/llm_translation/test_anthropic_completion.py
View file

@ -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,
)