From c16a12fdec8a7c275fa57ff5ddbf55f2dd5a6ebf Mon Sep 17 00:00:00 2001 From: Luca Beurer-Kellner Date: Fri, 16 May 2025 20:02:43 +0200 Subject: [PATCH] adjust guardrail failure address --- tests/integration/mcp/test_mcp.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/integration/mcp/test_mcp.py b/tests/integration/mcp/test_mcp.py index d0c7250..ab3dd05 100644 --- a/tests/integration/mcp/test_mcp.py +++ b/tests/integration/mcp/test_mcp.py @@ -298,7 +298,7 @@ async def test_mcp_with_gateway_and_blocking_guardrails( assert len(annotations) == 1 assert ( annotations[0]["content"] == "get_last_message_from_user is called" - and annotations[0]["address"] == "messages.0.tool_calls.0" + and annotations[0]["address"] == "messages.2.tool_calls.0" ) assert annotations[0]["extra_metadata"]["source"] == "guardrails-error" assert annotations[0]["extra_metadata"]["guardrail"]["action"] == "block"