Update litellm/proxy/guardrails/guardrail_hooks/deepkeep/deepkeep.py

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
This commit is contained in:
DK-yaniv 2026-08-09 17:39:02 +03:00 • committed by GitHub
parent 2f2e1ab9b4
commit e54a19f27c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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