chore: User and Assistant messages are grouped together.

This commit is contained in:
2026-03-04 18:47:17 -05:00
parent cfd286da40
commit ae14feac2f

View File

@@ -18,7 +18,6 @@ async def main():
if prompt := st.chat_input("What is up?"): if prompt := st.chat_input("What is up?"):
st.chat_message("user").markdown(prompt) st.chat_message("user").markdown(prompt)
st.session_state.messages.append({"role": "user", "content": prompt})
agent_client = AgentClient() agent_client = AgentClient()
response = await agent_client.send_message(prompt) response = await agent_client.send_message(prompt)
@@ -38,6 +37,12 @@ async def main():
with st.chat_message("assistant"): with st.chat_message("assistant"):
st.markdown(raw_response) st.markdown(raw_response)
st.session_state.messages.append({
"role": "user",
"content": prompt
})
st.session_state.messages.append({ st.session_state.messages.append({
"role": "assistant", "role": "assistant",
"content": raw_response "content": raw_response