diff --git a/litellm/proxy/guardrails/guardrail_hooks/deepkeep/deepkeep.py b/litellm/proxy/guardrails/guardrail_hooks/deepkeep/deepkeep.py
index 7021e6beffd..b47008884ae 100644
--- a/litellm/proxy/guardrails/guardrail_hooks/deepkeep/deepkeep.py
+++ b/litellm/proxy/guardrails/guardrail_hooks/deepkeep/deepkeep.py
@@ -259,7 +259,9 @@ class DeepKeepGuardrail(CustomGuardrail):
             return_inputs["tool_calls"] = response_json["tool_calls"]
         elif tool_calls is not None:
             return_inputs["tool_calls"] = tool_calls
-        if structured_messages is not None:
+        if response_json.get("structured_messages") is not None:
+            return_inputs["structured_messages"] = response_json["structured_messages"]
+        elif structured_messages is not None:
             return_inputs["structured_messages"] = structured_messages
         return return_inputs