add dataset field to check request

2026-06-06 21:23:55 +02:00 · 2025-07-15 11:00:36 +02:00
parent 31b128950b
commit b8fbb37935
1 changed files with 1 additions and 0 deletions
@@ -148,6 +148,7 @@ async def check_guardrails(
                    "messages": messages[:-1] + [new_message],
                    "policies": [g.content for g in guardrails],
                    "parameters": context.guardrails_parameters or {},
+                    "dataset_name": context.dataset_name,
                },
                headers={
                    "Authorization": context.get_guardrailing_authorization(),