From bd6c91e8063b33bc83800078a5ac6f816f3a4352 Mon Sep 17 00:00:00 2001
From: JR Boos <j.robert.boos@gmail.com>
Date: Mon, 27 Jul 2026 11:03:24 -0400
Subject: [PATCH] feat(shields): add shields to `/streaming_query`

---
 src/utils/agents/streaming.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/utils/agents/streaming.py b/src/utils/agents/streaming.py
index 768bdcef3..02b09a85c 100644
--- a/src/utils/agents/streaming.py
+++ b/src/utils/agents/streaming.py
@@ -121,7 +121,11 @@ async def retrieve_agent_response_generator(
             )
 
         agent = build_agent(
-            context.client, responses_params, configuration, no_tools=no_tools
+            context.client,
+            responses_params,
+            configuration,
+            shields=context.query_request.shield_ids,
+            no_tools=no_tools,
         )
 
         return (