scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-19 00:01:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 122

fixing mcp tests

Browse source
This commit is contained in:
yuneng-jiang 2026-02-12 11:42:41 -08:00
parent 206b6bad41
commit fd5bbe070b
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tests/mcp_tests/test_mcp_chat_completions.py
View file

@ -19,7 +19,7 @@ async def test_acompletion_mcp_auto_exec(monkeypatch):
inputSchema={"type": "object", "properties": {}},
)
async def fake_process(user_api_key_auth, mcp_tools_with_litellm_proxy):
async def fake_process(user_api_key_auth, mcp_tools_with_litellm_proxy, **kwargs):
return [dummy_tool], {"local_search": "local"}
async def fake_execute(**kwargs):
@ -95,7 +95,7 @@ async def test_acompletion_mcp_respects_manual_approval(monkeypatch):
inputSchema={"type": "object", "properties": {}},
)
async def fake_process(user_api_key_auth, mcp_tools_with_litellm_proxy):
async def fake_process(user_api_key_auth, mcp_tools_with_litellm_proxy, **kwargs):
return [dummy_tool], {"local_search": "local"}
async def fake_execute(**kwargs):
@ -170,7 +170,7 @@ async def test_completion_mcp_with_streaming_no_timeout_error(monkeypatch):
inputSchema={"type": "object", "properties": {}},
)
async def fake_process(user_api_key_auth, mcp_tools_with_litellm_proxy):
async def fake_process(user_api_key_auth, mcp_tools_with_litellm_proxy, **kwargs):
return [dummy_tool], {"local_search": "local"}
async def fake_execute(**kwargs):
@ -470,7 +470,7 @@ async def test_mcp_metadata_in_streaming_final_chunk(monkeypatch):
inputSchema={"type": "object", "properties": {}},
)
async def fake_process(user_api_key_auth, mcp_tools_with_litellm_proxy):
async def fake_process(user_api_key_auth, mcp_tools_with_litellm_proxy, **kwargs):
return [dummy_tool], {"local_search": "local"}
async def fake_execute(**kwargs):
@ -793,7 +793,7 @@ async def test_mcp_streaming_metadata_ordering(monkeypatch):
inputSchema={"type": "object", "properties": {}},
)
async def fake_process(user_api_key_auth, mcp_tools_with_litellm_proxy):
async def fake_process(user_api_key_auth, mcp_tools_with_litellm_proxy, **kwargs):
return [dummy_tool], {"local_search": "local"}
async def fake_execute(**kwargs):