fix(tabsContext.tsx): add missing import statement for updateEdgesHandleIds function

fix(tabsContext.tsx): update processFlowEdges function to call updateEdgesHandleIds and update the edges in the flow data
refactor(reactflowUtils.ts): update updateEdgesHandleIds function to return the updated edges instead of using setEdges function

src/frontend/src/contexts/tabsContext.tsx

@@ -27,6 +27,7 @@ import {
 import { TabsContextType, TabsState } from "../types/tabs";
 import {
   addVersionToDuplicates,
+  updateEdgesHandleIds,
   updateIds,
   updateTemplate,
 } from "../utils/reactflowUtils";
@@ -149,8 +150,10 @@ export function TabsProvider({ children }: { children: ReactNode }) {
     });
   }
 
-  function processFlowEdges(flow) {
+  function processFlowEdges(flow: FlowType) {
     if (!flow.data || !flow.data.edges) return;
+    const newEdges = updateEdgesHandleIds(flow.data);
+    flow.data.edges = newEdges;
     flow.data.edges.forEach((edge) => {
       edge.className = "";
       edge.style = { stroke: "#555" };

src/frontend/src/types/utils/reactflowUtils.ts

@@ -12,5 +12,4 @@ export type cleanEdgesType = {
 export type updateEdgesHandleIdsType = {
   nodes: NodeType[];
   edges: Edge[];
-  setEdges: (edges: Edge[]) => void;
 };

src/frontend/src/utils/reactflowUtils.ts

@@ -235,8 +235,7 @@ export function addVersionToDuplicates(flow: FlowType, flows: FlowType[]) {
 export function updateEdgesHandleIds({
   edges,
   nodes,
-  setEdges,
-}: updateEdgesHandleIdsType) {
+}: updateEdgesHandleIdsType): Edge[] {
   let newEdges = _.cloneDeep(edges);
   newEdges.forEach((edge) => {
     const sourceNodeId = edge.source;
@@ -268,7 +267,7 @@ export function updateEdgesHandleIds({
     edge.sourceHandle = JSON.stringify(newSource);
     edge.targetHandle = JSON.stringify(newTarget);
   });
-  setEdges(newEdges);
+  return newEdges;
 }
 
 export function getConnectedNodes(