scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-01 02:02:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 91

test(mcp): isolate reload cooldown from shared discovery

Browse source
This commit is contained in:
Joshua Valluru 2026-09-26 17:51:43 -07:00
parent f6c85ddf5d
commit f41fb256ba
1 changed files with 6 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py
View file

@ -6393,16 +6393,13 @@ class TestEnsureUpstreamInitializeInstructionsCached:
@pytest.mark.asyncio
async def test_reload_resets_probe_cooldown(self):
"""load_servers_from_config clears the negative-cache map so reloads re-probe."""
from litellm.proxy._experimental.mcp_server.mcp_server_manager import (
global_mcp_server_manager,
)
from litellm.proxy._experimental.mcp_server.mcp_server_manager import MCPServerManager
manager: Final = MCPServerManager()
manager._upstream_initialize_instructions_probed_at["reload-target"] = 1.0
await manager.load_servers_from_config({})
assert "reload-target" not in manager._upstream_initialize_instructions_probed_at
global_mcp_server_manager._upstream_initialize_instructions_probed_at["reload-target"] = 1.0
try:
await global_mcp_server_manager.load_servers_from_config({})
assert "reload-target" not in global_mcp_server_manager._upstream_initialize_instructions_probed_at
finally:
global_mcp_server_manager._upstream_initialize_instructions_probed_at.pop("reload-target", None)
class TestGatewayCreateInitializationOptions: