jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix/workflow use nodes hooks (#21822)

Browse source
This commit is contained in:
zxhlyh 2025-07-02 17:48:23 +08:00 • committed by GitHub
commit 68f41bbaa8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 74 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

9
web/app/components/workflow-app/components/workflow-header/features-trigger.tsx
View file

@ -3,7 +3,7 @@ import {
useCallback,
useMemo,
} from 'react'
import { useNodes } from 'reactflow'
import { useStore as useReactflowStore } from 'reactflow'
import { RiApps2AddLine } from '@remixicon/react'
import { useTranslation } from 'react-i18next'
import {
@ -22,7 +22,6 @@ import {
BlockEnum,
InputVarType,
} from '@/app/components/workflow/types'
import type { StartNodeType } from '@/app/components/workflow/nodes/start/types'
import { useToastContext } from '@/app/components/base/toast'
import { usePublishWorkflow, useResetWorkflowVersionHistory } from '@/service/use-workflow'
import type { PublishWorkflowParams } from '@/types/workflow'
@ -42,9 +41,9 @@ const FeaturesTrigger = () => {
const publishedAt = useStore(s => s.publishedAt)
const draftUpdatedAt = useStore(s => s.draftUpdatedAt)
const toolPublished = useStore(s => s.toolPublished)
const nodes = useNodes<StartNodeType>()
const startNode = nodes.find(node => node.data.type === BlockEnum.Start)
const startVariables = startNode?.data.variables
const startVariables = useReactflowStore(
s => s.getNodes().find(node => node.data.type === BlockEnum.Start)?.data.variables,
)
const fileSettings = useFeatures(s => s.features.file)
const variables = useMemo(() => {
const data = startVariables || []