refactor: use bivariance to normalize node metadata types

This commit is contained in:
zhsama
2026-01-23 06:37:21 +08:00
parent ef8d0f497d
commit 4707a319e5
11 changed files with 157 additions and 67 deletions

View File

@ -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