Update main.py

main.py

@@ -460,10 +460,8 @@ async def start():
     cl.user_session.set("runnable", runnable)
 
 @literal_client.step(type="run")
-async def construction_NCS(competenceList):
-    chat_profile = cl.user_session.get("chat_profile")
-    chatProfile = chat_profile.split(' - ')
-    context = await contexte(competenceList, chatProfile[1])
+async def construction_NCS(competenceList, chatProfile):
+    context = await contexte(competenceList, chatProfile)
     emploisST = context.to_string(index = False)
     if chatProfile[1] == 'ROMESKILLS':
         romeListArray = cl.user_session.get("codeRomeArray")
@@ -529,8 +527,8 @@ async def recuperation_contexte(getNote):
     return getNote + " :\n" + getContext
 @cl.step(type="retrieval")
 async def contexte(competence, chatProfile):
-    chat_profile = cl.user_session.get("chat_profile")
-    chatProfile = chat_profile.split(' - ')
+    #chat_profile = cl.user_session.get("chat_profile")
+    #chatProfile = chat_profile.split(' - ')
     if chatProfile == 'ROMESKILLS':
         results = await creation_liste_code_Rome(competence, cl.user_session.get("categorie"))
     else:
@@ -888,7 +886,10 @@ async def setup_agent(settings):
     if not cl.user_session.get("saveMemory"):
         cl.user_session.set("saveMemory", "")
         
-    await construction_NCS(competenceList)
+    chat_profile = cl.user_session.get("chat_profile")
+    chatProfile = chat_profile.split(' - ')
+    
+    await construction_NCS(competenceList, chatProfile[1])
     
     contextChat = cl.user_session.get("contextChatBot")
     if not contextChat: