From e6c5e89f09637b8d9ebca6895d781663f12646d6 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期二, 17 十二月 2024 16:43:55 +0800
Subject: [PATCH] dify 文档智能
---
app/api/dialog.py | 34 +++++++++++++++++++++++++++++++---
1 files changed, 31 insertions(+), 3 deletions(-)
diff --git a/app/api/dialog.py b/app/api/dialog.py
index e50cfd3..28bd1df 100644
--- a/app/api/dialog.py
+++ b/app/api/dialog.py
@@ -2,10 +2,10 @@
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
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
dialog_router = APIRouter()
@@ -14,8 +14,36 @@
async def dialog_list(current: int,
pageSize: int,
keyword: str = "",
+ 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=200, msg="", data=await get_dialog_list(db, current_user.id, keyword, pageSize, current))
+ 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)
+ 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