Function create_api_key_pool_from_env

contrib/multilingual/api_pool.py:552–619

Build an :class:`ApiKeyPool` from environment variables. Reads ``SKILLSPECTOR_API_KEYS`` — a newline- or semicolon-delimited list of ``key|base_url|model`` entries. Also supports a fallback format where multiple keys are specified via sequentially numbered env vars ``OPENAI_API_KEY

(
    max_concurrent_per_key: int = _DEFAULT_MAX_CONCURRENT_PER_KEY,
)

def create_api_key_pool_from_env(
    max_concurrent_per_key: int = _DEFAULT_MAX_CONCURRENT_PER_KEY,
) -> ApiKeyPool | None:
    """Build an :class:`ApiKeyPool` from environment variables.

    Reads ``SKILLSPECTOR_API_KEYS`` — a newline- or semicolon-delimited list
    of ``key|base_url|model`` entries.

    Also supports a fallback format where multiple keys are specified via
    sequentially numbered env vars ``OPENAI_API_KEY``, ``OPENAI_API_KEY_2``,
    etc.

    Parameters
    ----------
    max_concurrent_per_key :
        Maximum simultaneous requests allowed per key (default 5).
        With 10 keys this gives 50 aggregate slots.

    Returns
    -------
    ApiKeyPool or None
        ``None`` when no multi-key configuration is detected.
    """
    keys: list[ApiKey] = []

    raw = os.environ.get(_API_KEYS_ENV, "").strip()
    if raw:
        for line in raw.replace(";", "\n").splitlines():
            line = line.strip()
            if not line or line.startswith("#"):
                continue
            parts = line.split("|")
            if len(parts) < 1:
                continue
            key_str = parts[0].strip()
            base_url = parts[1].strip() if len(parts) > 1 else None
            model = parts[2].strip() if len(parts) > 2 else "gpt-5.4"
            keys.append(ApiKey(
                key=key_str, base_url=base_url, model=model,
                max_concurrent=max_concurrent_per_key,
            ))

    if not keys:
        base = os.environ.get("OPENAI_API_KEY", "").strip()
        base_url = os.environ.get("OPENAI_BASE_URL", None)
        if base:
            keys.append(ApiKey(
                key=base, base_url=base_url, model="gpt-5.4",
                max_concurrent=max_concurrent_per_key,
            ))
        for idx in range(2, 10):
            extra = os.environ.get(f"OPENAI_API_KEY_{idx}", "").strip()
            if not extra:
                break
            keys.append(ApiKey(
                key=extra, base_url=base_url, model="gpt-5.4",
                max_concurrent=max_concurrent_per_key,
            ))