问题优化

zhaoqingang

2024-12-13 f2c43713b16e821f4a1ae97804c60f18734d5f6a

 app/service/dialog.py

@@ -27,8 +27,9 @@
        query = query.filter(DialogModel.name.like('%{}%'.format(keyword)))

    if status:
        print(status)
        # print(status)
        query = query.filter(DialogModel.status == status)
    query = query.order_by(DialogModel.update_date.desc())
    total = query.count()
    if page_size:
        query = query.limit(page_size).offset((page_index - 1) * page_size)
@@ -86,3 +87,28 @@
                                                      ConversationModel.dialog_id.__eq__(dialog_id)).order_by(
        ConversationModel.update_time.desc()).all()
    return [i.to_json() for i in session_list]



async def create_dialog_service(db, dialog_id, dialog_name, description, icon, dialog_type, mode, user_id):
    try:
        dialog_model = DialogModel(id=dialog_id,name=dialog_name, description=description,icon=icon, dialog_type=dialog_type, tenant_id=user_id, agent_id=dialog_id, mode=mode)
        db.add(dialog_model)
        db.commit()
        db.refresh(dialog_model)
    except Exception as e:
        logger.error(e)
        db.rollback()
        return False
    return True


async def update_dialog_status_service(db, dialog_id, status):
    try:
        DialogModel(id=dialog_id).update({"status":status})
        db.commit()
    except Exception as e:
        logger.error(e)
        db.rollback()
        return False
    return True