don't

Author: KRRT7

codeflash/code_utils/checkpoint.py

@@ -8,7 +8,7 @@
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, Optional
 
-import click
+from rich.prompt import Confirm
 
 if TYPE_CHECKING:
     import argparse
@@ -139,7 +139,7 @@ def ask_should_use_checkpoint_get_functions(args: argparse.Namespace) -> Optiona
     previous_checkpoint_functions = None
     if args.all and (sys.platform == "linux" or sys.platform == "darwin") and Path("/tmp").is_dir():  # noqa: S108 #TODO: use the temp dir from codeutils-compat.py
         previous_checkpoint_functions = get_all_historical_functions(args.module_root, Path("/tmp"))  # noqa: S108
-        if previous_checkpoint_functions and click.confirm(
+        if previous_checkpoint_functions and Confirm.ask(
             "Previous Checkpoint detected from an incomplete optimization run, shall I continue the optimization from that point?",
             default=True,
         ):

codeflash/code_utils/config_parser.py

@@ -26,7 +26,7 @@ def find_pyproject_toml(config_file: Path | None = None) -> Path:
             return config_file
         # Search for pyproject.toml in the parent directories
         dir_path = dir_path.parent
-    msg = f"Could not find pyproject.toml in the current directory {Path.cwd()} or any of the parent directories. Please create it by running `poetry init`, or pass the path to pyproject.toml with the --config-file argument."
+    msg = f"Could not find pyproject.toml in the current directory {Path.cwd()} or any of the parent directories. Please create it by running `codeflash init`, or pass the path to pyproject.toml with the --config-file argument."
 
     raise ValueError(msg)
 

codeflash/code_utils/git_utils.py

@@ -12,10 +12,9 @@
 from typing import TYPE_CHECKING
 
 import git
-import inquirer
+from rich.prompt import Confirm
 from unidiff import PatchSet
 
-from codeflash.cli_cmds.cli_common import inquirer_wrapper
 from codeflash.cli_cmds.console import logger
 from codeflash.code_utils.config_consts import N_CANDIDATES
 
@@ -109,11 +108,9 @@ def check_running_in_git_repo(module_root: str) -> bool:
 
 def confirm_proceeding_with_no_git_repo() -> str | bool:
     if sys.__stdin__.isatty():
-        return inquirer_wrapper(
-            inquirer.confirm,
-            message="WARNING: I did not find a git repository for your code. If you proceed with running codeflash, "
-            "optimized code will"
-            " be written over your current code and you could irreversibly lose your current code. Proceed?",
+        return Confirm.ask(
+            "WARNING: I did not find a git repository for your code. If you proceed with running codeflash, "
+            "optimized code will be written over your current code and you could irreversibly lose your current code. Proceed?",
             default=False,
         )
     # continue running on non-interactive environments, important for GitHub actions
@@ -130,11 +127,9 @@ def check_and_push_branch(repo: git.Repo, wait_for_push: bool = False) -> bool:
         if not sys.__stdin__.isatty():
             logger.warning("Non-interactive shell detected. Branch will not be pushed.")
             return False
-        if sys.__stdin__.isatty() and inquirer_wrapper(
-            inquirer.confirm,
-            message=f"⚡️ In order for me to create PRs, your current branch needs to be pushed. Do you want to push "
-            f"the branch"
-            f"'{current_branch}' to the remote repository?",
+        if sys.__stdin__.isatty() and Confirm.ask(
+            f"⚡️ In order for me to create PRs, your current branch needs to be pushed. Do you want to push "
+            f"the branch '{current_branch}' to the remote repository?",
             default=False,
         ):
             origin.push(current_branch)

tests/test_git_utils.py

@@ -64,7 +64,7 @@ def test_check_running_in_git_repo_not_in_git_repo_non_interactive(self, mock_is
 
     @patch("codeflash.code_utils.git_utils.git.Repo")
     @patch("codeflash.code_utils.git_utils.sys.__stdin__.isatty", return_value=True)
-    @patch("codeflash.code_utils.git_utils.inquirer.confirm", return_value=True)
+    @patch("codeflash.code_utils.git_utils.Confirm.ask", return_value=True)
     def test_check_and_push_branch(self, mock_confirm, mock_isatty, mock_repo):
         mock_repo_instance = mock_repo.return_value
         mock_repo_instance.active_branch.name = "test-branch"