Fixup

mattecasu · Feb 21, 2024 · f47bb3b · f47bb3b
1 parent ce247f7
commit f47bb3b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rag/docs_rag/utils.py b/rag/docs_rag/utils.py
@@ -40,7 +40,7 @@ def get_ui(generate_output):
         st.chat_message("user").write(text)
         with st.spinner("🤔..."):
             answer, answer_to_store = itertools.tee(generate_output(text))
-            st.chat_message("assistant").write(answer)
+            st.chat_message("assistant").write_stream(answer)
         stored_answer = ''.join(answer_to_store)
         st.session_state.messages.append({"role": "user", "content": text})
         st.session_state.messages.append({"role": "assistant", "content": stored_answer})