move lightrag_webui folder to top directory

lightrag_webui/src/stores/graph.ts

@@ -0,0 +1,130 @@
+import { create } from 'zustand'
+import { createSelectors } from '@/lib/utils'
+import { DirectedGraph } from 'graphology'
+
+export type RawNodeType = {
+  id: string
+  labels: string[]
+  properties: Record<string, any>
+
+  size: number
+  x: number
+  y: number
+  color: string
+
+  degree: number
+}
+
+export type RawEdgeType = {
+  id: string
+  source: string
+  target: string
+  type: string
+  properties: Record<string, any>
+
+  dynamicId: string
+}
+
+export class RawGraph {
+  nodes: RawNodeType[] = []
+  edges: RawEdgeType[] = []
+  nodeIdMap: Record<string, number> = {}
+  edgeIdMap: Record<string, number> = {}
+  edgeDynamicIdMap: Record<string, number> = {}
+
+  getNode = (nodeId: string) => {
+    const nodeIndex = this.nodeIdMap[nodeId]
+    if (nodeIndex !== undefined) {
+      return this.nodes[nodeIndex]
+    }
+    return undefined
+  }
+
+  getEdge = (edgeId: string, dynamicId: boolean = true) => {
+    const edgeIndex = dynamicId ? this.edgeDynamicIdMap[edgeId] : this.edgeIdMap[edgeId]
+    if (edgeIndex !== undefined) {
+      return this.edges[edgeIndex]
+    }
+    return undefined
+  }
+
+  buildDynamicMap = () => {
+    this.edgeDynamicIdMap = {}
+    for (let i = 0; i < this.edges.length; i++) {
+      const edge = this.edges[i]
+      this.edgeDynamicIdMap[edge.dynamicId] = i
+    }
+  }
+}
+
+interface GraphState {
+  selectedNode: string | null
+  focusedNode: string | null
+  selectedEdge: string | null
+  focusedEdge: string | null
+
+  rawGraph: RawGraph | null
+  sigmaGraph: DirectedGraph | null
+
+  moveToSelectedNode: boolean
+
+  setSelectedNode: (nodeId: string | null, moveToSelectedNode?: boolean) => void
+  setFocusedNode: (nodeId: string | null) => void
+  setSelectedEdge: (edgeId: string | null) => void
+  setFocusedEdge: (edgeId: string | null) => void
+  clearSelection: () => void
+  reset: () => void
+
+  setMoveToSelectedNode: (moveToSelectedNode: boolean) => void
+
+  setRawGraph: (rawGraph: RawGraph | null) => void
+  setSigmaGraph: (sigmaGraph: DirectedGraph | null) => void
+}
+
+const useGraphStoreBase = create<GraphState>()((set) => ({
+  selectedNode: null,
+  focusedNode: null,
+  selectedEdge: null,
+  focusedEdge: null,
+
+  moveToSelectedNode: false,
+
+  rawGraph: null,
+  sigmaGraph: null,
+
+  setSelectedNode: (nodeId: string | null, moveToSelectedNode?: boolean) =>
+    set({ selectedNode: nodeId, moveToSelectedNode }),
+  setFocusedNode: (nodeId: string | null) => set({ focusedNode: nodeId }),
+  setSelectedEdge: (edgeId: string | null) => set({ selectedEdge: edgeId }),
+  setFocusedEdge: (edgeId: string | null) => set({ focusedEdge: edgeId }),
+  clearSelection: () =>
+    set({
+      selectedNode: null,
+      focusedNode: null,
+      selectedEdge: null,
+      focusedEdge: null
+    }),
+  reset: () =>
+    set({
+      selectedNode: null,
+      focusedNode: null,
+      selectedEdge: null,
+      focusedEdge: null,
+      rawGraph: null,
+      sigmaGraph: null,
+      moveToSelectedNode: false
+    }),
+
+  setRawGraph: (rawGraph: RawGraph | null) =>
+    set({
+      rawGraph
+    }),
+
+  setSigmaGraph: (sigmaGraph: DirectedGraph | null) => set({ sigmaGraph }),
+
+  setMoveToSelectedNode: (moveToSelectedNode?: boolean) => set({ moveToSelectedNode })
+}))
+
+const useGraphStore = createSelectors(useGraphStoreBase)
+
+export { useGraphStore }

lightrag_webui/src/stores/settings.ts

@@ -0,0 +1,88 @@
+import { create } from 'zustand'
+import { persist, createJSONStorage } from 'zustand/middleware'
+import { createSelectors } from '@/lib/utils'
+import { defaultQueryLabel } from '@/lib/constants'
+
+type Theme = 'dark' | 'light' | 'system'
+
+interface SettingsState {
+  theme: Theme
+  setTheme: (theme: Theme) => void
+
+  showPropertyPanel: boolean
+  showNodeSearchBar: boolean
+
+  showNodeLabel: boolean
+  enableNodeDrag: boolean
+
+  showEdgeLabel: boolean
+  enableHideUnselectedEdges: boolean
+  enableEdgeEvents: boolean
+
+  queryLabel: string
+  setQueryLabel: (queryLabel: string) => void
+
+  enableHealthCheck: boolean
+  setEnableHealthCheck: (enable: boolean) => void
+
+  apiKey: string | null
+  setApiKey: (key: string | null) => void
+}
+
+const useSettingsStoreBase = create<SettingsState>()(
+  persist(
+    (set) => ({
+      theme: 'system',
+
+      showPropertyPanel: true,
+      showNodeSearchBar: true,
+
+      showNodeLabel: true,
+      enableNodeDrag: true,
+
+      showEdgeLabel: false,
+      enableHideUnselectedEdges: true,
+      enableEdgeEvents: false,
+
+      queryLabel: defaultQueryLabel,
+      enableHealthCheck: true,
+
+      apiKey: null,
+
+      setTheme: (theme: Theme) => set({ theme }),
+
+      setQueryLabel: (queryLabel: string) =>
+        set({
+          queryLabel
+        }),
+
+      setEnableHealthCheck: (enable: boolean) => set({ enableHealthCheck: enable }),
+
+      setApiKey: (apiKey: string | null) => set({ apiKey })
+    }),
+    {
+      name: 'settings-storage',
+      storage: createJSONStorage(() => localStorage),
+      version: 4,
+      migrate: (state: any, version: number) => {
+        if (version < 2) {
+          state.showEdgeLabel = false
+        }
+        if (version < 3) {
+          state.queryLabel = defaultQueryLabel
+        }
+        if (version < 4) {
+          state.showPropertyPanel = true
+          state.showNodeSearchBar = true
+          state.showNodeLabel = true
+          state.enableHealthCheck = true
+          state.apiKey = null
+        }
+      }
+    }
+  )
+)
+
+const useSettingsStore = createSelectors(useSettingsStoreBase)
+
+export { useSettingsStore, type Theme }

lightrag_webui/src/stores/state.ts

@@ -0,0 +1,59 @@
+import { create } from 'zustand'
+import { createSelectors } from '@/lib/utils'
+import { checkHealth, LightragStatus } from '@/api/lightrag'
+
+interface BackendState {
+  health: boolean
+  message: string | null
+  messageTitle: string | null
+
+  status: LightragStatus | null
+
+  lastCheckTime: number
+
+  check: () => Promise<boolean>
+  clear: () => void
+  setErrorMessage: (message: string, messageTitle: string) => void
+}
+
+const useBackendStateStoreBase = create<BackendState>()((set) => ({
+  health: true,
+  message: null,
+  messageTitle: null,
+  lastCheckTime: Date.now(),
+  status: null,
+
+  check: async () => {
+    const health = await checkHealth()
+    if (health.status === 'healthy') {
+      set({
+        health: true,
+        message: null,
+        messageTitle: null,
+        lastCheckTime: Date.now(),
+        status: health
+      })
+      return true
+    }
+    set({
+      health: false,
+      message: health.message,
+      messageTitle: 'Backend Health Check Error!',
+      lastCheckTime: Date.now(),
+      status: null
+    })
+    return false
+  },
+
+  clear: () => {
+    set({ health: true, message: null, messageTitle: null })
+  },
+
+  setErrorMessage: (message: string, messageTitle: string) => {
+    set({ health: false, message, messageTitle })
+  }
+}))
+
+const useBackendState = createSelectors(useBackendStateStoreBase)
+
+export { useBackendState }