@@ -2,13 +2,13 @@ import React, { useCallback, useEffect, useState } from "react";
22import { ArrowHeadType , Elements , getMarkerEnd , ReactFlowProvider } from "react-flow-renderer" ;
33import { Meta , StoryFn } from "@storybook/react" ;
44
5- import { EdgeLabel , EdgeLabelObject , ReactFlow } from "./../../../../../index" ;
6- import { EdgeDefault , EdgeDefaultDataProps as EdgeData } from "./../EdgeDefault" ;
5+ import { EdgeDefaultProps , EdgeLabel , EdgeLabelObject , ReactFlowExtended } from "./../../../../../index" ;
76import { edgeTypes } from "./../../../../cmem/react-flow/configuration/workflow" ;
7+ import { EdgeDefault , EdgeDefaultDataProps as EdgeData } from "./../EdgeDefault" ;
88
99const EdgeDefaultDataProps = ( data : EdgeData ) => {
1010 // this is only a mock to get it as sub element in the table
11- return < > </ > ;
11+ return < > { data . intent } </ > ;
1212} ;
1313
1414export default {
@@ -23,7 +23,7 @@ export default {
2323 } ,
2424} as Meta < typeof EdgeDefault > ;
2525
26- const EdgeDefaultExample = ( args : any ) => {
26+ const EdgeDefaultExample = ( args : EdgeDefaultProps ) => {
2727 const [ reactflowInstance , setReactflowInstance ] = useState ( null ) ;
2828 const [ elements , setElements ] = useState ( [ ] as Elements ) ;
2929
@@ -66,7 +66,7 @@ const EdgeDefaultExample = (args: any) => {
6666
6767 return (
6868 < ReactFlowProvider >
69- < ReactFlow
69+ < ReactFlowExtended
7070 elements = { elements }
7171 style = { { height : "400px" } }
7272 onLoad = { onLoad }
0 commit comments