From 73335e4aafb59279b002b4eb65cf79357a042c67 Mon Sep 17 00:00:00 2001
From: Xi Yan <xiyan@meta.com>
Date: Wed, 27 Nov 2024 15:31:57 -0800
Subject: [PATCH] playground

---
 llama_stack/distribution/ui/page/playground/chat.py | 6 ++++++
 1 file changed, 6 insertions(+)

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 = []