mirror of
https://github.com/GoodStartLabs/AI_Diplomacy.git
synced 2026-04-19 12:58:09 +00:00
540c2003e8
The PowerEnum correctly handles some misspellings. It can be easily expanded to handle more within the _POWER_ALIASES dict.
48 lines
1.8 KiB
Python
48 lines
1.8 KiB
Python
import datetime
|
|
from pydantic_settings import BaseSettings
|
|
from pathlib import Path
|
|
import warnings
|
|
|
|
|
|
class Configuration(BaseSettings):
|
|
DEBUG: bool = False
|
|
log_file_path: Path | None = None
|
|
DEEPSEEK_API_KEY: str | None = None
|
|
OPENAI_API_KEY: str | None = None
|
|
ANTHROPIC_API_KEY: str | None = None
|
|
GEMINI_API_KEY: str | None = None
|
|
OPENROUTER_API_KEY: str | None = None
|
|
|
|
def __init__(self, power_name, **kwargs):
|
|
super().__init__(**kwargs)
|
|
self.log_file_path = Path(f"./logs/{datetime.datetime.now().strftime('%d-%m-%y_%H:%M')}/{power_name}.txt")
|
|
# Make the path absolute, gets rid of weirdness of calling this in different places
|
|
self.log_file_path = self.log_file_path.resolve()
|
|
self.log_file_path.parent.mkdir(parents=True, exist_ok=True)
|
|
self.log_file_path.touch(exist_ok=True)
|
|
|
|
self._validate_api_keys()
|
|
|
|
def _validate_api_keys(self):
|
|
"""Validate API keys at startup and issue warnings for missing keys"""
|
|
api_keys = [
|
|
"DEEPSEEK_API_KEY",
|
|
"OPENAI_API_KEY",
|
|
"ANTHROPIC_API_KEY",
|
|
"GEMINI_API_KEY",
|
|
"OPENROUTER_API_KEY",
|
|
]
|
|
|
|
for key in api_keys:
|
|
value = super().__getattribute__(key)
|
|
if not value or (isinstance(value, str) and len(value) == 0):
|
|
warnings.warn(f"API key '{key}' is not set or is empty", UserWarning)
|
|
|
|
def __getattribute__(self, name):
|
|
"""Override to check for empty API keys at access time"""
|
|
value = super().__getattribute__(name)
|
|
|
|
if name.endswith("_KEY") and (not value or (isinstance(value, str) and len(value) == 0)):
|
|
raise ValueError(f"API key '{name}' is not set or is empty. Please configure it before use.")
|
|
|
|
return value
|