Resolve the correct API key env var(s) based on model provider prefix.
(self)
| 216 | |
| 217 | return config |
| 218 | |
| 219 | # Snapshot env vars at class-load time |
| 220 | _original_env = { |
| 221 | k: os.environ.get(k, "") |
| 222 | for k in [ |
| 223 | "OPENROUTER_API_KEY", |
| 224 | "ANTHROPIC_API_KEY", |
| 225 | "OPENAI_API_KEY", |
| 226 | "GEMINI_API_KEY", |
| 227 | ] |
| 228 | } |
| 229 | |
| 230 | def _resolve_api_key_env(self) -> dict[str, str]: |
| 231 | """Resolve the correct API key env var(s) based on model provider prefix.""" |
| 232 | model_id = self.model_name or self._default_model |
| 233 | provider_name = model_id.split("/", 1)[0] if "/" in model_id else "openrouter" |
| 234 | |
| 235 | env_var = _PROVIDER_KEY_MAP.get(provider_name, "OPENROUTER_API_KEY") |