mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-07 08:26:10 +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:
parent
cc2dff0581
commit
c3fb5e1ea5
1 changed files with 2 additions and 0 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue