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)

 app/api/v2/chat.py

@@ -179,7 +179,7 @@
async def api_get_complex_model(current_user: UserModel = Depends(get_current_user), db: Session = Depends(get_db)): #  current_user: UserModel = Depends(get_current_user)
    status_code = http_200
    data = await service_get_complex_model(db)
    if data:
    if not data:
        status_code = http_500
    return Response(data, media_type="application/json", status_code=status_code)


 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 == 1:
        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()]

 app/service/v2/chat.py

@@ -725,7 +725,7 @@

async def service_complex_model(db, chat_type, model_type, model_name, model_provider):
    if chat_type == 1 and model_type == 1:
        return await set_dialog_model(db, complex_knowledge_chat, model_name)
        return await set_dialog_model(db, complex_knowledge_chat, model_name,model_provider)
    elif chat_type == 1 and model_type == 2:
        return await set_dialog_model(db, complex_knowledge_chat_deep, model_name, model_provider)
    else: