From 71d696dc9ea608080935ced871613da7c883d73a Mon Sep 17 00:00:00 2001
From: Hans Raaf <hara@oderwat.de>
Date: Sat, 4 Nov 2023 05:04:37 +0100
Subject: [PATCH] I added a "/retry" command to retry for getting another
 answer. (#188)

* I added a "/retry" command to retry for getting another answer.

- Implemented to pop messages until hitting the last user message. Then
  extracting the users last message and sending it again. This will also
  work with state files and after manually popping messages.
- Updated the README to include /retry
- Update the README for "pop" with parameter and changed default to 3 as
  this will pop "function/assistant/user" which is the usual turn
  around.

* disclaimer

---------

Co-authored-by: Charles Packer <packercharles@gmail.com>
---
 README.md      |  6 ++++--
 memgpt/main.py | 16 ++++++++++++++--
 2 files changed, 18 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 82cd1192..da0677da 100644
--- a/README.md
+++ b/README.md
@@ -295,8 +295,10 @@ While using MemGPT via the CLI (not Discord!) you can run various commands:
   view the last <count> messages (all if <count> is omitted)
 /memory
   print the current contents of agent memory
-/pop
-  undo the last message in the conversation
+/pop <count>
+  undo the last messages in the conversation. It defaults to 3, which usually is one turn around in the conversation
+/retry
+  pops the last answer and tries to get another one
 /rethink <text>
   will replace the inner dialog of the last assistant message with the <text> to help shaping the conversation 
 /rewrite
diff --git a/memgpt/main.py b/memgpt/main.py
index fda6155d..e93ed085 100644
--- a/memgpt/main.py
+++ b/memgpt/main.py
@@ -504,12 +504,23 @@ async def run_agent_loop(memgpt_agent, first, no_verify=False, cfg=None, strip_u
                 elif user_input.lower() == "/pop" or user_input.lower().startswith("/pop "):
                     # Check if there's an additional argument that's an integer
                     command = user_input.strip().split()
-                    amount = int(command[1]) if len(command) > 1 and command[1].isdigit() else 2
+                    amount = int(command[1]) if len(command) > 1 and command[1].isdigit() else 3
                     print(f"Popping last {amount} messages from stack")
                     for _ in range(min(amount, len(memgpt_agent.messages))):
                         memgpt_agent.messages.pop()
                     continue
 
+                elif user_input.lower() == "/retry":
+                    # TODO this needs to also modify the persistence manager
+                    print(f"Retrying for another answer")
+                    while len(memgpt_agent.messages) > 0:
+                        if memgpt_agent.messages[-1].get("role") == "user":
+                            # we want to pop up to the last user message and send it again
+                            user_message = memgpt_agent.messages[-1].get("content")
+                            memgpt_agent.messages.pop()
+                            break
+                        memgpt_agent.messages.pop()
+
                 elif user_input.lower() == "/rethink" or user_input.lower().startswith("/rethink "):
                     # TODO this needs to also modify the persistence manager
                     if len(user_input) < len("/rethink "):
@@ -615,7 +626,8 @@ USER_COMMANDS = [
     ("/load", "load a saved checkpoint"),
     ("/dump <count>", "view the last <count> messages (all if <count> is omitted)"),
     ("/memory", "print the current contents of agent memory"),
-    ("/pop", "undo the last message in the conversation"),
+    ("/pop <count>", "undo <count> messages in the conversation (default is 3)"),
+    ("/retry", "pops the last answer and tries to get another one"),
     ("/rethink <text>", "changes the inner thoughts of the last agent message"),
     ("/rewrite <text>", "changes the reply of the last agent message"),
     ("/heartbeat", "send a heartbeat system message to the agent"),