From 0de80fafbe210df3cb3dd2887c642a62b53f1f30 Mon Sep 17 00:00:00 2001 From: Luca Beurer-Kellner Date: Wed, 7 May 2025 13:51:08 +0200 Subject: [PATCH] test: adjust addresses --- tests/integration/mcp/test_mcp_stdio.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/integration/mcp/test_mcp_stdio.py b/tests/integration/mcp/test_mcp_stdio.py index 9d01966..79b2b67 100644 --- a/tests/integration/mcp/test_mcp_stdio.py +++ b/tests/integration/mcp/test_mcp_stdio.py @@ -162,12 +162,12 @@ async def test_mcp_stdio_with_gateway_and_logging_guardrails( for annotation in annotations: if ( annotation["content"] == "food in ToolOutput" - and annotation["address"] == "messages.1.content.0.text:22-26" + and annotation["address"] == "messages.3.content.0.text:22-26" ): food_annotation = annotation elif ( annotation["content"] == "get_last_message_from_user is called" - and annotation["address"] == "messages.0.tool_calls.0" + and annotation["address"] == "messages.2.tool_calls.0" ): tool_call_annotation = annotation assert food_annotation is not None, "Missing 'food in ToolOutput' annotation" @@ -255,6 +255,6 @@ async def test_mcp_stdio_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"