From 21fc69503ad18bb6b88e973cc203e231c2cd82a2 Mon Sep 17 00:00:00 2001 From: Yuneng Jiang Date: Tue, 8 Sep 2026 18:03:58 -0700 Subject: [PATCH] test: preserve captured MCP toolset update fields --- .../proxy/_experimental/mcp_server/test_mcp_partial_update.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_partial_update.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_partial_update.py index 29e84331f6f..669e094fee4 100644 --- a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_partial_update.py +++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_partial_update.py @@ -878,8 +878,8 @@ async def _run_toolset_update(payload: dict) -> dict: mock_prisma = _mock_toolset_prisma() await update_mcp_toolset(mock_prisma, UpdateMCPToolsetRequest.model_validate(payload), "test-user") written = dict(mock_prisma.db.litellm_mcptoolsettable.update.call_args[1]["data"]) - assert written.pop("updated_by") == "test-user" - return written + assert written["updated_by"] == "test-user" + return {name: value for name, value in written.items() if name != "updated_by"} @pytest.mark.asyncio