114 lines
3.9 KiB
Python
114 lines
3.9 KiB
Python
"""
|
|
Workspace management routes for LightRAG server.
|
|
"""
|
|
|
|
import logging
|
|
from typing import List, Optional
|
|
from fastapi import APIRouter, Depends, HTTPException, Query, Request
|
|
from pydantic import BaseModel, Field
|
|
from lightrag.api.utils_api import get_combined_auth_dependency
|
|
from lightrag.api.workspace_manager import WorkspaceManager
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
router = APIRouter(
|
|
prefix="/workspaces",
|
|
tags=["workspaces"],
|
|
)
|
|
|
|
|
|
class WorkspaceCreateRequest(BaseModel):
|
|
name: str = Field(..., min_length=1, max_length=100, description="Workspace name (alphanumeric, underscores, hyphens)")
|
|
|
|
|
|
class WorkspaceRenameRequest(BaseModel):
|
|
new_name: str = Field(..., min_length=1, max_length=100, description="New workspace name (alphanumeric, underscores, hyphens)")
|
|
|
|
|
|
class WorkspaceResponse(BaseModel):
|
|
name: str
|
|
path: str
|
|
|
|
|
|
def get_workspace_manager(request: Request):
|
|
"""Dependency to get workspace manager instance."""
|
|
# Access workspace manager from app state via request
|
|
return request.app.state.workspace_manager
|
|
|
|
|
|
@router.get("/", response_model=List[WorkspaceResponse])
|
|
async def list_workspaces(
|
|
search: Optional[str] = Query(None, description="Search workspaces by keyword in name"),
|
|
workspace_manager: WorkspaceManager = Depends(get_workspace_manager),
|
|
):
|
|
"""List all existing workspaces with optional keyword search."""
|
|
try:
|
|
workspaces = workspace_manager.list_workspaces()
|
|
|
|
# Filter by search keyword if provided
|
|
if search:
|
|
search_lower = search.lower()
|
|
workspaces = [ws for ws in workspaces if search_lower in ws.lower()]
|
|
|
|
return [
|
|
WorkspaceResponse(name=ws, path=str(workspace_manager.base_working_dir / ws))
|
|
for ws in workspaces
|
|
]
|
|
except Exception as e:
|
|
logger.error(f"Error listing workspaces: {e}")
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@router.post("/", response_model=WorkspaceResponse)
|
|
async def create_workspace(
|
|
create_request: WorkspaceCreateRequest,
|
|
workspace_manager: WorkspaceManager = Depends(get_workspace_manager),
|
|
):
|
|
"""Create a new workspace."""
|
|
try:
|
|
workspace_manager.create_workspace(create_request.name)
|
|
return WorkspaceResponse(
|
|
name=create_request.name,
|
|
path=str(workspace_manager.base_working_dir / create_request.name)
|
|
)
|
|
except ValueError as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
except Exception as e:
|
|
logger.error(f"Error creating workspace: {e}")
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@router.patch("/{workspace_name}", response_model=WorkspaceResponse)
|
|
async def rename_workspace(
|
|
workspace_name: str,
|
|
rename_request: WorkspaceRenameRequest,
|
|
workspace_manager: WorkspaceManager = Depends(get_workspace_manager),
|
|
):
|
|
"""Rename a workspace."""
|
|
try:
|
|
await workspace_manager.rename_workspace(workspace_name, rename_request.new_name)
|
|
return WorkspaceResponse(
|
|
name=rename_request.new_name,
|
|
path=str(workspace_manager.base_working_dir / rename_request.new_name)
|
|
)
|
|
except ValueError as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
except Exception as e:
|
|
logger.error(f"Error renaming workspace: {e}")
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@router.delete("/{workspace_name}")
|
|
async def delete_workspace(
|
|
workspace_name: str,
|
|
workspace_manager: WorkspaceManager = Depends(get_workspace_manager),
|
|
):
|
|
"""Delete a workspace."""
|
|
try:
|
|
await workspace_manager.delete_workspace(workspace_name)
|
|
return {"message": f"Workspace '{workspace_name}' deleted successfully"}
|
|
except ValueError as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
except Exception as e:
|
|
logger.error(f"Error deleting workspace: {e}")
|
|
raise HTTPException(status_code=500, detail=str(e)) |