fix: clear the chat history

umbertogriffo · Jan 8, 2024 · 2babff6 · 2babff6
1 parent 6293149
commit 2babff6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/chatbot/chatbot_app.py b/chatbot/chatbot_app.py
@@ -56,13 +56,14 @@ def init_welcome_message() -> None:
         st.write("How can I help you today?")
 
 
-def init_messages() -> None:
+def init_chat_history(conversational_retrieval: ConversationRetrieval) -> None:
     """
     Initialize chat history.
     """
     clear_button = st.sidebar.button("Clear Conversation", key="clear")
     if clear_button or "messages" not in st.session_state:
         st.session_state.messages = []
+        conversational_retrieval.get_chat_history().clear()
 
 
 def display_messages_from_history():
@@ -83,7 +84,7 @@ def main(parameters) -> None:
     init_page(root_folder)
     llm = load_llm(client, model, model_folder)
     conversational_retrieval = load_conversational_retrieval(_llm=llm)
-    init_messages()
+    init_chat_history(conversational_retrieval)
     init_welcome_message()
     display_messages_from_history()