chore: User and Assistant messages are grouped together.

2026-03-04 18:47:17 -05:00
parent cfd286da40
commit ae14feac2f
1 changed files with 6 additions and 1 deletions
--- a/src/ui/src/ui/app.py
+++ b/src/ui/src/ui/app.py
@@ -18,7 +18,6 @@ async def main():

    if prompt := st.chat_input("What is up?"):
        st.chat_message("user").markdown(prompt)
-        st.session_state.messages.append({"role": "user", "content": prompt})

        agent_client = AgentClient()
        response = await agent_client.send_message(prompt)
@@ -38,6 +37,12 @@ async def main():

        with st.chat_message("assistant"):
            st.markdown(raw_response)
+
+        st.session_state.messages.append({
+            "role": "user",
+            "content": prompt
+        })
+
        st.session_state.messages.append({
            "role": "assistant",
            "content": raw_response