rag-gateway/app/api/dialog.py

# coding:utf-8

from fastapi import APIRouter, Depends
from app.api import Response, get_current_user, ResponseList
from app.models import RoleParameter, GroupModel, KnowledgeModel
from app.models.base_model import get_db
from app.models.user_model import UserModel
from app.service.dialog import get_dialog_list

dialog_router = APIRouter()


@dialog_router.get("/list", response_model=Response)
async def dialog_list(current: int,
                      pageSize: int,
                      keyword: str = "",
                      current_user: UserModel = Depends(get_current_user),
                      db=Depends(get_db)):
    if current and not pageSize:
        return ResponseList(code=400, msg="缺少参数")
    return Response(code=200, msg="", data=await get_dialog_list(db, current_user.id, keyword, pageSize, current))