optimize rendering chat bubble

Author: daltonnyx

AgentCrew/modules/chat/message/handler.py

@@ -196,7 +196,7 @@ async def get_assistant_response(
                 if chunk_text:
                     # End thinking when chunk_text start
                     if not end_thinking and start_thinking:
-                        self._notify("thinking_completed")
+                        self._notify("thinking_completed", thinking_content)
                         end_thinking = True
                     # Notify about response progress
                     if not self.agent.is_streaming():
@@ -232,7 +232,7 @@ async def get_assistant_response(
                     },
                 )
                 self._messages_append(assistant_message)
-                self._notify("assistant_message_added", assistant_message)
+                self._notify("assistant_message_added", assistant_response)
 
                 # This should allows YOLO can be configured on-the-fly without recalled to config too many times
                 config_management = ConfigManagement()

AgentCrew/modules/gui/components/message_handlers.py

@@ -51,6 +51,7 @@ def handle_user_message_created(self, data):
 
     def handle_response_completed(self, data):
         """Handle response completion."""
+        self.chat_window.chunk_buffer = data
         if self.chat_window.current_response_bubble:
             self.chat_window.current_response_bubble.message_index = (
                 len(self.chat_window.message_handler.streamline_messages) - 1
@@ -105,6 +106,8 @@ def _render_buffered_chunks(self):
             self.chat_window.chat_components.display_response_chunk(
                 self.chat_window.chunk_buffer
             )
+            # Release the buffer after rendering
+            self.chat_window.chunk_buffer = ""
 
     def _render_buffered_thinking(self):
         """Render the latest buffered thinking chunk."""
@@ -116,3 +119,4 @@ def _render_buffered_thinking(self):
             self.chat_window.chat_components.display_thinking_chunk(
                 self.chat_window.thinking_buffer
             )
+            self.chat_window.thinking_buffer = ""  # Clear the buffer after rendering