From ae14feac2f0f8a0060e2c89bde83004acc0d4c11 Mon Sep 17 00:00:00 2001 From: aserrador Date: Wed, 4 Mar 2026 18:47:17 -0500 Subject: [PATCH] chore: User and Assistant messages are grouped together. --- src/ui/src/ui/app.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/ui/src/ui/app.py b/src/ui/src/ui/app.py index 1678625..4feeebb 100644 --- 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