Merge pull request AbanteAI#26 from kvaky/keep-cmd-history

Add command history
thorstone137 · Feb 5, 2024 · d33af5a · d33af5a
2 parents 7b20e78 + fd2ed16
commit d33af5a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/rawdog/__main__.py b/src/rawdog/__main__.py
@@ -1,9 +1,11 @@
 import argparse
 import io
+import readline
 
 from contextlib import redirect_stdout
 
 from rawdog.llm_client import LLMClient
+from rawdog.utils import history_file
 
 
 llm_client = LLMClient()  # Will prompt for API key if not found
@@ -65,6 +67,10 @@ def main():
     parser.add_argument('--dry-run', action='store_true', help='Print the script before executing and prompt for confirmation.')
     args = parser.parse_args()
 
+    if history_file.exists():
+        readline.read_history_file(history_file)
+    readline.set_history_length(1000)
+
     if len(args.prompt) > 0:
         rawdog(" ".join(args.prompt))
     else:
@@ -73,6 +79,8 @@ def main():
             try:
                 print("\nWhat can I do for you? (Ctrl-C to exit)")
                 prompt = input("> ")
+                # Save history after each command to avoid losing it in case of crash
+                readline.write_history_file(history_file)
                 print("")
                 rawdog(prompt, verbose=args.dry_run)
             except KeyboardInterrupt:

diff --git a/src/rawdog/utils.py b/src/rawdog/utils.py
@@ -6,6 +6,8 @@
 rawdog_dir = Path.home() / ".rawdog"
 rawdog_dir.mkdir(exist_ok=True)
 
+# Command history file
+history_file = rawdog_dir / "cmdline_history"
 
 # Config
 config_path = rawdog_dir / "config.yaml"