Yijia-Xiao
e3bc872982
Clear the deferred full-repo lint backlog so the whole tree passes the strict ruff select (E,W,F,I,B,UP,C4,SIM). Mechanical fixes dominate: import sorting, pep585/604 annotations, dropped dead imports, and whitespace. The few semantic changes are behavior-preserving: declare __all__ on the agent_utils and alpha_vantage re-export hubs; expand 'from x import *' to explicit names; use immutable tuple defaults instead of mutable list defaults; contextlib.suppress for try/except/pass; and narrow an over-broad assertRaises.
54 lines
2.2 KiB
Python
54 lines
2.2 KiB
Python
"""Canonical provider -> API-key env-var mapping.
|
|
|
|
A single source of truth for which environment variable holds the API
|
|
key for each supported LLM provider. Used by the CLI's interactive key
|
|
prompt (cli/utils.ensure_api_key) and by anything else that needs to
|
|
ask "does this provider require a key, and which env var is it?".
|
|
|
|
When adding a new provider, register its env var here so the CLI flow
|
|
prompts for it automatically instead of failing on first API call.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
PROVIDER_API_KEY_ENV: dict[str, str | None] = {
|
|
"openai": "OPENAI_API_KEY",
|
|
"anthropic": "ANTHROPIC_API_KEY",
|
|
"google": "GOOGLE_API_KEY",
|
|
"azure": "AZURE_OPENAI_API_KEY",
|
|
# Bedrock authenticates via the AWS credential chain, not a single key env.
|
|
"bedrock": None,
|
|
"xai": "XAI_API_KEY",
|
|
"deepseek": "DEEPSEEK_API_KEY",
|
|
# Dual-region providers each carry their own account; keys are not
|
|
# interchangeable between the international and China endpoints.
|
|
"qwen": "DASHSCOPE_API_KEY",
|
|
"qwen-cn": "DASHSCOPE_CN_API_KEY",
|
|
"glm": "ZHIPU_API_KEY",
|
|
"glm-cn": "ZHIPU_CN_API_KEY",
|
|
"minimax": "MINIMAX_API_KEY",
|
|
"minimax-cn": "MINIMAX_CN_API_KEY",
|
|
"openrouter": "OPENROUTER_API_KEY",
|
|
# Additional hosted OpenAI-compatible providers (model is user-specified).
|
|
# kimi -> Moonshot AI; nvidia -> NVIDIA NIM.
|
|
"mistral": "MISTRAL_API_KEY",
|
|
"kimi": "MOONSHOT_API_KEY",
|
|
"groq": "GROQ_API_KEY",
|
|
"nvidia": "NVIDIA_API_KEY",
|
|
# Local runtimes do not authenticate.
|
|
"ollama": None,
|
|
# Generic OpenAI-compatible endpoint: the client reads this when set (keyed
|
|
# relays), but it is marked key-optional in the provider registry so the CLI
|
|
# never forces a prompt and keyless local servers still work.
|
|
"openai_compatible": "OPENAI_COMPATIBLE_API_KEY",
|
|
}
|
|
|
|
|
|
def get_api_key_env(provider: str) -> str | None:
|
|
"""Return the env var name for `provider`'s API key, or None if not applicable.
|
|
|
|
Unknown providers also return None — callers should treat that as
|
|
"no key check possible" rather than as "no key required".
|
|
"""
|
|
return PROVIDER_API_KEY_ENV.get(provider.lower())
|