feat: add language version support across multiple language implementations

Author: HeshamHM28

codeflash/api/aiservice.py

@@ -129,8 +129,7 @@ def optimize_code(
         experiment_metadata: ExperimentMetadata | None = None,
         *,
         language: str = "python",
-        language_version: str
-        | None = None,  # TODO:{claude} add language version to the language support and it should be cached
+        language_version: str | None = None,
         module_system: str | None = None,
         is_async: bool = False,
         n_candidates: int = 5,

codeflash/languages/base.py

@@ -322,6 +322,11 @@ def dir_excludes(self) -> frozenset[str]:
         """
         ...
 
+    @property
+    def language_version(self) -> str | None:
+        """The detected language version (e.g., "17" for Java, "ES2022" for JS)."""
+        ...
+
     # === Discovery ===
 
     def discover_functions(

codeflash/languages/java/support.py

@@ -64,6 +64,7 @@ def __init__(self) -> None:
         self._analyzer = get_java_analyzer()
         self.line_profiler_agent_arg: str | None = None
         self.line_profiler_warmup_iterations: int = 0
+        self._language_version: str | None = None
 
     @property
     def language(self) -> Language:
@@ -93,6 +94,10 @@ def default_file_extension(self) -> str:
     def dir_excludes(self) -> frozenset[str]:
         return frozenset({"target", "build", ".gradle", ".mvn", ".idea"})
 
+    @property
+    def language_version(self) -> str | None:
+        return self._language_version
+
     def postprocess_generated_tests(
         self, generated_tests: GeneratedTestsList, test_framework: str, project_root: Path, source_file_path: Path
     ) -> GeneratedTestsList:
@@ -364,6 +369,8 @@ def ensure_runtime_environment(self, project_root: Path) -> bool:
         if config is None:
             return False
 
+        self._language_version = config.java_version
+
         # For now, assume the runtime is available
         # A full implementation would check/install the JAR
         return True

codeflash/languages/javascript/support.py

@@ -68,6 +68,10 @@ def comment_prefix(self) -> str:
     def dir_excludes(self) -> frozenset[str]:
         return frozenset({"node_modules", "dist", "build", ".next", ".nuxt", "coverage", ".cache", ".turbo", ".vercel"})
 
+    @property
+    def language_version(self) -> str | None:
+        return None
+
     # === Discovery ===
 
     def discover_functions(

codeflash/languages/python/support.py

@@ -107,6 +107,10 @@ def dir_excludes(self) -> frozenset[str]:
             }
         )
 
+    @property
+    def language_version(self) -> str | None:
+        return None
+
     # === Discovery ===
 
     def discover_functions(

codeflash/optimization/function_optimizer.py

@@ -1451,6 +1451,7 @@ def process_single_candidate(
                             optimized_line_profiler_results=best_optimization.line_profiler_test_results["str_out"],
                             function_references=function_references,
                             language=self.function_to_optimize.language,
+                            language_version=self.language_support.language_version,
                         )
                     ],
                 )
@@ -1512,6 +1513,7 @@ def determine_best_candidate(
             else None,
             is_numerical_code=self.is_numerical_code and not self.args.no_jit_opts,
             language=self.function_to_optimize.language,
+            language_version=self.language_support.language_version,
         )
 
         processor = CandidateProcessor(
@@ -2139,6 +2141,7 @@ def generate_optimizations(
             self.function_trace_id[:-4] + "EXP0" if run_experiment else self.function_trace_id,
             ExperimentMetadata(id=self.experiment_id, group="control") if run_experiment else None,
             language=self.function_to_optimize.language,
+            language_version=self.language_support.language_version,
             is_async=self.function_to_optimize.is_async,
             n_candidates=n_candidates,
             is_numerical_code=is_numerical_code,
@@ -2165,6 +2168,7 @@ def generate_optimizations(
                 self.function_trace_id[:-4] + "EXP1",
                 ExperimentMetadata(id=self.experiment_id, group="experiment"),
                 language=self.function_to_optimize.language,
+                language_version=self.language_support.language_version,
                 is_async=self.function_to_optimize.is_async,
                 n_candidates=n_candidates,
             )

codeflash/verification/verifier.py

@@ -7,7 +7,7 @@
 
 from codeflash.cli_cmds.console import logger
 from codeflash.code_utils.code_utils import get_run_tmp_file, module_name_from_file_path
-from codeflash.languages import is_java, is_javascript
+from codeflash.languages import current_language_support, is_java, is_javascript
 from codeflash.verification.verification_utils import ModifyInspiredTests, delete_multiple_if_name_main
 
 if TYPE_CHECKING:
@@ -68,6 +68,7 @@ def generate_tests(
         trace_id=function_trace_id,
         test_index=test_index,
         language=function_to_optimize.language,
+        language_version=current_language_support().language_version,
         module_system=project_module_system,
         is_numerical_code=is_numerical_code,
     )