feat: knowledge pipeline (#25360)

Signed-off-by: -LAN- <laipz8200@outlook.com>
Co-authored-by: twwu <twwu@dify.ai>
Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com>
Co-authored-by: jyong <718720800@qq.com>
Co-authored-by: Wu Tianwei <30284043+WTW0313@users.noreply.github.com>
Co-authored-by: QuantumGhost <obelisk.reg+git@gmail.com>
Co-authored-by: lyzno1 <yuanyouhuilyz@gmail.com>
Co-authored-by: quicksand <quicksandzn@gmail.com>
Co-authored-by: Jyong <76649700+JohnJyong@users.noreply.github.com>
Co-authored-by: lyzno1 <92089059+lyzno1@users.noreply.github.com>
Co-authored-by: zxhlyh <jasonapring2015@outlook.com>
Co-authored-by: Yongtao Huang <yongtaoh2022@gmail.com>
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Co-authored-by: Joel <iamjoel007@gmail.com>
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Co-authored-by: nite-knite <nkCoding@gmail.com>
Co-authored-by: Hanqing Zhao <sherry9277@gmail.com>
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
Co-authored-by: Harry <xh001x@hotmail.com>

web/app/components/workflow/nodes/tool/default.ts

@@ -1,27 +1,27 @@
-import { BlockEnum } from '../../types'
-import type { NodeDefault } from '../../types'
+import { genNodeMetaData } from '@/app/components/workflow/utils'
+import { BlockEnum, VarType } from '@/app/components/workflow/types'
+import type { NodeDefault, ToolWithProvider } from '../../types'
 import type { ToolNodeType } from './types'
 import { VarType as VarKindType } from '@/app/components/workflow/nodes/tool/types'
-import { ALL_CHAT_AVAILABLE_BLOCKS, ALL_COMPLETION_AVAILABLE_BLOCKS } from '@/app/components/workflow/blocks'
+import { TOOL_OUTPUT_STRUCT } from '../../constants'
+import { CollectionType } from '@/app/components/tools/types'
+import { canFindTool } from '@/utils'
+import { getMatchedSchemaType } from '../_base/components/variable/use-match-schema-type'
 
 const i18nPrefix = 'workflow.errorMsg'
 
+const metaData = genNodeMetaData({
+  sort: -1,
+  type: BlockEnum.Tool,
+  helpLinkUri: 'tools',
+})
 const nodeDefault: NodeDefault<ToolNodeType> = {
+  metaData,
   defaultValue: {
     tool_parameters: {},
     tool_configurations: {},
     tool_node_version: '2',
   },
-  getAvailablePrevNodes(isChatMode: boolean) {
-    const nodes = isChatMode
-      ? ALL_CHAT_AVAILABLE_BLOCKS
-      : ALL_COMPLETION_AVAILABLE_BLOCKS.filter(type => type !== BlockEnum.End)
-    return nodes
-  },
-  getAvailableNextNodes(isChatMode: boolean) {
-    const nodes = isChatMode ? ALL_CHAT_AVAILABLE_BLOCKS : ALL_COMPLETION_AVAILABLE_BLOCKS
-    return nodes
-  },
   checkValid(payload: ToolNodeType, t: any, moreDataForCheckValid: any) {
     const { toolInputsSchema, toolSettingSchema, language, notAuthed } = moreDataForCheckValid
     let errorMessages = ''
@@ -66,6 +66,65 @@ const nodeDefault: NodeDefault<ToolNodeType> = {
       errorMessage: errorMessages,
     }
   },
+  getOutputVars(payload: ToolNodeType, allPluginInfoList: Record<string, ToolWithProvider[]>, _ragVars: any, { schemaTypeDefinitions } = { schemaTypeDefinitions: [] }) {
+    const { provider_id, provider_type } = payload
+    let currentTools: ToolWithProvider[] = []
+    switch (provider_type) {
+      case CollectionType.builtIn:
+        currentTools = allPluginInfoList.buildInTools ?? []
+        break
+      case CollectionType.custom:
+        currentTools = allPluginInfoList.customTools ?? []
+        break
+      case CollectionType.workflow:
+        currentTools = allPluginInfoList.workflowTools ?? []
+        break
+      case CollectionType.mcp:
+        currentTools = allPluginInfoList.mcpTools ?? []
+        break
+      default:
+        currentTools = []
+    }
+    const currCollection = currentTools.find(item => canFindTool(item.id, provider_id))
+    const currTool = currCollection?.tools.find(tool => tool.name === payload.tool_name)
+    const output_schema = currTool?.output_schema
+    let res: any[] = []
+    if (!output_schema || !output_schema.properties) {
+      res = TOOL_OUTPUT_STRUCT
+    }
+    else {
+      const outputSchema: any[] = []
+      Object.keys(output_schema.properties).forEach((outputKey) => {
+        const output = output_schema.properties[outputKey]
+        const dataType = output.type
+        const schemaType = getMatchedSchemaType(output, schemaTypeDefinitions)
+        let type = dataType === 'array'
+          ? `Array[${output.items?.type ? output.items.type.slice(0, 1).toLocaleLowerCase() + output.items.type.slice(1) : 'Unknown'}]`
+          : `${output.type ? output.type.slice(0, 1).toLocaleLowerCase() + output.type.slice(1) : 'Unknown'}`
+
+        if (type === VarType.object && schemaType === 'file')
+          type = VarType.file
+
+        outputSchema.push({
+          variable: outputKey,
+          type,
+          description: output.description,
+          schemaType,
+          children: output.type === 'object' ? {
+            schema: {
+              type: 'object',
+              properties: output.properties,
+            },
+          } : undefined,
+        })
+      })
+      res = [
+        ...TOOL_OUTPUT_STRUCT,
+        ...outputSchema,
+      ]
+    }
+    return res
+  },
 }
 
 export default nodeDefault