From fd5bbe070b8d717ee8eca6d0adfdad5d608abd24 Mon Sep 17 00:00:00 2001 From: yuneng-jiang Date: Thu, 12 Feb 2026 11:42:41 -0800 Subject: [PATCH] fixing mcp tests --- tests/mcp_tests/test_mcp_chat_completions.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/mcp_tests/test_mcp_chat_completions.py b/tests/mcp_tests/test_mcp_chat_completions.py index 0617dcb2e42..9010e8c0d29 100644 --- a/tests/mcp_tests/test_mcp_chat_completions.py +++ b/tests/mcp_tests/test_mcp_chat_completions.py @@ -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):