Update main.py

main.py

@@ -809,6 +809,7 @@ async def construction_FCS(romeListArray,settings):
 
 @literal_client.step(type="run")
 async def construction_NCS(romeListArray):
+    print(romeListArray[0])
     context = await contexte(romeListArray)
     emploisST = cl.user_session.get("EmploiST")
     memory = ConversationBufferMemory(return_messages=True)
@@ -898,7 +899,7 @@ async def recuperation_contexte(getNote):
     return getNote + " :\n" + getContext
 @cl.step(type="retrieval")
 async def contexte(romeListArray):
-    results = await API_connexion(romeListArray)
+    results = await API_FranceTravail(romeListArray)
     index = await vectorDatabase_connexion()
     emplois = []
     for i in range(0,len(results)):
@@ -1154,7 +1155,7 @@ async def vectorDatabase_connexion():
     return index
     
 @cl.step(type="tool")
-async def API_connexion(romeListArray):
+async def API_FranceTravail(romeListArray):
     client = Api(client_id=os.environ['POLE_EMPLOI_CLIENT_ID'],
                 client_secret=os.environ['POLE_EMPLOI_CLIENT_SECRET'])
     todayDate = datetime.datetime.today()
@@ -1162,7 +1163,6 @@ async def API_connexion(romeListArray):
     start_dt = todayDate.replace(day=1, month=month, year=year)
     end_dt = datetime.datetime.today()
     params = {"motsCles": romeListArray[0],'lieux':'75D','minCreationDate': dt_to_str_iso(start_dt),'maxCreationDate': dt_to_str_iso(end_dt),'range':'0-149'}
-    print(romeListArray[0])
     search_on_big_data = client.search(params=params)
     results = search_on_big_data["resultats"]
     return results