Update main.py

main.py

@@ -149,6 +149,10 @@ async def Search(input, categorie):
 async def setup_conversationalChain():
     model = await LLModel()
     retriever = await Retriever(cl.user_session.get("selectRequest"))
+    ########## Chain with streaming ##########
+    message_history = ChatMessageHistory()
+    memory = ConversationBufferMemory(memory_key="chat_history",output_key="answer",chat_memory=message_history,return_messages=True)
+    
     qa = ConversationalRetrievalChain.from_llm(
         model,
         memory=cl.user_session.get("memory"),
@@ -218,9 +222,6 @@ async def on_chat_start():
         await cl.Message(f"Vous pouvez requêter sur la thématique : {res.get('value')}").send()
         cl.user_session.set("selectRequest", res.get("name"))
         
-    ########## Chain with streaming ##########
-    message_history = ChatMessageHistory()
-    memory = ConversationBufferMemory(memory_key="chat_history",output_key="answer",chat_memory=message_history,return_messages=True)
     await setup_conversationalChain()
     
 @cl.on_chat_resume