From 2babff6ccf61082455ed8e1ce6f8f2c1f11c3c41 Mon Sep 17 00:00:00 2001
From: Umberto Griffo <1609440+umbertogriffo@users.noreply.github.com>
Date: Mon, 8 Jan 2024 19:11:47 +0000
Subject: [PATCH] fix: clear the chat history

---
 chatbot/chatbot_app.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/chatbot/chatbot_app.py b/chatbot/chatbot_app.py
index fdee9f0..77ef3f8 100644
--- 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()