TODO liste avant LLM

Affichée juste avant dans un System message

AgentReact/utils/nodes.py

@@ -66,10 +66,17 @@ def user_prompt(state: CustomState):
     return {'messages': messages}# Je passe unen liste, devrait écraser tous les messages précédent au lieu d'ajouter à la liste du State
 
 
-def LLM_central(state: MessagesState):
+def LLM_central(state: CustomState):
     """Noeud qui s'occupe de gérer les appels au LLM"""
     # Initialisation du LLM
     model = llm.bind_tools(getTools())
+    #print(state)
+
+    if "todo" in state.keys(): # S'il y a des TODO, je l'ajoute avant le prompt au LLM
+        if len(state['todo'])>0:
+            sysmsg = SystemMessage(f"Voici la liste des tâches en cours : {str([f"{i}: {str(todo)}\n" for i,todo in enumerate(state['todo'])])}")
+            print(sysmsg.content)
+            return {"messages": [model.invoke(state["messages"] + [AIMessage('.'), sysmsg])]} # AIMessage pour que Msitrail ne refuse pas la requête avec un 400
     
     # Appel du LLM 
     return {"messages": [model.invoke(state["messages"])]}
@@ -142,7 +149,7 @@ def should_shorten(state: CustomState)->str:
     return 'réduire contexte'
 
 # fonction de routage : Après reponse_question, si le LLM veut appeler un outil, on va au tool_node
-def should_continue(state: MessagesState):
+def should_continue(state: CustomState):
     """
     Vérifier s'il y a un appel aux outils dans le dernier message
     """