Merge branch 'main' of github.com:codeflash-ai/codeflash into multi-language

Author: mohammedahmed18

codeflash/api/cfapi.py

@@ -171,6 +171,8 @@ def suggest_changes(
     replay_tests: str = "",
     concolic_tests: str = "",
     optimization_review: str = "",
+    original_line_profiler: str | None = None,
+    optimized_line_profiler: str | None = None,
 ) -> Response:
     """Suggest changes to a pull request.
 
@@ -182,6 +184,8 @@ def suggest_changes(
     :param file_changes: A dictionary of file changes.
     :param pr_comment: The pull request comment object, containing the optimization explanation, best runtime, etc.
     :param generated_tests: The generated tests.
+    :param original_line_profiler: Line profiler results for original code (markdown format).
+    :param optimized_line_profiler: Line profiler results for optimized code (markdown format).
     :return: The response object.
     """
     payload = {
@@ -197,7 +201,10 @@ def suggest_changes(
         "replayTests": replay_tests,
         "concolicTests": concolic_tests,
         "optimizationReview": optimization_review,  # impact keyword left for legacy reasons, touches js/ts code
+        "originalLineProfiler": original_line_profiler,
+        "optimizedLineProfiler": optimized_line_profiler,
     }
+
     return make_cfapi_request(endpoint="/suggest-pr-changes", method="POST", payload=payload)
 
 
@@ -214,6 +221,8 @@ def create_pr(
     replay_tests: str = "",
     concolic_tests: str = "",
     optimization_review: str = "",
+    original_line_profiler: str | None = None,
+    optimized_line_profiler: str | None = None,
 ) -> Response:
     """Create a pull request, targeting the specified branch. (usually 'main').
 
@@ -223,6 +232,8 @@ def create_pr(
     :param file_changes: A dictionary of file changes.
     :param pr_comment: The pull request comment object, containing the optimization explanation, best runtime, etc.
     :param generated_tests: The generated tests.
+    :param original_line_profiler: Line profiler results for original code (markdown format).
+    :param optimized_line_profiler: Line profiler results for optimized code (markdown format).
     :return: The response object.
     """
     # convert Path objects to strings
@@ -239,7 +250,10 @@ def create_pr(
         "replayTests": replay_tests,
         "concolicTests": concolic_tests,
         "optimizationReview": optimization_review,  # Impact keyword left for legacy reasons, it touches js/ts codebase
+        "originalLineProfiler": original_line_profiler,
+        "optimizedLineProfiler": optimized_line_profiler,
     }
+
     return make_cfapi_request(endpoint="/create-pr", method="POST", payload=payload)
 
 
@@ -269,6 +283,8 @@ def create_staging(
     concolic_tests: str,
     root_dir: Path,
     optimization_review: str = "",
+    original_line_profiler: str | None = None,
+    optimized_line_profiler: str | None = None,
 ) -> Response:
     """Create a staging pull request, targeting the specified branch. (usually 'staging').
 
@@ -279,6 +295,8 @@ def create_staging(
     :param generated_original_test_source: Generated tests for the original function.
     :param function_trace_id: Unique identifier for this optimization trace.
     :param coverage_message: Coverage report or summary.
+    :param original_line_profiler: Line profiler results for original code (markdown format).
+    :param optimized_line_profiler: Line profiler results for optimized code (markdown format).
     :return: The response object from the backend.
     """
     relative_path = explanation.file_path.relative_to(root_dir).as_posix()
@@ -310,6 +328,8 @@ def create_staging(
         "replayTests": replay_tests,
         "concolicTests": concolic_tests,
         "optimizationReview": optimization_review,  # Impact keyword left for legacy reasons, it touches js/ts codebase
+        "originalLineProfiler": original_line_profiler,
+        "optimizedLineProfiler": optimized_line_profiler,
     }
 
     return make_cfapi_request(endpoint="/create-staging", method="POST", payload=payload)

codeflash/optimization/function_optimizer.py

@@ -1790,8 +1790,8 @@ def generate_optimizations(
         self,
         read_writable_code: CodeStringsMarkdown,
         read_only_context_code: str,
-        run_experiment: bool = False,  # noqa: FBT001, FBT002
-        is_numerical_code: bool | None = None,  # noqa: FBT001
+        run_experiment: bool = False,
+        is_numerical_code: bool | None = None,
     ) -> Result[tuple[OptimizationSet, str], str]:
         """Generate optimization candidates for the function. Backend handles multi-model diversity."""
         n_candidates = get_effort_value(EffortKeys.N_OPTIMIZER_CANDIDATES, self.effort)
@@ -2174,6 +2174,8 @@ def process_review(
             "replay_tests": replay_tests,
             "concolic_tests": concolic_tests,
             "language": self.function_to_optimize.language,
+            "original_line_profiler": original_code_baseline.line_profile_results.get("str_out", ""),
+            "optimized_line_profiler": best_optimization.line_profiler_test_results.get("str_out", ""),
         }
 
         raise_pr = not self.args.no_pr

codeflash/result/create_pr.py

@@ -285,6 +285,8 @@ def check_create_pr(
     root_dir: Path,
     git_remote: Optional[str] = None,
     optimization_review: str = "",
+    original_line_profiler: str | None = None,
+    optimized_line_profiler: str | None = None,
 ) -> None:
     pr_number: Optional[int] = env_utils.get_pr_number()
     git_repo = git.Repo(search_parent_directories=True)
@@ -329,6 +331,8 @@ def check_create_pr(
             replay_tests=replay_tests,
             concolic_tests=concolic_tests,
             optimization_review=optimization_review,
+            original_line_profiler=original_line_profiler,
+            optimized_line_profiler=optimized_line_profiler,
         )
         if response.ok:
             logger.info(f"Suggestions were successfully made to PR #{pr_number}")
@@ -381,6 +385,8 @@ def check_create_pr(
             replay_tests=replay_tests,
             concolic_tests=concolic_tests,
             optimization_review=optimization_review,
+            original_line_profiler=original_line_profiler,
+            optimized_line_profiler=optimized_line_profiler,
         )
         if response.ok:
             pr_id = response.text

tests/test_languages/test_js_code_replacer.py

@@ -1679,7 +1679,7 @@ def test_code_replacer_for_class_method(ts_support, temp_project):
 
     /**
      * Find duplicates in the data array.
-     * Intentionally inefficient O(n²) implementation.
+     * Intentionally inefficient implementation.
      * @returns Array of duplicate values
      */
     findDuplicates(): T[] {
@@ -1698,7 +1698,7 @@ def test_code_replacer_for_class_method(ts_support, temp_project):
 
     /**
      * Sort the data using bubble sort.
-     * Intentionally inefficient O(n²) implementation.
+     * Intentionally inefficient implementation.
      * @returns Sorted copy of the data
      */
     sortData(): T[] {
@@ -1718,7 +1718,7 @@ def test_code_replacer_for_class_method(ts_support, temp_project):
 
     /**
      * Get unique values from the data.
-     * Intentionally inefficient O(n²) implementation.
+     * Intentionally inefficient implementation.
      * @returns Array of unique values
      */
     getUnique(): T[] {
@@ -1843,7 +1843,7 @@ class DataProcessor<T> {
 
     /**
      * Sort the data using bubble sort.
-     * Intentionally inefficient O(n²) implementation.
+     * Intentionally inefficient implementation.
      * @returns Sorted copy of the data
      */
     sortData(): T[] {
@@ -1863,7 +1863,7 @@ class DataProcessor<T> {
 
     /**
      * Get unique values from the data.
-     * Intentionally inefficient O(n²) implementation.
+     * Intentionally inefficient implementation.
      * @returns Array of unique values
      */
     getUnique(): T[] {