delete all the checkpoint files for a given module root at the end of a successful run

Author: misrasaurabh1

codeflash/code_utils/checkpoint.py

@@ -81,6 +81,21 @@ def _update_metadata_timestamp(self) -> None:
             f.write(json.dumps(metadata) + "\n")
             f.write(rest_content)
 
+    def cleanup(self) -> None:
+        """Unlink all the checkpoint files for this module_root."""
+        to_delete = []
+        self.checkpoint_path.unlink(missing_ok=True)
+
+        for file in self.checkpoint_dir.glob("codeflash_checkpoint_*.jsonl"):
+            with file.open() as f:
+                # Skip the first line (metadata)
+                first_line = next(f)
+                metadata = json.loads(first_line)
+                if metadata.get("module_root", str(self.module_root)) == str(self.module_root):
+                    to_delete.append(file)
+        for file in to_delete:
+            file.unlink(missing_ok=True)
+
 
 def get_all_historical_functions(module_root: Path, checkpoint_dir: Path) -> dict[str, dict[str, str]]:
     """Get information about all processed functions, regardless of status.

codeflash/optimization/optimizer.py

@@ -258,6 +258,8 @@ def run(self) -> None:
                         console.rule()
                         continue
             ph("cli-optimize-run-finished", {"optimizations_found": optimizations_found})
+            if self.functions_checkpoint:
+                self.functions_checkpoint.cleanup()
             if optimizations_found == 0:
                 logger.info("❌ No optimizations found.")
             elif self.args.all: