This commit is contained in:
Timothy J. Baek 2024-07-24 11:25:07 +01:00
parent 3b487cfae9
commit edff071cd2

View File

@ -265,19 +265,17 @@ class ChatTable:
).all() ).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( return [
map( ChatTitleIdResponse.model_validate(
lambda row: ChatTitleIdResponse.model_validate( {
{ "id": chat[0],
"id": row[0], "title": chat[1],
"title": row[1], "updated_at": chat[2],
"updated_at": row[2], "created_at": chat[3],
"created_at": row[3], }
}
),
all_chats,
) )
) for chat in 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