load.py

Go to the documentation of this file.

# Copyright (c) 2025-present Aquin Labs Private Limited. All Rights Reserved.
# This file is part of the Aquin Engine. Unauthorized copying, modification,
# distribution, or use of this file, via any medium, is strictly prohibited.
# Proprietary and confidential. See LICENSE for terms.
 
"""Load Recipe YAML (or JSON) and resolve paths relative to the file."""
 
from __future__ import annotations
 
import json
from pathlib import Path
from typing import Any
 
from aquin.recipe.schema import Recipe, RecipeError
 
 
def _parse(path: Path) -> dict[str, Any]:
    raw = path.read_text(encoding="utf-8")
    if path.suffix.lower() == ".json":
        data = json.loads(raw)
    else:
        try:
            import yaml
        except ImportError as exc:
            raise RecipeError(
                "YAML recipes need PyYAML (`pip install pyyaml`) or use a .json recipe."
            ) from exc
        data = yaml.safe_load(raw)
    if not isinstance(data, dict):
        raise RecipeError(f"{path} is not a mapping.")
    return data
 
 
def _resolve_path(base: Path, value: Any) -> Any:
    if not isinstance(value, str) or not value:
        return value
    p = Path(value).expanduser()
    if p.is_absolute():
        return str(p)
    return str((base / p).resolve())
 
 
def load_recipe(path: str | Path) -> Recipe:
    recipe_path = Path(path).expanduser().resolve()
    if not recipe_path.is_file():
        raise RecipeError(f"Recipe not found: {recipe_path}")
    data = _parse(recipe_path)
    base = recipe_path.parent
    data_block = data.get("data")
    if isinstance(data_block, dict) and data_block.get("path"):
        data_block = dict(data_block)
        data_block["path"] = _resolve_path(base, data_block["path"])
        data["data"] = data_block
    ev = data.get("eval")
    if isinstance(ev, dict) and ev.get("probes"):
        ev = dict(ev)
        ev["probes"] = _resolve_path(base, ev["probes"])
        data["eval"] = ev
    return Recipe.from_dict(data, source=str(recipe_path))