Continuer la conversation

Permet de maintenir la conversation au lieu de la stopper après un message sans outils

AgentReact/utils/InterruptPayload.py

@@ -59,7 +59,7 @@ class InterruptPayload():
     def __human_prompt_display(self):
         print("=== L'AGENT DEMANDE DES CONSIGNES! ===\n")
 
-        print("Veuillez saisir un prompt pour l'agent...\n")
+        print("Veuillez saisir un prompt pour l'agent, ou 'exit' pour terminer ici...\n")
         prompt = input("Prompt...")
 
         self.__fields = {'prompt': prompt}

AgentReact/utils/nodes.py

@@ -61,10 +61,14 @@ def user_prompt(state: CustomState):
         ).get("prompt")
     ) # Récupérer un prompt
 
+    end = False # Permet de mettre fin à l'exécution du modèle
+    if user_message.content.lower().strip() == "exit":
+        end = True
+
     messages.append(sys_message) # Rajout des nouveaux messages dans le système
     messages.append(user_message)
 
-    return {'messages': messages}# Je passe unen liste, devrait écraser tous les messages précédent au lieu d'ajouter à la liste du State
+    return {'stop': end, '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: CustomState):

AgentReact/utils/state.py

@@ -10,6 +10,8 @@ class CustomState(MessagesState):
 
     lastSummarizedMessage: int # Index du message où l'on s'était arrêté de résumer
 
+    stop: bool # Permet d'indiquer la fin de l'exécution de l'agent
+
     # TODO: Ajouter la source des documents sélectionnés pour la fin du rapport ?