Merge pull request #1626 from codeflash-ai/codeflash/optimize-pr1199-2026-02-20T22.33.34

⚡️ Speed up function `find_helper_functions` by 7,426% in PR #1199 (`omni-java`)

Author: misrasaurabh1

codeflash/languages/__init__.py

@@ -38,7 +38,6 @@
     reset_current_language,
     set_current_language,
 )
-
 from codeflash.languages.registry import (
     detect_project_language,
     get_language_support,

codeflash/languages/java/context.py

@@ -660,6 +660,10 @@ def find_helper_functions(
     helper_files = find_helper_files(function.file_path, project_root, max_depth, analyzer)
 
     for file_path in helper_files:
+        # Skip non-existent files early to avoid expensive exception handling
+        if not file_path.exists():
+            continue
+
         try:
             source = file_path.read_text(encoding="utf-8")
             file_functions = discover_functions_from_source(source, file_path, analyzer=analyzer)
@@ -713,6 +717,10 @@ def _find_same_class_helpers(function: FunctionToOptimize, analyzer: JavaAnalyze
     if not function.class_name:
         return helpers
 
+    # Check if file exists before trying to read it
+    if not function.file_path.exists():
+        return helpers
+
     try:
         source = function.file_path.read_text(encoding="utf-8")
         source_bytes = source.encode("utf8")

codeflash/languages/registry.py

@@ -53,10 +53,10 @@ def _ensure_languages_registered() -> None:
         from codeflash.languages.python import support as _
 
     with contextlib.suppress(ImportError):
-        from codeflash.languages.javascript import support as _  # noqa: F401
+        from codeflash.languages.javascript import support as _
 
     with contextlib.suppress(ImportError):
-        from codeflash.languages.java import support as _  # noqa: F401
+        from codeflash.languages.java import support as _
 
     _languages_registered = True