From 20bbed188fd7833d14801dc202fa3cdbb085ca2b Mon Sep 17 00:00:00 2001
From: xuyonghao <898441624@qq.com>
Date: 星期五, 20 十二月 2024 11:51:34 +0800
Subject: [PATCH] 优化excel合并脚本

---
 app/api/dialog.py |   21 +++++++++++++++++++--
 1 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/app/api/dialog.py b/app/api/dialog.py
index 44030a2..6e97966 100644
--- a/app/api/dialog.py
+++ b/app/api/dialog.py
@@ -5,7 +5,8 @@
 from app.models.dialog_model import dialogData, dialogDataUpdate
 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
 
 dialog_router = APIRouter()
 
@@ -25,7 +26,7 @@
 
 @dialog_router.post("/create", response_model=Response)
 async def create_dialog_api(dialog: dialogData, current_user: UserModel = Depends(get_current_user), db=Depends(get_db)):
-    is_create = await create_dialog_service(db, dialog.id, dialog.name, dialog.description, dialog.icon, dialog.dialogType, current_user.id)
+    is_create = await create_dialog_service(db, dialog.id, dialog.name, dialog.description, dialog.icon, dialog.dialogType, dialog.mode,current_user.id)
     if not is_create:
         return Response(code=500, msg="role create failure", data={})
     return Response(code=200, msg="role create success", data={})
@@ -46,4 +47,20 @@
     is_create = await update_dialog_status_service(db, dialog.id, dialog.status)
     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)
+    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