From cc2c33fee59d08f361dafc269a1479ad4f4b69b2 Mon Sep 17 00:00:00 2001
From: Mahmut CAVDAR <4072246+mcavdar@users.noreply.github.com>
Date: Sun, 16 Nov 2025 07:25:00 +0100
Subject: [PATCH 1/3] Refactor safety response handling in guardrails.mdx

---
 src/oss/langchain/guardrails.mdx | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/src/oss/langchain/guardrails.mdx b/src/oss/langchain/guardrails.mdx
index 34fef04a72..67facb7b46 100644
--- a/src/oss/langchain/guardrails.mdx
+++ b/src/oss/langchain/guardrails.mdx
@@ -484,13 +484,7 @@ class SafetyGuardrailMiddleware(AgentMiddleware):
         result = self.safety_model.invoke([{"role": "user", "content": safety_prompt}])
 
         if "UNSAFE" in result.content:
-            return {
-                "messages": [{
-                    "role": "assistant",
-                    "content": "I cannot provide that response. Please rephrase your request."
-                }],
-                "jump_to": "end"
-            }
+            last_message.content = "I cannot provide that response. Please rephrase your request."
 
         return None
 

From 1282c2d2019be2126fedbe89e339ff21b324b040 Mon Sep 17 00:00:00 2001
From: Mahmut CAVDAR <4072246+mcavdar@users.noreply.github.com>
Date: Wed, 19 Nov 2025 19:24:25 +0000
Subject: [PATCH 2/3] Refactor safety response handling in guardrails.mdx for
 js

---
 src/oss/langchain/guardrails.mdx | 17 ++---------------
 1 file changed, 2 insertions(+), 15 deletions(-)

diff --git a/src/oss/langchain/guardrails.mdx b/src/oss/langchain/guardrails.mdx
index 67facb7b46..9bd48a994f 100644
--- a/src/oss/langchain/guardrails.mdx
+++ b/src/oss/langchain/guardrails.mdx
@@ -531,13 +531,7 @@ def safety_guardrail(state: AgentState, runtime: Runtime) -> dict[str, Any] | No
     result = safety_model.invoke([{"role": "user", "content": safety_prompt}])
 
     if "UNSAFE" in result.content:
-        return {
-            "messages": [{
-                "role": "assistant",
-                "content": "I cannot provide that response. Please rephrase your request."
-            }],
-            "jump_to": "end"
-        }
+        last_message.content = "I cannot provide that response. Please rephrase your request."
 
     return None
 
@@ -590,14 +584,7 @@ const safetyGuardrailMiddleware = () => {
       ]);
 
       if (result.content.toString().includes("UNSAFE")) {
-        return {
-          messages: [
-            new AIMessage(
-              "I cannot provide that response. Please rephrase your request."
-            )
-          ],
-          jumpTo: "end",
-        };
+        lastMessage.content = "I cannot provide that response. Please rephrase your request.";
       }
 
       return;

From ae990bb2a0af6bb3571027566a6da855091c525f Mon Sep 17 00:00:00 2001
From: Mahmut CAVDAR <4072246+mcavdar@users.noreply.github.com>
Date: Thu, 20 Nov 2025 07:30:26 +0000
Subject: [PATCH 3/3] Revert JavaScript change: restore previous implementation

---
 src/oss/langchain/guardrails.mdx | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/oss/langchain/guardrails.mdx b/src/oss/langchain/guardrails.mdx
index 9bd48a994f..46ca584f6c 100644
--- a/src/oss/langchain/guardrails.mdx
+++ b/src/oss/langchain/guardrails.mdx
@@ -584,7 +584,14 @@ const safetyGuardrailMiddleware = () => {
       ]);
 
       if (result.content.toString().includes("UNSAFE")) {
-        lastMessage.content = "I cannot provide that response. Please rephrase your request.";
+        return {
+          messages: [
+            new AIMessage(
+              "I cannot provide that response. Please rephrase your request."
+            )
+          ],
+          jumpTo: "end",
+        };
       }
 
       return;