refactor(types): use same structure for nodes and edges

packages/react/src/hooks/useEdges.ts

@@ -11,8 +11,8 @@ const edgesSelector = (state: ReactFlowState) => state.edges;
  * @public
  * @returns An array of edges
  */
-export function useEdges<EdgeData>(): Edge<EdgeData>[] {
-  const edges = useStore(edgesSelector, shallow);
+export function useEdges<EdgeType extends Edge = Edge>(): EdgeType[] {
+  const edges = useStore(edgesSelector, shallow) as EdgeType[];
 
   return edges;
 }

packages/react/src/hooks/useOnInitHandler.ts

@@ -1,15 +1,17 @@
 import { useEffect, useRef } from 'react';
 
 import { useReactFlow } from './useReactFlow';
-import type { OnInit, Node } from '../types';
+import type { OnInit, Node, Edge } from '../types';
 
 /**
  * Hook for calling onInit handler.
  *
  * @internal
  */
-export function useOnInitHandler<NodeType extends Node = Node>(onInit: OnInit<NodeType> | undefined) {
-  const rfInstance = useReactFlow<NodeType>();
+export function useOnInitHandler<NodeType extends Node = Node, EdgeType extends Edge = Edge>(
+  onInit: OnInit<NodeType, EdgeType> | undefined
+) {
+  const rfInstance = useReactFlow<NodeType, EdgeType>();
   const isInitialized = useRef<boolean>(false);
 
   useEffect(() => {