diff --git a/src/components/ConnectionLine/index.tsx b/src/components/ConnectionLine/index.tsx index bb9161bb..e3c71f9e 100644 --- a/src/components/ConnectionLine/index.tsx +++ b/src/components/ConnectionLine/index.tsx @@ -79,6 +79,8 @@ export default ({ targetPosition={targetPosition} connectionLineType={connectionLineType} connectionLineStyle={connectionLineStyle} + sourceNode={sourceNode} + sourceHandle={sourceHandle} /> ); diff --git a/src/types/index.ts b/src/types/index.ts index 96247795..8006aa1a 100644 --- a/src/types/index.ts +++ b/src/types/index.ts @@ -303,6 +303,8 @@ export type ConnectionLineComponentProps = { targetPosition?: Position; connectionLineStyle?: CSSProperties; connectionLineType: ConnectionLineType; + sourceNode?: Node; + sourceHandle?: HandleElement; }; export type ConnectionLineComponent = React.ComponentType;