Code Llama

oceanbio · Aug 28, 2023 · 54a2b9c · 54a2b9c
1 parent 4211ad6
commit 54a2b9c
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/interpreter/llama_2.py b/interpreter/llama_2.py
@@ -176,8 +176,11 @@ def supports_metal():
     return llama_2
 
 def confirm_action(message):
-    # Print message with newlines on either side (aesthetic choice)
-    print('', Markdown(f"{message} (y/n)"), '')
-    response = input().strip().lower()
-    print('') # <- Aesthetic choice
-    return response == 'y'
+    question = [
+        inquirer.Confirm('confirm',
+                         message=message,
+                         default=True),
+    ]
+
+    answers = inquirer.prompt(question)
+    return answers['confirm']