"""Takumi helpers (minimal EB1 facade).

Canonical code lives in ``keiro.ensemble``; this package re-exports
for backward compatibility.
"""

from keiro.ensemble.eb1 import (
    DEFAULT_CANDIDATES,
    DEFAULT_CODEX_PLAN,
    DEFAULT_EB2_CODEX_PLAN,
    DEFAULT_JUDGE_MODEL,
    DEFAULT_PRO_PLAN,
    run_eb1,
    run_eb1_codex,
    run_eb1_pro,
)
from keiro.ensemble.eb1_a import get_eb1_a_runtime, run_eb1_a
from keiro.ensemble.eb1_swe import (
    COMPLEXITY_MODEL_MAP,
    FAST_COMPLEXITY_MODEL_MAP,
    PRO_COMPLEXITY_MODEL_MAP,
    get_routing_decision,
    run_eb1_swe_fast_preview,
    run_eb1_swe_preview,
    run_eb1_swe_pro_preview,
)

__all__ = [
    # Core EB1
    "DEFAULT_CANDIDATES",
    "DEFAULT_CODEX_PLAN",
    "DEFAULT_EB2_CODEX_PLAN",
    "DEFAULT_JUDGE_MODEL",
    "DEFAULT_PRO_PLAN",
    "run_eb1",
    "run_eb1_codex",
    "run_eb1_pro",
    "run_eb1_a",
    "get_eb1_a_runtime",
    # SWE-bench routing
    "COMPLEXITY_MODEL_MAP",
    "FAST_COMPLEXITY_MODEL_MAP",
    "PRO_COMPLEXITY_MODEL_MAP",
    "get_routing_decision",
    "run_eb1_swe_preview",
    "run_eb1_swe_fast_preview",
    "run_eb1_swe_pro_preview",
]
