fix: harden worktree creation and improve subagent mode support

codeflash/code_utils/git_worktree_utils.py

@@ -13,7 +13,7 @@
 
 from codeflash.cli_cmds.console import logger
 from codeflash.code_utils.compat import codeflash_cache_dir
-from codeflash.code_utils.git_utils import check_running_in_git_repo, git_root_dir
+from codeflash.code_utils.git_utils import git_root_dir
 
 worktree_dirs = codeflash_cache_dir / "worktrees"
 patches_dir = codeflash_cache_dir / "patches"
@@ -53,10 +53,7 @@ def create_worktree_snapshot_commit(worktree_dir: Path, commit_message: str) ->
             cw.remove_option("user", "email")
 
 
-def create_detached_worktree(module_root: Path) -> Optional[Path]:
-    if not check_running_in_git_repo(module_root):
-        logger.warning("Module is not in a git repository. Skipping worktree creation.")
-        return None
+def create_detached_worktree() -> Optional[Path]:
     git_root = git_root_dir()
     current_time_str = time.strftime("%Y%m%d-%H%M%S")
     worktree_dir = worktree_dirs / f"{git_root.name}-{current_time_str}"

codeflash/discovery/functions_to_optimize.py

@@ -28,7 +28,7 @@
 from codeflash.discovery.discover_unit_tests import discover_unit_tests
 from codeflash.languages.language_enum import Language
 from codeflash.languages.registry import get_language_support, get_supported_extensions, is_language_supported
-from codeflash.lsp.helpers import is_LSP_enabled
+from codeflash.lsp.helpers import is_LSP_enabled, is_subagent_mode
 from codeflash.models.function_types import FunctionParent, FunctionToOptimize
 from codeflash.telemetry.posthog_cf import ph
 
@@ -759,8 +759,7 @@ def was_function_previously_optimized(
         Tuple of (filtered_functions_dict, remaining_count)
 
     """
-    if is_LSP_enabled():
-        # was_function_previously_optimized is for the checking the optimization duplicates in the github action, no need to do this in the LSP mode
+    if is_LSP_enabled() or is_subagent_mode():
         return False
 
     # Check optimization status if repository info is provided

codeflash/languages/function_optimizer.py

@@ -1297,6 +1297,8 @@ def process_single_candidate(
         # Display runtime information
         if is_LSP_enabled():
             lsp_log(LspMarkdownMessage(markdown=tree_to_markdown(tree)))
+        elif is_subagent_mode():
+            logger.info(tree_to_markdown(tree))
         else:
             console.print(tree)
         if self.args.benchmark and benchmark_tree:

codeflash/languages/javascript/test_runner.py

@@ -1134,9 +1134,9 @@ def run_jest_benchmarking_tests(
         # Create result with combined stdout
         result = subprocess.CompletedProcess(args=result.args, returncode=result.returncode, stdout=stdout, stderr="")
         if result.returncode != 0:
-            logger.info(f"Jest benchmarking failed with return code {result.returncode}")
-            logger.info(f"Jest benchmarking stdout: {result.stdout}")
-            logger.info(f"Jest benchmarking stderr: {result.stderr}")
+            logger.debug(f"Jest benchmarking failed with return code {result.returncode}")
+            logger.debug(f"Jest benchmarking stdout: {result.stdout}")
+            logger.debug(f"Jest benchmarking stderr: {result.stderr}")
 
     except subprocess.TimeoutExpired:
         logger.warning(f"Jest benchmarking timed out after {total_timeout}s")

codeflash/optimization/optimizer.py

@@ -2,6 +2,7 @@
 
 import copy
 import os
+import sys
 import tempfile
 import time
 from collections import defaultdict
@@ -780,15 +781,19 @@ def worktree_mode(self) -> None:
         if self.current_worktree:
             return
 
-        if check_running_in_git_repo(self.args.module_root):
-            worktree_dir = create_detached_worktree(self.args.module_root)
-            if worktree_dir is None:
-                logger.warning("Failed to create worktree. Skipping optimization.")
-                return
-            self.current_worktree = worktree_dir
-            self.mirror_paths_for_worktree_mode(worktree_dir)
-            # make sure the tests dir is created in the worktree, this can happen if the original tests dir is empty
-            Path(self.args.tests_root).mkdir(parents=True, exist_ok=True)
+        if not check_running_in_git_repo(self.args.module_root):
+            logger.error("Worktree creation failed because the current directory is not part of a Git repository.")
+            sys.exit(1)
+
+        worktree_dir = create_detached_worktree()
+        if worktree_dir is None:
+            logger.error("Failed to create a worktree.")
+            sys.exit(1)
+
+        self.current_worktree = worktree_dir
+        self.mirror_paths_for_worktree_mode(worktree_dir)
+        # make sure the tests dir is created in the worktree, this can happen if the original tests dir is empty
+        Path(self.args.tests_root).mkdir(parents=True, exist_ok=True)
 
     def mirror_paths_for_worktree_mode(self, worktree_dir: Path) -> None:
         original_args = copy.deepcopy(self.args)