9189a7bf26
* I made dump showing more message and added a count (the last x) There seem to be some changes about the implementation so that the current dump message helper functions do not show a lot of useful info. I changed it so that you can `dump 5` (last 5 messages) and that it will print user readable output. This lets you get some more understanding about what is going on. As some messages are still not shown I also show the index (reverse) of the printed message, so one can see what to "pop" to reach a special point without geting into the drumpraw. * black * patch --------- Co-authored-by: Charles Packer <packercharles@gmail.com>
59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
import subprocess
|
|
import sys
|
|
|
|
subprocess.check_call([sys.executable, "-m", "pip", "install", "pexpect"])
|
|
import pexpect
|
|
|
|
|
|
TIMEOUT = 30 # seconds
|
|
|
|
|
|
def test_legacy_cli_sequence():
|
|
# Start the CLI process
|
|
child = pexpect.spawn("memgpt --first --strip_ui")
|
|
|
|
child.expect("Continue with legacy CLI?", timeout=TIMEOUT)
|
|
# Send 'Y' followed by newline
|
|
child.sendline("Y")
|
|
|
|
# Since .memgpt is empty, should jump immediately to "Which model?"
|
|
child.expect("Which model would you like to use?", timeout=TIMEOUT)
|
|
child.sendline()
|
|
|
|
child.expect("Which persona would you like MemGPT to use?", timeout=TIMEOUT)
|
|
child.sendline()
|
|
|
|
child.expect("Which user would you like to use?", timeout=TIMEOUT)
|
|
child.sendline()
|
|
|
|
child.expect("Would you like to preload anything into MemGPT's archival memory?", timeout=TIMEOUT)
|
|
child.sendline() # Default No
|
|
|
|
child.expect("Testing messaging functionality", timeout=TIMEOUT)
|
|
child.expect("Enter your message", timeout=TIMEOUT)
|
|
child.sendline() # Send empty message
|
|
|
|
child.expect("Try again!", timeout=TIMEOUT) # Empty message
|
|
child.sendline("/save")
|
|
|
|
child.expect("Saved checkpoint", timeout=TIMEOUT)
|
|
child.sendline("/load")
|
|
|
|
child.expect("Loaded persistence manager", timeout=TIMEOUT)
|
|
|
|
child.sendline("/dump") # just testing no-crash
|
|
# child.expect("", timeout=TIMEOUT)
|
|
child.sendline("/dump 3") # just testing no-crash
|
|
|
|
child.sendline("/exit")
|
|
child.expect("Finished.", timeout=TIMEOUT)
|
|
|
|
child.expect(pexpect.EOF, timeout=TIMEOUT) # Wait for child to exit
|
|
child.close()
|
|
assert child.isalive() is False, "CLI should have terminated."
|
|
assert child.exitstatus == 0, "CLI did not exit cleanly."
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_legacy_cli_sequence()
|