How to Stream Messages with Assistant Icon?

LLMs and AI
1

Currently the message streams in a placeholder, disappears, then reappears with assistant icon. I want the message to stream WITH the assistant chat bubble from the start.

# Current code
message_placeholder = st.empty()
full_response = ""

for response in current_agent.run(prompt, **kwargs):
    full_response += response.content
    message_placeholder.markdown(full_response)

st.session_state.messages.append(...)
message_placeholder.empty()
display_message(...)