From 64df5015148bf8ab3e717dd4eb70f33a3a3c3aac Mon Sep 17 00:00:00 2001
From: JR Boos <j.robert.boos@gmail.com>
Date: Tue, 28 Jul 2026 11:04:06 -0400
Subject: [PATCH] feat(shields): add shields to `/query`

---
 src/app/endpoints/query.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/app/endpoints/query.py b/src/app/endpoints/query.py
index fcbb57a11..47e3fc28e 100644
--- a/src/app/endpoints/query.py
+++ b/src/app/endpoints/query.py
@@ -212,6 +212,7 @@ async def query_endpoint_handler(
         moderation_result,
         endpoint_path,
         compaction.original_input if compaction.compacted else None,
+        shield_ids=query_request.shield_ids,
         no_tools=bool(query_request.no_tools),
     )