Update main.py

main.py

@@ -154,41 +154,26 @@ async def start():
         CONDENSE_QUESTION_PROMPT = PromptTemplate.from_template(_template)
         
         ########## Chain with streaming ##########
-        #message_history = ChatMessageHistory()
-        #memory = ConversationBufferMemory(
-        #    memory_key="chat_history",
-        #    output_key="answer",
-        #    chat_memory=message_history,
-        #    return_messages=True,
-        #)
-        #llm = ChatAnthropic(temperature=0)
-        
+        message_history = ChatMessageHistory()
+        memory = ConversationBufferMemory(
+            memory_key="chat_history",
+            output_key="answer",
+            chat_memory=message_history,
+            return_messages=True,
+        )
         streaming_llm = ChatAnthropic(
             streaming=True,
             temperature=1,
             max_tokens=4000
         )
-        #question_generator = LLMChain(llm=llm, prompt=CONDENSE_QUESTION_PROMPT)
-        #doc_chain = load_qa_chain(streaming_llm, chain_type="stuff")
-        #relevant=retriever_to_cache()
-        memory = ConversationSummaryBufferMemory(
-            llm=streaming_llm,
-            output_key='answer',
-            memory_key='chat_history',
-            return_messages=True
-        )
         qa = ConversationalRetrievalChain.from_llm(
             streaming_llm,
             chain_type="stuff",
             retriever=retriever_to_cache(),
-            #combine_docs_chain=doc_chain,
-            #question_generator=question_generator,
             memory=memory,
             return_source_documents=True,
-            get_chat_history=lambda h : h,
             verbose=False
         )
-    
         cl.user_session.set("conversation_chain", qa)
     elif chatProfile[0] == 'Emplois':
         poleemploi = cl.TaskList()