From 08c8e8c9a4d65677de6a493446a605d70efee631 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期二, 10 十二月 2024 16:32:07 +0800
Subject: [PATCH] 12.10 16
---
app/api/v2/public_api.py | 20 ++++++++++++++++----
1 files changed, 16 insertions(+), 4 deletions(-)
diff --git a/app/api/public_api.py b/app/api/v2/public_api.py
similarity index 77%
rename from app/api/public_api.py
rename to app/api/v2/public_api.py
index 06a37dc..380daf5 100644
--- a/app/api/public_api.py
+++ b/app/api/v2/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={})
\ No newline at end of file
+ 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={})
\ No newline at end of file
--
Gitblit v1.8.0