Update main.py

main.py

@@ -111,7 +111,7 @@ async def on_message(message: cl.Message):
     )
     index = pc.Index(index_name)
     xq = embeddings.embed_query(message.content)
-    xc = index.query(vector=xq, filter={"categorie": {"$eq": "bibliographie-OPP-DGDIN"}},top_k=150, include_metadata=True)
+    xc = index.query(vector=xq, filter={"categorie": {"$eq": "bibliographie-OPP-DGDIN"}},top_k=120, include_metadata=True)
     context_p = ""
     for result in xc['matches']:
         context_p = context_p + result['metadata']['text']
@@ -139,7 +139,7 @@ async def on_message(message: cl.Message):
                     cl.Text(name="Sources", content=sources_text, display="inline")
                 )
         
-    async with cl.Step(type="run", name="QA Assistant"):
+    async with cl.Step(type="run", name="Réponse de Mistral"):
         async for chunk in runnable.astream(
             {"question": message.content, "context":context_p},
             config=RunnableConfig(callbacks=[