early exit

Author: KRRT7

codeflash/code_utils/env_utils.py

@@ -26,12 +26,12 @@ def check_formatter_installed(formatter_cmds: list[str], exit_on_failure: bool =
         return True
 
     exe_name = cmd_tokens[0]
-    command_str = " ".join(formatter_cmds)
+    command_str = " ".join(formatter_cmds).replace(" $file", "")
 
     if shutil.which(exe_name) is None:
         logger.error(
-            f"⚠️  Formatter command not found: {command_str}\n"
-            f"Please install '{exe_name}' or update the 'formatter-cmds' in your pyproject.toml config."
+            f"Could not find formatter: {command_str}\n"
+            f"Please install it or update 'formatter-cmds' in your codeflash configuration"
         )
         return False
 
@@ -44,15 +44,12 @@ def check_formatter_installed(formatter_cmds: list[str], exit_on_failure: bool =
             return True
     except FileNotFoundError:
         logger.error(
-            f"⚠️  Formatter command not found: {command_str}\n"
-            f"Please install '{exe_name}' or update the 'formatter-cmds' in your pyproject.toml config."
+            f"Could not find formatter: {command_str}\n"
+            f"Please install it or update 'formatter-cmds' in your codeflash configuration"
         )
         return False
     except Exception as e:
-        logger.error(
-            f"⚠️  Error running formatter '{command_str}': {e}\n"
-            f"Please verify the 'formatter-cmds' in your pyproject.toml config."
-        )
+        logger.error(f"Formatter failed to run: {command_str}\nError: {e}")
         return False
 
 

codeflash/main.py

@@ -9,6 +9,7 @@
 from codeflash.cli_cmds.cli import parse_args, process_pyproject_config
 from codeflash.cli_cmds.cmd_init import CODEFLASH_LOGO, ask_run_end_to_end_test
 from codeflash.cli_cmds.console import paneled_text
+from codeflash.code_utils import env_utils
 from codeflash.code_utils.checkpoint import ask_should_use_checkpoint_get_functions
 from codeflash.code_utils.config_parser import parse_config_file
 from codeflash.code_utils.version_check import check_for_newer_minor_version
@@ -39,6 +40,8 @@ def main() -> None:
         ask_run_end_to_end_test(args)
     else:
         args = process_pyproject_config(args)
+        if not env_utils.check_formatter_installed(args.formatter_cmds):
+            return
         args.previous_checkpoint_functions = ask_should_use_checkpoint_get_functions(args)
         init_sentry(not args.disable_telemetry, exclude_errors=True)
         posthog_cf.initialize_posthog(not args.disable_telemetry)

codeflash/optimization/optimizer.py

@@ -260,8 +260,6 @@ def run(self) -> None:
         console.rule()
         if not env_utils.ensure_codeflash_api_key():
             return
-        if not env_utils.check_formatter_installed(self.args.formatter_cmds):
-            return
         if self.args.no_draft and is_pr_draft():
             logger.warning("PR is in draft mode, skipping optimization")
             return