Update main.py

main.py

@@ -148,7 +148,7 @@ async def Search(input, categorie):
 @cl.step(type="llm")
 async def setup_conversationalChain():
     model = await LLModel()
-    retriever = await Retriever(res.get("name"))
+    retriever = await Retriever(cl.user_session.get("selectRequest"))
     qa = ConversationalRetrievalChain.from_llm(
         model,
         memory=memory,
@@ -160,6 +160,23 @@ async def setup_conversationalChain():
     cl.user_session.set("runnable", qa)
     cl.user_session.set("memory", memory)
 
+@cl.step(type="tool")
+async def switch(value):
+    if value == "Pédagogie durable":
+        return "bibliographie-OPP-DGDIN"
+    elif value == "Lieux d'apprentissage":
+        return "bibliographie-OPP-DGDIN"
+    elif value == "Journée de La Pédagogie":
+        return "year"
+    elif value == "Compétences du CFA Descartes":
+        return "skills"
+    elif value == "Formations Gustave Eiffel":
+        return "OF"
+    elif value == "Vidéos paroles de confiné.es":
+        return "videosTC"
+    elif value == "Offres d'emploi France Travail":
+        return "offreST"
+        
 @cl.on_chat_start
 async def on_chat_start():
     await cl.Message(f"> REVIEWSTREAM").send()
@@ -211,6 +228,11 @@ async def on_chat_resume(thread: ThreadDict):
     memory = ConversationBufferMemory(return_messages=True)
     root_messages = [m for m in thread["steps"] if m["parentId"] == None]
     for message in root_messages:
+        if message["type"] == "assistant_message" and message["output"][29:38] == "thématique":
+            resName = await switch(message["output"][42:])
+            cl.user_session.set("selectRequest", resName)
+            print(message["output"])
+        
         if message["type"] == "user_message":
             memory.chat_memory.add_user_message(message["output"])
         else: