agents/cliproxyapi/env.py

"""Build env-var dicts (or YAML-config snippets) that point an SDK at the proxy.

Two SDK shapes are covered today; add more here as agents arrive:

    anthropic_env(ep, model)   → Anthropic SDK / Claude Code CLI
    openai_env(ep, model)      → OpenAI SDK / Codex CLI

Plus `openai_yaml_block(ep, model)` for agents whose config files take
`base_url` / `api_key` / `model` fields directly (e.g. MLEvolve).

Usage from any agent runner:

    from agents.cliproxyapi import ProxyEndpoint, anthropic_env
    ep = ProxyEndpoint.from_env()
    subprocess.run(cmd, env={**os.environ, **anthropic_env(ep, model="...")})
"""

from __future__ import annotations

from .endpoint import ProxyEndpoint


def anthropic_env(ep: ProxyEndpoint, model: str | None = None) -> dict[str, str]:
    """Env vars consumed by anthropic-python and claude-code.

    The Anthropic SDK appends `/v1/messages` to ANTHROPIC_BASE_URL itself,
    so we hand it the proxy root (no trailing path).
    """
    env = {
        "ANTHROPIC_BASE_URL": ep.base_url(),
        "ANTHROPIC_API_KEY": ep.api_key,
        "ANTHROPIC_AUTH_TOKEN": ep.api_key,
    }
    if model:
        env["ANTHROPIC_MODEL"] = model
    return env


def openai_env(ep: ProxyEndpoint, model: str | None = None) -> dict[str, str]:
    """Env vars consumed by openai-python, codex-cli, and many compatible SDKs.

    The OpenAI SDK appends `/chat/completions` (and other paths) to
    OPENAI_BASE_URL, so we include the `/v1` prefix here.
    """
    env = {
        "OPENAI_BASE_URL": f"{ep.base_url()}/v1",
        "OPENAI_API_KEY": ep.api_key,
        "OPENAI_API_BASE": f"{ep.base_url()}/v1",  # legacy var, still common
    }
    if model:
        env["OPENAI_MODEL"] = model
    return env


def openai_yaml_block(ep: ProxyEndpoint, model: str) -> dict[str, str]:
    """Three-key dict for configs that name the proxy directly (e.g. MLEvolve).

    Returns:
        {"model": ..., "base_url": ".../v1", "api_key": ...}
    """
    return {
        "model": model,
        "base_url": f"{ep.base_url()}/v1",
        "api_key": ep.api_key,
    }