会话菜单增加

zhaoqingang

2024-12-10 c7d5be908e24ee7677140c1cf983836f2e413cf0

 app/service/dialog.py

@@ -5,7 +5,7 @@
from Log import logger


async def get_dialog_list(db, user_id):
async def get_dialog_list(db, user_id, keyword, page_size, page_index):
    user = db.query(UserModel).filter(UserModel.id == user_id).first()
    if user is None:
        return {"rows": []}
@@ -23,7 +23,22 @@
                group_dialog_table.c.group_id.in_(group_list)
            )
        )
    return {"rows": [kld.to_json() for kld in query.all()]}
    if keyword:
        query = query.filter(DialogModel.name.like('%{}%'.format(keyword)))
    total = query.count()
    if page_size:
        query = query.limit(page_size).offset((page_index - 1) * page_size)

    rows = []
    user_id_set = set()
    for kld in query.all():
        user_id_set.add(kld.tenant_id)
        rows.append(kld.to_json())
    print(rows)
    user_dict = {i.id: i.to_dict() for i in db.query(UserModel).filter(UserModel.id.in_(user_id_set)).all()}
    for r in rows:
        r["user"] = user_dict.get(r["user_id"], {})
    return {"total":  total, "rows": rows}


async def update_session_history(db, data: dict, user_id):