chore: User and Assistant messages are grouped together.
This commit is contained in:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user