code/self_healing_finetuning/orchestrator/provider_routing.py

from __future__ import annotations

from providers.hf_managed import HFManagedProvider
from providers.local_native import LocalNativeProvider
from providers.pod_cloud import PodCloudProvider


PROVIDERS = {
    "hf_managed": HFManagedProvider,
    "pod_cloud": PodCloudProvider,
    "local_native": LocalNativeProvider,
}


def get_provider(name: str):
    try:
        return PROVIDERS[name]()
    except KeyError as exc:
        raise ValueError(f"unknown provider: {name}") from exc


def validate_route(train_provider: str, infer_provider: str, *, allow_mixed: bool = True) -> list[str]:
    errors: list[str] = []
    if train_provider not in PROVIDERS:
        errors.append(f"unknown train provider: {train_provider}")
    if infer_provider not in PROVIDERS:
        errors.append(f"unknown inference provider: {infer_provider}")
    if not allow_mixed and train_provider != infer_provider:
        errors.append("mixed training/inference providers are disabled")
    return errors