首页智能体中心过滤掉自研类型

zhaoqingang

2025-04-11 e078028f1a8da34f3cc2cb5095b8e103a996f553

 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()]
@@ -245,6 +247,7 @@
            if app_dialog:
                dialog.name = app_dialog["name"]
                dialog.description = app_dialog["description"]
                dialog.kb_ids = app_dialog["kb_ids"]
                dialog.update_date = datetime.now()
                db.add(dialog)
                db.commit()