code/self_healing_finetuning/manifests/validators.py

from __future__ import annotations

from typing import Any


REQUIRED_FIELDS = {
    "dataset_manifest": ["dataset_manifest_id", "dataset_name", "license_status", "schema", "splits"],
    "model_manifest": ["model_id", "base_model", "base_strategy", "provider", "train_mode", "evaluation", "lineage"],
    "run_manifest": ["run_id", "model_candidate", "execution", "created_at"],
    "deployment_manifest": ["deployment_id", "run_id", "env", "infer_provider", "status"],
    "promotion_manifest": ["run_id", "env", "status", "approvals", "rollback_target"],
}


def validate_manifest(manifest_type: str, payload: dict[str, Any]) -> list[str]:
    errors: list[str] = []
    for field in REQUIRED_FIELDS.get(manifest_type, []):
        if field not in payload:
            errors.append(f"missing required field: {field}")
    return errors