From c3fb5e1ea563239adf76a5c4c5f9fdd1ea7ec6a3 Mon Sep 17 00:00:00 2001 From: jquinter Date: Mon, 16 Feb 2026 12:12:48 -0300 Subject: [PATCH] Update tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com> --- .../proxy/_experimental/mcp_server/test_mcp_server.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py index 30a917bb22..8996ef1b7e 100644 --- a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py +++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py @@ -30,9 +30,11 @@ def cleanup_mcp_global_state(): ) # Clear before test global_mcp_server_manager.registry.clear() + global_mcp_server_manager.tool_name_to_mcp_server_name_mapping.clear() yield # Clear after test global_mcp_server_manager.registry.clear() + global_mcp_server_manager.tool_name_to_mcp_server_name_mapping.clear() except ImportError: # MCP not available, skip cleanup yield