OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-05-19 12:51:35 +00:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

chore: formatting

Browse Source
This commit is contained in:
Aryan Kothari 2024-07-22 14:45:47 -04:00
parent a0667dfd1b
commit f531a51e91
1 changed files with 18 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
backend/apps/webui/models/chats.py
View File

@ -244,13 +244,14 @@ class ChatTable:
.all() .all()
) )
return [ChatModel.model_validate(chat) for chat in all_chats] return [ChatModel.model_validate(chat) for chat in all_chats]
def get_chat_title_id_list_by_user_id( def get_chat_title_id_list_by_user_id(
self, self,
user_id: str, user_id: str,
include_archived: bool = False, include_archived: bool = False,
skip: int = 0, skip: int = 0,
limit: int = 50, limit: int = 50,
) -> List[ChatTitleIdResponse]: ) -> List[ChatTitleIdResponse]:
with get_db() as db: with get_db() as db:
query = db.query(Chat).filter_by(user_id=user_id) query = db.query(Chat).filter_by(user_id=user_id)
if not include_archived: if not include_archived:
@ -259,17 +260,24 @@ class ChatTable:
all_chats = ( all_chats = (
query.order_by(Chat.updated_at.desc()) query.order_by(Chat.updated_at.desc())
# limit cols # limit cols
.with_entities(Chat.id, Chat.title, Chat.updated_at, Chat.created_at) .with_entities(
.all() Chat.id, Chat.title, Chat.updated_at, Chat.created_at
).all()
) )
# result has to be destrctured from sqlalchemy `row` and mapped to a dict since the `ChatModel`is not the returned dataclass. # result has to be destrctured from sqlalchemy `row` and mapped to a dict since the `ChatModel`is not the returned dataclass.
return list(map(lambda row: ChatTitleIdResponse.model_validate({ return list(
"id": row[0], map(
"title": row[1], lambda row: ChatTitleIdResponse.model_validate(
"updated_at": row[2], {
"created_at": row[3] "id": row[0],
}), all_chats)) "title": row[1],
"updated_at": row[2],
"created_at": row[3],
}
),
all_chats,
)
)
def get_chat_list_by_chat_ids( def get_chat_list_by_chat_ids(
self, chat_ids: List[str], skip: int = 0, limit: int = 50 self, chat_ids: List[str], skip: int = 0, limit: int = 50