diff --git a/llama_stack/distribution/ui/page/playground/chat.py b/llama_stack/distribution/ui/page/playground/chat.py index b27e6b768..210211f7a 100644 --- a/llama_stack/distribution/ui/page/playground/chat.py +++ b/llama_stack/distribution/ui/page/playground/chat.py @@ -60,10 +60,16 @@ with st.sidebar: help="Initial instructions given to the AI to set its behavior and context", ) + # Add clear chat button to sidebar + if st.button("Clear Chat", use_container_width=True): + st.session_state.messages = [] + st.rerun() + # Main chat interface st.title("🦙 Chat") + # Initialize chat history if "messages" not in st.session_state: st.session_state.messages = []