From b4eea78aff28a1506794e6eb8f9c32c9a4c08f98 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Sat, 27 Sep 2025 04:06:42 -0500 Subject: [PATCH] refac --- backend/open_webui/models/messages.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/backend/open_webui/models/messages.py b/backend/open_webui/models/messages.py index 197befa061..2e70b8a4d7 100644 --- a/backend/open_webui/models/messages.py +++ b/backend/open_webui/models/messages.py @@ -222,7 +222,7 @@ class MessageTable: def get_messages_by_parent_id( self, channel_id: str, parent_id: str, skip: int = 0, limit: int = 50 - ) -> list[MessageModel]: + ) -> list[MessageReplyToResponse]: with get_db() as db: message = db.get(Message, parent_id) @@ -242,7 +242,19 @@ class MessageTable: if len(all_messages) < limit: all_messages.append(message) - return [MessageModel.model_validate(message) for message in all_messages] + return [ + MessageReplyToResponse.model_validate( + { + **MessageModel.model_validate(message).model_dump(), + "reply_to_message": ( + self.get_message_by_id(message.reply_to_id).model_dump() + if message.reply_to_id + else None + ), + } + ) + for message in all_messages + ] def update_message_by_id( self, id: str, form_data: MessageForm