mirror of
https://github.com/langgenius/dify.git
synced 2026-05-06 10:28:10 +08:00
refactor: use bivariance to normalize node metadata types
This commit is contained in:
@ -1,4 +1,5 @@
|
||||
import type { AvailableNodesMetaData } from '@/app/components/workflow/hooks-store/store'
|
||||
import type { CommonNodeType, NodeDefault, NodeDefaultBase } from '@/app/components/workflow/types'
|
||||
import { useMemo } from 'react'
|
||||
import { useTranslation } from 'react-i18next'
|
||||
import { WORKFLOW_COMMON_NODES } from '@/app/components/workflow/constants/node'
|
||||
@ -29,25 +30,47 @@ export const useAvailableNodesMetaData = () => {
|
||||
'/use-dify/knowledge/knowledge-pipeline/knowledge-pipeline-orchestration',
|
||||
), [docLink])
|
||||
|
||||
const availableNodesMetaData = useMemo(() => mergedNodesMetaData.map((node) => {
|
||||
const { metaData } = node
|
||||
const title = t(`blocks.${metaData.type}`, { ns: 'workflow' })
|
||||
const description = t(`blocksAbout.${metaData.type}`, { ns: 'workflow' })
|
||||
return {
|
||||
...node,
|
||||
metaData: {
|
||||
const availableNodesMetaData = useMemo<NodeDefaultBase[]>(() => {
|
||||
const toNodeDefaultBase = (
|
||||
node: NodeDefault<CommonNodeType>,
|
||||
metaData: NodeDefaultBase['metaData'],
|
||||
defaultValue: Partial<CommonNodeType>,
|
||||
): NodeDefaultBase => {
|
||||
return {
|
||||
...node,
|
||||
metaData,
|
||||
defaultValue,
|
||||
checkValid: (payload: CommonNodeType, translator, moreDataForCheckValid) => {
|
||||
// normalize validator signature for shared metadata storage.
|
||||
return node.checkValid(payload, translator, moreDataForCheckValid)
|
||||
},
|
||||
getOutputVars: node.getOutputVars
|
||||
? (payload: CommonNodeType, allPluginInfoList, ragVariables, utils) => {
|
||||
// normalize output var signature for shared metadata storage.
|
||||
return node.getOutputVars!(payload, allPluginInfoList, ragVariables, utils)
|
||||
}
|
||||
: undefined,
|
||||
}
|
||||
}
|
||||
|
||||
return mergedNodesMetaData.map((node) => {
|
||||
// normalize per-node defaults into a shared metadata shape.
|
||||
const typedNode = node as NodeDefault<CommonNodeType>
|
||||
const { metaData } = typedNode
|
||||
const title = t(`blocks.${metaData.type}`, { ns: 'workflow' })
|
||||
const description = t(`blocksAbout.${metaData.type}`, { ns: 'workflow' })
|
||||
return toNodeDefaultBase(typedNode, {
|
||||
...metaData,
|
||||
title,
|
||||
description,
|
||||
helpLinkUri,
|
||||
},
|
||||
defaultValue: {
|
||||
...node.defaultValue,
|
||||
}, {
|
||||
...typedNode.defaultValue,
|
||||
type: metaData.type,
|
||||
title,
|
||||
},
|
||||
}
|
||||
}), [mergedNodesMetaData, t])
|
||||
})
|
||||
})
|
||||
}, [mergedNodesMetaData, t, helpLinkUri])
|
||||
|
||||
const availableNodesMetaDataMap = useMemo(() => availableNodesMetaData.reduce((acc, node) => {
|
||||
acc![node.metaData.type] = node
|
||||
|
||||
Reference in New Issue
Block a user