From 0c100ac0b1bf6260aee083a664daf98570a3c7f1 Mon Sep 17 00:00:00 2001
From: StyleZhang <jasonapring2015@outlook.com>
Date: Tue, 19 Mar 2024 15:06:47 +0800
Subject: [PATCH] fix node

---
 web/app/components/workflow/hooks/use-edges-interactions.ts | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/web/app/components/workflow/hooks/use-edges-interactions.ts b/web/app/components/workflow/hooks/use-edges-interactions.ts
index 56d67c336e..1d621051a2 100644
--- a/web/app/components/workflow/hooks/use-edges-interactions.ts
+++ b/web/app/components/workflow/hooks/use-edges-interactions.ts
@@ -13,6 +13,7 @@ import type {
   Edge,
   Node,
 } from '../types'
+import { BlockEnum } from '../types'
 import { getNodesConnectedSourceOrTargetHandleIdsMap } from '../utils'
 import { useNodesSyncDraft } from './use-nodes-sync-draft'
 
@@ -168,6 +169,10 @@ export const useEdgesInteractions = () => {
         sourceHandle: 'source',
         target: nodeId,
         targetHandle,
+        data: {
+          sourceType: nodes.find(node => node.id === targetHandle)?.data.type,
+          targetType: BlockEnum.VariableAssigner,
+        },
       }
     })