Update message handling for complete state

In messagesStore.ts, use updateMessage for complete messages instead of updateMessagePartial. In component.py, set stored_message.properties.state to 'complete' when a complete message is received. These changes improve message state management and consistency.

Author: edwinjosechittilappilly

src/frontend/src/stores/messagesStore.ts

@@ -27,7 +27,7 @@ export const useMessagesStore = create<MessagesStoreType>((set, get) => ({
         get().updateMessagePartial(messageWithoutText);
       } else {
         // For complete messages, replace entirely
-        get().updateMessagePartial(message);
+        get().updateMessage(message);
       }
       return;
     }

src/lfx/src/lfx/custom/custom_component/component.py

@@ -1591,6 +1591,8 @@ async def send_message(self, message: Message, id_: str | None = None, *, skip_d
                 ):
                     complete_message = await self._stream_message(message.text, stored_message)
                     stored_message.text = complete_message
+                    if complete_message:
+                        stored_message.properties.state = "complete"
                     stored_message = await self._update_stored_message(stored_message)
                 else:
                     # Only send message event for non-streaming messages