12.10 16

zhaoqingang

2024-12-10 08c8e8c9a4d65677de6a493446a605d70efee631

 app/api/v2/public_api.py

File was renamed from app/api/public_api.py
@@ -8,8 +8,9 @@
from app.config.const import IMAGE_TO_TEXT, DOCUMENT_TO_CLEANING, DOCUMENT_TO_REPORT, DIFY, BISHENG, RAGFLOW
from app.models.base_model import get_db
from app.models.public_api_model import DfToken, AppRegister
from app.service.common.api_token import DfTokenDao
from app.service.common.app_register import AppRegisterDao
from app.service.v2.api_token import DfTokenDao
from app.service.v2.app_register import AppRegisterDao
from app.service.v2.initialize_data import dialog_menu_sync
from app.task.sync_resources import sync_knowledge, sync_dialog, sync_agent, sync_llm, sync_resource

public_api = APIRouter()
@@ -47,7 +48,7 @@


@public_api.get("/sync/resource", response_model=Response)
async def user_group_list(resource_type:int, db=Depends(get_db)):
async def sync_resource_data(resource_type:int, db=Depends(get_db)):
    if resource_type == 1:
        await sync_knowledge()
    elif resource_type == 2:
@@ -59,4 +60,15 @@
    else:
        await sync_resource()

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


@public_api.get("/sync/dialog_menu", response_model=Response)
async def sync_dialog_menu(db=Depends(get_db)):
    try:
        await dialog_menu_sync(db)
    except Exception as e:
        logger.error(e)
        return Response(code=500, msg="failed", data={})

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