proxy/providers.py

from dataclasses import dataclass
from typing import Callable


@dataclass
class Provider:
    name: str
    base_url: str
    auth_header: str
    auth_value: Callable[[str], str]
    handles_202: bool
    env_real_key: str


PROVIDERS: dict[str, Provider] = {
    "nvidia": Provider(
        name="nvidia",
        base_url="https://integrate.api.nvidia.com",
        auth_header="Authorization",
        auth_value=lambda key: f"Bearer {key}",
        handles_202=True,
        env_real_key="REAL_NVIDIA_KEY",
    ),
}