from __future__ import annotations

from pathlib import Path
from typing import Any

import yaml

PROMPT_ROOT = Path(__file__).resolve().parent / "prompt"

def load_prompt(agent_name: str, language: str = "zh") -> dict[str, Any]:
    lang = "zh" if language.lower().startswith("zh") else "zh"
    path = PROMPT_ROOT / lang / f"{agent_name}.yaml"
    if not path.exists():
        raise FileNotFoundError(f"Prompt file not found: {path}")
    with path.open("r", encoding="utf-8") as f:
        data = yaml.safe_load(f) or {}
    if not isinstance(data, dict):
        raise ValueError(f"Prompt file must contain a YAML object: {path}")
    return data