rag-gateway/app/api/sync_data.py

# coding:utf-8

from fastapi import APIRouter, Depends
from app.api import Response, get_current_user
from app.models.base_model import get_db
from app.models.user_model import UserModel
from app.task.sync_resources import sync_knowledge, sync_dialog, sync_agent, sync_resource, sync_llm

sync_router = APIRouter()


@sync_router.get("/data", response_model=Response)
async def user_group_list(resource_type:int, current_user: UserModel = Depends(get_current_user),
                          db=Depends(get_db)):
    if resource_type == 1:
        await sync_knowledge()
    elif resource_type == 2:
        await sync_dialog()
    elif resource_type == 3:
        await sync_agent()
    elif resource_type == 4:
        await sync_llm()
    else:
        await sync_resource()

    return Response(code=200, msg="", data={})