Modification prompt et max_iterations

nodes.py

@@ -17,7 +17,7 @@ from prompts import df_prompt, feed_back_prompt, reflection_prompt
 
 
 llm = ChatOpenAI(model="gpt-4o-mini")
-MAX_GENERATIONS = 2
+MAX_GENERATIONS = 3
 MAX_ROWS: int = 200
 
 class Query(BaseModel): 

prompts.py

@@ -5,15 +5,15 @@ Created on Sun Oct 13 10:32:26 2024
 
 @author: legalchain
 """
-
+# TODO : ajouter la signification de chaque jugement. 
 df_prompt = [
     (
         "system",
         "Tu travailles avec dees pandas dataframes en Python. Le nom du dataframe est  `df`."
         " Voici les instructions à connaître pour effectuer une recherche performante :"
-        "1. La colonne 'jugement_nature' renseigne sur le type de jugement. C'est dans cette colonne que tu peux trouver les sociétés ou les personnes en liquidation judiciaire ou en redressement judiciaire"
+        "1. La colonne 'jugement_nature' renseigne sur le type de jugement. Le type de jugement obéit à une nomenclature strict que tu dois respecter. Par exemple, clôture s'écrit 'clôture' et non 'cloture'"
         " La liste des valeurs que peut prendre ce champ est la suivante : {nature_jugement}"
-        "2. La colonne 'activite' indique le type d'activité comme restauration, hotelerie, avocat, etc. Ce champ est libre. Il faudra souvent élargir la recherche initiale pour trouver la bonne activité." 
+        "2. La colonne 'activite' indique le type d'activité comme restauration, hôtelerie, avocat, etc. Ce champ est libre. Il faudra souvent élargir la recherche initiale pour trouver la bonne activité." 
         "3. Les colonnes 'departement_nom_officiel, 'region_nom_officiel' servent à localiser l'entreprise concernée"
         " Voici la liste des valeurs possibles pour les régions : \n {region} \n"
         " Voici la liste des valeurs possibles pour les départements : \n  {dep} \n"