mirror of
https://github.com/TauricResearch/TradingAgents.git
synced 2026-06-16 21:06:15 +03:00
9f7abfcbd5
Adds a canonical PROVIDER_API_KEY_ENV mapping (14 providers including the three dual-region pairs) and an ensure_api_key() helper. When the selected provider's key is absent from the environment, the CLI prompts via questionary.password, writes the value to .env via python-dotenv's set_key (preserves existing lines), and exports it into os.environ so the run continues without restart. Wired into cli/main.py right after the region prompts so qwen-cn, glm-cn, and minimax-cn each check their own region-specific key. openai_client refactored to consult the same mapping, eliminating its private duplicate of provider→env-var data.
45 lines
1.6 KiB
Python
45 lines
1.6 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
|
|
|
|
from typing import Optional
|
|
|
|
|
|
PROVIDER_API_KEY_ENV: dict[str, Optional[str]] = {
|
|
"openai": "OPENAI_API_KEY",
|
|
"anthropic": "ANTHROPIC_API_KEY",
|
|
"google": "GOOGLE_API_KEY",
|
|
"azure": "AZURE_OPENAI_API_KEY",
|
|
"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",
|
|
# Local runtimes do not authenticate.
|
|
"ollama": None,
|
|
}
|
|
|
|
|
|
def get_api_key_env(provider: str) -> Optional[str]:
|
|
"""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())
|