Continuer la conversation

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

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):