From 3b1bf7da6771bd1d6852d3dcc1f906d5ae5c95d1 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期三, 12 三月 2025 13:49:12 +0800
Subject: [PATCH] 增加内容优化

---
 app/api/dialog.py |   68 ++++++++++++++++++++++++++++++++--
 1 files changed, 64 insertions(+), 4 deletions(-)

diff --git a/app/api/dialog.py b/app/api/dialog.py
index cb2b1e4..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 import RoleParameter, GroupModel, KnowledgeModel
+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
+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()
 
@@ -14,10 +15,69 @@
 async def dialog_list(current: int,
                       pageSize: int,
                       keyword: str = "",
-                      label: int =0,
+                      label: int = 0,
                       status: str ="",
                       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))
+
+
+@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, 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={})
+
+
+@dialog_router.get("/update", response_model=Response)
+async def change_dialog_data(dialog: dialogDataUpdate, 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)
+    # if not is_create:
+    #     return Response(code=500, msg="role create failure", data={})
+    return Response(code=200, msg="role create success", data={})
+
+
+@dialog_router.put("/status", response_model=Response)
+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, 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