selfrelease
fad458b2a7
- UIUX 文档：填充 19 个缺口（多主体画像/健康度/AI+看板/增长域/洞察域/创始人端/OODA/助推/商密） - UIUX 文档：插入 6 个新章节（十四~十九），旧章节重编号为二十~三十一，更新目录和交叉引用 - 作业指导书 x5：导航改为 6 域分组，新增 Context Bar/工作模式/Insight Rail/决策线程/多工作区等 UI 概念 - 新建 docs/2-task-uiux.md：50 个代码落地开发任务，按 P0-P6 分优先级 + 8 Sprint 规划 - 后端/前端：大量新增模型、路由、组件（来自之前 Phase 开发）
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
"""RAG 知识库路由。"""
|
|
|
|
from fastapi import APIRouter, Depends, Query
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.core.database import get_db
|
|
from app.core.dependencies import get_current_user
|
|
from app.models.user import User
|
|
from app.schemas.common import ApiResponse, success
|
|
from app.services.rag import semantic_search, build_context
|
|
|
|
router = APIRouter(prefix="/knowledge", tags=["knowledge"])
|
|
|
|
|
|
@router.get("/search", response_model=ApiResponse[list])
|
|
async def search_knowledge(
|
|
q: str = Query(..., min_length=1),
|
|
top_k: int = Query(default=5, ge=1, le=20),
|
|
db: AsyncSession = Depends(get_db),
|
|
user: User = Depends(get_current_user),
|
|
):
|
|
"""语义搜索知识库。"""
|
|
results = await semantic_search(db, user.tenant_id, q, top_k)
|
|
return success(data=results)
|
|
|
|
|
|
@router.post("/context", response_model=ApiResponse[str])
|
|
async def get_context(req: dict, db: AsyncSession = Depends(get_db), user: User = Depends(get_current_user)):
|
|
"""构建 RAG 上下文。"""
|
|
results = await semantic_search(db, user.tenant_id, req.get("query", ""))
|
|
context = await build_context(results)
|
|
return success(data=context)
|