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()