From e078028f1a8da34f3cc2cb5095b8e103a996f553 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期五, 11 四月 2025 10:16:42 +0800
Subject: [PATCH] 首页智能体中心过滤掉自研类型
---
app/service/dialog.py | 6 ++++--
app/api/dialog.py | 3 ++-
2 files changed, 6 insertions(+), 3 deletions(-)
diff --git a/app/api/dialog.py b/app/api/dialog.py
index b11ef36..dfc5f57 100644
--- a/app/api/dialog.py
+++ b/app/api/dialog.py
@@ -17,13 +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 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))
+ 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)
diff --git a/app/service/dialog.py b/app/service/dialog.py
index 82b2b22..cd79559 100644
--- a/app/service/dialog.py
+++ b/app/service/dialog.py
@@ -5,7 +5,7 @@
from app.config.agent_base_url import DF_CHAT_PARAMETERS, DF_CHAT_API_KEY
from app.config.config import settings
-from app.config.const import Dialog_STATSU_DELETE, DF_TYPE, Dialog_STATSU_ON, workflow_server, RG_TYPE
+from app.config.const import Dialog_STATSU_DELETE, DF_TYPE, Dialog_STATSU_ON, workflow_server, RG_TYPE, basic_chat
from app.models import KnowledgeModel, GroupModel, DialogModel, ConversationModel, group_dialog_table, LabelWorkerModel, \
LabelModel, ApiTokenModel
from app.models.user_model import UserModel, UserTokenModel
@@ -15,7 +15,7 @@
from app.task.fetch_agent import get_one_from_ragflow_dialog
-async def get_dialog_list(db, user_id, keyword, label, status, page_size, page_index):
+async def get_dialog_list(db, user_id, keyword, label, status, page_size, page_index, mode):
user = db.query(UserModel).filter(UserModel.id == user_id).first()
if user is None:
return {"rows": []}
@@ -24,6 +24,8 @@
query = query.filter(DialogModel.status == status)
else:
query = query.filter(DialogModel.status != Dialog_STATSU_DELETE)
+ if mode:
+ query = query.filter(DialogModel.mode != basic_chat)
id_list = []
# if label:
# id_list = [i.object_id for i in db.query(LabelWorkerModel).filter(LabelWorkerModel.label_id==label).all()]
--
Gitblit v1.8.0