linting issue and handle file name in code_print for repaired candidate

Author: mohammedahmed18

codeflash/api/aiservice.py

@@ -307,7 +307,14 @@ def optimize_python_code_repair(self, request: AIServiceCodeRepairRequest) -> Op
         """
         console.rule()
         try:
-            response = self.make_ai_service_request("/code_repair", payload=request, timeout=120)
+            payload = {
+                "optimization_id": request.optimization_id,
+                "original_source_code": request.original_source_code,
+                "modified_source_code": request.modified_source_code,
+                "trace_id": request.trace_id,
+                "test_diffs": request.test_diffs,
+            }
+            response = self.make_ai_service_request("/code_repair", payload=payload, timeout=120)
         except (requests.exceptions.RequestException, TypeError) as e:
             logger.exception(f"Error generating optimization repair: {e}")
             ph("cli-optimize-error-caught", {"error": str(e)})

codeflash/optimization/function_optimizer.py

@@ -703,7 +703,7 @@ def determine_best_candidate(
                             async_throughput=candidate_result.async_throughput,
                         )
                         valid_optimizations.append(best_optimization)
-                        # # queue corresponding refined optimization for best optimization
+                        # queue corresponding refined optimization for best optimization
                         if not candidate.optimization_id.endswith("refi"):
                             future_all_refinements.append(
                                 self.refine_optimizations(
@@ -1896,7 +1896,11 @@ def run_optimized_candidate(  # noqa: PLR0911
                     if not new_candidate:
                         return Failure("Code repair failed to generate a valid candidate."), candidate
 
-                code_print(new_candidate.source_code.flat)
+                code_print(
+                    new_candidate.source_code.flat,
+                    file_name=f"candidate_{optimization_candidate_index}.py",
+                    function_name=self.function_to_optimize.function_name,
+                )
                 normalized_code = normalize_code(new_candidate.source_code.flat.strip())
                 self.ast_code_to_id[normalized_code] = {
                     "optimization_id": new_candidate.optimization_id,