Browse Source

fix: channel thread

Timothy Jaeryang Baek 4 months ago
parent
commit
b2b56d14cb
1 changed files with 5 additions and 3 deletions
  1. 5 3
      backend/open_webui/models/messages.py

+ 5 - 3
backend/open_webui/models/messages.py

@@ -189,9 +189,11 @@ class MessageTable:
                 .all()
             )
 
-            return [
-                MessageModel.model_validate(message) for message in all_messages
-            ] + [MessageModel.model_validate(message)]
+            # If length of all_messages is less than limit, then add the parent message
+            if len(all_messages) < limit:
+                all_messages.append(message)
+
+            return [MessageModel.model_validate(message) for message in all_messages]
 
     def update_message_by_id(
         self, id: str, form_data: MessageForm