youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-04 17:38:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: hide output tools and improve JSON formatting for structured output

Browse Source
This commit is contained in:
Stream
2026-02-04 15:12:34 +08:00
parent d84aaff825
commit 4466688e97
1 changed files with 21 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
api/core/workflow/nodes/llm/node.py
View File

@ -2261,25 +2261,32 @@ class LLMNode(Node[LLMNodeData]):
# Add tool call to pending list for model segment
buffers.pending_tool_calls.append(ToolCall(id=tool_call_id, name=tool_name, arguments=tool_arguments))
yield ToolCallChunkEvent(
selector=[self._node_id, "generation", "tool_calls"],
chunk=tool_arguments,
tool_call=ToolCall(
id=tool_call_id,
name=tool_name,
arguments=tool_arguments,
icon=tool_icon,
icon_dark=tool_icon_dark,
),
is_final=False,
)
output_tool_names = {OUTPUT_TEXT_TOOL, FINAL_OUTPUT_TOOL, FINAL_STRUCTURED_OUTPUT_TOOL}
if tool_name not in output_tool_names:
yield ToolCallChunkEvent(
selector=[self._node_id, "generation", "tool_calls"],
chunk=tool_arguments,
tool_call=ToolCall(
id=tool_call_id,
name=tool_name,
arguments=tool_arguments,
icon=tool_icon,
icon_dark=tool_icon_dark,
),
is_final=False,
)
if tool_name in output_tool_names:
content = ""
if tool_name in (OUTPUT_TEXT_TOOL, FINAL_OUTPUT_TOOL):
content = payload.tool_args["text"]
elif tool_name == FINAL_STRUCTURED_OUTPUT_TOOL:
content = json.dumps(payload.tool_args["data"], ensure_ascii=False)
content = json.dumps(
payload.tool_args["data"],
ensure_ascii=False,
indent=2
)
if content:
yield StreamChunkEvent(
@ -2289,7 +2296,7 @@ class LLMNode(Node[LLMNodeData]):
)
yield StreamChunkEvent(
selector=[self._node_id, "generation", "content"],
chunk=content,
chunk=f"```json\n{content}\n```",
is_final=False,
)