Update main.py

main.py

@@ -784,9 +784,10 @@ async def contexte(romeListArray):
     context = emplois_list.replace('[','').replace(']','').replace('{','').replace('}','')
     ficheMetier = await Fiche_metier("https://candidat.francetravail.fr/marche-du-travail/fichemetierrome.blocficherome.telechargerpdf?codeRome=", romeListArray[0])
     ficheMetiersCompetencesSavoirs = await Fiche_metier_competences_savoirs(romeListArray[0])
-    metierSecteurContexteTravail = await Metier_secteur_contexte_travail(romeListArray[0])
+    #metierSecteurContexteTravail = await Metier_secteur_contexte_travail(romeListArray[0])
     cl.user_session.set("EmploiST", context)
-    return "Fiche métier Compétences Savoirs :\n" + ficheMetiersCompetencesSavoirs + "\nMetier secteur contexte au travail :\n" + metierSecteurContexteTravail + "\nListe des emplois issus de France Travail :\n" + context
+    return "Fiche métier Compétences Savoirs :\n" + ficheMetiersCompetencesSavoirs + "\nListe des emplois issus de France Travail :\n" + context
+    #return "Fiche métier Compétences Savoirs :\n" + ficheMetiersCompetencesSavoirs + "\nMetier secteur contexte au travail :\n" + metierSecteurContexteTravail + "\nListe des emplois issus de France Travail :\n" + context
     #return "Liste des emplois issus de France Travail :\n" + context
     #return "\nMetier secteur contexte au travail :\n" + metierSecteurContexteTravail + "\nListe des emplois issus de France Travail :\n" + context
 @cl.step(type="tool")
@@ -925,7 +926,7 @@ async def IA():
     repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
 
     llm = HuggingFaceEndpoint(
-        repo_id=repo_id, max_new_tokens=8000, temperature=0.5, streaming=True
+        repo_id=repo_id, max_new_tokens=5000, temperature=0.5, streaming=True
     )
     return llm