From e078028f1a8da34f3cc2cb5095b8e103a996f553 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期五, 11 四月 2025 10:16:42 +0800
Subject: [PATCH] 首页智能体中心过滤掉自研类型
---
app/api/dialog.py | 30 ++++++++++++++++++++++++------
1 files changed, 24 insertions(+), 6 deletions(-)
diff --git a/app/api/dialog.py b/app/api/dialog.py
index 6e97966..dfc5f57 100644
--- a/app/api/dialog.py
+++ b/app/api/dialog.py
@@ -2,11 +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, \
- delete_dialog_service, update_dialog_icon_service
+ delete_dialog_service, update_dialog_icon_service, get_dialog_manage_list, sync_dialog_service
dialog_router = APIRouter()
@@ -17,11 +17,14 @@
keyword: str = "",
label: int = 0,
status: str ="",
+ mode: int = 0,
current_user: UserModel = Depends(get_current_user),
db=Depends(get_db)):
if current and not pageSize:
- return ResponseList(code=400, msg="缂哄皯鍙傛暟")
- return Response(code=200, msg="", data=await get_dialog_list(db, current_user.id, keyword, label, status, pageSize, current))
+ 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,mode))
@dialog_router.post("/create", response_model=Response)
@@ -44,7 +47,7 @@
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={})
@@ -60,7 +63,22 @@
@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)
+ 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