gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: Decouple user management from history service by removing the User model, using string user_ids, and renaming history tables.

Browse source
This commit is contained in:
dangddt 2025-12-02 17:28:51 +07:00
parent 854dc67c12
commit d924577cc5
13 changed files with 1520 additions and 2067 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
lightrag/api/auth.py
View file

@ -15,6 +15,7 @@ load_dotenv(dotenv_path=".env", override=False)
class TokenPayload(BaseModel): class TokenPayload(BaseModel):
sub: str # Username sub: str # Username
user_id: str # User ID
exp: datetime # Expiration time exp: datetime # Expiration time
role: str = "user" # User role, default is regular user role: str = "user" # User role, default is regular user
metadata: dict = {} # Additional metadata metadata: dict = {} # Additional metadata
@ -30,8 +31,13 @@ class AuthHandler:
auth_accounts = global_args.auth_accounts auth_accounts = global_args.auth_accounts
if auth_accounts: if auth_accounts:
for account in auth_accounts.split(","): for account in auth_accounts.split(","):
username, password = account.split(":", 1) parts = account.split(":")
self.accounts[username] = password if len(parts) == 3:
username, password, user_id = parts
else:
username, password = parts
user_id = username # Default user_id to username if not provided
self.accounts[username] = {"password": password, "user_id": user_id}
def create_token( def create_token(
self, self,
@ -63,9 +69,14 @@ class AuthHandler:
expire = datetime.utcnow() + timedelta(hours=expire_hours) expire = datetime.utcnow() + timedelta(hours=expire_hours)
# Get user_id from accounts or use username
user_id = username
if username in self.accounts and isinstance(self.accounts[username], dict):
user_id = self.accounts[username].get("user_id", username)
# Create payload # Create payload
payload = TokenPayload( payload = TokenPayload(
sub=username, exp=expire, role=role, metadata=metadata or {} sub=username, user_id=user_id, exp=expire, role=role, metadata=metadata or {}
) )
return jwt.encode(payload.dict(), self.secret, algorithm=self.algorithm) return jwt.encode(payload.dict(), self.secret, algorithm=self.algorithm)
@ -96,6 +107,7 @@ class AuthHandler:
# Return complete payload instead of just username # Return complete payload instead of just username
return { return {
"username": payload["sub"], "username": payload["sub"],
"user_id": payload.get("user_id", payload["sub"]),
"role": payload.get("role", "user"), "role": payload.get("role", "user"),
"metadata": payload.get("metadata", {}), "metadata": payload.get("metadata", {}),
"exp": expire_time, "exp": expire_time,

3
lightrag/api/lightrag_server.py
View file

@ -1161,7 +1161,8 @@ def create_app(args):
"webui_description": webui_description, "webui_description": webui_description,
} }
username = form_data.username username = form_data.username
if auth_handler.accounts.get(username) != form_data.password: account = auth_handler.accounts.get(username)
if not account or account["password"] != form_data.password:
raise HTTPException(status_code=401, detail="Incorrect credentials") raise HTTPException(status_code=401, detail="Incorrect credentials")
# Regular user login # Regular user login

67
lightrag/api/routers/history_routes.py
View file

@ -1,11 +1,9 @@
from fastapi import APIRouter, Depends, HTTPException, Security, status from fastapi import APIRouter, Depends, HTTPException, Header
from fastapi.security import OAuth2PasswordBearer
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from typing import List, Optional from typing import List, Optional
from uuid import UUID from uuid import UUID
import sys import sys
import os import os
from lightrag.api.auth import auth_handler
# Ensure service module is in path (similar to query_routes.py) # Ensure service module is in path (similar to query_routes.py)
project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../../")) project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../../"))
@ -17,7 +15,6 @@ try:
from app.core.database import get_db from app.core.database import get_db
from app.services.history_manager import HistoryManager from app.services.history_manager import HistoryManager
from app.models.schemas import SessionResponse, SessionCreate, ChatMessageResponse from app.models.schemas import SessionResponse, SessionCreate, ChatMessageResponse
from app.models.models import User
except ImportError: except ImportError:
# Fallback if service not found (shouldn't happen if setup is correct) # Fallback if service not found (shouldn't happen if setup is correct)
get_db = None get_db = None
@ -25,79 +22,45 @@ except ImportError:
SessionResponse = None SessionResponse = None
SessionCreate = None SessionCreate = None
ChatMessageResponse = None ChatMessageResponse = None
User = None
router = APIRouter() router = APIRouter()
oauth2_scheme = OAuth2PasswordBearer(tokenUrl="login", auto_error=False)
def check_dependencies(): def check_dependencies():
if not HistoryManager: if not HistoryManager:
raise HTTPException(status_code=503, detail="History service not available") raise HTTPException(status_code=503, detail="History service not available")
async def get_current_user( async def get_current_user_id(
token: str = Security(oauth2_scheme), x_user_id: Optional[str] = Header(None, alias="X-User-ID")
db: Session = Depends(get_db) ) -> str:
) -> User: # Prefer X-User-ID, default to default_user
check_dependencies() uid = x_user_id
if not uid:
if not token: # Fallback to default user if no header provided (for backward compatibility or dev)
# If no token provided, try to use default user if configured or allowed # Or raise error if strict
# For now, we'll return the default user for backward compatibility if needed, return "default_user"
# but ideally we should require auth. return uid
# Let's check if we have a default user
user = db.query(User).filter(User.username == "default_user").first()
if not user:
user = User(username="default_user", email="default@example.com")
db.add(user)
db.commit()
db.refresh(user)
return user
try:
user_data = auth_handler.validate_token(token)
username = user_data["username"]
user = db.query(User).filter(User.username == username).first()
if not user:
# Create user if not exists (auto-registration on first login)
# In a real app you might want to fetch email from token metadata or require explicit registration
user = User(username=username, email=f"{username}@example.com")
db.add(user)
db.commit()
db.refresh(user)
return user
except HTTPException:
raise
except Exception as e:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail="Could not validate credentials",
headers={"WWW-Authenticate": "Bearer"},
)
@router.get("/sessions", response_model=List[SessionResponse], tags=["History"]) @router.get("/sessions", response_model=List[SessionResponse], tags=["History"])
def list_sessions( def list_sessions(
skip: int = 0, skip: int = 0,
limit: int = 20, limit: int = 20,
db: Session = Depends(get_db), db: Session = Depends(get_db),
current_user: User = Depends(get_current_user) current_user_id: str = Depends(get_current_user_id)
): ):
check_dependencies() check_dependencies()
manager = HistoryManager(db) manager = HistoryManager(db)
sessions = manager.list_sessions(user_id=current_user.id, skip=skip, limit=limit) sessions = manager.list_sessions(user_id=current_user_id, skip=skip, limit=limit)
return sessions return sessions
@router.post("/sessions", response_model=SessionResponse, tags=["History"]) @router.post("/sessions", response_model=SessionResponse, tags=["History"])
def create_session( def create_session(
session_in: SessionCreate, session_in: SessionCreate,
db: Session = Depends(get_db), db: Session = Depends(get_db),
current_user: User = Depends(get_current_user) current_user_id: str = Depends(get_current_user_id)
): ):
check_dependencies() check_dependencies()
manager = HistoryManager(db) manager = HistoryManager(db)
return manager.create_session(user_id=current_user.id, title=session_in.title) return manager.create_session(user_id=current_user_id, title=session_in.title)
@router.get("/sessions/{session_id}/history", response_model=List[ChatMessageResponse], tags=["History"]) @router.get("/sessions/{session_id}/history", response_model=List[ChatMessageResponse], tags=["History"])
def get_session_history( def get_session_history(

128
lightrag/api/routers/query_routes.py
View file

@ -3,17 +3,19 @@ This module contains all query-related routes for the LightRAG API.
""" """
import json import json
from typing import Any, Dict, List, Literal, Optional import logging
from fastapi import APIRouter, Depends, HTTPException import os
from lightrag.base import QueryParam import sys
import time
import uuid
from typing import Any, Dict, List, Literal, Optional, Union
from fastapi import APIRouter, Body, Depends, Header, HTTPException, Query
from fastapi.responses import StreamingResponse
from lightrag.api.utils_api import get_combined_auth_dependency from lightrag.api.utils_api import get_combined_auth_dependency
from lightrag.base import QueryParam
from lightrag.utils import logger from lightrag.utils import logger
from pydantic import BaseModel, Field, field_validator from pydantic import BaseModel, Field, field_validator
import sys
import os
import uuid
import time
import json
# Add the project root to sys.path to allow importing 'service' # Add the project root to sys.path to allow importing 'service'
# Assuming this file is at lightrag/api/routers/query_routes.py # Assuming this file is at lightrag/api/routers/query_routes.py
@ -28,13 +30,17 @@ if service_dir not in sys.path:
try: try:
from app.core.database import SessionLocal from app.core.database import SessionLocal
from app.services.history_manager import HistoryManager from app.services.history_manager import HistoryManager
from app.models.models import User from app.models.schemas import ChatMessageResponse
except ImportError as e: except ImportError as e:
# Fallback or handle error if service module is not found logger.error(f"Warning: Could not import service module. History logging will be disabled. Error: {e}")
print(f"Warning: Could not import service module. History logging will be disabled. Error: {e}") print(f"CRITICAL ERROR: Could not import service module: {e}", file=sys.stderr)
import traceback
traceback.print_exc()
SessionLocal = None SessionLocal = None
HistoryManager = None HistoryManager = None
User = None QueryRequest = None
QueryResponse = None
ChatMessageResponse = None
router = APIRouter(tags=["query"]) router = APIRouter(tags=["query"])
@ -354,7 +360,10 @@ def create_query_routes(rag, api_key: Optional[str] = None, top_k: int = 60):
}, },
}, },
) )
async def query_text(request: QueryRequest): async def query_text(
request: QueryRequest,
x_user_id: Optional[str] = Header(None, alias="X-User-ID")
):
""" """
Comprehensive RAG query endpoint with non-streaming response. Parameter "stream" is ignored. Comprehensive RAG query endpoint with non-streaming response. Parameter "stream" is ignored.
@ -441,6 +450,7 @@ def create_query_routes(rag, api_key: Optional[str] = None, top_k: int = 60):
param.stream = False param.stream = False
# Unified approach: always use aquery_llm for both cases # Unified approach: always use aquery_llm for both cases
start_time = time.time()
result = await rag.aquery_llm(request.query, param=param) result = await rag.aquery_llm(request.query, param=param)
# Extract LLM response and references from unified result # Extract LLM response and references from unified result
@ -484,67 +494,81 @@ def create_query_routes(rag, api_key: Optional[str] = None, top_k: int = 60):
final_response = QueryResponse(response=response_content, references=None) final_response = QueryResponse(response=response_content, references=None)
# --- LOGGING START --- # --- LOGGING START ---
logger.info(f"DEBUG: SessionLocal={SessionLocal}, HistoryManager={HistoryManager}")
if SessionLocal and HistoryManager: if SessionLocal and HistoryManager:
try: try:
logger.info("DEBUG: Entering logging block")
db = SessionLocal() db = SessionLocal()
manager = HistoryManager(db) manager = HistoryManager(db)
# 1. Get or Create User (Default) # 1. Get User ID from Header (or default)
user = db.query(User).filter(User.username == "default_user").first() current_user_id = x_user_id or "default_user"
if not user:
user = User(username="default_user", email="default@example.com")
db.add(user)
db.commit()
db.refresh(user)
# 2. Handle Session # 2. Handle Session
session_uuid = None session_uuid = None
if request.session_id: if request.session_id:
try: try:
session_uuid = uuid.UUID(request.session_id) temp_uuid = uuid.UUID(request.session_id)
# Verify session exists # Verify session exists
if not manager.get_session(session_uuid): if manager.get_session(temp_uuid):
# If provided ID doesn't exist, create it with that ID if possible or just create new session_uuid = temp_uuid
# For simplicity, let's create a new one if it doesn't exist but we can't force ID easily with current manager else:
# Let's just create a new session if not found or use the provided one if we trust it. logger.warning(f"Session {request.session_id} not found. Creating new session.")
# Actually, manager.create_session generates ID.
# Let's just create a new session if the provided one is invalid/not found,
# OR we can just create a new session if session_id is NOT provided.
# If session_id IS provided, we assume it exists.
pass
except ValueError: except ValueError:
pass logger.warning(f"Invalid session ID format: {request.session_id}")
if not session_uuid: if not session_uuid:
# Create new session # Create new session
session = manager.create_session(user_id=user.id, title=request.query[:50]) session = manager.create_session(user_id=current_user_id, title=request.query[:50])
session_uuid = session.id session_uuid = session.id
# Calculate processing time
end_time = time.time()
processing_time = end_time - start_time
# Calculate token counts
try:
import tiktoken
enc = tiktoken.get_encoding("cl100k_base")
query_tokens = len(enc.encode(request.query))
response_tokens = len(enc.encode(response_content))
except ImportError:
# Fallback approximation
query_tokens = len(request.query) // 4
response_tokens = len(response_content) // 4
except Exception as e:
logger.warning(f"Error calculating tokens: {e}")
query_tokens = len(request.query) // 4
response_tokens = len(response_content) // 4
# 3. Log User Message # 3. Log User Message
manager.save_message( manager.save_message(
session_id=session_uuid, session_id=session_uuid,
role="user", role="user",
content=request.query content=request.query,
token_count=query_tokens,
processing_time=None # User message processing time is negligible/not applicable in this context
) )
# 4. Log Assistant Message # 4. Log Assistant Message
ai_msg = manager.save_message( ai_msg = manager.save_message(
session_id=session_uuid, session_id=session_uuid,
role="assistant", role="assistant",
content=response_content content=response_content,
token_count=response_tokens,
processing_time=processing_time
) )
# 5. Log Citations # 5. Log Citations
if references: if references:
# Convert references to dict format expected by save_citations
# references is a list of ReferenceItem (pydantic) or dicts?
# In the code above: references = data.get("references", []) which are dicts.
# Then enriched_references are also dicts.
manager.save_citations(ai_msg.id, references) manager.save_citations(ai_msg.id, references)
db.close() db.close()
except Exception as log_exc: except Exception as log_exc:
print(f"Error logging history: {log_exc}") print(f"Error logging history: {log_exc}", file=sys.stderr)
import traceback
traceback.print_exc()
logger.error(f"Error logging history: {log_exc}", exc_info=True)
# Don't fail the request if logging fails # Don't fail the request if logging fails
# --- LOGGING END --- # --- LOGGING END ---
@ -633,7 +657,10 @@ def create_query_routes(rag, api_key: Optional[str] = None, top_k: int = 60):
}, },
}, },
) )
async def query_text_stream(request: QueryRequest): async def query_text_stream(
request: QueryRequest,
x_user_id: Optional[str] = Header(None, alias="X-User-ID")
):
""" """
Advanced RAG query endpoint with flexible streaming response. Advanced RAG query endpoint with flexible streaming response.
@ -848,24 +875,23 @@ def create_query_routes(rag, api_key: Optional[str] = None, top_k: int = 60):
db = SessionLocal() db = SessionLocal()
manager = HistoryManager(db) manager = HistoryManager(db)
# 1. Get or Create User # 1. Get User ID
user = db.query(User).filter(User.username == "default_user").first() current_user_id = x_user_id or "default_user"
if not user:
user = User(username="default_user", email="default@example.com")
db.add(user)
db.commit()
db.refresh(user)
# 2. Handle Session # 2. Handle Session
session_uuid = None session_uuid = None
if request.session_id: if request.session_id:
try: try:
session_uuid = uuid.UUID(request.session_id) temp_uuid = uuid.UUID(request.session_id)
if manager.get_session(temp_uuid):
session_uuid = temp_uuid
else:
logger.warning(f"Session {request.session_id} not found. Creating new session.")
except ValueError: except ValueError:
pass logger.warning(f"Invalid session ID format: {request.session_id}")
if not session_uuid or not manager.get_session(session_uuid): if not session_uuid:
session = manager.create_session(user_id=user.id, title=request.query[:50]) session = manager.create_session(user_id=current_user_id, title=request.query[:50])
session_uuid = session.id session_uuid = session.id
# 3. Log User Message # 3. Log User Message

19
lightrag_webui/src/api/lightrag.ts
View file

@ -8,6 +8,17 @@ import { navigationService } from '@/services/navigation'
import { useSettingsStore } from '@/stores/settings' import { useSettingsStore } from '@/stores/settings'
import axios, { AxiosError } from 'axios' import axios, { AxiosError } from 'axios'
const getUserIdFromToken = (token: string): string | null => {
try {
const parts = token.split('.')
if (parts.length !== 3) return null
const payload = JSON.parse(atob(parts[1]))
return payload.user_id || payload.sub || null
} catch (e) {
return null
}
}
// Types // Types
export type LightragNodeType = { export type LightragNodeType = {
id: string id: string
@ -299,6 +310,10 @@ axiosInstance.interceptors.request.use((config) => {
// Always include token if it exists, regardless of path // Always include token if it exists, regardless of path
if (token) { if (token) {
config.headers['Authorization'] = `Bearer ${token}` config.headers['Authorization'] = `Bearer ${token}`
const userId = getUserIdFromToken(token)
if (userId) {
config.headers['X-User-ID'] = userId
}
} }
if (apiKey) { if (apiKey) {
config.headers['X-API-Key'] = apiKey config.headers['X-API-Key'] = apiKey
@ -418,6 +433,10 @@ export const queryTextStream = async (
} }
if (token) { if (token) {
headers['Authorization'] = `Bearer ${token}` headers['Authorization'] = `Bearer ${token}`
const userId = getUserIdFromToken(token)
if (userId) {
headers['X-User-ID'] = userId
}
} }
if (apiKey) { if (apiKey) {
headers['X-API-Key'] = apiKey headers['X-API-Key'] = apiKey

188
lightrag_webui/src/stores/state.ts
View file

@ -1,8 +1,8 @@
import { create } from 'zustand'
import { createSelectors } from '@/lib/utils'
import { checkHealth, LightragStatus } from '@/api/lightrag' import { checkHealth, LightragStatus } from '@/api/lightrag'
import { useSettingsStore } from './settings'
import { healthCheckInterval } from '@/lib/constants' import { healthCheckInterval } from '@/lib/constants'
import { createSelectors } from '@/lib/utils'
import { create } from 'zustand'
import { useSettingsStore } from './settings'
interface BackendState { interface BackendState {
health: boolean health: boolean
@ -26,18 +26,26 @@ interface BackendState {
} }
interface AuthState { interface AuthState {
isAuthenticated: boolean; isAuthenticated: boolean
isGuestMode: boolean; // Add guest mode flag isGuestMode: boolean // Add guest mode flag
coreVersion: string | null; coreVersion: string | null
apiVersion: string | null; apiVersion: string | null
username: string | null; // login username username: string | null // login username
webuiTitle: string | null; // Custom title userId: string | null // user id
webuiDescription: string | null; // Title description webuiTitle: string | null // Custom title
webuiDescription: string | null // Title description
login: (token: string, isGuest?: boolean, coreVersion?: string | null, apiVersion?: string | null, webuiTitle?: string | null, webuiDescription?: string | null) => void; login: (
logout: () => void; token: string,
setVersion: (coreVersion: string | null, apiVersion: string | null) => void; isGuest?: boolean,
setCustomTitle: (webuiTitle: string | null, webuiDescription: string | null) => void; coreVersion?: string | null,
apiVersion?: string | null,
webuiTitle?: string | null,
webuiDescription?: string | null
) => void
logout: () => void
setVersion: (coreVersion: string | null, apiVersion: string | null) => void
setCustomTitle: (webuiTitle: string | null, webuiDescription: string | null) => void
} }
const useBackendStateStoreBase = create<BackendState>()((set, get) => ({ const useBackendStateStoreBase = create<BackendState>()((set, get) => ({
@ -56,18 +64,17 @@ const useBackendStateStoreBase = create<BackendState>()((set, get) => ({
if (health.status === 'healthy') { if (health.status === 'healthy') {
// Update version information if health check returns it // Update version information if health check returns it
if (health.core_version || health.api_version) { if (health.core_version || health.api_version) {
useAuthStore.getState().setVersion( useAuthStore.getState().setVersion(health.core_version || null, health.api_version || null)
health.core_version || null,
health.api_version || null
);
} }
// Update custom title information if health check returns it // Update custom title information if health check returns it
if ('webui_title' in health || 'webui_description' in health) { if ('webui_title' in health || 'webui_description' in health) {
useAuthStore.getState().setCustomTitle( useAuthStore
.getState()
.setCustomTitle(
'webui_title' in health ? (health.webui_title ?? null) : null, 'webui_title' in health ? (health.webui_title ?? null) : null,
'webui_description' in health ? (health.webui_description ?? null) : null 'webui_description' in health ? (health.webui_description ?? null) : null
); )
} }
// Extract and store backend max graph nodes limit // Extract and store backend max graph nodes limit
@ -156,36 +163,51 @@ const useBackendState = createSelectors(useBackendStateStoreBase)
export { useBackendState } export { useBackendState }
const parseTokenPayload = (token: string): { sub?: string; role?: string } => { const parseTokenPayload = (token: string): { sub?: string; role?: string; user_id?: string } => {
try { try {
// JWT tokens are in the format: header.payload.signature // JWT tokens are in the format: header.payload.signature
const parts = token.split('.'); const parts = token.split('.')
if (parts.length !== 3) return {}; if (parts.length !== 3) return {}
const payload = JSON.parse(atob(parts[1])); const payload = JSON.parse(atob(parts[1]))
return payload; return payload
} catch (e) { } catch (e) {
console.error('Error parsing token payload:', e); console.error('Error parsing token payload:', e)
return {}; return {}
} }
}; }
const getUsernameFromToken = (token: string): string | null => { const getUsernameFromToken = (token: string): string | null => {
const payload = parseTokenPayload(token); const payload = parseTokenPayload(token)
return payload.sub || null; return payload.sub || null
}; }
const getUserIdFromToken = (token: string): string | null => {
const payload = parseTokenPayload(token)
return payload.user_id || payload.sub || null
}
const isGuestToken = (token: string): boolean => { const isGuestToken = (token: string): boolean => {
const payload = parseTokenPayload(token); const payload = parseTokenPayload(token)
return payload.role === 'guest'; return payload.role === 'guest'
}; }
const initAuthState = (): { isAuthenticated: boolean; isGuestMode: boolean; coreVersion: string | null; apiVersion: string | null; username: string | null; webuiTitle: string | null; webuiDescription: string | null } => { const initAuthState = (): {
const token = localStorage.getItem('LIGHTRAG-API-TOKEN'); isAuthenticated: boolean
const coreVersion = localStorage.getItem('LIGHTRAG-CORE-VERSION'); isGuestMode: boolean
const apiVersion = localStorage.getItem('LIGHTRAG-API-VERSION'); coreVersion: string | null
const webuiTitle = localStorage.getItem('LIGHTRAG-WEBUI-TITLE'); apiVersion: string | null
const webuiDescription = localStorage.getItem('LIGHTRAG-WEBUI-DESCRIPTION'); username: string | null
const username = token ? getUsernameFromToken(token) : null; userId: string | null
webuiTitle: string | null
webuiDescription: string | null
} => {
const token = localStorage.getItem('LIGHTRAG-API-TOKEN')
const coreVersion = localStorage.getItem('LIGHTRAG-CORE-VERSION')
const apiVersion = localStorage.getItem('LIGHTRAG-API-VERSION')
const webuiTitle = localStorage.getItem('LIGHTRAG-WEBUI-TITLE')
const webuiDescription = localStorage.getItem('LIGHTRAG-WEBUI-DESCRIPTION')
const username = token ? getUsernameFromToken(token) : null
const userId = token ? getUserIdFromToken(token) : null
if (!token) { if (!token) {
return { return {
@ -194,9 +216,10 @@ const initAuthState = (): { isAuthenticated: boolean; isGuestMode: boolean; core
coreVersion: coreVersion, coreVersion: coreVersion,
apiVersion: apiVersion, apiVersion: apiVersion,
username: null, username: null,
userId: null,
webuiTitle: webuiTitle, webuiTitle: webuiTitle,
webuiDescription: webuiDescription, webuiDescription: webuiDescription
}; }
} }
return { return {
@ -205,14 +228,15 @@ const initAuthState = (): { isAuthenticated: boolean; isGuestMode: boolean; core
coreVersion: coreVersion, coreVersion: coreVersion,
apiVersion: apiVersion, apiVersion: apiVersion,
username: username, username: username,
userId: userId,
webuiTitle: webuiTitle, webuiTitle: webuiTitle,
webuiDescription: webuiDescription, webuiDescription: webuiDescription
}; }
}; }
export const useAuthStore = create<AuthState>(set => { export const useAuthStore = create<AuthState>((set) => {
// Get initial state from localStorage // Get initial state from localStorage
const initialState = initAuthState(); const initialState = initAuthState()
return { return {
isAuthenticated: initialState.isAuthenticated, isAuthenticated: initialState.isAuthenticated,
@ -220,97 +244,109 @@ export const useAuthStore = create<AuthState>(set => {
coreVersion: initialState.coreVersion, coreVersion: initialState.coreVersion,
apiVersion: initialState.apiVersion, apiVersion: initialState.apiVersion,
username: initialState.username, username: initialState.username,
userId: initialState.userId,
webuiTitle: initialState.webuiTitle, webuiTitle: initialState.webuiTitle,
webuiDescription: initialState.webuiDescription, webuiDescription: initialState.webuiDescription,
login: (token, isGuest = false, coreVersion = null, apiVersion = null, webuiTitle = null, webuiDescription = null) => { login: (
localStorage.setItem('LIGHTRAG-API-TOKEN', token); token,
isGuest = false,
coreVersion = null,
apiVersion = null,
webuiTitle = null,
webuiDescription = null
) => {
localStorage.setItem('LIGHTRAG-API-TOKEN', token)
if (coreVersion) { if (coreVersion) {
localStorage.setItem('LIGHTRAG-CORE-VERSION', coreVersion); localStorage.setItem('LIGHTRAG-CORE-VERSION', coreVersion)
} }
if (apiVersion) { if (apiVersion) {
localStorage.setItem('LIGHTRAG-API-VERSION', apiVersion); localStorage.setItem('LIGHTRAG-API-VERSION', apiVersion)
} }
if (webuiTitle) { if (webuiTitle) {
localStorage.setItem('LIGHTRAG-WEBUI-TITLE', webuiTitle); localStorage.setItem('LIGHTRAG-WEBUI-TITLE', webuiTitle)
} else { } else {
localStorage.removeItem('LIGHTRAG-WEBUI-TITLE'); localStorage.removeItem('LIGHTRAG-WEBUI-TITLE')
} }
if (webuiDescription) { if (webuiDescription) {
localStorage.setItem('LIGHTRAG-WEBUI-DESCRIPTION', webuiDescription); localStorage.setItem('LIGHTRAG-WEBUI-DESCRIPTION', webuiDescription)
} else { } else {
localStorage.removeItem('LIGHTRAG-WEBUI-DESCRIPTION'); localStorage.removeItem('LIGHTRAG-WEBUI-DESCRIPTION')
} }
const username = getUsernameFromToken(token); const username = getUsernameFromToken(token)
const userId = getUserIdFromToken(token)
set({ set({
isAuthenticated: true, isAuthenticated: true,
isGuestMode: isGuest, isGuestMode: isGuest,
username: username, username: username,
userId: userId,
coreVersion: coreVersion, coreVersion: coreVersion,
apiVersion: apiVersion, apiVersion: apiVersion,
webuiTitle: webuiTitle, webuiTitle: webuiTitle,
webuiDescription: webuiDescription, webuiDescription: webuiDescription
}); })
}, },
logout: () => { logout: () => {
localStorage.removeItem('LIGHTRAG-API-TOKEN'); localStorage.removeItem('LIGHTRAG-API-TOKEN')
const coreVersion = localStorage.getItem('LIGHTRAG-CORE-VERSION'); const coreVersion = localStorage.getItem('LIGHTRAG-CORE-VERSION')
const apiVersion = localStorage.getItem('LIGHTRAG-API-VERSION'); const apiVersion = localStorage.getItem('LIGHTRAG-API-VERSION')
const webuiTitle = localStorage.getItem('LIGHTRAG-WEBUI-TITLE'); const webuiTitle = localStorage.getItem('LIGHTRAG-WEBUI-TITLE')
const webuiDescription = localStorage.getItem('LIGHTRAG-WEBUI-DESCRIPTION'); const webuiDescription = localStorage.getItem('LIGHTRAG-WEBUI-DESCRIPTION')
set({ set({
isAuthenticated: false, isAuthenticated: false,
isGuestMode: false, isGuestMode: false,
username: null, username: null,
userId: null,
coreVersion: coreVersion, coreVersion: coreVersion,
apiVersion: apiVersion, apiVersion: apiVersion,
webuiTitle: webuiTitle, webuiTitle: webuiTitle,
webuiDescription: webuiDescription, webuiDescription: webuiDescription
}); })
}, },
setVersion: (coreVersion, apiVersion) => { setVersion: (coreVersion, apiVersion) => {
// Update localStorage // Update localStorage
if (coreVersion) { if (coreVersion) {
localStorage.setItem('LIGHTRAG-CORE-VERSION', coreVersion); localStorage.setItem('LIGHTRAG-CORE-VERSION', coreVersion)
} }
if (apiVersion) { if (apiVersion) {
localStorage.setItem('LIGHTRAG-API-VERSION', apiVersion); localStorage.setItem('LIGHTRAG-API-VERSION', apiVersion)
} }
// Update state // Update state
set({ set({
coreVersion: coreVersion, coreVersion: coreVersion,
apiVersion: apiVersion apiVersion: apiVersion
}); })
}, },
setCustomTitle: (webuiTitle, webuiDescription) => { setCustomTitle: (webuiTitle, webuiDescription) => {
// Update localStorage // Update localStorage
if (webuiTitle) { if (webuiTitle) {
localStorage.setItem('LIGHTRAG-WEBUI-TITLE', webuiTitle); localStorage.setItem('LIGHTRAG-WEBUI-TITLE', webuiTitle)
} else { } else {
localStorage.removeItem('LIGHTRAG-WEBUI-TITLE'); localStorage.removeItem('LIGHTRAG-WEBUI-TITLE')
} }
if (webuiDescription) { if (webuiDescription) {
localStorage.setItem('LIGHTRAG-WEBUI-DESCRIPTION', webuiDescription); localStorage.setItem('LIGHTRAG-WEBUI-DESCRIPTION', webuiDescription)
} else { } else {
localStorage.removeItem('LIGHTRAG-WEBUI-DESCRIPTION'); localStorage.removeItem('LIGHTRAG-WEBUI-DESCRIPTION')
} }
// Update state // Update state
set({ set({
webuiTitle: webuiTitle, webuiTitle: webuiTitle,
webuiDescription: webuiDescription webuiDescription: webuiDescription
}); })
} }
}; }
}); })

4
pyproject.toml
View file

@ -44,6 +44,8 @@ dependencies = [
"psycopg2-binary", "psycopg2-binary",
"openai", "openai",
"httpx", "httpx",
"redis",
"pydantic-settings",
] ]
[project.optional-dependencies] [project.optional-dependencies]
@ -99,6 +101,8 @@ api = [
"python-docx>=0.8.11,<2.0.0", # DOCX processing "python-docx>=0.8.11,<2.0.0", # DOCX processing
"python-pptx>=0.6.21,<2.0.0", # PPTX processing "python-pptx>=0.6.21,<2.0.0", # PPTX processing
"sqlalchemy>=2.0.0,<3.0.0", "sqlalchemy>=2.0.0,<3.0.0",
"pydantic_settings",
"neo4j>=5.0.0,<7.0.0",
] ]
# Advanced document processing engine (optional) # Advanced document processing engine (optional)

0
service/app/__init__.py Normal file
View file

26
service/app/api/routes.py
View file

@ -32,17 +32,13 @@ def create_session(session_in: SessionCreate, db: Session = Depends(get_db)):
# Actually, let's just create a user on the fly for this session if we don't have auth. # Actually, let's just create a user on the fly for this session if we don't have auth.
manager = HistoryManager(db) manager = HistoryManager(db)
# Check if we have any user, if not create one. # User logic removed
from app.models.models import User # Using a fixed UUID for demonstration purposes. In a real application,
user = db.query(User).first() # this would come from an authenticated user.
if not user: fixed_user_id = UUID("00000000-0000-0000-0000-000000000001")
user = User(username="default_user", email="default@example.com")
db.add(user)
db.commit()
db.refresh(user)
session = manager.create_session( session = manager.create_session(
user_id=user.id, user_id=fixed_user_id,
title=session_in.title, title=session_in.title,
rag_config=session_in.rag_config rag_config=session_in.rag_config
) )
@ -51,13 +47,13 @@ def create_session(session_in: SessionCreate, db: Session = Depends(get_db)):
@router.get("/sessions", response_model=List[SessionResponse]) @router.get("/sessions", response_model=List[SessionResponse])
def list_sessions(skip: int = 0, limit: int = 100, db: Session = Depends(get_db)): def list_sessions(skip: int = 0, limit: int = 100, db: Session = Depends(get_db)):
manager = HistoryManager(db) manager = HistoryManager(db)
# Again, need user_id. Using the default user strategy. # User logic removed
from app.models.models import User pass
user = db.query(User).first() # Using a fixed UUID for demonstration purposes. In a real application,
if not user: # this would come from an authenticated user.
return [] fixed_user_id = UUID("00000000-0000-0000-0000-000000000001")
sessions = manager.list_sessions(user_id=user.id, skip=skip, limit=limit) sessions = manager.list_sessions(user_id=fixed_user_id, skip=skip, limit=limit)
return sessions return sessions
@router.get("/sessions/{session_id}/history") @router.get("/sessions/{session_id}/history")

25
service/app/models/models.py
View file

@ -5,37 +5,24 @@ from sqlalchemy.orm import relationship
from sqlalchemy.sql import func from sqlalchemy.sql import func
from app.core.database import Base from app.core.database import Base
class User(Base):
__tablename__ = "users"
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
username = Column(String(50), unique=True, nullable=False)
email = Column(String(255), unique=True, nullable=False, index=True)
full_name = Column(String(100), nullable=True)
created_at = Column(DateTime(timezone=True), server_default=func.now())
is_active = Column(Boolean, default=True)
sessions = relationship("ChatSession", back_populates="user", cascade="all, delete-orphan")
class ChatSession(Base): class ChatSession(Base):
__tablename__ = "chat_sessions" __tablename__ = "lightrag_chat_sessions_history"
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4) id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
user_id = Column(UUID(as_uuid=True), ForeignKey("users.id", ondelete="CASCADE"), nullable=False, index=True) user_id = Column(String(255), nullable=False, index=True)
title = Column(String(255), nullable=True) title = Column(String(255), nullable=True)
rag_config = Column(JSON, default={}) rag_config = Column(JSON, default={})
summary = Column(Text, nullable=True) summary = Column(Text, nullable=True)
last_message_at = Column(DateTime(timezone=True), server_default=func.now(), index=True) last_message_at = Column(DateTime(timezone=True), server_default=func.now(), index=True)
created_at = Column(DateTime(timezone=True), server_default=func.now()) created_at = Column(DateTime(timezone=True), server_default=func.now())
user = relationship("User", back_populates="sessions")
messages = relationship("ChatMessage", back_populates="session", cascade="all, delete-orphan") messages = relationship("ChatMessage", back_populates="session", cascade="all, delete-orphan")
class ChatMessage(Base): class ChatMessage(Base):
__tablename__ = "chat_messages" __tablename__ = "lightrag_chat_messages_history"
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4) id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
session_id = Column(UUID(as_uuid=True), ForeignKey("chat_sessions.id", ondelete="CASCADE"), nullable=False) session_id = Column(UUID(as_uuid=True), ForeignKey("lightrag_chat_sessions_history.id", ondelete="CASCADE"), nullable=False)
role = Column(String(20), nullable=False) # user, assistant, system role = Column(String(20), nullable=False) # user, assistant, system
content = Column(Text, nullable=False) content = Column(Text, nullable=False)
token_count = Column(Integer, nullable=True) token_count = Column(Integer, nullable=True)
@ -46,10 +33,10 @@ class ChatMessage(Base):
citations = relationship("MessageCitation", back_populates="message", cascade="all, delete-orphan") citations = relationship("MessageCitation", back_populates="message", cascade="all, delete-orphan")
class MessageCitation(Base): class MessageCitation(Base):
__tablename__ = "message_citations" __tablename__ = "lightrag_message_citations_history"
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4) id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
message_id = Column(UUID(as_uuid=True), ForeignKey("chat_messages.id", ondelete="CASCADE"), nullable=False) message_id = Column(UUID(as_uuid=True), ForeignKey("lightrag_chat_messages_history.id", ondelete="CASCADE"), nullable=False)
source_doc_id = Column(String(255), nullable=False, index=True) source_doc_id = Column(String(255), nullable=False, index=True)
file_path = Column(Text, nullable=False) file_path = Column(Text, nullable=False)
chunk_content = Column(Text, nullable=True) chunk_content = Column(Text, nullable=True)

9
service/app/services/history_manager.py
View file

@ -34,7 +34,7 @@ class HistoryManager:
return list(reversed(context)) return list(reversed(context))
def create_session(self, user_id: uuid.UUID, title: str = None, rag_config: dict = None) -> ChatSession: def create_session(self, user_id: str, title: str = None, rag_config: dict = None) -> ChatSession:
session = ChatSession( session = ChatSession(
user_id=user_id, user_id=user_id,
title=title, title=title,
@ -48,7 +48,7 @@ class HistoryManager:
def get_session(self, session_id: uuid.UUID) -> Optional[ChatSession]: def get_session(self, session_id: uuid.UUID) -> Optional[ChatSession]:
return self.db.query(ChatSession).filter(ChatSession.id == session_id).first() return self.db.query(ChatSession).filter(ChatSession.id == session_id).first()
def list_sessions(self, user_id: uuid.UUID, skip: int = 0, limit: int = 100) -> List[ChatSession]: def list_sessions(self, user_id: str, skip: int = 0, limit: int = 100) -> List[ChatSession]:
return ( return (
self.db.query(ChatSession) self.db.query(ChatSession)
.filter(ChatSession.user_id == user_id) .filter(ChatSession.user_id == user_id)
@ -80,11 +80,12 @@ class HistoryManager:
def save_citations(self, message_id: uuid.UUID, citations: List[Dict]): def save_citations(self, message_id: uuid.UUID, citations: List[Dict]):
for cit in citations: for cit in citations:
content = "\n".join(cit.get("content", []))
citation = MessageCitation( citation = MessageCitation(
message_id=message_id, message_id=message_id,
source_doc_id=cit.get("source_doc_id", "unknown"), source_doc_id=cit.get("reference_id", "unknown"),
file_path=cit.get("file_path", "unknown"), file_path=cit.get("file_path", "unknown"),
chunk_content=cit.get("chunk_content"), chunk_content=content,
relevance_score=cit.get("relevance_score") relevance_score=cit.get("relevance_score")
) )
self.db.add(citation) self.db.add(citation)

38
service/init_db.py
View file

@ -9,7 +9,7 @@ from dotenv import load_dotenv
load_dotenv(os.path.join(os.path.dirname(os.path.abspath(__file__)), "../.env")) load_dotenv(os.path.join(os.path.dirname(os.path.abspath(__file__)), "../.env"))
from app.core.database import engine, Base, SessionLocal from app.core.database import engine, Base, SessionLocal
from app.models.models import User, ChatSession, ChatMessage, MessageCitation # Import models to register them from app.models.models import ChatSession, ChatMessage, MessageCitation # Import models to register them
from app.core.config import settings from app.core.config import settings
logging.basicConfig(level=logging.INFO) logging.basicConfig(level=logging.INFO)
@ -22,39 +22,9 @@ def init_db():
logger.info("Tables created successfully!") logger.info("Tables created successfully!")
# Create default users from AUTH_ACCOUNTS # Create default users from AUTH_ACCOUNTS
if settings.AUTH_ACCOUNTS: # User table removed, so we don't need to create users anymore.
db = SessionLocal() # Logic kept as comment or removed.
try: logger.info("Database initialized.")
accounts = settings.AUTH_ACCOUNTS.split(',')
for account in accounts:
if ':' in account:
username, password = account.split(':', 1)
username = username.strip()
# Check if user exists
existing_user = db.query(User).filter(User.username == username).first()
if not existing_user:
logger.info(f"Creating default user: {username}")
# Note: In a real app, password should be hashed.
# For now, we are just creating the user record.
# The User model doesn't have a password field in the provided schema,
# so we might need to add it or just store the user for now.
# Looking at models.py, User has: username, email, full_name. No password.
# I will use username as email for now if email is required.
new_user = User(
username=username,
email=f"{username}@example.com",
full_name=username
)
db.add(new_user)
else:
logger.info(f"User {username} already exists.")
db.commit()
logger.info("Default users processed.")
except Exception as e:
logger.error(f"Error creating default users: {e}")
db.rollback()
finally:
db.close()
except Exception as e: except Exception as e:
logger.error(f"Error creating tables: {e}") logger.error(f"Error creating tables: {e}")

3056
uv.lock generated
View file

File diff suppressed because it is too large Load diff