From 282a631b9ceee9a634ee1d93751a5254ed37ccef Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期二, 18 三月 2025 10:10:48 +0800 Subject: [PATCH] 首页知识库对话-rg --- app/api/dialog.py | 42 ++++++++++++++++++++++++++++++++++++++---- 1 files changed, 38 insertions(+), 4 deletions(-) diff --git a/app/api/dialog.py b/app/api/dialog.py index 28bd1df..b11ef36 100644 --- a/app/api/dialog.py +++ b/app/api/dialog.py @@ -2,10 +2,11 @@ from fastapi import APIRouter, Depends from app.api import Response, get_current_user, ResponseList -from app.models.dialog_model import dialogData, dialogDataUpdate +from app.models.dialog_model import dialogData, dialogDataUpdate, dialogList from app.models.base_model import get_db from app.models.user_model import UserModel -from app.service.dialog import get_dialog_list, create_dialog_service, update_dialog_status_service +from app.service.dialog import get_dialog_list, create_dialog_service, update_dialog_status_service, \ + delete_dialog_service, update_dialog_icon_service, get_dialog_manage_list, sync_dialog_service dialog_router = APIRouter() @@ -19,7 +20,9 @@ current_user: UserModel = Depends(get_current_user), db=Depends(get_db)): if current and not pageSize: - return ResponseList(code=400, msg="缂哄皯鍙傛暟") + return Response(code=400, msg="缂哄皯鍙傛暟") + if status and status not in ["0", "1"]: + return Response(code=400, msg="Parameter status exception") return Response(code=200, msg="", data=await get_dialog_list(db, current_user.id, keyword, label, status, pageSize, current)) @@ -43,7 +46,38 @@ async def change_dialog_status(dialog: dialogDataUpdate, current_user: UserModel = Depends(get_current_user), db=Depends(get_db)): if dialog.status not in ["0", "1"]: return Response(code=400, msg="invalid parameter value", data={}) - is_create = await update_dialog_status_service(db, dialog.id, dialog.status) + is_create = await update_dialog_status_service(db, dialog.id, dialog.status, current_user.id) + if not is_create: + return Response(code=500, msg="dialog update failure", data={}) + return Response(code=200, msg="dialog update success", data={}) + + +@dialog_router.get("/delete", response_model=Response) +async def sync_knowledge_api(dialogId: str, current_user: UserModel = Depends(get_current_user), db=Depends(get_db)): + is_delete = await delete_dialog_service(db, dialogId) + if not is_delete: + return Response(code=500, msg="knowledge delete failure", data={}) + return Response(code=200, msg="knowledge delete success", data={}) + + +@dialog_router.put("/update_icon", response_model=Response) +async def change_dialog_icon(dialog: dialogDataUpdate, current_user: UserModel = Depends(get_current_user), db=Depends(get_db)): + is_create = await update_dialog_icon_service(db, dialog.id, dialog.icon, dialog.name, dialog.description) + if not is_create: + return Response(code=500, msg="dialog update failure", data={}) + return Response(code=200, msg="dialog update success", data={}) + + +@dialog_router.post("/dialog_list", response_model=Response) +async def dialog_list_api(dialog:dialogList, + current_user: UserModel = Depends(get_current_user), + db=Depends(get_db)): + return Response(code=200, msg="", data=await get_dialog_manage_list(db, current_user.id, dialog.keyword, dialog.label, dialog.status, dialog.pageSize, dialog.current, dialog.mode)) + + +@dialog_router.get("/sync", response_model=Response) +async def sync_dialog_api(dialogId: str, current_user: UserModel = Depends(get_current_user), db=Depends(get_db)): + is_create = await sync_dialog_service(db, dialogId) if not is_create: return Response(code=500, msg="dialog update failure", data={}) return Response(code=200, msg="dialog update success", data={}) \ No newline at end of file -- Gitblit v1.8.0