diff --git a/ui/src/workflow/nodes/ai-chat-node/index.vue b/ui/src/workflow/nodes/ai-chat-node/index.vue
index 32347b2d07e..61198a648f3 100644
--- a/ui/src/workflow/nodes/ai-chat-node/index.vue
+++ b/ui/src/workflow/nodes/ai-chat-node/index.vue
@@ -144,7 +144,14 @@
           />
         </el-form-item>
         <el-form-item
-          v-if="[WorkflowMode.Application, WorkflowMode.ApplicationLoop].includes(workflowMode)"
+          v-if="
+            [
+              WorkflowMode.Application,
+              WorkflowMode.ApplicationLoop,
+              WorkflowMode.Tool,
+              WorkflowMode.ToolLoop,
+            ].includes(workflowMode)
+          "
         >
           <template #label>
             <div class="flex-between">
@@ -449,7 +456,14 @@
         </el-form-item>
         <el-form-item
           @click.prevent
-          v-if="[WorkflowMode.Application, WorkflowMode.ApplicationLoop].includes(workflowMode)"
+          v-if="
+            [
+              WorkflowMode.Application,
+              WorkflowMode.ApplicationLoop,
+              WorkflowMode.Tool,
+              WorkflowMode.ToolLoop,
+            ].includes(workflowMode)
+          "
         >
           <template #label>
             <div class="flex align-center">
diff --git a/ui/src/workflow/nodes/image-generate/index.vue b/ui/src/workflow/nodes/image-generate/index.vue
index f3e3c6aba41..d20f44921e3 100644
--- a/ui/src/workflow/nodes/image-generate/index.vue
+++ b/ui/src/workflow/nodes/image-generate/index.vue
@@ -120,7 +120,14 @@
         <el-form-item
           :label="$t('workflow.nodes.aiChatNode.returnContent.label')"
           @click.prevent
-          v-if="[WorkflowMode.Application, WorkflowMode.ApplicationLoop].includes(workflowMode)"
+          v-if="
+            [
+              WorkflowMode.Application,
+              WorkflowMode.ApplicationLoop,
+              WorkflowMode.Tool,
+              WorkflowMode.ToolLoop,
+            ].includes(workflowMode)
+          "
         >
           <template #label>
             <div class="flex align-center">
diff --git a/ui/src/workflow/nodes/image-to-video/index.vue b/ui/src/workflow/nodes/image-to-video/index.vue
index a8633ab1eca..1c9fd57d2f5 100644
--- a/ui/src/workflow/nodes/image-to-video/index.vue
+++ b/ui/src/workflow/nodes/image-to-video/index.vue
@@ -162,7 +162,14 @@
         <el-form-item
           :label="$t('workflow.nodes.aiChatNode.returnContent.label')"
           @click.prevent
-          v-if="[WorkflowMode.Application, WorkflowMode.ApplicationLoop].includes(workflowMode)"
+          v-if="
+            [
+              WorkflowMode.Application,
+              WorkflowMode.ApplicationLoop,
+              WorkflowMode.Tool,
+              WorkflowMode.ToolLoop,
+            ].includes(workflowMode)
+          "
         >
           <template #label>
             <div class="flex align-center">
diff --git a/ui/src/workflow/nodes/image-understand/index.vue b/ui/src/workflow/nodes/image-understand/index.vue
index 24194522de0..47c21a35e78 100644
--- a/ui/src/workflow/nodes/image-understand/index.vue
+++ b/ui/src/workflow/nodes/image-understand/index.vue
@@ -115,7 +115,14 @@
           />
         </el-form-item>
         <el-form-item
-          v-if="[WorkflowMode.Application, WorkflowMode.ApplicationLoop].includes(workflowMode)"
+          v-if="
+            [
+              WorkflowMode.Application,
+              WorkflowMode.ApplicationLoop,
+              WorkflowMode.Tool,
+              WorkflowMode.ToolLoop,
+            ].includes(workflowMode)
+          "
         >
           <template #label>
             <div class="flex-between">
@@ -165,7 +172,14 @@
         <el-form-item
           :label="$t('workflow.nodes.aiChatNode.returnContent.label')"
           @click.prevent
-          v-if="[WorkflowMode.Application, WorkflowMode.ApplicationLoop].includes(workflowMode)"
+          v-if="
+            [
+              WorkflowMode.Application,
+              WorkflowMode.ApplicationLoop,
+              WorkflowMode.Tool,
+              WorkflowMode.ToolLoop,
+            ].includes(workflowMode)
+          "
         >
           <template #label>
             <div class="flex align-center">
diff --git a/ui/src/workflow/nodes/question-node/index.vue b/ui/src/workflow/nodes/question-node/index.vue
index 859ad1a3abb..c140ba0c0fe 100644
--- a/ui/src/workflow/nodes/question-node/index.vue
+++ b/ui/src/workflow/nodes/question-node/index.vue
@@ -120,7 +120,14 @@
         <el-form-item
           :label="$t('workflow.nodes.aiChatNode.returnContent.label')"
           @click.prevent
-          v-if="[WorkflowMode.Application, WorkflowMode.ApplicationLoop].includes(workflowMode)"
+          v-if="
+            [
+              WorkflowMode.Application,
+              WorkflowMode.ApplicationLoop,
+              WorkflowMode.Tool,
+              WorkflowMode.ToolLoop,
+            ].includes(workflowMode)
+          "
         >
           <template #label>
             <div class="flex align-center">
diff --git a/ui/src/workflow/nodes/reply-node/index.vue b/ui/src/workflow/nodes/reply-node/index.vue
index 6c91ca7e2bf..562fed725e1 100644
--- a/ui/src/workflow/nodes/reply-node/index.vue
+++ b/ui/src/workflow/nodes/reply-node/index.vue
@@ -43,7 +43,14 @@
           />
         </el-form-item>
         <el-form-item
-          v-if="[WorkflowMode.Application, WorkflowMode.ApplicationLoop].includes(workflowMode)"
+          v-if="
+            [
+              WorkflowMode.Application,
+              WorkflowMode.ApplicationLoop,
+              WorkflowMode.Tool,
+              WorkflowMode.ToolLoop,
+            ].includes(workflowMode)
+          "
           :label="$t('workflow.nodes.aiChatNode.returnContent.label')"
           @click.prevent
         >
diff --git a/ui/src/workflow/nodes/speech-to-text-node/index.vue b/ui/src/workflow/nodes/speech-to-text-node/index.vue
index 76f3aca045e..e1e6ac5ff7b 100644
--- a/ui/src/workflow/nodes/speech-to-text-node/index.vue
+++ b/ui/src/workflow/nodes/speech-to-text-node/index.vue
@@ -81,7 +81,14 @@
         <el-form-item
           :label="$t('workflow.nodes.aiChatNode.returnContent.label')"
           @click.prevent
-          v-if="[WorkflowMode.Application, WorkflowMode.ApplicationLoop].includes(workflowMode)"
+          v-if="
+            [
+              WorkflowMode.Application,
+              WorkflowMode.ApplicationLoop,
+              WorkflowMode.Tool,
+              WorkflowMode.ToolLoop,
+            ].includes(workflowMode)
+          "
         >
           <template #label>
             <div class="flex align-center">
diff --git a/ui/src/workflow/nodes/text-to-speech-node/index.vue b/ui/src/workflow/nodes/text-to-speech-node/index.vue
index eb8469ae9d1..1a8ab3cd5b2 100644
--- a/ui/src/workflow/nodes/text-to-speech-node/index.vue
+++ b/ui/src/workflow/nodes/text-to-speech-node/index.vue
@@ -80,7 +80,14 @@
         <el-form-item
           :label="$t('workflow.nodes.aiChatNode.returnContent.label')"
           @click.prevent
-          v-if="[WorkflowMode.Application, WorkflowMode.ApplicationLoop].includes(workflowMode)"
+          v-if="
+            [
+              WorkflowMode.Application,
+              WorkflowMode.ApplicationLoop,
+              WorkflowMode.Tool,
+              WorkflowMode.ToolLoop,
+            ].includes(workflowMode)
+          "
         >
           <template #label>
             <div class="flex align-center">
diff --git a/ui/src/workflow/nodes/text-to-video/index.vue b/ui/src/workflow/nodes/text-to-video/index.vue
index bf4f9579a62..b54c1cad4d8 100644
--- a/ui/src/workflow/nodes/text-to-video/index.vue
+++ b/ui/src/workflow/nodes/text-to-video/index.vue
@@ -120,7 +120,14 @@
         <el-form-item
           :label="$t('workflow.nodes.aiChatNode.returnContent.label')"
           @click.prevent
-          v-if="[WorkflowMode.Application, WorkflowMode.ApplicationLoop].includes(workflowMode)"
+          v-if="
+            [
+              WorkflowMode.Application,
+              WorkflowMode.ApplicationLoop,
+              WorkflowMode.Tool,
+              WorkflowMode.ToolLoop,
+            ].includes(workflowMode)
+          "
         >
           <template #label>
             <div class="flex align-center">
diff --git a/ui/src/workflow/nodes/tool-lib-node/index.vue b/ui/src/workflow/nodes/tool-lib-node/index.vue
index 94aba8e9c51..23a668d2498 100644
--- a/ui/src/workflow/nodes/tool-lib-node/index.vue
+++ b/ui/src/workflow/nodes/tool-lib-node/index.vue
@@ -73,14 +73,19 @@
       <el-form-item
         :label="$t('workflow.nodes.aiChatNode.returnContent.label')"
         @click.prevent
-        v-if="[WorkflowMode.Application, WorkflowMode.ApplicationLoop].includes(workflowMode)"
+        v-if="
+          [
+            WorkflowMode.Application,
+            WorkflowMode.ApplicationLoop,
+            WorkflowMode.Tool,
+            WorkflowMode.ToolLoop,
+          ].includes(workflowMode)
+        "
       >
         <template #label>
           <div class="flex align-center">
             <div class="mr-4">
-              <span>{{
-                $t('workflow.nodes.aiChatNode.returnContent.label')
-              }}</span>
+              <span>{{ $t('workflow.nodes.aiChatNode.returnContent.label') }}</span>
             </div>
             <el-tooltip effect="dark" placement="right" popper-class="max-w-200">
               <template #content>
diff --git a/ui/src/workflow/nodes/tool-node/index.vue b/ui/src/workflow/nodes/tool-node/index.vue
index 830537e2b0a..3758324ae92 100644
--- a/ui/src/workflow/nodes/tool-node/index.vue
+++ b/ui/src/workflow/nodes/tool-node/index.vue
@@ -101,7 +101,14 @@
       <el-form-item
         :label="$t('workflow.nodes.aiChatNode.returnContent.label')"
         @click.prevent
-        v-if="[WorkflowMode.Application, WorkflowMode.ApplicationLoop].includes(workflowMode)"
+        v-if="
+          [
+            WorkflowMode.Application,
+            WorkflowMode.ApplicationLoop,
+            WorkflowMode.Tool,
+            WorkflowMode.ToolLoop,
+          ].includes(workflowMode)
+        "
       >
         <template #label>
           <div class="flex align-center">
diff --git a/ui/src/workflow/nodes/video-understand/index.vue b/ui/src/workflow/nodes/video-understand/index.vue
index 9cfef952e10..101a4b39c99 100644
--- a/ui/src/workflow/nodes/video-understand/index.vue
+++ b/ui/src/workflow/nodes/video-understand/index.vue
@@ -115,7 +115,14 @@
           />
         </el-form-item>
         <el-form-item
-          v-if="[WorkflowMode.Application, WorkflowMode.ApplicationLoop].includes(workflowMode)"
+          v-if="
+            [
+              WorkflowMode.Application,
+              WorkflowMode.ApplicationLoop,
+              WorkflowMode.Tool,
+              WorkflowMode.ToolLoop,
+            ].includes(workflowMode)
+          "
         >
           <template #label>
             <div class="flex-between">
@@ -164,7 +171,14 @@
         <el-form-item
           :label="$t('workflow.nodes.aiChatNode.returnContent.label')"
           @click.prevent
-          v-if="[WorkflowMode.Application, WorkflowMode.ApplicationLoop].includes(workflowMode)"
+          v-if="
+            [
+              WorkflowMode.Application,
+              WorkflowMode.ApplicationLoop,
+              WorkflowMode.Tool,
+              WorkflowMode.ToolLoop,
+            ].includes(workflowMode)
+          "
         >
           <template #label>
             <div class="flex align-center">