move lightrag_webui folder to top directory

2025-02-13 17:29:51 +08:00
parent c50c581767
commit 85e6989abc
52 changed files with 0 additions and 0 deletions
--- a/lightrag_webui/src/components/FocusOnNode.tsx
+++ b/lightrag_webui/src/components/FocusOnNode.tsx
@@ -0,0 +1,31 @@
+import { useCamera, useSigma } from '@react-sigma/core'
+import { useEffect } from 'react'
+import { useGraphStore } from '@/stores/graph'
+
+/**
+ * Component that highlights a node and centers the camera on it.
+ */
+const FocusOnNode = ({ node, move }: { node: string | null; move?: boolean }) => {
+  const sigma = useSigma()
+  const { gotoNode } = useCamera()
+
+  /**
+   * When the selected item changes, highlighted the node and center the camera on it.
+   */
+  useEffect(() => {
+    if (!node) return
+    sigma.getGraph().setNodeAttribute(node, 'highlighted', true)
+    if (move) {
+      gotoNode(node)
+      useGraphStore.getState().setMoveToSelectedNode(false)
+    }
+
+    return () => {
+      sigma.getGraph().setNodeAttribute(node, 'highlighted', false)
+    }
+  }, [node, move, sigma, gotoNode])
+
+  return null
+}
+
+export default FocusOnNode
--- a/lightrag_webui/src/components/FullScreenControl.tsx
+++ b/lightrag_webui/src/components/FullScreenControl.tsx
@@ -0,0 +1,27 @@
+import { useFullScreen } from '@react-sigma/core'
+import { MaximizeIcon, MinimizeIcon } from 'lucide-react'
+import { controlButtonVariant } from '@/lib/constants'
+import Button from '@/components/ui/Button'
+
+/**
+ * Component that toggles full screen mode.
+ */
+const FullScreenControl = () => {
+  const { isFullScreen, toggle } = useFullScreen()
+
+  return (
+    <>
+      {isFullScreen ? (
+        <Button variant={controlButtonVariant} onClick={toggle} tooltip="Windowed" size="icon">
+          <MinimizeIcon />
+        </Button>
+      ) : (
+        <Button variant={controlButtonVariant} onClick={toggle} tooltip="Full Screen" size="icon">
+          <MaximizeIcon />
+        </Button>
+      )}
+    </>
+  )
+}
+
+export default FullScreenControl
--- a/lightrag_webui/src/components/GraphControl.tsx
+++ b/lightrag_webui/src/components/GraphControl.tsx
@@ -0,0 +1,185 @@
+import { useLoadGraph, useRegisterEvents, useSetSettings, useSigma } from '@react-sigma/core'
+// import { useLayoutCircular } from '@react-sigma/layout-circular'
+import { useLayoutForceAtlas2 } from '@react-sigma/layout-forceatlas2'
+import { useEffect } from 'react'
+
+// import useRandomGraph, { EdgeType, NodeType } from '@/hooks/useRandomGraph'
+import useLightragGraph, { EdgeType, NodeType } from '@/hooks/useLightragGraph'
+import useTheme from '@/hooks/useTheme'
+import * as Constants from '@/lib/constants'
+
+import { useSettingsStore } from '@/stores/settings'
+import { useGraphStore } from '@/stores/graph'
+
+const isButtonPressed = (ev: MouseEvent | TouchEvent) => {
+  if (ev.type.startsWith('mouse')) {
+    if ((ev as MouseEvent).buttons !== 0) {
+      return true
+    }
+  }
+  return false
+}
+
+const GraphControl = ({ disableHoverEffect }: { disableHoverEffect?: boolean }) => {
+  const { lightrageGraph } = useLightragGraph()
+  const sigma = useSigma<NodeType, EdgeType>()
+  const registerEvents = useRegisterEvents<NodeType, EdgeType>()
+  const setSettings = useSetSettings<NodeType, EdgeType>()
+  const loadGraph = useLoadGraph<NodeType, EdgeType>()
+  const { assign: assignLayout } = useLayoutForceAtlas2({
+    iterations: 20
+  })
+
+  const { theme } = useTheme()
+  const hideUnselectedEdges = useSettingsStore.use.enableHideUnselectedEdges()
+  const selectedNode = useGraphStore.use.selectedNode()
+  const focusedNode = useGraphStore.use.focusedNode()
+  const selectedEdge = useGraphStore.use.selectedEdge()
+  const focusedEdge = useGraphStore.use.focusedEdge()
+
+  /**
+   * When component mount
+   * => load the graph
+   */
+  useEffect(() => {
+    // Create & load the graph
+    const graph = lightrageGraph()
+    loadGraph(graph)
+    if (!(graph as any).__force_applied) {
+      assignLayout()
+      Object.assign(graph, { __force_applied: true })
+    }
+
+    const { setFocusedNode, setSelectedNode, setFocusedEdge, setSelectedEdge, clearSelection } =
+      useGraphStore.getState()
+
+    // Register the events
+    registerEvents({
+      enterNode: (event) => {
+        if (!isButtonPressed(event.event.original)) {
+          setFocusedNode(event.node)
+        }
+      },
+      leaveNode: (event) => {
+        if (!isButtonPressed(event.event.original)) {
+          setFocusedNode(null)
+        }
+      },
+      clickNode: (event) => {
+        setSelectedNode(event.node)
+        setSelectedEdge(null)
+      },
+      clickEdge: (event) => {
+        setSelectedEdge(event.edge)
+        setSelectedNode(null)
+      },
+      enterEdge: (event) => {
+        if (!isButtonPressed(event.event.original)) {
+          setFocusedEdge(event.edge)
+        }
+      },
+      leaveEdge: (event) => {
+        if (!isButtonPressed(event.event.original)) {
+          setFocusedEdge(null)
+        }
+      },
+      clickStage: () => clearSelection()
+    })
+  }, [assignLayout, loadGraph, registerEvents, lightrageGraph])
+
+  /**
+   * When component mount or hovered node change
+   * => Setting the sigma reducers
+   */
+  useEffect(() => {
+    const isDarkTheme = theme === 'dark'
+    const labelColor = isDarkTheme ? Constants.labelColorDarkTheme : undefined
+    const edgeColor = isDarkTheme ? Constants.edgeColorDarkTheme : undefined
+
+    setSettings({
+      nodeReducer: (node, data) => {
+        const graph = sigma.getGraph()
+        const newData: NodeType & {
+          labelColor?: string
+          borderColor?: string
+        } = { ...data, highlighted: data.highlighted || false, labelColor }
+
+        if (!disableHoverEffect) {
+          newData.highlighted = false
+          const _focusedNode = focusedNode || selectedNode
+          const _focusedEdge = focusedEdge || selectedEdge
+
+          if (_focusedNode) {
+            if (node === _focusedNode || graph.neighbors(_focusedNode).includes(node)) {
+              newData.highlighted = true
+              if (node === selectedNode) {
+                newData.borderColor = Constants.nodeBorderColorSelected
+              }
+            }
+          } else if (_focusedEdge) {
+            if (graph.extremities(_focusedEdge).includes(node)) {
+              newData.highlighted = true
+              newData.size = 3
+            }
+          } else {
+            return newData
+          }
+
+          if (newData.highlighted) {
+            if (isDarkTheme) {
+              newData.labelColor = Constants.LabelColorHighlightedDarkTheme
+            }
+          } else {
+            newData.color = Constants.nodeColorDisabled
+          }
+        }
+        return newData
+      },
+      edgeReducer: (edge, data) => {
+        const graph = sigma.getGraph()
+        const newData = { ...data, hidden: false, labelColor, color: edgeColor }
+
+        if (!disableHoverEffect) {
+          const _focusedNode = focusedNode || selectedNode
+
+          if (_focusedNode) {
+            if (hideUnselectedEdges) {
+              if (!graph.extremities(edge).includes(_focusedNode)) {
+                newData.hidden = true
+              }
+            } else {
+              if (graph.extremities(edge).includes(_focusedNode)) {
+                newData.color = Constants.edgeColorHighlighted
+              }
+            }
+          } else {
+            if (focusedEdge || selectedEdge) {
+              if (edge === selectedEdge) {
+                newData.color = Constants.edgeColorSelected
+              } else if (edge === focusedEdge) {
+                newData.color = Constants.edgeColorHighlighted
+              } else if (hideUnselectedEdges) {
+                newData.hidden = true
+              }
+            }
+          }
+        }
+        return newData
+      }
+    })
+  }, [
+    selectedNode,
+    focusedNode,
+    selectedEdge,
+    focusedEdge,
+    setSettings,
+    sigma,
+    disableHoverEffect,
+    theme,
+    hideUnselectedEdges
+  ])
+
+  return null
+}
+
+export default GraphControl
--- a/lightrag_webui/src/components/GraphLabels.tsx
+++ b/lightrag_webui/src/components/GraphLabels.tsx
@@ -0,0 +1,87 @@
+import { useCallback, useState } from 'react'
+import { AsyncSelect } from '@/components/ui/AsyncSelect'
+import { getGraphLabels } from '@/api/lightrag'
+import { useSettingsStore } from '@/stores/settings'
+import MiniSearch from 'minisearch'
+
+const GraphLabels = () => {
+  const label = useSettingsStore.use.queryLabel()
+  const [labels, setLabels] = useState<{
+    labels: string[]
+    searchEngine: MiniSearch | null
+  }>({
+    labels: [],
+    searchEngine: null
+  })
+  const [fetched, setFetched] = useState(false)
+
+  const fetchData = useCallback(
+    async (query?: string): Promise<string[]> => {
+      let _labels = labels.labels
+      let _searchEngine = labels.searchEngine
+
+      if (!fetched || !_searchEngine) {
+        _labels = ['*'].concat(await getGraphLabels())
+
+        // Ensure query label exists
+        if (!_labels.includes(useSettingsStore.getState().queryLabel)) {
+          useSettingsStore.getState().setQueryLabel(_labels[0])
+        }
+
+        // Create search engine
+        _searchEngine = new MiniSearch({
+          idField: 'id',
+          fields: ['value'],
+          searchOptions: {
+            prefix: true,
+            fuzzy: 0.2,
+            boost: {
+              label: 2
+            }
+          }
+        })
+
+        // Add documents
+        const documents = _labels.map((str, index) => ({ id: index, value: str }))
+        _searchEngine.addAll(documents)
+
+        setLabels({
+          labels: _labels,
+          searchEngine: _searchEngine
+        })
+        setFetched(true)
+      }
+      if (!query) {
+        return _labels
+      }
+
+      // Search labels
+      return _searchEngine.search(query).map((result) => _labels[result.id])
+    },
+    [labels, fetched, setLabels, setFetched]
+  )
+
+  const setQueryLabel = useCallback((label: string) => {
+    useSettingsStore.getState().setQueryLabel(label)
+  }, [])
+
+  return (
+    <AsyncSelect<string>
+      className="ml-2"
+      triggerClassName="max-h-8"
+      searchInputClassName="max-h-8"
+      triggerTooltip="Select query label"
+      fetcher={fetchData}
+      renderOption={(item) => <div>{item}</div>}
+      getOptionValue={(item) => item}
+      getDisplayValue={(item) => <div>{item}</div>}
+      notFound={<div className="py-6 text-center text-sm">No labels found</div>}
+      label="Label"
+      placeholder="Search labels..."
+      value={label !== null ? label : ''}
+      onChange={setQueryLabel}
+    />
+  )
+}
+
+export default GraphLabels
--- a/lightrag_webui/src/components/GraphSearch.tsx
+++ b/lightrag_webui/src/components/GraphSearch.tsx
@@ -0,0 +1,133 @@
+import { FC, useCallback, useMemo } from 'react'
+import {
+  EdgeById,
+  NodeById,
+  GraphSearchInputProps,
+  GraphSearchContextProviderProps
+} from '@react-sigma/graph-search'
+import { AsyncSearch } from '@/components/ui/AsyncSearch'
+import { searchResultLimit } from '@/lib/constants'
+import { useGraphStore } from '@/stores/graph'
+import MiniSearch from 'minisearch'
+
+interface OptionItem {
+  id: string
+  type: 'nodes' | 'edges' | 'message'
+  message?: string
+}
+
+function OptionComponent(item: OptionItem) {
+  return (
+    <div>
+      {item.type === 'nodes' && <NodeById id={item.id} />}
+      {item.type === 'edges' && <EdgeById id={item.id} />}
+      {item.type === 'message' && <div>{item.message}</div>}
+    </div>
+  )
+}
+
+const messageId = '__message_item'
+const lastGraph: any = {
+  graph: null,
+  searchEngine: null
+}
+
+/**
+ * Component thats display the search input.
+ */
+export const GraphSearchInput = ({
+  onChange,
+  onFocus,
+  value
+}: {
+  onChange: GraphSearchInputProps['onChange']
+  onFocus?: GraphSearchInputProps['onFocus']
+  value?: GraphSearchInputProps['value']
+}) => {
+  const graph = useGraphStore.use.sigmaGraph()
+
+  const search = useMemo(() => {
+    if (lastGraph.graph == graph) {
+      return lastGraph.searchEngine
+    }
+    if (!graph || graph.nodes().length == 0) return
+
+    lastGraph.graph = graph
+
+    const searchEngine = new MiniSearch({
+      idField: 'id',
+      fields: ['label'],
+      searchOptions: {
+        prefix: true,
+        fuzzy: 0.2,
+        boost: {
+          label: 2
+        }
+      }
+    })
+
+    // Add documents
+    const documents = graph.nodes().map((id: string) => ({
+      id: id,
+      label: graph.getNodeAttribute(id, 'label')
+    }))
+    searchEngine.addAll(documents)
+
+    lastGraph.searchEngine = searchEngine
+    return searchEngine
+  }, [graph])
+
+  /**
+   * Loading the options while the user is typing.
+   */
+  const loadOptions = useCallback(
+    async (query?: string): Promise<OptionItem[]> => {
+      if (onFocus) onFocus(null)
+      if (!query || !search) return []
+      const result: OptionItem[] = search.search(query).map((result) => ({
+        id: result.id,
+        type: 'nodes'
+      }))
+
+      // prettier-ignore
+      return result.length <= searchResultLimit
+        ? result
+        : [
+          ...result.slice(0, searchResultLimit),
+          {
+            type: 'message',
+            id: messageId,
+            message: `And ${result.length - searchResultLimit} others`
+          }
+        ]
+    },
+    [search, onFocus]
+  )
+
+  return (
+    <AsyncSearch
+      className="bg-background/60 w-24 rounded-xl border-1 opacity-60 backdrop-blur-lg transition-all hover:w-fit hover:opacity-100"
+      fetcher={loadOptions}
+      renderOption={OptionComponent}
+      getOptionValue={(item) => item.id}
+      value={value && value.type !== 'message' ? value.id : null}
+      onChange={(id) => {
+        if (id !== messageId) onChange(id ? { id, type: 'nodes' } : null)
+      }}
+      onFocus={(id) => {
+        if (id !== messageId && onFocus) onFocus(id ? { id, type: 'nodes' } : null)
+      }}
+      label={'item'}
+      placeholder="Search nodes..."
+    />
+  )
+}
+
+/**
+ * Component that display the search.
+ */
+const GraphSearch: FC<GraphSearchInputProps & GraphSearchContextProviderProps> = ({ ...props }) => {
+  return <GraphSearchInput {...props} />
+}
+
+export default GraphSearch
--- a/lightrag_webui/src/components/LayoutsControl.tsx
+++ b/lightrag_webui/src/components/LayoutsControl.tsx
@@ -0,0 +1,179 @@
+import { useSigma } from '@react-sigma/core'
+import { animateNodes } from 'sigma/utils'
+import { useLayoutCirclepack } from '@react-sigma/layout-circlepack'
+import { useLayoutCircular } from '@react-sigma/layout-circular'
+import { LayoutHook, LayoutWorkerHook, WorkerLayoutControlProps } from '@react-sigma/layout-core'
+import { useLayoutForce, useWorkerLayoutForce } from '@react-sigma/layout-force'
+import { useLayoutForceAtlas2, useWorkerLayoutForceAtlas2 } from '@react-sigma/layout-forceatlas2'
+import { useLayoutNoverlap, useWorkerLayoutNoverlap } from '@react-sigma/layout-noverlap'
+import { useLayoutRandom } from '@react-sigma/layout-random'
+import { useCallback, useMemo, useState, useEffect } from 'react'
+
+import Button from '@/components/ui/Button'
+import { Popover, PopoverContent, PopoverTrigger } from '@/components/ui/Popover'
+import { Command, CommandGroup, CommandItem, CommandList } from '@/components/ui/Command'
+import { controlButtonVariant } from '@/lib/constants'
+
+import { GripIcon, PlayIcon, PauseIcon } from 'lucide-react'
+
+type LayoutName =
+  | 'Circular'
+  | 'Circlepack'
+  | 'Random'
+  | 'Noverlaps'
+  | 'Force Directed'
+  | 'Force Atlas'
+
+const WorkerLayoutControl = ({ layout, autoRunFor }: WorkerLayoutControlProps) => {
+  const sigma = useSigma()
+  const { stop, start, isRunning } = layout
+
+  /**
+   * Init component when Sigma or component settings change.
+   */
+  useEffect(() => {
+    if (!sigma) {
+      return
+    }
+
+    // we run the algo
+    let timeout: number | null = null
+    if (autoRunFor !== undefined && autoRunFor > -1 && sigma.getGraph().order > 0) {
+      start()
+      // set a timeout to stop it
+      timeout =
+        autoRunFor > 0
+          ? window.setTimeout(() => { stop() }, autoRunFor) // prettier-ignore
+          : null
+    }
+
+    //cleaning
+    return () => {
+      stop()
+      if (timeout) {
+        clearTimeout(timeout)
+      }
+    }
+  }, [autoRunFor, start, stop, sigma])
+
+  return (
+    <Button
+      size="icon"
+      onClick={() => (isRunning ? stop() : start())}
+      tooltip={isRunning ? 'Stop the layout animation' : 'Start the layout animation'}
+      variant={controlButtonVariant}
+    >
+      {isRunning ? <PauseIcon /> : <PlayIcon />}
+    </Button>
+  )
+}
+
+/**
+ * Component that controls the layout of the graph.
+ */
+const LayoutsControl = () => {
+  const sigma = useSigma()
+  const [layout, setLayout] = useState<LayoutName>('Circular')
+  const [opened, setOpened] = useState<boolean>(false)
+
+  const layoutCircular = useLayoutCircular()
+  const layoutCirclepack = useLayoutCirclepack()
+  const layoutRandom = useLayoutRandom()
+  const layoutNoverlap = useLayoutNoverlap({ settings: { margin: 1 } })
+  const layoutForce = useLayoutForce({ maxIterations: 20 })
+  const layoutForceAtlas2 = useLayoutForceAtlas2({ iterations: 20 })
+  const workerNoverlap = useWorkerLayoutNoverlap()
+  const workerForce = useWorkerLayoutForce()
+  const workerForceAtlas2 = useWorkerLayoutForceAtlas2()
+
+  const layouts = useMemo(() => {
+    return {
+      Circular: {
+        layout: layoutCircular
+      },
+      Circlepack: {
+        layout: layoutCirclepack
+      },
+      Random: {
+        layout: layoutRandom
+      },
+      Noverlaps: {
+        layout: layoutNoverlap,
+        worker: workerNoverlap
+      },
+      'Force Directed': {
+        layout: layoutForce,
+        worker: workerForce
+      },
+      'Force Atlas': {
+        layout: layoutForceAtlas2,
+        worker: workerForceAtlas2
+      }
+    } as { [key: string]: { layout: LayoutHook; worker?: LayoutWorkerHook } }
+  }, [
+    layoutCirclepack,
+    layoutCircular,
+    layoutForce,
+    layoutForceAtlas2,
+    layoutNoverlap,
+    layoutRandom,
+    workerForce,
+    workerNoverlap,
+    workerForceAtlas2
+  ])
+
+  const runLayout = useCallback(
+    (newLayout: LayoutName) => {
+      console.debug(newLayout)
+      const { positions } = layouts[newLayout].layout
+      animateNodes(sigma.getGraph(), positions(), { duration: 500 })
+      setLayout(newLayout)
+    },
+    [layouts, sigma]
+  )
+
+  return (
+    <>
+      <div>
+        {layouts[layout] && 'worker' in layouts[layout] && (
+          <WorkerLayoutControl layout={layouts[layout].worker!} />
+        )}
+      </div>
+      <div>
+        <Popover open={opened} onOpenChange={setOpened}>
+          <PopoverTrigger asChild>
+            <Button
+              size="icon"
+              variant={controlButtonVariant}
+              onClick={() => setOpened((e: boolean) => !e)}
+              tooltip="Layout Graph"
+            >
+              <GripIcon />
+            </Button>
+          </PopoverTrigger>
+          <PopoverContent side="right" align="center" className="p-1">
+            <Command>
+              <CommandList>
+                <CommandGroup>
+                  {Object.keys(layouts).map((name) => (
+                    <CommandItem
+                      onSelect={() => {
+                        runLayout(name as LayoutName)
+                      }}
+                      key={name}
+                      className="cursor-pointer text-xs"
+                    >
+                      {name}
+                    </CommandItem>
+                  ))}
+                </CommandGroup>
+              </CommandList>
+            </Command>
+          </PopoverContent>
+        </Popover>
+      </div>
+    </>
+  )
+}
+
+export default LayoutsControl
--- a/lightrag_webui/src/components/MessageAlert.tsx
+++ b/lightrag_webui/src/components/MessageAlert.tsx
@@ -0,0 +1,55 @@
+import { Alert, AlertDescription, AlertTitle } from '@/components/ui/Alert'
+import { useBackendState } from '@/stores/state'
+import { useEffect, useState } from 'react'
+import { cn } from '@/lib/utils'
+
+// import Button from '@/components/ui/Button'
+// import { controlButtonVariant } from '@/lib/constants'
+
+import { AlertCircle } from 'lucide-react'
+
+const MessageAlert = () => {
+  const health = useBackendState.use.health()
+  const message = useBackendState.use.message()
+  const messageTitle = useBackendState.use.messageTitle()
+  const [isMounted, setIsMounted] = useState(false)
+
+  useEffect(() => {
+    setTimeout(() => {
+      setIsMounted(true)
+    }, 50)
+  }, [])
+
+  return (
+    <Alert
+      variant={health ? 'default' : 'destructive'}
+      className={cn(
+        'bg-background/90 absolute top-2 left-1/2 flex w-auto -translate-x-1/2 transform items-center gap-4 shadow-md backdrop-blur-lg transition-all duration-500 ease-in-out',
+        isMounted ? 'translate-y-0 opacity-100' : '-translate-y-20 opacity-0'
+      )}
+    >
+      {!health && (
+        <div>
+          <AlertCircle className="size-4" />
+        </div>
+      )}
+      <div>
+        <AlertTitle className="font-bold">{messageTitle}</AlertTitle>
+        <AlertDescription>{message}</AlertDescription>
+      </div>
+      {/* <div className="flex">
+        <div className="flex-auto" />
+        <Button
+          size="sm"
+          variant={controlButtonVariant}
+          className="text-primary max-h-8 border !p-2 text-xs"
+          onClick={() => useBackendState.getState().clear()}
+        >
+          Close
+        </Button>
+      </div> */}
+    </Alert>
+  )
+}
+
+export default MessageAlert
--- a/lightrag_webui/src/components/PropertiesView.tsx
+++ b/lightrag_webui/src/components/PropertiesView.tsx
@@ -0,0 +1,231 @@
+import { useEffect, useState } from 'react'
+import { useGraphStore, RawNodeType, RawEdgeType } from '@/stores/graph'
+import Text from '@/components/ui/Text'
+import useLightragGraph from '@/hooks/useLightragGraph'
+
+/**
+ * Component that view properties of elements in graph.
+ */
+const PropertiesView = () => {
+  const { getNode, getEdge } = useLightragGraph()
+  const selectedNode = useGraphStore.use.selectedNode()
+  const focusedNode = useGraphStore.use.focusedNode()
+  const selectedEdge = useGraphStore.use.selectedEdge()
+  const focusedEdge = useGraphStore.use.focusedEdge()
+
+  const [currentElement, setCurrentElement] = useState<NodeType | EdgeType | null>(null)
+  const [currentType, setCurrentType] = useState<'node' | 'edge' | null>(null)
+
+  useEffect(() => {
+    let type: 'node' | 'edge' | null = null
+    let element: RawNodeType | RawEdgeType | null = null
+    if (focusedNode) {
+      type = 'node'
+      element = getNode(focusedNode)
+    } else if (selectedNode) {
+      type = 'node'
+      element = getNode(selectedNode)
+    } else if (focusedEdge) {
+      type = 'edge'
+      element = getEdge(focusedEdge, true)
+    } else if (selectedEdge) {
+      type = 'edge'
+      element = getEdge(selectedEdge, true)
+    }
+
+    if (element) {
+      if (type == 'node') {
+        setCurrentElement(refineNodeProperties(element as any))
+      } else {
+        setCurrentElement(refineEdgeProperties(element as any))
+      }
+      setCurrentType(type)
+    } else {
+      setCurrentElement(null)
+      setCurrentType(null)
+    }
+  }, [
+    focusedNode,
+    selectedNode,
+    focusedEdge,
+    selectedEdge,
+    setCurrentElement,
+    setCurrentType,
+    getNode,
+    getEdge
+  ])
+
+  if (!currentElement) {
+    return <></>
+  }
+  return (
+    <div className="bg-background/80 max-w-xs rounded-lg border-2 p-2 text-xs backdrop-blur-lg">
+      {currentType == 'node' ? (
+        <NodePropertiesView node={currentElement as any} />
+      ) : (
+        <EdgePropertiesView edge={currentElement as any} />
+      )}
+    </div>
+  )
+}
+
+type NodeType = RawNodeType & {
+  relationships: {
+    type: string
+    id: string
+    label: string
+  }[]
+}
+
+type EdgeType = RawEdgeType & {
+  sourceNode?: RawNodeType
+  targetNode?: RawNodeType
+}
+
+const refineNodeProperties = (node: RawNodeType): NodeType => {
+  const state = useGraphStore.getState()
+  const relationships = []
+
+  if (state.sigmaGraph && state.rawGraph) {
+    for (const edgeId of state.sigmaGraph.edges(node.id)) {
+      const edge = state.rawGraph.getEdge(edgeId, true)
+      if (edge) {
+        const isTarget = node.id === edge.source
+        const neighbourId = isTarget ? edge.target : edge.source
+        const neighbour = state.rawGraph.getNode(neighbourId)
+        if (neighbour) {
+          relationships.push({
+            type: isTarget ? 'Target' : 'Source',
+            id: neighbourId,
+            label: neighbour.labels.join(', ')
+          })
+        }
+      }
+    }
+  }
+  return {
+    ...node,
+    relationships
+  }
+}
+
+const refineEdgeProperties = (edge: RawEdgeType): EdgeType => {
+  const state = useGraphStore.getState()
+  const sourceNode = state.rawGraph?.getNode(edge.source)
+  const targetNode = state.rawGraph?.getNode(edge.target)
+  return {
+    ...edge,
+    sourceNode,
+    targetNode
+  }
+}
+
+const PropertyRow = ({
+  name,
+  value,
+  onClick,
+  tooltip
+}: {
+  name: string
+  value: any
+  onClick?: () => void
+  tooltip?: string
+}) => {
+  return (
+    <div className="flex items-center gap-2">
+      <label className="text-primary/60 tracking-wide">{name}</label>:
+      <Text
+        className="hover:bg-primary/20 rounded p-1 text-ellipsis"
+        tooltipClassName="max-w-80"
+        text={value}
+        tooltip={tooltip || value}
+        side="left"
+        onClick={onClick}
+      />
+    </div>
+  )
+}
+
+const NodePropertiesView = ({ node }: { node: NodeType }) => {
+  return (
+    <div className="flex flex-col gap-2">
+      <label className="text-md pl-1 font-bold tracking-wide text-sky-300">Node</label>
+      <div className="bg-primary/5 max-h-96 overflow-auto rounded p-1">
+        <PropertyRow name={'Id'} value={node.id} />
+        <PropertyRow
+          name={'Labels'}
+          value={node.labels.join(', ')}
+          onClick={() => {
+            useGraphStore.getState().setSelectedNode(node.id, true)
+          }}
+        />
+        <PropertyRow name={'Degree'} value={node.degree} />
+      </div>
+      <label className="text-md pl-1 font-bold tracking-wide text-yellow-400/90">Properties</label>
+      <div className="bg-primary/5 max-h-96 overflow-auto rounded p-1">
+        {Object.keys(node.properties)
+          .sort()
+          .map((name) => {
+            return <PropertyRow key={name} name={name} value={node.properties[name]} />
+          })}
+      </div>
+      {node.relationships.length > 0 && (
+        <>
+          <label className="text-md pl-1 font-bold tracking-wide text-teal-600/90">
+            Relationships
+          </label>
+          <div className="bg-primary/5 max-h-96 overflow-auto rounded p-1">
+            {node.relationships.map(({ type, id, label }) => {
+              return (
+                <PropertyRow
+                  key={id}
+                  name={type}
+                  value={label}
+                  onClick={() => {
+                    useGraphStore.getState().setSelectedNode(id, true)
+                  }}
+                />
+              )
+            })}
+          </div>
+        </>
+      )}
+    </div>
+  )
+}
+
+const EdgePropertiesView = ({ edge }: { edge: EdgeType }) => {
+  return (
+    <div className="flex flex-col gap-2">
+      <label className="text-md pl-1 font-bold tracking-wide text-teal-600">Relationship</label>
+      <div className="bg-primary/5 max-h-96 overflow-auto rounded p-1">
+        <PropertyRow name={'Id'} value={edge.id} />
+        <PropertyRow name={'Type'} value={edge.type} />
+        <PropertyRow
+          name={'Source'}
+          value={edge.sourceNode ? edge.sourceNode.labels.join(', ') : edge.source}
+          onClick={() => {
+            useGraphStore.getState().setSelectedNode(edge.source, true)
+          }}
+        />
+        <PropertyRow
+          name={'Target'}
+          value={edge.targetNode ? edge.targetNode.labels.join(', ') : edge.target}
+          onClick={() => {
+            useGraphStore.getState().setSelectedNode(edge.target, true)
+          }}
+        />
+      </div>
+      <label className="text-md pl-1 font-bold tracking-wide text-yellow-400/90">Properties</label>
+      <div className="bg-primary/5 max-h-96 overflow-auto rounded p-1">
+        {Object.keys(edge.properties)
+          .sort()
+          .map((name) => {
+            return <PropertyRow key={name} name={name} value={edge.properties[name]} />
+          })}
+      </div>
+    </div>
+  )
+}
+
+export default PropertiesView
--- a/lightrag_webui/src/components/Settings.tsx
+++ b/lightrag_webui/src/components/Settings.tsx
@@ -0,0 +1,211 @@
+import { Popover, PopoverContent, PopoverTrigger } from '@/components/ui/Popover'
+import { Checkbox } from '@/components/ui/Checkbox'
+import Button from '@/components/ui/Button'
+import Separator from '@/components/ui/Separator'
+import Input from '@/components/ui/Input'
+import { useState, useCallback, useEffect } from 'react'
+import { controlButtonVariant } from '@/lib/constants'
+import { useSettingsStore } from '@/stores/settings'
+import { useBackendState } from '@/stores/state'
+
+import { SettingsIcon } from 'lucide-react'
+
+/**
+ * Component that displays a checkbox with a label.
+ */
+const LabeledCheckBox = ({
+  checked,
+  onCheckedChange,
+  label
+}: {
+  checked: boolean
+  onCheckedChange: () => void
+  label: string
+}) => {
+  return (
+    <div className="flex items-center gap-2">
+      <Checkbox checked={checked} onCheckedChange={onCheckedChange} />
+      <label
+        htmlFor="terms"
+        className="text-sm leading-none font-medium peer-disabled:cursor-not-allowed peer-disabled:opacity-70"
+      >
+        {label}
+      </label>
+    </div>
+  )
+}
+
+/**
+ * Component that displays a popover with settings options.
+ */
+export default function Settings() {
+  const [opened, setOpened] = useState<boolean>(false)
+  const [tempApiKey, setTempApiKey] = useState<string>('') // 用于临时存储输入的API Key
+
+  const showPropertyPanel = useSettingsStore.use.showPropertyPanel()
+  const showNodeSearchBar = useSettingsStore.use.showNodeSearchBar()
+  const showNodeLabel = useSettingsStore.use.showNodeLabel()
+
+  const enableEdgeEvents = useSettingsStore.use.enableEdgeEvents()
+  const enableNodeDrag = useSettingsStore.use.enableNodeDrag()
+  const enableHideUnselectedEdges = useSettingsStore.use.enableHideUnselectedEdges()
+  const showEdgeLabel = useSettingsStore.use.showEdgeLabel()
+
+  const enableHealthCheck = useSettingsStore.use.enableHealthCheck()
+  const apiKey = useSettingsStore.use.apiKey()
+
+  useEffect(() => {
+    setTempApiKey(apiKey || '')
+  }, [apiKey, opened])
+
+  const setEnableNodeDrag = useCallback(
+    () => useSettingsStore.setState((pre) => ({ enableNodeDrag: !pre.enableNodeDrag })),
+    []
+  )
+  const setEnableEdgeEvents = useCallback(
+    () => useSettingsStore.setState((pre) => ({ enableEdgeEvents: !pre.enableEdgeEvents })),
+    []
+  )
+  const setEnableHideUnselectedEdges = useCallback(
+    () =>
+      useSettingsStore.setState((pre) => ({
+        enableHideUnselectedEdges: !pre.enableHideUnselectedEdges
+      })),
+    []
+  )
+  const setShowEdgeLabel = useCallback(
+    () =>
+      useSettingsStore.setState((pre) => ({
+        showEdgeLabel: !pre.showEdgeLabel
+      })),
+    []
+  )
+
+  //
+  const setShowPropertyPanel = useCallback(
+    () => useSettingsStore.setState((pre) => ({ showPropertyPanel: !pre.showPropertyPanel })),
+    []
+  )
+
+  const setShowNodeSearchBar = useCallback(
+    () => useSettingsStore.setState((pre) => ({ showNodeSearchBar: !pre.showNodeSearchBar })),
+    []
+  )
+
+  const setShowNodeLabel = useCallback(
+    () => useSettingsStore.setState((pre) => ({ showNodeLabel: !pre.showNodeLabel })),
+    []
+  )
+
+  const setEnableHealthCheck = useCallback(
+    () => useSettingsStore.setState((pre) => ({ enableHealthCheck: !pre.enableHealthCheck })),
+    []
+  )
+
+  const setApiKey = useCallback(async () => {
+    useSettingsStore.setState({ apiKey: tempApiKey || null })
+    await useBackendState.getState().check()
+    setOpened(false)
+  }, [tempApiKey])
+
+  const handleTempApiKeyChange = useCallback(
+    (e: React.ChangeEvent<HTMLInputElement>) => {
+      setTempApiKey(e.target.value)
+    },
+    [setTempApiKey]
+  )
+
+  return (
+    <Popover open={opened} onOpenChange={setOpened}>
+      <PopoverTrigger asChild>
+        <Button variant={controlButtonVariant} tooltip="Settings" size="icon">
+          <SettingsIcon />
+        </Button>
+      </PopoverTrigger>
+      <PopoverContent
+        side="right"
+        align="start"
+        className="mb-2 p-2"
+        onCloseAutoFocus={(e) => e.preventDefault()}
+      >
+        <div className="flex flex-col gap-2">
+          <LabeledCheckBox
+            checked={showPropertyPanel}
+            onCheckedChange={setShowPropertyPanel}
+            label="Show Property Panel"
+          />
+          <LabeledCheckBox
+            checked={showNodeSearchBar}
+            onCheckedChange={setShowNodeSearchBar}
+            label="Show Search Bar"
+          />
+
+          <Separator />
+
+          <LabeledCheckBox
+            checked={showNodeLabel}
+            onCheckedChange={setShowNodeLabel}
+            label="Show Node Label"
+          />
+          <LabeledCheckBox
+            checked={enableNodeDrag}
+            onCheckedChange={setEnableNodeDrag}
+            label="Node Draggable"
+          />
+
+          <Separator />
+
+          <LabeledCheckBox
+            checked={showEdgeLabel}
+            onCheckedChange={setShowEdgeLabel}
+            label="Show Edge Label"
+          />
+          <LabeledCheckBox
+            checked={enableHideUnselectedEdges}
+            onCheckedChange={setEnableHideUnselectedEdges}
+            label="Hide Unselected Edges"
+          />
+          <LabeledCheckBox
+            checked={enableEdgeEvents}
+            onCheckedChange={setEnableEdgeEvents}
+            label="Edge Events"
+          />
+
+          <Separator />
+
+          <LabeledCheckBox
+            checked={enableHealthCheck}
+            onCheckedChange={setEnableHealthCheck}
+            label="Health Check"
+          />
+
+          <Separator />
+
+          <div className="flex flex-col gap-2">
+            <label className="text-sm font-medium">API Key</label>
+            <form className="flex h-6 gap-2" onSubmit={(e) => e.preventDefault()}>
+              <div className="w-0 flex-1">
+                <Input
+                  type="password"
+                  value={tempApiKey}
+                  onChange={handleTempApiKeyChange}
+                  placeholder="Enter your API key"
+                  className="max-h-full w-full min-w-0"
+                  autoComplete="off"
+                />
+              </div>
+              <Button
+                onClick={setApiKey}
+                variant="outline"
+                size="sm"
+                className="max-h-full shrink-0"
+              >
+                Save
+              </Button>
+            </form>
+          </div>
+        </div>
+      </PopoverContent>
+    </Popover>
+  )
+}
--- a/lightrag_webui/src/components/StatusCard.tsx
+++ b/lightrag_webui/src/components/StatusCard.tsx
@@ -0,0 +1,65 @@
+import { LightragStatus } from '@/api/lightrag'
+
+const StatusCard = ({ status }: { status: LightragStatus | null }) => {
+  if (!status) {
+    return <div className="text-muted-foreground text-sm">Status information unavailable</div>
+  }
+
+  return (
+    <div className="min-w-[300px] space-y-3 text-sm">
+      <div className="space-y-1">
+        <h4 className="font-medium">Storage Info</h4>
+        <div className="text-muted-foreground grid grid-cols-2 gap-1">
+          <span>Working Directory:</span>
+          <span className="truncate">{status.working_directory}</span>
+          <span>Input Directory:</span>
+          <span className="truncate">{status.input_directory}</span>
+          <span>Indexed Files:</span>
+          <span>{status.indexed_files_count}</span>
+        </div>
+      </div>
+
+      <div className="space-y-1">
+        <h4 className="font-medium">LLM Configuration</h4>
+        <div className="text-muted-foreground grid grid-cols-2 gap-1">
+          <span>LLM Binding:</span>
+          <span>{status.configuration.llm_binding}</span>
+          <span>LLM Binding Host:</span>
+          <span>{status.configuration.llm_binding_host}</span>
+          <span>LLM Model:</span>
+          <span>{status.configuration.llm_model}</span>
+          <span>Max Tokens:</span>
+          <span>{status.configuration.max_tokens}</span>
+        </div>
+      </div>
+
+      <div className="space-y-1">
+        <h4 className="font-medium">Embedding Configuration</h4>
+        <div className="text-muted-foreground grid grid-cols-2 gap-1">
+          <span>Embedding Binding:</span>
+          <span>{status.configuration.embedding_binding}</span>
+          <span>Embedding Binding Host:</span>
+          <span>{status.configuration.embedding_binding_host}</span>
+          <span>Embedding Model:</span>
+          <span>{status.configuration.embedding_model}</span>
+        </div>
+      </div>
+
+      <div className="space-y-1">
+        <h4 className="font-medium">Storage Configuration</h4>
+        <div className="text-muted-foreground grid grid-cols-2 gap-1">
+          <span>KV Storage:</span>
+          <span>{status.configuration.kv_storage}</span>
+          <span>Doc Status Storage:</span>
+          <span>{status.configuration.doc_status_storage}</span>
+          <span>Graph Storage:</span>
+          <span>{status.configuration.graph_storage}</span>
+          <span>Vector Storage:</span>
+          <span>{status.configuration.vector_storage}</span>
+        </div>
+      </div>
+    </div>
+  )
+}
+
+export default StatusCard
--- a/lightrag_webui/src/components/StatusIndicator.tsx
+++ b/lightrag_webui/src/components/StatusIndicator.tsx
@@ -0,0 +1,48 @@
+import { cn } from '@/lib/utils'
+import { useBackendState } from '@/stores/state'
+import { useEffect, useState } from 'react'
+import { Popover, PopoverContent, PopoverTrigger } from '@/components/ui/Popover'
+import StatusCard from '@/components/StatusCard'
+
+const StatusIndicator = () => {
+  const health = useBackendState.use.health()
+  const lastCheckTime = useBackendState.use.lastCheckTime()
+  const status = useBackendState.use.status()
+  const [animate, setAnimate] = useState(false)
+
+  // listen to health change
+  useEffect(() => {
+    setAnimate(true)
+    const timer = setTimeout(() => setAnimate(false), 300)
+    return () => clearTimeout(timer)
+  }, [lastCheckTime])
+
+  return (
+    <div className="fixed right-4 bottom-4 flex items-center gap-2 opacity-80 select-none">
+      <Popover>
+        <PopoverTrigger asChild>
+          <div className="flex cursor-help items-center gap-2">
+            <div
+              className={cn(
+                'h-3 w-3 rounded-full transition-all duration-300',
+                'shadow-[0_0_8px_rgba(0,0,0,0.2)]',
+                health ? 'bg-green-500' : 'bg-red-500',
+                animate && 'scale-125',
+                animate && health && 'shadow-[0_0_12px_rgba(34,197,94,0.4)]',
+                animate && !health && 'shadow-[0_0_12px_rgba(239,68,68,0.4)]'
+              )}
+            />
+            <span className="text-muted-foreground text-xs">
+              {health ? 'Connected' : 'Disconnected'}
+            </span>
+          </div>
+        </PopoverTrigger>
+        <PopoverContent className="w-auto" side="top" align="end">
+          <StatusCard status={status} />
+        </PopoverContent>
+      </Popover>
+    </div>
+  )
+}
+
+export default StatusIndicator
--- a/lightrag_webui/src/components/ThemeProvider.tsx
+++ b/lightrag_webui/src/components/ThemeProvider.tsx
@@ -0,0 +1,57 @@
+import { createContext, useEffect, useState } from 'react'
+import { Theme, useSettingsStore } from '@/stores/settings'
+
+type ThemeProviderProps = {
+  children: React.ReactNode
+}
+
+type ThemeProviderState = {
+  theme: Theme
+  setTheme: (theme: Theme) => void
+}
+
+const initialState: ThemeProviderState = {
+  theme: 'system',
+  setTheme: () => null
+}
+
+const ThemeProviderContext = createContext<ThemeProviderState>(initialState)
+
+/**
+ * Component that provides the theme state and setter function to its children.
+ */
+export default function ThemeProvider({ children, ...props }: ThemeProviderProps) {
+  const [theme, setTheme] = useState<Theme>(useSettingsStore.getState().theme)
+
+  useEffect(() => {
+    const root = window.document.documentElement
+    root.classList.remove('light', 'dark')
+
+    if (theme === 'system') {
+      const systemTheme = window.matchMedia('(prefers-color-scheme: dark)').matches
+        ? 'dark'
+        : 'light'
+      root.classList.add(systemTheme)
+      setTheme(systemTheme)
+      return
+    }
+
+    root.classList.add(theme)
+  }, [theme])
+
+  const value = {
+    theme,
+    setTheme: (theme: Theme) => {
+      useSettingsStore.getState().setTheme(theme)
+      setTheme(theme)
+    }
+  }
+
+  return (
+    <ThemeProviderContext.Provider {...props} value={value}>
+      {children}
+    </ThemeProviderContext.Provider>
+  )
+}
+
+export { ThemeProviderContext }
--- a/lightrag_webui/src/components/ThemeToggle.tsx
+++ b/lightrag_webui/src/components/ThemeToggle.tsx
@@ -0,0 +1,37 @@
+import Button from '@/components/ui/Button'
+import useTheme from '@/hooks/useTheme'
+import { MoonIcon, SunIcon } from 'lucide-react'
+import { useCallback } from 'react'
+import { controlButtonVariant } from '@/lib/constants'
+
+/**
+ * Component that toggles the theme between light and dark.
+ */
+export default function ThemeToggle() {
+  const { theme, setTheme } = useTheme()
+  const setLight = useCallback(() => setTheme('light'), [setTheme])
+  const setDark = useCallback(() => setTheme('dark'), [setTheme])
+
+  if (theme === 'dark') {
+    return (
+      <Button
+        onClick={setLight}
+        variant={controlButtonVariant}
+        tooltip="Switch to light theme"
+        size="icon"
+      >
+        <MoonIcon />
+      </Button>
+    )
+  }
+  return (
+    <Button
+      onClick={setDark}
+      variant={controlButtonVariant}
+      tooltip="Switch to dark theme"
+      size="icon"
+    >
+      <SunIcon />
+    </Button>
+  )
+}
--- a/lightrag_webui/src/components/ZoomControl.tsx
+++ b/lightrag_webui/src/components/ZoomControl.tsx
@@ -0,0 +1,37 @@
+import { useCamera } from '@react-sigma/core'
+import { useCallback } from 'react'
+import Button from '@/components/ui/Button'
+import { ZoomInIcon, ZoomOutIcon, FullscreenIcon } from 'lucide-react'
+import { controlButtonVariant } from '@/lib/constants'
+
+/**
+ * Component that provides zoom controls for the graph viewer.
+ */
+const ZoomControl = () => {
+  const { zoomIn, zoomOut, reset } = useCamera({ duration: 200, factor: 1.5 })
+
+  const handleZoomIn = useCallback(() => zoomIn(), [zoomIn])
+  const handleZoomOut = useCallback(() => zoomOut(), [zoomOut])
+  const handleResetZoom = useCallback(() => reset(), [reset])
+
+  return (
+    <>
+      <Button variant={controlButtonVariant} onClick={handleZoomIn} tooltip="Zoom In" size="icon">
+        <ZoomInIcon />
+      </Button>
+      <Button variant={controlButtonVariant} onClick={handleZoomOut} tooltip="Zoom Out" size="icon">
+        <ZoomOutIcon />
+      </Button>
+      <Button
+        variant={controlButtonVariant}
+        onClick={handleResetZoom}
+        tooltip="Reset Zoom"
+        size="icon"
+      >
+        <FullscreenIcon />
+      </Button>
+    </>
+  )
+}
+
+export default ZoomControl
--- a/lightrag_webui/src/components/ui/Alert.tsx
+++ b/lightrag_webui/src/components/ui/Alert.tsx
@@ -0,0 +1,49 @@
+import * as React from 'react'
+import { cva, type VariantProps } from 'class-variance-authority'
+
+import { cn } from '@/lib/utils'
+
+const alertVariants = cva(
+  'relative w-full rounded-lg border px-4 py-3 text-sm [&>svg+div]:translate-y-[-3px] [&>svg]:absolute [&>svg]:left-4 [&>svg]:top-4 [&>svg]:text-foreground [&>svg~*]:pl-7',
+  {
+    variants: {
+      variant: {
+        default: 'bg-background text-foreground',
+        destructive:
+          'border-destructive/50 text-destructive dark:border-destructive [&>svg]:text-destructive'
+      }
+    },
+    defaultVariants: {
+      variant: 'default'
+    }
+  }
+)
+
+const Alert = React.forwardRef<
+  HTMLDivElement,
+  React.HTMLAttributes<HTMLDivElement> & VariantProps<typeof alertVariants>
+>(({ className, variant, ...props }, ref) => (
+  <div ref={ref} role="alert" className={cn(alertVariants({ variant }), className)} {...props} />
+))
+Alert.displayName = 'Alert'
+
+const AlertTitle = React.forwardRef<HTMLParagraphElement, React.HTMLAttributes<HTMLHeadingElement>>(
+  ({ className, ...props }, ref) => (
+    <h5
+      ref={ref}
+      className={cn('mb-1 leading-none font-medium tracking-tight', className)}
+      {...props}
+    />
+  )
+)
+AlertTitle.displayName = 'AlertTitle'
+
+const AlertDescription = React.forwardRef<
+  HTMLParagraphElement,
+  React.HTMLAttributes<HTMLParagraphElement>
+>(({ className, ...props }, ref) => (
+  <div ref={ref} className={cn('text-sm [&_p]:leading-relaxed', className)} {...props} />
+))
+AlertDescription.displayName = 'AlertDescription'
+
+export { Alert, AlertTitle, AlertDescription }
--- a/lightrag_webui/src/components/ui/AsyncSearch.tsx
+++ b/lightrag_webui/src/components/ui/AsyncSearch.tsx
@@ -0,0 +1,243 @@
+import { useState, useEffect, useCallback } from 'react'
+import { Loader2 } from 'lucide-react'
+import { useDebounce } from '@/hooks/useDebounce'
+
+import { cn } from '@/lib/utils'
+import {
+  Command,
+  CommandEmpty,
+  CommandGroup,
+  CommandInput,
+  CommandItem,
+  CommandList
+} from '@/components/ui/Command'
+
+export interface Option {
+  value: string
+  label: string
+  disabled?: boolean
+  description?: string
+  icon?: React.ReactNode
+}
+
+export interface AsyncSearchProps<T> {
+  /** Async function to fetch options */
+  fetcher: (query?: string) => Promise<T[]>
+  /** Preload all data ahead of time */
+  preload?: boolean
+  /** Function to filter options */
+  filterFn?: (option: T, query: string) => boolean
+  /** Function to render each option */
+  renderOption: (option: T) => React.ReactNode
+  /** Function to get the value from an option */
+  getOptionValue: (option: T) => string
+  /** Custom not found message */
+  notFound?: React.ReactNode
+  /** Custom loading skeleton */
+  loadingSkeleton?: React.ReactNode
+  /** Currently selected value */
+  value: string | null
+  /** Callback when selection changes */
+  onChange: (value: string) => void
+  /** Callback when focus changes */
+  onFocus: (value: string) => void
+  /** Label for the select field */
+  label: string
+  /** Placeholder text when no selection */
+  placeholder?: string
+  /** Disable the entire select */
+  disabled?: boolean
+  /** Custom width for the popover */
+  width?: string | number
+  /** Custom class names */
+  className?: string
+  /** Custom trigger button class names */
+  triggerClassName?: string
+  /** Custom no results message */
+  noResultsMessage?: string
+  /** Allow clearing the selection */
+  clearable?: boolean
+}
+
+export function AsyncSearch<T>({
+  fetcher,
+  preload,
+  filterFn,
+  renderOption,
+  getOptionValue,
+  notFound,
+  loadingSkeleton,
+  label,
+  placeholder = 'Select...',
+  value,
+  onChange,
+  onFocus,
+  disabled = false,
+  className,
+  noResultsMessage
+}: AsyncSearchProps<T>) {
+  const [mounted, setMounted] = useState(false)
+  const [open, setOpen] = useState(false)
+  const [options, setOptions] = useState<T[]>([])
+  const [loading, setLoading] = useState(false)
+  const [error, setError] = useState<string | null>(null)
+  const [selectedValue, setSelectedValue] = useState(value)
+  const [focusedValue, setFocusedValue] = useState<string | null>(null)
+  const [searchTerm, setSearchTerm] = useState('')
+  const debouncedSearchTerm = useDebounce(searchTerm, preload ? 0 : 150)
+  const [originalOptions, setOriginalOptions] = useState<T[]>([])
+
+  useEffect(() => {
+    setMounted(true)
+    setSelectedValue(value)
+  }, [value])
+
+  // Effect for initial fetch
+  useEffect(() => {
+    const initializeOptions = async () => {
+      try {
+        setLoading(true)
+        setError(null)
+        // If we have a value, use it for the initial search
+        const data = value !== null ? await fetcher(value) : []
+        setOriginalOptions(data)
+        setOptions(data)
+      } catch (err) {
+        setError(err instanceof Error ? err.message : 'Failed to fetch options')
+      } finally {
+        setLoading(false)
+      }
+    }
+
+    if (!mounted) {
+      initializeOptions()
+    }
+  }, [mounted, fetcher, value])
+
+  useEffect(() => {
+    const fetchOptions = async () => {
+      try {
+        setLoading(true)
+        setError(null)
+        const data = await fetcher(debouncedSearchTerm)
+        setOriginalOptions(data)
+        setOptions(data)
+      } catch (err) {
+        setError(err instanceof Error ? err.message : 'Failed to fetch options')
+      } finally {
+        setLoading(false)
+      }
+    }
+
+    if (!mounted) {
+      fetchOptions()
+    } else if (!preload) {
+      fetchOptions()
+    } else if (preload) {
+      if (debouncedSearchTerm) {
+        setOptions(
+          originalOptions.filter((option) =>
+            filterFn ? filterFn(option, debouncedSearchTerm) : true
+          )
+        )
+      } else {
+        setOptions(originalOptions)
+      }
+    }
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+  }, [fetcher, debouncedSearchTerm, mounted, preload, filterFn])
+
+  const handleSelect = useCallback(
+    (currentValue: string) => {
+      if (currentValue !== selectedValue) {
+        setSelectedValue(currentValue)
+        onChange(currentValue)
+      }
+      setOpen(false)
+    },
+    [selectedValue, setSelectedValue, setOpen, onChange]
+  )
+
+  const handleFocus = useCallback(
+    (currentValue: string) => {
+      if (currentValue !== focusedValue) {
+        setFocusedValue(currentValue)
+        onFocus(currentValue)
+      }
+    },
+    [focusedValue, setFocusedValue, onFocus]
+  )
+
+  return (
+    <div
+      className={cn(disabled && 'cursor-not-allowed opacity-50', className)}
+      onFocus={() => {
+        setOpen(true)
+      }}
+      onBlur={() => setOpen(false)}
+    >
+      <Command shouldFilter={false} className="bg-transparent">
+        <div>
+          <CommandInput
+            placeholder={placeholder}
+            value={searchTerm}
+            className="max-h-8"
+            onValueChange={(value) => {
+              setSearchTerm(value)
+              if (value && !open) setOpen(true)
+            }}
+          />
+          {loading && options.length > 0 && (
+            <div className="absolute top-1/2 right-2 flex -translate-y-1/2 transform items-center">
+              <Loader2 className="h-4 w-4 animate-spin" />
+            </div>
+          )}
+        </div>
+        <CommandList className="max-h-auto" hidden={!open || debouncedSearchTerm.length === 0}>
+          {error && <div className="text-destructive p-4 text-center">{error}</div>}
+          {loading && options.length === 0 && (loadingSkeleton || <DefaultLoadingSkeleton />)}
+          {!loading &&
+            !error &&
+            options.length === 0 &&
+            (notFound || (
+              <CommandEmpty>{noResultsMessage ?? `No ${label.toLowerCase()} found.`}</CommandEmpty>
+            ))}
+          <CommandGroup>
+            {options.map((option, idx) => (
+              <>
+                <CommandItem
+                  key={getOptionValue(option) + `${idx}`}
+                  value={getOptionValue(option)}
+                  onSelect={handleSelect}
+                  onMouseEnter={() => handleFocus(getOptionValue(option))}
+                  className="truncate"
+                >
+                  {renderOption(option)}
+                </CommandItem>
+                {idx !== options.length - 1 && (
+                  <div key={idx} className="bg-foreground/10 h-[1px]" />
+                )}
+              </>
+            ))}
+          </CommandGroup>
+        </CommandList>
+      </Command>
+    </div>
+  )
+}
+
+function DefaultLoadingSkeleton() {
+  return (
+    <CommandGroup>
+      <CommandItem disabled>
+        <div className="flex w-full items-center gap-2">
+          <div className="bg-muted h-6 w-6 animate-pulse rounded-full" />
+          <div className="flex flex-1 flex-col gap-1">
+            <div className="bg-muted h-4 w-24 animate-pulse rounded" />
+            <div className="bg-muted h-3 w-16 animate-pulse rounded" />
+          </div>
+        </div>
+      </CommandItem>
+    </CommandGroup>
+  )
+}
--- a/lightrag_webui/src/components/ui/AsyncSelect.tsx
+++ b/lightrag_webui/src/components/ui/AsyncSelect.tsx
@@ -0,0 +1,268 @@
+import { useState, useEffect, useCallback } from 'react'
+import { Check, ChevronsUpDown, Loader2 } from 'lucide-react'
+import { useDebounce } from '@/hooks/useDebounce'
+
+import { cn } from '@/lib/utils'
+import Button from '@/components/ui/Button'
+import {
+  Command,
+  CommandEmpty,
+  CommandGroup,
+  CommandInput,
+  CommandItem,
+  CommandList
+} from '@/components/ui/Command'
+import { Popover, PopoverContent, PopoverTrigger } from '@/components/ui/Popover'
+
+export interface Option {
+  value: string
+  label: string
+  disabled?: boolean
+  description?: string
+  icon?: React.ReactNode
+}
+
+export interface AsyncSelectProps<T> {
+  /** Async function to fetch options */
+  fetcher: (query?: string) => Promise<T[]>
+  /** Preload all data ahead of time */
+  preload?: boolean
+  /** Function to filter options */
+  filterFn?: (option: T, query: string) => boolean
+  /** Function to render each option */
+  renderOption: (option: T) => React.ReactNode
+  /** Function to get the value from an option */
+  getOptionValue: (option: T) => string
+  /** Function to get the display value for the selected option */
+  getDisplayValue: (option: T) => React.ReactNode
+  /** Custom not found message */
+  notFound?: React.ReactNode
+  /** Custom loading skeleton */
+  loadingSkeleton?: React.ReactNode
+  /** Currently selected value */
+  value: string
+  /** Callback when selection changes */
+  onChange: (value: string) => void
+  /** Label for the select field */
+  label: string
+  /** Placeholder text when no selection */
+  placeholder?: string
+  /** Disable the entire select */
+  disabled?: boolean
+  /** Custom width for the popover *
+  width?: string | number
+  /** Custom class names */
+  className?: string
+  /** Custom trigger button class names */
+  triggerClassName?: string
+  /** Custom search input class names */
+  searchInputClassName?: string
+  /** Custom no results message */
+  noResultsMessage?: string
+  /** Custom trigger tooltip */
+  triggerTooltip?: string
+  /** Allow clearing the selection */
+  clearable?: boolean
+}
+
+export function AsyncSelect<T>({
+  fetcher,
+  preload,
+  filterFn,
+  renderOption,
+  getOptionValue,
+  getDisplayValue,
+  notFound,
+  loadingSkeleton,
+  label,
+  placeholder = 'Select...',
+  value,
+  onChange,
+  disabled = false,
+  className,
+  triggerClassName,
+  searchInputClassName,
+  noResultsMessage,
+  triggerTooltip,
+  clearable = true
+}: AsyncSelectProps<T>) {
+  const [mounted, setMounted] = useState(false)
+  const [open, setOpen] = useState(false)
+  const [options, setOptions] = useState<T[]>([])
+  const [loading, setLoading] = useState(false)
+  const [error, setError] = useState<string | null>(null)
+  const [selectedValue, setSelectedValue] = useState(value)
+  const [selectedOption, setSelectedOption] = useState<T | null>(null)
+  const [searchTerm, setSearchTerm] = useState('')
+  const debouncedSearchTerm = useDebounce(searchTerm, preload ? 0 : 150)
+  const [originalOptions, setOriginalOptions] = useState<T[]>([])
+
+  useEffect(() => {
+    setMounted(true)
+    setSelectedValue(value)
+  }, [value])
+
+  // Initialize selectedOption when options are loaded and value exists
+  useEffect(() => {
+    if (value && options.length > 0) {
+      const option = options.find((opt) => getOptionValue(opt) === value)
+      if (option) {
+        setSelectedOption(option)
+      }
+    }
+  }, [value, options, getOptionValue])
+
+  // Effect for initial fetch
+  useEffect(() => {
+    const initializeOptions = async () => {
+      try {
+        setLoading(true)
+        setError(null)
+        // If we have a value, use it for the initial search
+        const data = await fetcher(value)
+        setOriginalOptions(data)
+        setOptions(data)
+      } catch (err) {
+        setError(err instanceof Error ? err.message : 'Failed to fetch options')
+      } finally {
+        setLoading(false)
+      }
+    }
+
+    if (!mounted) {
+      initializeOptions()
+    }
+  }, [mounted, fetcher, value])
+
+  useEffect(() => {
+    const fetchOptions = async () => {
+      try {
+        setLoading(true)
+        setError(null)
+        const data = await fetcher(debouncedSearchTerm)
+        setOriginalOptions(data)
+        setOptions(data)
+      } catch (err) {
+        setError(err instanceof Error ? err.message : 'Failed to fetch options')
+      } finally {
+        setLoading(false)
+      }
+    }
+
+    if (!mounted) {
+      fetchOptions()
+    } else if (!preload) {
+      fetchOptions()
+    } else if (preload) {
+      if (debouncedSearchTerm) {
+        setOptions(
+          originalOptions.filter((option) =>
+            filterFn ? filterFn(option, debouncedSearchTerm) : true
+          )
+        )
+      } else {
+        setOptions(originalOptions)
+      }
+    }
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+  }, [fetcher, debouncedSearchTerm, mounted, preload, filterFn])
+
+  const handleSelect = useCallback(
+    (currentValue: string) => {
+      const newValue = clearable && currentValue === selectedValue ? '' : currentValue
+      setSelectedValue(newValue)
+      setSelectedOption(options.find((option) => getOptionValue(option) === newValue) || null)
+      onChange(newValue)
+      setOpen(false)
+    },
+    [selectedValue, onChange, clearable, options, getOptionValue]
+  )
+
+  return (
+    <Popover open={open} onOpenChange={setOpen}>
+      <PopoverTrigger asChild>
+        <Button
+          variant="outline"
+          role="combobox"
+          aria-expanded={open}
+          className={cn(
+            'justify-between',
+            disabled && 'cursor-not-allowed opacity-50',
+            triggerClassName
+          )}
+          disabled={disabled}
+          tooltip={triggerTooltip}
+          side="bottom"
+        >
+          {selectedOption ? getDisplayValue(selectedOption) : placeholder}
+          <ChevronsUpDown className="opacity-50" size={10} />
+        </Button>
+      </PopoverTrigger>
+      <PopoverContent className={cn('p-0', className)} onCloseAutoFocus={(e) => e.preventDefault()}>
+        <Command shouldFilter={false}>
+          <div className="relative w-full border-b">
+            <CommandInput
+              placeholder={`Search ${label.toLowerCase()}...`}
+              value={searchTerm}
+              onValueChange={(value) => {
+                setSearchTerm(value)
+              }}
+              className={searchInputClassName}
+            />
+            {loading && options.length > 0 && (
+              <div className="absolute top-1/2 right-2 flex -translate-y-1/2 transform items-center">
+                <Loader2 className="h-4 w-4 animate-spin" />
+              </div>
+            )}
+          </div>
+          <CommandList>
+            {error && <div className="text-destructive p-4 text-center">{error}</div>}
+            {loading && options.length === 0 && (loadingSkeleton || <DefaultLoadingSkeleton />)}
+            {!loading &&
+              !error &&
+              options.length === 0 &&
+              (notFound || (
+                <CommandEmpty>
+                  {noResultsMessage ?? `No ${label.toLowerCase()} found.`}
+                </CommandEmpty>
+              ))}
+            <CommandGroup>
+              {options.map((option) => (
+                <CommandItem
+                  key={getOptionValue(option)}
+                  value={getOptionValue(option)}
+                  onSelect={handleSelect}
+                  className="truncate"
+                >
+                  {renderOption(option)}
+                  <Check
+                    className={cn(
+                      'ml-auto h-3 w-3',
+                      selectedValue === getOptionValue(option) ? 'opacity-100' : 'opacity-0'
+                    )}
+                  />
+                </CommandItem>
+              ))}
+            </CommandGroup>
+          </CommandList>
+        </Command>
+      </PopoverContent>
+    </Popover>
+  )
+}
+
+function DefaultLoadingSkeleton() {
+  return (
+    <CommandGroup>
+      <CommandItem disabled>
+        <div className="flex w-full items-center gap-2">
+          <div className="bg-muted h-6 w-6 animate-pulse rounded-full" />
+          <div className="flex flex-1 flex-col gap-1">
+            <div className="bg-muted h-4 w-24 animate-pulse rounded" />
+            <div className="bg-muted h-3 w-16 animate-pulse rounded" />
+          </div>
+        </div>
+      </CommandItem>
+    </CommandGroup>
+  )
+}
--- a/lightrag_webui/src/components/ui/Button.tsx
+++ b/lightrag_webui/src/components/ui/Button.tsx
@@ -0,0 +1,77 @@
+import * as React from 'react'
+import { Slot } from '@radix-ui/react-slot'
+import { cva, type VariantProps } from 'class-variance-authority'
+import { Tooltip, TooltipContent, TooltipProvider, TooltipTrigger } from '@/components/ui/Tooltip'
+import { cn } from '@/lib/utils'
+
+const buttonVariants = cva(
+  'inline-flex items-center justify-center gap-2 whitespace-nowrap rounded-md text-sm font-medium ring-offset-background transition-colors focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-ring focus-visible:ring-offset-2 disabled:pointer-events-none disabled:opacity-50 [&_svg]:pointer-events-none [&_svg]:size-4 [&_svg]:shrink-0',
+  {
+    variants: {
+      variant: {
+        default: 'bg-primary text-primary-foreground hover:bg-primary/90',
+        destructive: 'bg-destructive text-destructive-foreground hover:bg-destructive/90',
+        outline: 'border border-input bg-background hover:bg-accent hover:text-accent-foreground',
+        secondary: 'bg-secondary text-secondary-foreground hover:bg-secondary/80',
+        ghost: 'hover:bg-accent hover:text-accent-foreground',
+        link: 'text-primary underline-offset-4 hover:underline'
+      },
+      size: {
+        default: 'h-10 px-4 py-2',
+        sm: 'h-9 rounded-md px-3',
+        lg: 'h-11 rounded-md px-8',
+        icon: 'size-8'
+      }
+    },
+    defaultVariants: {
+      variant: 'default',
+      size: 'default'
+    }
+  }
+)
+
+interface ButtonProps
+  extends React.ButtonHTMLAttributes<HTMLButtonElement>,
+    VariantProps<typeof buttonVariants> {
+  asChild?: boolean
+  side?: 'top' | 'right' | 'bottom' | 'left'
+  tooltip?: string
+}
+
+const Button = React.forwardRef<HTMLButtonElement, ButtonProps>(
+  ({ className, variant, tooltip, size, side = 'right', asChild = false, ...props }, ref) => {
+    const Comp = asChild ? Slot : 'button'
+    if (!tooltip) {
+      return (
+        <Comp
+          className={cn(buttonVariants({ variant, size, className }), 'cursor-pointer')}
+          ref={ref}
+          {...props}
+        />
+      )
+    }
+
+    return (
+      <TooltipProvider>
+        <Tooltip>
+          <TooltipTrigger asChild>
+            <Comp
+              className={cn(buttonVariants({ variant, size, className }), 'cursor-pointer')}
+              ref={ref}
+              {...props}
+            />
+          </TooltipTrigger>
+          <TooltipContent side={side}>{tooltip}</TooltipContent>
+        </Tooltip>
+      </TooltipProvider>
+    )
+  }
+)
+Button.displayName = 'Button'
+
+export type ButtonVariantType = Exclude<
+  NonNullable<Parameters<typeof buttonVariants>[0]>['variant'],
+  undefined
+>
+
+export default Button
--- a/lightrag_webui/src/components/ui/Checkbox.tsx
+++ b/lightrag_webui/src/components/ui/Checkbox.tsx
@@ -0,0 +1,26 @@
+import * as React from 'react'
+import * as CheckboxPrimitive from '@radix-ui/react-checkbox'
+import { Check } from 'lucide-react'
+
+import { cn } from '@/lib/utils'
+
+const Checkbox = React.forwardRef<
+  React.ComponentRef<typeof CheckboxPrimitive.Root>,
+  React.ComponentPropsWithoutRef<typeof CheckboxPrimitive.Root>
+>(({ className, ...props }, ref) => (
+  <CheckboxPrimitive.Root
+    ref={ref}
+    className={cn(
+      'peer border-primary ring-offset-background focus-visible:ring-ring data-[state=checked]:bg-primary data-[state=checked]:text-primary-foreground h-4 w-4 shrink-0 rounded-sm border focus-visible:ring-2 focus-visible:ring-offset-2 focus-visible:outline-none disabled:cursor-not-allowed disabled:opacity-50',
+      className
+    )}
+    {...props}
+  >
+    <CheckboxPrimitive.Indicator className={cn('flex items-center justify-center text-current')}>
+      <Check className="h-4 w-4" />
+    </CheckboxPrimitive.Indicator>
+  </CheckboxPrimitive.Root>
+))
+Checkbox.displayName = CheckboxPrimitive.Root.displayName
+
+export { Checkbox }
--- a/lightrag_webui/src/components/ui/Command.tsx
+++ b/lightrag_webui/src/components/ui/Command.tsx
@@ -0,0 +1,143 @@
+import * as React from 'react'
+import { type DialogProps } from '@radix-ui/react-dialog'
+import { Command as CommandPrimitive } from 'cmdk'
+import { Search } from 'lucide-react'
+
+import { cn } from '@/lib/utils'
+import { Dialog, DialogContent } from './Dialog'
+
+const Command = React.forwardRef<
+  React.ComponentRef<typeof CommandPrimitive>,
+  React.ComponentPropsWithoutRef<typeof CommandPrimitive>
+>(({ className, ...props }, ref) => (
+  <CommandPrimitive
+    ref={ref}
+    className={cn(
+      'bg-popover text-popover-foreground flex h-full w-full flex-col overflow-hidden rounded-md',
+      className
+    )}
+    {...props}
+  />
+))
+Command.displayName = CommandPrimitive.displayName
+
+const CommandDialog = ({ children, ...props }: DialogProps) => {
+  return (
+    <Dialog {...props}>
+      <DialogContent className="overflow-hidden p-0 shadow-lg">
+        <Command className="[&_[cmdk-group-heading]]:text-muted-foreground [&_[cmdk-group-heading]]:px-2 [&_[cmdk-group-heading]]:font-medium [&_[cmdk-group]]:px-2 [&_[cmdk-group]:not([hidden])_~[cmdk-group]]:pt-0 [&_[cmdk-input-wrapper]_svg]:h-5 [&_[cmdk-input-wrapper]_svg]:w-5 [&_[cmdk-input]]:h-12 [&_[cmdk-item]]:px-2 [&_[cmdk-item]]:py-3 [&_[cmdk-item]_svg]:h-5 [&_[cmdk-item]_svg]:w-5">
+          {children}
+        </Command>
+      </DialogContent>
+    </Dialog>
+  )
+}
+
+const CommandInput = React.forwardRef<
+  React.ComponentRef<typeof CommandPrimitive.Input>,
+  React.ComponentPropsWithoutRef<typeof CommandPrimitive.Input>
+>(({ className, ...props }, ref) => (
+  // eslint-disable-next-line react/no-unknown-property
+  <div className="flex items-center border-b px-3" cmdk-input-wrapper="">
+    <Search className="mr-2 h-4 w-4 shrink-0 opacity-50" />
+    <CommandPrimitive.Input
+      ref={ref}
+      className={cn(
+        'placeholder:text-muted-foreground flex h-11 w-full rounded-md bg-transparent py-3 text-sm outline-none disabled:cursor-not-allowed disabled:opacity-50',
+        className
+      )}
+      {...props}
+    />
+  </div>
+))
+
+CommandInput.displayName = CommandPrimitive.Input.displayName
+
+const CommandList = React.forwardRef<
+  React.ComponentRef<typeof CommandPrimitive.List>,
+  React.ComponentPropsWithoutRef<typeof CommandPrimitive.List>
+>(({ className, ...props }, ref) => (
+  <CommandPrimitive.List
+    ref={ref}
+    className={cn('max-h-[300px] overflow-x-hidden overflow-y-auto', className)}
+    {...props}
+  />
+))
+
+CommandList.displayName = CommandPrimitive.List.displayName
+
+const CommandEmpty = React.forwardRef<
+  React.ComponentRef<typeof CommandPrimitive.Empty>,
+  React.ComponentPropsWithoutRef<typeof CommandPrimitive.Empty>
+>((props, ref) => (
+  <CommandPrimitive.Empty ref={ref} className="py-6 text-center text-sm" {...props} />
+))
+
+CommandEmpty.displayName = CommandPrimitive.Empty.displayName
+
+const CommandGroup = React.forwardRef<
+  React.ComponentRef<typeof CommandPrimitive.Group>,
+  React.ComponentPropsWithoutRef<typeof CommandPrimitive.Group>
+>(({ className, ...props }, ref) => (
+  <CommandPrimitive.Group
+    ref={ref}
+    className={cn(
+      'text-foreground [&_[cmdk-group-heading]]:text-muted-foreground overflow-hidden p-1 [&_[cmdk-group-heading]]:px-2 [&_[cmdk-group-heading]]:py-1.5 [&_[cmdk-group-heading]]:text-xs [&_[cmdk-group-heading]]:font-medium',
+      className
+    )}
+    {...props}
+  />
+))
+
+CommandGroup.displayName = CommandPrimitive.Group.displayName
+
+const CommandSeparator = React.forwardRef<
+  React.ComponentRef<typeof CommandPrimitive.Separator>,
+  React.ComponentPropsWithoutRef<typeof CommandPrimitive.Separator>
+>(({ className, ...props }, ref) => (
+  <CommandPrimitive.Separator
+    ref={ref}
+    className={cn('bg-border -mx-1 h-px', className)}
+    {...props}
+  />
+))
+CommandSeparator.displayName = CommandPrimitive.Separator.displayName
+
+const CommandItem = React.forwardRef<
+  React.ComponentRef<typeof CommandPrimitive.Item>,
+  React.ComponentPropsWithoutRef<typeof CommandPrimitive.Item>
+>(({ className, ...props }, ref) => (
+  <CommandPrimitive.Item
+    ref={ref}
+    className={cn(
+      // eslint-disable-next-line @stylistic/js/quotes
+      "data-[selected='true']:bg-accent data-[selected=true]:text-accent-foreground relative flex cursor-default items-center gap-2 rounded-sm px-2 py-1.5 text-sm outline-none select-none data-[disabled=true]:pointer-events-none data-[disabled=true]:opacity-50 [&_svg]:pointer-events-none [&_svg]:size-4 [&_svg]:shrink-0",
+      className
+    )}
+    {...props}
+  />
+))
+
+CommandItem.displayName = CommandPrimitive.Item.displayName
+
+const CommandShortcut = ({ className, ...props }: React.HTMLAttributes<HTMLSpanElement>) => {
+  return (
+    <span
+      className={cn('text-muted-foreground ml-auto text-xs tracking-widest', className)}
+      {...props}
+    />
+  )
+}
+CommandShortcut.displayName = 'CommandShortcut'
+
+export {
+  Command,
+  CommandDialog,
+  CommandInput,
+  CommandList,
+  CommandEmpty,
+  CommandGroup,
+  CommandItem,
+  CommandShortcut,
+  CommandSeparator
+}
--- a/lightrag_webui/src/components/ui/Dialog.tsx
+++ b/lightrag_webui/src/components/ui/Dialog.tsx
@@ -0,0 +1,102 @@
+import * as React from 'react'
+import * as DialogPrimitive from '@radix-ui/react-dialog'
+import { X } from 'lucide-react'
+
+import { cn } from '@/lib/utils'
+
+const Dialog = DialogPrimitive.Root
+
+const DialogTrigger = DialogPrimitive.Trigger
+
+const DialogPortal = DialogPrimitive.Portal
+
+const DialogClose = DialogPrimitive.Close
+
+const DialogOverlay = React.forwardRef<
+  React.ComponentRef<typeof DialogPrimitive.Overlay>,
+  React.ComponentPropsWithoutRef<typeof DialogPrimitive.Overlay>
+>(({ className, ...props }, ref) => (
+  <DialogPrimitive.Overlay
+    ref={ref}
+    className={cn(
+      'data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 fixed inset-0 z-50 bg-black/80',
+      className
+    )}
+    {...props}
+  />
+))
+DialogOverlay.displayName = DialogPrimitive.Overlay.displayName
+
+const DialogContent = React.forwardRef<
+  React.ComponentRef<typeof DialogPrimitive.Content>,
+  React.ComponentPropsWithoutRef<typeof DialogPrimitive.Content>
+>(({ className, children, ...props }, ref) => (
+  <DialogPortal>
+    <DialogOverlay />
+    <DialogPrimitive.Content
+      ref={ref}
+      className={cn(
+        'bg-background data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 data-[state=closed]:zoom-out-95 data-[state=open]:zoom-in-95 data-[state=closed]:slide-out-to-left-1/2 data-[state=closed]:slide-out-to-top-[48%] data-[state=open]:slide-in-from-left-1/2 data-[state=open]:slide-in-from-top-[48%] fixed top-[50%] left-[50%] z-50 grid w-full max-w-lg translate-x-[-50%] translate-y-[-50%] gap-4 border p-6 shadow-lg duration-200 sm:rounded-lg',
+        className
+      )}
+      {...props}
+    >
+      {children}
+      <DialogPrimitive.Close className="ring-offset-background focus:ring-ring data-[state=open]:bg-accent data-[state=open]:text-muted-foreground absolute top-4 right-4 rounded-sm opacity-70 transition-opacity hover:opacity-100 focus:ring-2 focus:ring-offset-2 focus:outline-none disabled:pointer-events-none">
+        <X className="h-4 w-4" />
+        <span className="sr-only">Close</span>
+      </DialogPrimitive.Close>
+    </DialogPrimitive.Content>
+  </DialogPortal>
+))
+DialogContent.displayName = DialogPrimitive.Content.displayName
+
+const DialogHeader = ({ className, ...props }: React.HTMLAttributes<HTMLDivElement>) => (
+  <div className={cn('flex flex-col space-y-1.5 text-center sm:text-left', className)} {...props} />
+)
+DialogHeader.displayName = 'DialogHeader'
+
+const DialogFooter = ({ className, ...props }: React.HTMLAttributes<HTMLDivElement>) => (
+  <div
+    className={cn('flex flex-col-reverse sm:flex-row sm:justify-end sm:space-x-2', className)}
+    {...props}
+  />
+)
+DialogFooter.displayName = 'DialogFooter'
+
+const DialogTitle = React.forwardRef<
+  React.ElementRef<typeof DialogPrimitive.Title>,
+  React.ComponentPropsWithoutRef<typeof DialogPrimitive.Title>
+>(({ className, ...props }, ref) => (
+  <DialogPrimitive.Title
+    ref={ref}
+    className={cn('text-lg leading-none font-semibold tracking-tight', className)}
+    {...props}
+  />
+))
+DialogTitle.displayName = DialogPrimitive.Title.displayName
+
+const DialogDescription = React.forwardRef<
+  React.ElementRef<typeof DialogPrimitive.Description>,
+  React.ComponentPropsWithoutRef<typeof DialogPrimitive.Description>
+>(({ className, ...props }, ref) => (
+  <DialogPrimitive.Description
+    ref={ref}
+    className={cn('text-muted-foreground text-sm', className)}
+    {...props}
+  />
+))
+DialogDescription.displayName = DialogPrimitive.Description.displayName
+
+export {
+  Dialog,
+  DialogPortal,
+  DialogOverlay,
+  DialogClose,
+  DialogTrigger,
+  DialogContent,
+  DialogHeader,
+  DialogFooter,
+  DialogTitle,
+  DialogDescription
+}
--- a/lightrag_webui/src/components/ui/Input.tsx
+++ b/lightrag_webui/src/components/ui/Input.tsx
@@ -0,0 +1,21 @@
+import * as React from 'react'
+import { cn } from '@/lib/utils'
+
+const Input = React.forwardRef<HTMLInputElement, React.ComponentProps<'input'>>(
+  ({ className, type, ...props }, ref) => {
+    return (
+      <input
+        type={type}
+        className={cn(
+          'border-input file:text-foreground placeholder:text-muted-foreground focus-visible:ring-ring flex h-9 rounded-md border bg-transparent px-3 py-1 text-base shadow-sm transition-colors file:border-0 file:bg-transparent file:text-sm file:font-medium focus-visible:ring-1 focus-visible:outline-none disabled:cursor-not-allowed disabled:opacity-50 md:text-sm',
+          className
+        )}
+        ref={ref}
+        {...props}
+      />
+    )
+  }
+)
+Input.displayName = 'Input'
+
+export default Input
--- a/lightrag_webui/src/components/ui/Popover.tsx
+++ b/lightrag_webui/src/components/ui/Popover.tsx
@@ -0,0 +1,29 @@
+import * as React from 'react'
+import * as PopoverPrimitive from '@radix-ui/react-popover'
+
+import { cn } from '@/lib/utils'
+
+const Popover = PopoverPrimitive.Root
+
+const PopoverTrigger = PopoverPrimitive.Trigger
+
+const PopoverContent = React.forwardRef<
+  React.ComponentRef<typeof PopoverPrimitive.Content>,
+  React.ComponentPropsWithoutRef<typeof PopoverPrimitive.Content>
+>(({ className, align = 'center', sideOffset = 4, ...props }, ref) => (
+  <PopoverPrimitive.Portal>
+    <PopoverPrimitive.Content
+      ref={ref}
+      align={align}
+      sideOffset={sideOffset}
+      className={cn(
+        'bg-popover text-popover-foreground data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 data-[state=closed]:zoom-out-95 data-[state=open]:zoom-in-95 data-[side=bottom]:slide-in-from-top-2 data-[side=left]:slide-in-from-right-2 data-[side=right]:slide-in-from-left-2 data-[side=top]:slide-in-from-bottom-2 z-50 rounded-md border p-4 shadow-md outline-none',
+        className
+      )}
+      {...props}
+    />
+  </PopoverPrimitive.Portal>
+))
+PopoverContent.displayName = PopoverPrimitive.Content.displayName
+
+export { Popover, PopoverTrigger, PopoverContent }
--- a/lightrag_webui/src/components/ui/Separator.tsx
+++ b/lightrag_webui/src/components/ui/Separator.tsx
@@ -0,0 +1,24 @@
+import * as React from 'react'
+import * as SeparatorPrimitive from '@radix-ui/react-separator'
+
+import { cn } from '@/lib/utils'
+
+const Separator = React.forwardRef<
+  React.ComponentRef<typeof SeparatorPrimitive.Root>,
+  React.ComponentPropsWithoutRef<typeof SeparatorPrimitive.Root>
+>(({ className, orientation = 'horizontal', decorative = true, ...props }, ref) => (
+  <SeparatorPrimitive.Root
+    ref={ref}
+    decorative={decorative}
+    orientation={orientation}
+    className={cn(
+      'bg-border shrink-0',
+      orientation === 'horizontal' ? 'h-[1px] w-full' : 'h-full w-[1px]',
+      className
+    )}
+    {...props}
+  />
+))
+Separator.displayName = SeparatorPrimitive.Root.displayName
+
+export default Separator
--- a/lightrag_webui/src/components/ui/Text.tsx
+++ b/lightrag_webui/src/components/ui/Text.tsx
@@ -0,0 +1,49 @@
+import { Tooltip, TooltipContent, TooltipProvider, TooltipTrigger } from '@/components/ui/Tooltip'
+import { cn } from '@/lib/utils'
+
+const Text = ({
+  text,
+  className,
+  tooltipClassName,
+  tooltip,
+  side,
+  onClick
+}: {
+  text: string
+  className?: string
+  tooltipClassName?: string
+  tooltip?: string
+  side?: 'top' | 'right' | 'bottom' | 'left'
+  onClick?: () => void
+}) => {
+  if (!tooltip) {
+    return (
+      <label
+        className={cn(className, onClick !== undefined ? 'cursor-pointer' : undefined)}
+        onClick={onClick}
+      >
+        {text}
+      </label>
+    )
+  }
+
+  return (
+    <TooltipProvider delayDuration={200}>
+      <Tooltip>
+        <TooltipTrigger asChild>
+          <label
+            className={cn(className, onClick !== undefined ? 'cursor-pointer' : undefined)}
+            onClick={onClick}
+          >
+            {text}
+          </label>
+        </TooltipTrigger>
+        <TooltipContent side={side} className={tooltipClassName}>
+          {tooltip}
+        </TooltipContent>
+      </Tooltip>
+    </TooltipProvider>
+  )
+}
+
+export default Text
--- a/lightrag_webui/src/components/ui/Tooltip.tsx
+++ b/lightrag_webui/src/components/ui/Tooltip.tsx
@@ -0,0 +1,27 @@
+import * as React from 'react'
+import * as TooltipPrimitive from '@radix-ui/react-tooltip'
+import { cn } from '@/lib/utils'
+
+const TooltipProvider = TooltipPrimitive.Provider
+
+const Tooltip = TooltipPrimitive.Root
+
+const TooltipTrigger = TooltipPrimitive.Trigger
+
+const TooltipContent = React.forwardRef<
+  React.ComponentRef<typeof TooltipPrimitive.Content>,
+  React.ComponentPropsWithoutRef<typeof TooltipPrimitive.Content>
+>(({ className, sideOffset = 4, ...props }, ref) => (
+  <TooltipPrimitive.Content
+    ref={ref}
+    sideOffset={sideOffset}
+    className={cn(
+      'bg-popover text-popover-foreground animate-in fade-in-0 zoom-in-95 data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=closed]:zoom-out-95 data-[side=bottom]:slide-in-from-top-2 data-[side=left]:slide-in-from-right-2 data-[side=right]:slide-in-from-left-2 data-[side=top]:slide-in-from-bottom-2 z-50 overflow-hidden rounded-md border px-3 py-1.5 text-sm shadow-md',
+      className
+    )}
+    {...props}
+  />
+))
+TooltipContent.displayName = TooltipPrimitive.Content.displayName
+
+export { Tooltip, TooltipTrigger, TooltipContent, TooltipProvider }