bugfix

Author: aseembits93

codeflash/api/aiservice.py

@@ -559,6 +559,7 @@ def get_optimization_review(
         replay_tests: str,
         concolic_tests: str,  # noqa: ARG002
         calling_fn_details: str,
+        project_root_dir: Path,
     ) -> str:
         """Compute the optimization review of current Pull Request.
 
@@ -574,6 +575,7 @@ def get_optimization_review(
         root_dir: Path -> path of git directory
         concolic_tests: str -> concolic_tests (not used)
         calling_fn_details: str -> filenames and definitions of functions which call the function_to_optimize
+        project_root_dir: Path -> root dir of the project to calculate relative paths
 
         Returns:
         -------
@@ -583,7 +585,10 @@ def get_optimization_review(
         diff_str = "\n".join(
             [
                 unified_diff_strings(
-                    code1=original_code[p], code2=new_code[p], fromfile=Path(p).as_posix(), tofile=Path(p).as_posix()
+                    code1=original_code[p],
+                    code2=new_code[p],
+                    fromfile=Path(p).relative_to(project_root_dir).as_posix(),
+                    tofile=Path(p).relative_to(project_root_dir).as_posix(),
                 )
                 for p in original_code
                 if not is_zero_diff(original_code[p], new_code[p])

codeflash/optimization/function_optimizer.py

@@ -1506,7 +1506,7 @@ def process_review(
         # this will now run regardless of pr, staging review flags
         try:
             opt_review_response = self.aiservice_client.get_optimization_review(
-                **data, calling_fn_details=function_references
+                **data, calling_fn_details=function_references, project_root_dir=self.project_root
             )
         except Exception as e:
             logger.debug(f"optimization review response failed, investigate {e}")