mirror of
https://github.com/tiennm99/litellm.git
synced 2026-07-21 04:19:24 +00:00
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>
This commit is contained in:
co-authored by
greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
parent
cc2dff0581
commit
c3fb5e1ea5
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user