Spaces:
Sleeping
Sleeping
| from __future__ import annotations | |
| import os | |
| from fastapi import HTTPException | |
| from inference.config import get_inference_manifest | |
| # Legacy Python service env vars remain supported for credential fallback. | |
| _LEGACY_API_KEY_ENVS = ( | |
| "TABULAR_AGENT_API_KEY", | |
| "GOOGLE_API_KEY", | |
| "GEMINI_API_KEY", | |
| "WORD_REVIEW_API_KEY", | |
| ) | |
| def resolve_api_key(provider: str = "google-genai") -> str | None: | |
| manifest = get_inference_manifest() | |
| providers = manifest.get("providers") or {} | |
| provider_entry = providers.get(provider) or {} | |
| api_key_env = provider_entry.get("apiKeyEnv") | |
| if isinstance(api_key_env, str) and api_key_env.strip(): | |
| value = os.environ.get(api_key_env, "").strip() | |
| if value: | |
| return value | |
| for env_name in _LEGACY_API_KEY_ENVS: | |
| value = os.environ.get(env_name, "").strip() | |
| if value: | |
| return value | |
| return None | |
| def assert_api_key(provider: str = "proxy-openai") -> str: | |
| key = resolve_api_key(provider) | |
| if not key: | |
| raise HTTPException( | |
| status_code=500, | |
| detail="TabularAgent API key is not configured.", | |
| ) | |
| return key | |