mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-08 03:08:45 +00:00
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:
parent
2f2e1ab9b4
commit
e54a19f27c
1 changed files with 3 additions and 1 deletions
|
|
@ -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
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue