Help Needed: Linking UI Messages to Database for Retry Feature

[jjovalle99]

Description:
I'm working on implementing a Retry button for chat messages but I'm stuck on how to connect the UI elements to the database entries.

Current situation:

1. Chat history is stored in PostgreSQL
2. Backend is built on top of LangGraph (that is the reason for having 3 rows per message)
3. Messages can be identified in the database
4. UI displays message boxes for the conversation

Problem:

1. Can't link the message boxes in the UI to their corresponding database entries
2. This prevents implementing Retry functionality, especially for older messages

Questions:

1. How can I identify each message box? is there any nlux feature that can help me with this?
2. How can I associate each UI message box with its database entry?

Any suggestions or guidance would be greatly appreciated!