small changes

Author: mohammedahmed18

codeflash/api/aiservice.py

@@ -321,18 +321,17 @@ def optimize_python_code_repair(self, request: AIServiceCodeRepairRequest) -> Op
             return None
 
         if response.status_code == 200:
-            refined_optimization = response.json()
+            fixed_optimization = response.json()
             console.rule()
 
-            refinements = self._get_valid_candidates([refined_optimization])
-            if not refinements:
+            if not self._get_valid_candidates([fixed_optimization]):
                 logger.error("Code repair failed to generate a valid candidate.")
                 return None
 
             return OptimizedCandidate(
-                source_code=refinements[0].source_code,
-                explanation=refinements[0].explanation,
-                optimization_id=refinements[0].optimization_id[:-4] + "cdrp",
+                source_code=fixed_optimization["source_code"],
+                explanation=fixed_optimization["explanation"],
+                optimization_id=fixed_optimization["optimization_id"],
             )
 
         try:

codeflash/optimization/function_optimizer.py

@@ -1892,7 +1892,7 @@ def run_optimized_candidate(  # noqa: PLR0911
 
                 ai_service_client = self.aiservice_client if exp_type == "EXP0" else self.local_aiservice_client
 
-                with progress_bar("The test results are not matching, let me see if I can fix this"):
+                with progress_bar("Some of the test results are not matching, let me see if I can fix this"):
                     new_candidate = self.code_repair_optimizations(
                         original_source_code=code_context.read_writable_code.markdown,
                         modified_source_code=candidate.source_code.markdown,