47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
from fastapi import APIRouter, HTTPException
|
|
from pydantic import BaseModel, Field
|
|
|
|
from app.services.rag_service import RagService
|
|
|
|
router = APIRouter()
|
|
rag_service = RagService()
|
|
|
|
|
|
class QueryRequest(BaseModel):
|
|
query: str = Field(..., description="用户查询")
|
|
top_k: int = Field(8, description="返回条数")
|
|
|
|
|
|
class PathRequest(BaseModel):
|
|
source: str
|
|
target: str
|
|
|
|
|
|
@router.post("/query")
|
|
async def query(payload: QueryRequest):
|
|
"""优先调用 LightRAG，失败自动降级 Supabase"""
|
|
result = await rag_service.query(payload.query, payload.top_k)
|
|
return {"data": result}
|
|
|
|
|
|
@router.post("/graph")
|
|
async def graph(payload: QueryRequest):
|
|
result = await rag_service.graph(payload.query)
|
|
return {"data": result}
|
|
|
|
|
|
@router.post("/path")
|
|
async def path(payload: PathRequest):
|
|
result = await rag_service.path(payload.source, payload.target)
|
|
return {"data": result}
|
|
|
|
|
|
@router.get("/health/full")
|
|
async def health() -> dict:
|
|
"""最小化健康检查，后续补充依赖自检"""
|
|
try:
|
|
_ = await rag_service.query("ping", top_k=1)
|
|
return {"status": "ok"}
|
|
except Exception as exc: # noqa: BLE001
|
|
raise HTTPException(status_code=500, detail=str(exc))
|