Image messages are not shown after refreshing the chat

[jbolter]

Describe the bug
After generating an image the image appears in the chat correctly. However, once the page is refreshed the images do not appear. This is with SAVE_IMAGES set to true and I have confirmed the messages are in the database. I also noticed if I open the Chat Graph and select the last node in the graph then all the images suddenly appear in the chat.

To Reproduce
Steps to reproduce the behavior:

1. Go to agna.chat
2. Generate an image
3. Confirm the image is in the chat
4. Refresh the page
5. See that the image is gone

Expected behavior
Expected behavior is that the images are rendered correctly on initial load (after a refresh).

Desktop (please complete the following information):

• OS: MacOS
• Chrome
• v126.0.6478.183

Additional context
The message store it seems to lose the images in events.on(EVENTS.receiveMsgs, after this step const graph = toChatGraph(data.messages)
so when it's building the graph data.messages has all the messages but then when const fullPath = resolveChatPath(graph.tree, leaf) is run it does not have all the images in fullPath.

[jbolter]

Well I started a new chat locally and i'm not reproducing this now so I will have to try and get more specific steps to recreate this issue.

[jbolter]

If there is anything I can provide from the "broken" chat to help you debug let me know what that might be.

[jbolter]

Gathered some more info.... the images were appearing in a chat where I had only generated images. As soon as I sent a chat message the images go away after a refresh. The chat graph seems to have split as soon as I sent a message with just text.

[sceuick]

I'll fix this very soon.

[jbolter]

Any timeline on when this might be fixed? Thanks!

[jbolter]

Still hoping for this to be resolved 🤞

[jbolter]

@sceuick is this still something you're going to look at? thanks!