diff --git a/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowRunVisualizerEffect.tsx b/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowRunVisualizerEffect.tsx index 0693a51c50..aa3cfc96f8 100644 --- a/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowRunVisualizerEffect.tsx +++ b/packages/twenty-front/src/modules/workflow/workflow-diagram/components/WorkflowRunVisualizerEffect.tsx @@ -9,6 +9,7 @@ import { useWorkflowVersion } from '@/workflow/hooks/useWorkflowVersion'; import { flowComponentState } from '@/workflow/states/flowComponentState'; import { workflowVisualizerWorkflowIdComponentState } from '@/workflow/states/workflowVisualizerWorkflowIdComponentState'; import { workflowVisualizerWorkflowRunIdComponentState } from '@/workflow/states/workflowVisualizerWorkflowRunIdComponentState'; +import { workflowVisualizerWorkflowVersionIdComponentState } from '@/workflow/states/workflowVisualizerWorkflowVersionIdComponentState'; import { type WorkflowRunState } from '@/workflow/types/Workflow'; import { workflowDiagramComponentState } from '@/workflow/workflow-diagram/states/workflowDiagramComponentState'; import { workflowDiagramStatusComponentState } from '@/workflow/workflow-diagram/states/workflowDiagramStatusComponentState'; @@ -32,11 +33,16 @@ export const WorkflowRunVisualizerEffect = ({ const { getIcon } = useIcons(); const workflowRun = useWorkflowRun({ workflowRunId }); - const workflowVersion = useWorkflowVersion(workflowRun?.workflowVersionId); - const setWorkflowRunId = useSetRecoilComponentState( workflowVisualizerWorkflowRunIdComponentState, ); + + const workflowVersionId = workflowRun?.workflowVersionId; + const workflowVersion = useWorkflowVersion(workflowVersionId); + const setWorkflowVisualizerWorkflowVersionId = useSetRecoilComponentState( + workflowVisualizerWorkflowVersionIdComponentState, + ); + const workflowVisualizerWorkflowIdState = useRecoilComponentCallbackState( workflowVisualizerWorkflowIdComponentState, ); @@ -84,6 +90,14 @@ export const WorkflowRunVisualizerEffect = ({ setWorkflowVisualizerWorkflowId(workflowRun.workflowId); }, [setWorkflowVisualizerWorkflowId, workflowRun]); + useEffect(() => { + if (!isDefined(workflowVersionId)) { + return; + } + + setWorkflowVisualizerWorkflowVersionId(workflowVersionId); + }, [setWorkflowVisualizerWorkflowVersionId, workflowVersionId]); + const handleWorkflowRunDiagramGeneration = useRecoilCallback( ({ snapshot, set }) => ({