diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py index b743ccb6..2e091ac8 100644 --- a/lightrag/api/lightrag_server.py +++ b/lightrag/api/lightrag_server.py @@ -439,7 +439,7 @@ def create_app(args): }, "update_status": update_status, "core_version": core_version, - "api_version": __api_version__ + "api_version": __api_version__, } # Custom StaticFiles class to prevent caching of HTML files diff --git a/lightrag_webui/src/App.tsx b/lightrag_webui/src/App.tsx index b3e1acca..48d94ced 100644 --- a/lightrag_webui/src/App.tsx +++ b/lightrag_webui/src/App.tsx @@ -29,15 +29,15 @@ function App() { useEffect(() => { // Only execute if health check is enabled if (!enableHealthCheck) return; - + // Health check function const performHealthCheck = async () => { await useBackendState.getState().check(); }; - + // Execute immediately performHealthCheck(); - + // Set interval for periodic execution const interval = setInterval(performHealthCheck, healthCheckInterval * 1000); return () => clearInterval(interval); @@ -49,26 +49,26 @@ function App() { // Prevent duplicate calls in Vite dev mode if (versionCheckRef.current) return; versionCheckRef.current = true; - + // Check if version info was already obtained in login page const versionCheckedFromLogin = sessionStorage.getItem('VERSION_CHECKED_FROM_LOGIN') === 'true'; if (versionCheckedFromLogin) return; - + // Get version info const token = localStorage.getItem('LIGHTRAG-API-TOKEN'); if (!token) return; - + try { const status = await getAuthStatus(); if (status.core_version || status.api_version) { // Update version info while maintaining login state useAuthStore.getState().login( - token, - useAuthStore.getState().isGuestMode, - status.core_version, + token, + useAuthStore.getState().isGuestMode, + status.core_version, status.api_version ); - + // Set flag to indicate version info has been checked sessionStorage.setItem('VERSION_CHECKED_FROM_LOGIN', 'true'); } @@ -76,7 +76,7 @@ function App() { console.error('Failed to get version info:', error); } }; - + // Execute version check checkVersion(); }, []); // Empty dependency array ensures it only runs once on mount diff --git a/lightrag_webui/src/features/LoginPage.tsx b/lightrag_webui/src/features/LoginPage.tsx index 5b06c428..645e8fe9 100644 --- a/lightrag_webui/src/features/LoginPage.tsx +++ b/lightrag_webui/src/features/LoginPage.tsx @@ -35,7 +35,7 @@ const LoginPage = () => { return; } authCheckRef.current = true; - + try { // If already authenticated, redirect to home if (isAuthenticated) { @@ -45,13 +45,13 @@ const LoginPage = () => { // Check auth status const status = await getAuthStatus() - + // Set checkingAuth to false immediately after getAuthStatus // This allows the login page to render while other processing continues if (isMounted) { setCheckingAuth(false); } - + // Set session flag for version check to avoid duplicate checks in App component if (isMounted && (status.core_version || status.api_version)) { sessionStorage.setItem('VERSION_CHECKED_FROM_LOGIN', 'true'); @@ -107,7 +107,7 @@ const LoginPage = () => { // Check authentication mode const isGuestMode = response.auth_mode === 'disabled' login(response.access_token, isGuestMode, response.core_version, response.api_version) - + // Set session flag for version check if (response.core_version || response.api_version) { sessionStorage.setItem('VERSION_CHECKED_FROM_LOGIN', 'true'); diff --git a/lightrag_webui/src/stores/state.ts b/lightrag_webui/src/stores/state.ts index fb9db79e..a1926dbd 100644 --- a/lightrag_webui/src/stores/state.ts +++ b/lightrag_webui/src/stores/state.ts @@ -43,7 +43,7 @@ const useBackendStateStoreBase = create()((set) => ({ health.api_version || null ); } - + set({ health: true, message: null,