From eb7dc5436456ce7783cab5bf637cb0c322f38fef Mon Sep 17 00:00:00 2001
From: zhangzhanwei <zhanwei.zhang@fit2cloud.com>
Date: Thu, 4 Dec 2025 18:09:54 +0800
Subject: [PATCH] fix: Tool lib return button

---
 ui/src/workflow/nodes/tool-lib-node/index.vue | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/ui/src/workflow/nodes/tool-lib-node/index.vue b/ui/src/workflow/nodes/tool-lib-node/index.vue
index 996c3d2057a..afbdd9465a7 100644
--- a/ui/src/workflow/nodes/tool-lib-node/index.vue
+++ b/ui/src/workflow/nodes/tool-lib-node/index.vue
@@ -73,6 +73,7 @@
       <el-form-item
         :label="$t('workflow.nodes.aiChatNode.returnContent.label')"
         @click.prevent
+        v-if="[WorkflowMode.Application, WorkflowMode.ApplicationLoop].includes(workflowMode)"
       >
         <template #label>
           <div class="flex align-center">
@@ -100,9 +101,11 @@ import { useRoute } from 'vue-router'
 import NodeContainer from '@/workflow/common/NodeContainer.vue'
 import NodeCascader from '@/workflow/common/NodeCascader.vue'
 import type { FormInstance } from 'element-plus'
-import { ref, computed, onMounted } from 'vue'
+import { ref, computed, onMounted, inject } from 'vue'
 import { isLastNode } from '@/workflow/common/data'
 import { loadSharedApi } from '@/utils/dynamics-api/shared-api'
+import { WorkflowMode } from '@/enums/application'
+const workflowMode = (inject('workflowMode') as WorkflowMode) || WorkflowMode.Application
 
 const props = defineProps<{ nodeModel: any }>()