refactor: cli_input to handle both single and multi line

OpenInterpreter · Mar 31, 2024 · d348a00 · d348a00
1 parent cc6291f
commit d348a00
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/interpreter/terminal_interface/terminal_interface.py b/interpreter/terminal_interface/terminal_interface.py
@@ -76,14 +76,14 @@ def terminal_interface(interpreter, message):
         try:
             if interactive:
                 ### This is the primary input for Open Interpreter.
-                message = cli_input("> ").strip() if interpreter.multi_line else input("> ").strip()
 
                 try:
                     # This lets users hit the up arrow key for past messages
                     readline.add_history(message)
                 except:
                     # If the user doesn't have readline (may be the case on windows), that's fine
                     pass
+                message = cli_input("> ", interpreter.multi_line).strip()
 
         except KeyboardInterrupt:
             # Exit gracefully

diff --git a/interpreter/terminal_interface/utils/cli_input.py b/interpreter/terminal_interface/utils/cli_input.py
@@ -1,4 +1,6 @@
-def cli_input(prompt: str = "") -> str:
+def cli_input(prompt: str = "", multi_line=False) -> str:
+    if not multi_line:
+        return input(prompt)
     start_marker = "```"
     end_marker = "```"
     message = input(prompt)