feat(shields): add shields to /streaming_query

jrobertboos · jrobertboos · commit fb3fd5fbe0ea · 2026-07-28T18:55:50.000-04:00
diff --git 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 (

Original file line number	Diff line number	Diff line change
`@@ -121,7 +121,11 @@ async def retrieve_agent_response_generator(`
`121`	`121`	`)`
`122`	`122`
`123`	`123`	`agent = build_agent(`
`124`		`- context.client, responses_params, configuration, no_tools=no_tools`
	`124`	`+ context.client,`
	`125`	`+ responses_params,`
	`126`	`+ configuration,`
	`127`	`+ shields=context.query_request.shield_ids,`
	`128`	`+ no_tools=no_tools,`
`125`	`129`	`)`
`126`	`130`
`127`	`131`	`return (`