Update main.py

main.py

@@ -40,7 +40,7 @@ def retriever_to_cache():
     vectorstore = Pinecone.from_existing_index(
         index_name=index_name, embedding=embeddings
     )
-    retriever = vectorstore.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": .7, "k": 50,"filter": {'categorie': {'$eq': 'OF'}}})
+    retriever = vectorstore.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": .7, "k": 40,"filter": {'categorie': {'$eq': 'OF'}}})
     return retriever
 
 @cl.author_rename
@@ -128,7 +128,7 @@ async def start():
     
         await cl.Message(author="🌐🌐🌐",content="", elements=logo).send()
         await cl.Message(
-            author="🌐🌐🌐",content=f"Commencez à poser vos questions sur les données \"{chat_profile}\"\n\n💡Voici des exemples de question \n\t1️⃣ Basée sur les formations : Quelles sont toutes les formations licences générales?\n\t2️⃣ Basée sur les compétences : Quelles sont les compétences de la licence Economie et gestion?\n\t3️⃣ Basée sur les métiers : Quelles sont les métiers possibles de la licence Economie et gestion?\n\t4️⃣ Basée sur un souhait : Quelles formations si je veux travailler dans la vente?\n\t5️⃣ Basée sur un savoir-être : Quelles formations si j'aime travailler en équipe?\n\t6️⃣ Basée sur un état : Quelles formations si je suis créatif?\n\t7️⃣ Question multi-critère : Quelles sont les activités, les compétences et les métiers possibles de la licence Economie et gestion?"
+            author="🌐🌐🌐",content=f"Commencez à poser vos questions sur les données \"{chat_profile}\"\n\n💡Voici des exemples de question \n\t1️⃣ Basée sur les formations : Quelles sont toutes les formations licences générales?\n\t2️⃣ Basée sur les compétences : Quelles sont les compétences de la licence Economie et gestion?\n\t3️⃣ Basée sur les métiers : Quels sont les métiers possibles de la licence Economie et gestion?\n\t4️⃣ Basée sur un souhait : Quelles formations si je veux travailler dans la vente?\n\t5️⃣ Basée sur un savoir-être : Quelles formations si j'aime travailler en équipe?\n\t6️⃣ Basée sur un état : Quelles formations si je suis créatif?\n\t7️⃣ Question multi-critère : Quelles sont les activités, les compétences et les métiers possibles de la licence Economie et gestion?"
         ).send()
         settings = await cl.ChatSettings(
             [
@@ -169,11 +169,11 @@ async def start():
         )
         #question_generator = LLMChain(llm=llm, prompt=CONDENSE_QUESTION_PROMPT)
         #doc_chain = load_qa_chain(streaming_llm, chain_type="stuff")
-
+        relevant=retriever_to_cache()
         qa = ConversationalRetrievalChain.from_llm(
             streaming_llm,
             chain_type="stuff",
-            retriever=retriever_to_cache(),
+            retriever=relevant,
             #combine_docs_chain=doc_chain,
             #question_generator=question_generator,
             memory=memory,