54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from fastapi import APIRouter, HTTPException, status
|
||
|
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
from app.deps import AuthDep
|
||
|
|
from app.schemas.tasks import TaskStatusResponse
|
||
|
|
from app.services.lightrag_service import lightrag_service
|
||
|
|
from app.services.supabase_rest import supabase_rest
|
||
|
|
from app.services.task_tracker import task_tracker
|
||
|
|
from app.workers.tasks import lightrag_index_pipeline
|
||
|
|
|
||
|
|
router = APIRouter(prefix="/lightrag")
|
||
|
|
|
||
|
|
|
||
|
|
class IndexRequest(BaseModel):
|
||
|
|
document_id: str
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/index", response_model=TaskStatusResponse)
|
||
|
|
async def enqueue_lightrag_index(payload: IndexRequest, auth: AuthDep) -> TaskStatusResponse:
|
||
|
|
document = supabase_rest.select_one(
|
||
|
|
"documents", {"id": payload.document_id, "user_id": auth.user_id}
|
||
|
|
)
|
||
|
|
if not document:
|
||
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Document not found")
|
||
|
|
|
||
|
|
raw_text = str(document.get("raw_text") or "").strip()
|
||
|
|
if not raw_text:
|
||
|
|
raise HTTPException(
|
||
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
||
|
|
detail="文档尚未生成 raw_text，无法索引",
|
||
|
|
)
|
||
|
|
|
||
|
|
supabase_rest.update(
|
||
|
|
"documents", {"id": payload.document_id}, {"index_status": "pending"}
|
||
|
|
)
|
||
|
|
task = task_tracker.create_task(
|
||
|
|
user_id=auth.user_id, document_id=payload.document_id, task_type="index"
|
||
|
|
)
|
||
|
|
lightrag_index_pipeline.delay(task.task_id, payload.document_id, auth.user_id)
|
||
|
|
return task
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/health")
|
||
|
|
async def lightrag_health():
|
||
|
|
result = await lightrag_service.run_healthcheck()
|
||
|
|
if not result.get("ok"):
|
||
|
|
raise HTTPException(
|
||
|
|
status_code=status.HTTP_503_SERVICE_UNAVAILABLE,
|
||
|
|
detail=result.get("error", "LightRAG 未就绪"),
|
||
|
|
)
|
||
|
|
return result
|