Update main.py

main.py

@@ -41,7 +41,7 @@ os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
 repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
 
 model = HuggingFaceEndpoint(
-    repo_id=repo_id, max_new_tokens=5000, temperature=1.0, task="text2text-generation", streaming=True
+    repo_id=repo_id, max_new_tokens=5500, temperature=1.0, task="text2text-generation", streaming=True
 )
 os.environ['PINECONE_API_KEY'] = os.environ['PINECONE_API_KEY']
 embeddings = HuggingFaceEmbeddings()
@@ -183,11 +183,10 @@ async def on_message(message: cl.Message):
 
     await cl.Message(content=answer).send()
     search = vectorstore.similarity_search(message.content,k=50, filter={"categorie": {"$eq": "bibliographie-OPP-DGDIN"}})
-    if search:
+    if len(search) > 0:
         test = []
         sources_text = ""
         count = 0
-        search = cl.user_session.get("search")
         for i in range(0,len(search)):
             if search[i].metadata['Lien'] not in test:
                 if count <= 15: