Create Config dataclass for LLM options

This commit is contained in:
Marty Oehme 2025-06-06 10:22:14 +02:00
parent 3668cc3cbd
commit f96b6413e2
Signed by: Marty
GPG key ID: 4E535BC19C61886E
2 changed files with 28 additions and 3 deletions

22
prophet/config.py Normal file
View file

@ -0,0 +1,22 @@
import os
# Load environment variables from .env
from dataclasses import dataclass
from dotenv import load_dotenv
_ = load_dotenv()
@dataclass
class AiConfig:
API_KEY: str
@classmethod
def from_env(cls) -> "AiConfig":
API_KEY = os.getenv("GROQ_API_KEY", "")
if not API_KEY:
raise ValueError(f"{API_KEY} cannot be empty")
return cls(**{"API_KEY": API_KEY})