From 4466688e97215dd6056555ab769f6e099ef57732 Mon Sep 17 00:00:00 2001
From: Stream <Stream_2@qq.com>
Date: Wed, 4 Feb 2026 15:12:34 +0800
Subject: [PATCH] feat: hide output tools and improve JSON formatting for
 structured output

---
 api/core/workflow/nodes/llm/node.py | 35 +++++++++++++++++------------
 1 file changed, 21 insertions(+), 14 deletions(-)

diff --git a/api/core/workflow/nodes/llm/node.py b/api/core/workflow/nodes/llm/node.py
index 48969df559..9c503160b0 100644
--- a/api/core/workflow/nodes/llm/node.py
+++ b/api/core/workflow/nodes/llm/node.py
@@ -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,
                     )