Add metadata tracking to model pusher

src/mlpipeline/components/model_pusher.py

@@ -1,6 +1,8 @@
 import shutil
 import os
+import json
 from pathlib import Path
+from datetime import datetime
 from mlpipeline.entity import ModelPusherConfig, ModelPusherArtifact
 from mlpipeline.logging.logger import get_logger
 from mlpipeline.exception import ModelPusherException
@@ -18,6 +20,7 @@ class ModelPusher:
             logger.info("Starting model pusher")
             
             os.makedirs(self.config.model_registry_path, exist_ok=True)
+            os.makedirs(self.config.root_dir, exist_ok=True)
             
             model_source = Path(self.config.model_path)
             model_dest = Path(self.config.model_registry_path) / model_source.name
@@ -31,6 +34,18 @@ class ModelPusher:
             
             logger.info(f"Model pushed to: {model_dest}")
             
+            # Save push metadata
+            metadata = {
+                "pushed_at": datetime.now().isoformat(),
+                "source_path": str(model_source),
+                "destination_path": str(model_dest),
+                "status": "success"
+            }
+            
+            metadata_path = Path(self.config.root_dir) / "push_metadata.json"
+            with open(metadata_path, 'w') as f:
+                json.dump(metadata, f, indent=4)
+            
             return ModelPusherArtifact(
                 pushed_model_path=str(model_dest),
                 is_pushed=True,