From ae06dc165364d50e96bdfd11535444be64bdd59f Mon Sep 17 00:00:00 2001
From: parshvabhadra <parshva.bhadra@sevn.ai>
Date: Thu, 13 Feb 2025 09:21:28 +0000
Subject: [PATCH] fix: Improve RouterNode edge handle generation in flowSlice

- Update edge generation logic for RouterNode connections
- Ensure targetHandle follows correct node_id.handle_id format
- Handle edge cases with handle naming for router nodes
---
 frontend/src/store/flowSlice.ts | 36 ++++++++++++++++++++++++---------
 1 file changed, 27 insertions(+), 9 deletions(-)

diff --git a/frontend/src/store/flowSlice.ts b/frontend/src/store/flowSlice.ts
index d41b8900..1861f518 100644
--- a/frontend/src/store/flowSlice.ts
+++ b/frontend/src/store/flowSlice.ts
@@ -150,15 +150,33 @@ const flowSlice = createSlice({
                 return nodeObj
             })
 
-            let edges = links.map((link) => ({
-                id: uuidv4(),
-                key: uuidv4(),
-                selected: false,
-                source: link.source_id,
-                target: link.target_id,
-                sourceHandle: link.source_handle || link.source_id,
-                targetHandle: link.target_handle || link.source_id,
-            }))
+            let edges = links.map((link) => {
+                const sourceNode = nodes.find(node => node.id === link.source_id);
+                const isRouterNode = sourceNode?.node_type === 'RouterNode';
+                
+                // For router nodes, ensure targetHandle matches the format node_id.handle_id
+                let targetHandle = link.target_handle || link.source_id;
+                if (isRouterNode) {
+                    // If targetHandle contains a dot, take only what's after the dot
+                    if (targetHandle.includes('.')) {
+                        targetHandle = targetHandle.split('.').pop() || targetHandle;
+                    }
+                    // Ensure it has the correct prefix
+                    if (!targetHandle.startsWith(link.source_id + '.')) {
+                        targetHandle = `${link.source_id}.${targetHandle}`;
+                    }
+                }
+
+                return {
+                    id: uuidv4(),
+                    key: uuidv4(),
+                    selected: false,
+                    source: link.source_id,
+                    target: link.target_id,
+                    sourceHandle: link.source_handle || link.source_id,
+                    targetHandle,
+                }
+            })
             // deduplicate edges
             edges = edges.filter(
                 (edge, index, self) =>