Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-02-16 13:14:40 -06:00
parent 337109e99c
commit e5e39be90f
1 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/models/groups.py
+15 -7
View File
@@ -164,10 +164,14 @@ class GroupTable:
def get_groups(self, filter, db: Optional[Session] = None) -> list[GroupResponse]: def get_groups(self, filter, db: Optional[Session] = None) -> list[GroupResponse]:
with get_db_context(db) as db: with get_db_context(db) as db:
member_count = func.count(GroupMember.user_id).label("member_count") member_count = (
query = db.query(Group, member_count).outerjoin( select(func.count(GroupMember.user_id))
GroupMember, GroupMember.group_id == Group.id .where(GroupMember.group_id == Group.id)
.correlate(Group)
.scalar_subquery()
.label("member_count")
) )
query = db.query(Group, member_count)
if filter: if filter:
if "query" in filter: if "query" in filter:
@@ -218,7 +222,7 @@ class GroupTable:
) )
) )
results = query.group_by(Group.id).order_by(Group.updated_at.desc()).all() results = query.order_by(Group.updated_at.desc()).all()
return [ return [
GroupResponse.model_validate( GroupResponse.model_validate(
@@ -260,11 +264,15 @@ class GroupTable:
total = query.count() total = query.count()
member_count = func.count(GroupMember.user_id).label("member_count") member_count = (
select(func.count(GroupMember.user_id))
.where(GroupMember.group_id == Group.id)
.correlate(Group)
.scalar_subquery()
.label("member_count")
)
results = ( results = (
query.add_columns(member_count) query.add_columns(member_count)
.outerjoin(GroupMember, GroupMember.group_id == Group.id)
.group_by(Group.id)
.order_by(Group.updated_at.desc()) .order_by(Group.updated_at.desc())
.offset(skip) .offset(skip)
.limit(limit) .limit(limit)