from typing import Any, Dict
from database.db import DatabaseManager
from schemas.enums import DecisionStage


class CheckpointEngine:
    """Auto-saves state checkpoints during mission execution pipeline."""

    @staticmethod
    async def create_checkpoint(db: DatabaseManager, mission_id: str, stage: DecisionStage, state_data: Dict[str, Any]) -> str:
        return await db.save_checkpoint(mission_id, stage, state_data)