From d14fb9e6e20eef1d4cfc2650cad5efe9121dc116 Mon Sep 17 00:00:00 2001 From: yucheng Date: Tue, 29 Sep 2026 02:16:02 +0000 Subject: [PATCH] test(mcp): return one masked text per scanned string in the selected-guardrail REST test Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../proxy/_experimental/mcp_server/test_rest_endpoints.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_rest_endpoints.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_rest_endpoints.py index e82ab28bb4c..a27dd8d04bd 100644 --- a/tests/test_litellm/proxy/_experimental/mcp_server/test_rest_endpoints.py +++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_rest_endpoints.py @@ -3300,7 +3300,8 @@ async def test_request_selected_tool_specific_guardrail_applies_to_virtual_execu default_on=False, custom_code="def apply_guardrail(inputs, request_data, input_type):\n" ' if inputs.get("tools", [{}])[0].get("function", {}).get("name") == "execute":\n' - f' return {{"action": "{action}", "reason": "resolved tool blocked", "texts": ["redacted"]}}\n' + ' texts = [t.replace("confidential", "redacted") for t in inputs.get("texts", [])]\n' + f' return {{"action": "{action}", "reason": "resolved tool blocked", "texts": texts}}\n' " return allow()\n", ) manager: Final = mcp_server_manager.MCPServerManager()