From b8fbb379358e45be8ef675747e412fc2463ed9ea Mon Sep 17 00:00:00 2001
From: knielsen404 <kristian@invariantlabs.ai>
Date: Tue, 15 Jul 2025 11:00:36 +0200
Subject: [PATCH] add dataset field to check request

---
 gateway/integrations/guardrails.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/gateway/integrations/guardrails.py b/gateway/integrations/guardrails.py
index 425f084..bd5846a 100644
--- a/gateway/integrations/guardrails.py
+++ b/gateway/integrations/guardrails.py
@@ -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(),