diff --git a/src/FlowView/helper.tsx b/src/FlowView/helper.tsx index 96b4863..888edb7 100644 --- a/src/FlowView/helper.tsx +++ b/src/FlowView/helper.tsx @@ -165,15 +165,18 @@ export function getRenderEdges(edges: FlowViewEdge[]) { sourceHandle, targetHandle, className, + data, + id = `${source}-${target}`, } = edge; return { - id: `${source}-${target}`, + id, source, target, sourceHandle, targetHandle, type, + data, animated, label, className: getEdgeClsFromSelectType(select) + ' ' + className, diff --git a/src/constants.tsx b/src/constants.tsx index 78b9dc0..29f8fbf 100644 --- a/src/constants.tsx +++ b/src/constants.tsx @@ -72,6 +72,7 @@ export interface FlowViewEdge { label?: string; type?: EdgeType | string; className?: string; + data?: any; } export interface FlowViewProps {