From 9bb6c1346b3c16c388fc7ecbe0dd0f5836ba889f Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Wed, 27 Nov 2024 16:56:57 -0800 Subject: [PATCH] rag page --- llama_stack/distribution/ui/page/playground/rag.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/llama_stack/distribution/ui/page/playground/rag.py b/llama_stack/distribution/ui/page/playground/rag.py index 1b6fce5b3..d187caa09 100644 --- a/llama_stack/distribution/ui/page/playground/rag.py +++ b/llama_stack/distribution/ui/page/playground/rag.py @@ -166,18 +166,18 @@ def rag_chat_page(): # Display assistant response with st.chat_message("assistant"): - message_placeholder = st.empty() retrieval_message_placeholder = st.empty() + message_placeholder = st.empty() full_response = "" retrieval_response = "" for log in EventLogger().log(response): log.print() if log.role == "memory_retrieval": - retrieval_response += log.content.replace(">", "").strip() + retrieval_response += log.content.replace("====", "").strip() + retrieval_message_placeholder.info(retrieval_response) else: full_response += log.content message_placeholder.markdown(full_response + "▌") - # retrieval_message_placeholder.info(retrieval_response) message_placeholder.markdown(full_response) st.session_state.messages.append(