Update main.py

main.py

@@ -461,7 +461,9 @@ async def start():
 
 @literal_client.step(type="run")
 async def construction_NCS(competenceList):
-    context = await contexte(competenceList)
+    chat_profile = cl.user_session.get("chat_profile")
+    chatProfile = chat_profile.split(' - ')
+    context = await contexte(competenceList, chatProfile[1])
     emploisST = context.to_string(index = False)
     if chatProfile[1] == 'ROMESKILLS':
         romeListArray = cl.user_session.get("codeRomeArray")
@@ -526,8 +528,10 @@ async def recuperation_contexte(getNote):
     getContext = cl.user_session.get(getNote)
     return getNote + " :\n" + getContext
 @cl.step(type="retrieval")
-async def contexte(competence):
-    if chatProfile[1] == 'ROMESKILLS':
+async def contexte(competence, chatProfile):
+    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:
         results = await creation_liste_skills_Esco(competence, cl.user_session.get("categorie"))