add graph depth and layout iteration settings

lightrag_webui/src/stores/settings.ts

@@ -8,9 +8,7 @@ type Theme = 'dark' | 'light' | 'system'
 type Tab = 'documents' | 'knowledge-graph' | 'retrieval' | 'api'
 
 interface SettingsState {
-  theme: Theme
-  setTheme: (theme: Theme) => void
-
+  // Graph viewer settings
   showPropertyPanel: boolean
   showNodeSearchBar: boolean
 
@@ -21,23 +19,35 @@ interface SettingsState {
   enableHideUnselectedEdges: boolean
   enableEdgeEvents: boolean
 
+  graphQueryMaxDepth: number
+  setGraphQueryMaxDepth: (depth: number) => void
+
+  graphLayoutMaxIterations: number
+  setGraphLayoutMaxIterations: (iterations: number) => void
+
+  // Retrieval settings
   queryLabel: string
   setQueryLabel: (queryLabel: string) => void
 
-  enableHealthCheck: boolean
-  setEnableHealthCheck: (enable: boolean) => void
-
-  apiKey: string | null
-  setApiKey: (key: string | null) => void
-
-  currentTab: Tab
-  setCurrentTab: (tab: Tab) => void
-
   retrievalHistory: Message[]
   setRetrievalHistory: (history: Message[]) => void
 
   querySettings: Omit<QueryRequest, 'query'>
   updateQuerySettings: (settings: Partial<QueryRequest>) => void
+
+  // Auth settings
+  apiKey: string | null
+  setApiKey: (key: string | null) => void
+
+  // App settings
+  theme: Theme
+  setTheme: (theme: Theme) => void
+
+  enableHealthCheck: boolean
+  setEnableHealthCheck: (enable: boolean) => void
+
+  currentTab: Tab
+  setCurrentTab: (tab: Tab) => void
 }
 
 const useSettingsStoreBase = create<SettingsState>()(
@@ -55,7 +65,11 @@ const useSettingsStoreBase = create<SettingsState>()(
       enableHideUnselectedEdges: true,
       enableEdgeEvents: false,
 
+      graphQueryMaxDepth: 3,
+      graphLayoutMaxIterations: 10,
+
       queryLabel: defaultQueryLabel,
+
       enableHealthCheck: true,
 
       apiKey: null,
@@ -81,11 +95,18 @@ const useSettingsStoreBase = create<SettingsState>()(
 
       setTheme: (theme: Theme) => set({ theme }),
 
+      setGraphLayoutMaxIterations: (iterations: number) =>
+        set({
+          graphLayoutMaxIterations: iterations
+        }),
+
       setQueryLabel: (queryLabel: string) =>
         set({
           queryLabel
         }),
 
+      setGraphQueryMaxDepth: (depth: number) => set({ graphQueryMaxDepth: depth }),
+
       setEnableHealthCheck: (enable: boolean) => set({ enableHealthCheck: enable }),
 
       setApiKey: (apiKey: string | null) => set({ apiKey }),
@@ -102,7 +123,7 @@ const useSettingsStoreBase = create<SettingsState>()(
     {
       name: 'settings-storage',
       storage: createJSONStorage(() => localStorage),
-      version: 6,
+      version: 7,
       migrate: (state: any, version: number) => {
         if (version < 2) {
           state.showEdgeLabel = false
@@ -137,6 +158,10 @@ const useSettingsStoreBase = create<SettingsState>()(
           }
           state.retrievalHistory = []
         }
+        if (version < 7) {
+          state.graphQueryMaxDepth = 3
+          state.graphLayoutMaxIterations = 10
+        }
         return state
       }
     }