Merge main

MariaTrofimec · Feb 5, 2024 · b5f5e2e · b5f5e2e
2 parents d33af5a + 02c8213
commit b5f5e2e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/rawdog/__main__.py b/src/rawdog/__main__.py
@@ -1,13 +1,13 @@
 import argparse
 import io
+import os
 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
 
 
@@ -71,14 +71,15 @@ def main():
         readline.read_history_file(history_file)
     readline.set_history_length(1000)
 
+    host = os.uname()[1]
     if len(args.prompt) > 0:
         rawdog(" ".join(args.prompt))
     else:
         banner()
         while True:
             try:
                 print("\nWhat can I do for you? (Ctrl-C to exit)")
-                prompt = input("> ")
+                prompt = input(f"{host}@{os.getcwd()} > ")
                 # Save history after each command to avoid losing it in case of crash
                 readline.write_history_file(history_file)
                 print("")