1- import { type Column , type Model } from '@/api/client'
1+ import { type Column } from '@/api/client'
22import {
33 createContext ,
44 useState ,
99import { getNodeMap , hasActiveEdge , hasActiveEdgeConnector } from './help'
1010import { EnumSide } from './types'
1111import { type Node } from 'reactflow'
12- import type { Lineage } from '@/domain/lineage'
1312export interface Connections {
1413 left : string [ ]
1514 right : string [ ]
@@ -20,7 +19,9 @@ export type ActiveNodes = Set<string>
2019export type SelectedNodes = Set < string >
2120export type HighlightedNodes = Record < string , string [ ] >
2221
22+ type Lineage = any
2323type ModelSQLMeshModel = any
24+ type Model = any
2425
2526interface LineageFlow {
2627 lineage : Record < string , Lineage >
@@ -32,7 +33,7 @@ interface LineageFlow {
3233 selectedNodes : SelectedNodes
3334 selectedEdges : any [ ]
3435 models : Record < string , Model >
35- unknownModels : Set < string >
36+ unknownModels : Record < string , Model >
3637 connections : Map < string , Connections >
3738 withConnected : boolean
3839 withColumns : boolean
@@ -83,7 +84,7 @@ export const LineageFlowContext = createContext<LineageFlow>({
8384 mainNode : undefined ,
8485 activeEdges : new Map ( ) ,
8586 activeNodes : new Set ( ) ,
86- models : { } ,
87+ models : new Map ( ) ,
8788 unknownModels : new Set ( ) ,
8889 manuallySelectedColumn : undefined ,
8990 connections : new Map ( ) ,
0 commit comments