phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-02 08:44:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

rag page

Browse source
This commit is contained in:
Xi Yan 2024-11-27 16:56:57 -08:00
parent 2ecbbd92ed
commit 9bb6c1346b
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/distribution/ui/page/playground/rag.py
View file

@ -166,18 +166,18 @@ def rag_chat_page():
# Display assistant response # Display assistant response
with st.chat_message("assistant"): with st.chat_message("assistant"):
message_placeholder = st.empty()
retrieval_message_placeholder = st.empty() retrieval_message_placeholder = st.empty()
message_placeholder = st.empty()
full_response = "" full_response = ""
retrieval_response = "" retrieval_response = ""
for log in EventLogger().log(response): for log in EventLogger().log(response):
log.print() log.print()
if log.role == "memory_retrieval": if log.role == "memory_retrieval":
retrieval_response += log.content.replace(">", "").strip() retrieval_response += log.content.replace("====", "").strip()
retrieval_message_placeholder.info(retrieval_response)
else: else:
full_response += log.content full_response += log.content
message_placeholder.markdown(full_response + "▌") message_placeholder.markdown(full_response + "▌")
# retrieval_message_placeholder.info(retrieval_response)
message_placeholder.markdown(full_response) message_placeholder.markdown(full_response)
st.session_state.messages.append( st.session_state.messages.append(