From ac9442eb92e99bacaa4300d6edb07b738d1fb3bf Mon Sep 17 00:00:00 2001
From: Omar Abdelwahab <omara@fb.com>
Date: Thu, 6 Nov 2025 11:46:45 -0800
Subject: [PATCH] fix: update test_mcp to use authorization parameter instead
 of headers

Changed tool_defs in test_mcp_invocation to use 'authorization' parameter
instead of passing Authorization via headers dict for security compliance.
---
 tests/integration/tool_runtime/test_mcp.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/tests/integration/tool_runtime/test_mcp.py b/tests/integration/tool_runtime/test_mcp.py
index 59f558d2c..40c543977 100644
--- a/tests/integration/tool_runtime/test_mcp.py
+++ b/tests/integration/tool_runtime/test_mcp.py
@@ -81,9 +81,7 @@ def test_mcp_invocation(llama_stack_client, text_model_id, mcp_server):
             "server_label": test_toolgroup_id,
             "require_approval": "never",
             "allowed_tools": [tool.name for tool in tools_list],
-            "headers": {
-                "Authorization": f"Bearer {AUTH_TOKEN}",
-            },
+            "authorization": AUTH_TOKEN,
         }
     ]
     agent = Agent(