mirror of
https://github.com/langgenius/dify.git
synced 2026-07-24 21:18:35 +08:00
73 lines
3.3 KiB
TypeScript
73 lines
3.3 KiB
TypeScript
import { produce } from 'immer'
|
|
import { useHooksStore } from '@/app/components/workflow/hooks-store'
|
|
import { useConversationVarValues, useSysVarValues } from '@/service/use-workflow'
|
|
import { FlowType } from '@/types/common'
|
|
import { useStore } from '../store'
|
|
import { BlockEnum } from '../types'
|
|
|
|
const varsAppendStartNodeKeys = ['query', 'files']
|
|
const useInspectVarsCrud = () => {
|
|
const partOfNodesWithInspectVars = useStore((s) => s.nodesWithInspectVars)
|
|
const configsMap = useHooksStore((s) => s.configsMap)
|
|
const shouldSkipSharedVariableQueries =
|
|
configsMap?.flowType === FlowType.ragPipeline || configsMap?.flowType === FlowType.snippet
|
|
const variableFlowId = shouldSkipSharedVariableQueries ? '' : configsMap?.flowId
|
|
const { data: conversationVars } = useConversationVarValues(configsMap?.flowType, variableFlowId)
|
|
const { data: allSystemVars } = useSysVarValues(configsMap?.flowType, variableFlowId)
|
|
const { varsAppendStartNode, systemVars } = (() => {
|
|
if (allSystemVars?.length === 0) return { varsAppendStartNode: [], systemVars: [] }
|
|
const varsAppendStartNode =
|
|
allSystemVars?.filter(({ name }) => varsAppendStartNodeKeys.includes(name)) || []
|
|
const systemVars =
|
|
allSystemVars?.filter(({ name }) => !varsAppendStartNodeKeys.includes(name)) || []
|
|
return { varsAppendStartNode, systemVars }
|
|
})()
|
|
const nodesWithInspectVars = (() => {
|
|
if (!partOfNodesWithInspectVars || partOfNodesWithInspectVars.length === 0) return []
|
|
|
|
const nodesWithInspectVars = produce(partOfNodesWithInspectVars, (draft) => {
|
|
draft.forEach((nodeWithVars) => {
|
|
if (nodeWithVars.nodeType === BlockEnum.Start)
|
|
nodeWithVars.vars = [...nodeWithVars.vars, ...varsAppendStartNode]
|
|
})
|
|
})
|
|
return nodesWithInspectVars
|
|
})()
|
|
const hasNodeInspectVars = useHooksStore((s) => s.hasNodeInspectVars)
|
|
const hasSetInspectVar = useHooksStore((s) => s.hasSetInspectVar)
|
|
const fetchInspectVarValue = useHooksStore((s) => s.fetchInspectVarValue)
|
|
const editInspectVarValue = useHooksStore((s) => s.editInspectVarValue)
|
|
const renameInspectVarName = useHooksStore((s) => s.renameInspectVarName)
|
|
const appendNodeInspectVars = useHooksStore((s) => s.appendNodeInspectVars)
|
|
const deleteInspectVar = useHooksStore((s) => s.deleteInspectVar)
|
|
const deleteNodeInspectorVars = useHooksStore((s) => s.deleteNodeInspectorVars)
|
|
const deleteAllInspectorVars = useHooksStore((s) => s.deleteAllInspectorVars)
|
|
const isInspectVarEdited = useHooksStore((s) => s.isInspectVarEdited)
|
|
const resetToLastRunVar = useHooksStore((s) => s.resetToLastRunVar)
|
|
const invalidateSysVarValues = useHooksStore((s) => s.invalidateSysVarValues)
|
|
const resetConversationVar = useHooksStore((s) => s.resetConversationVar)
|
|
const invalidateConversationVarValues = useHooksStore((s) => s.invalidateConversationVarValues)
|
|
|
|
return {
|
|
conversationVars: conversationVars || [],
|
|
systemVars: systemVars || [],
|
|
nodesWithInspectVars,
|
|
hasNodeInspectVars,
|
|
hasSetInspectVar,
|
|
fetchInspectVarValue,
|
|
editInspectVarValue,
|
|
renameInspectVarName,
|
|
appendNodeInspectVars,
|
|
deleteInspectVar,
|
|
deleteNodeInspectorVars,
|
|
deleteAllInspectorVars,
|
|
isInspectVarEdited,
|
|
resetToLastRunVar,
|
|
invalidateSysVarValues,
|
|
resetConversationVar,
|
|
invalidateConversationVarValues,
|
|
}
|
|
}
|
|
|
|
export default useInspectVarsCrud
|