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

test(mcp): align concurrency probe with client error propagation

Browse source
This commit is contained in:
Joshua Valluru 2026-09-28 16:18:46 -07:00
parent d9239e94b3
commit 812e58a146
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_max_concurrent_requests.py
View file

@ -47,7 +47,7 @@ def _make_server(server_id: str, max_concurrent_requests: Optional[int]) -> MCPS
def _patch_client_with_tracker(manager: MCPServerManager, tracker: _ConcurrencyTracker):
async def fake_create_mcp_client(server, **kwargs):
class _ProbeClient:
async def call_tool(self, params, host_progress_callback=None, allow_input_required=False):
async def call_tool(self, params, host_progress_callback=None, allow_input_required=False, raise_on_error: bool = False):
tracker.enter(server.server_id)
try:
await asyncio.sleep(HOLD_SECONDS)