Tomas Kracmar
be319688f6
- Add LLM_API_VERSION config for Azure api-version query param - Detect Azure endpoints and use api-key header instead of Bearer - Handle base URLs that already include /chat/completions path - Update .env.example with Azure OpenAI guidance
86 lines
2.5 KiB
Python
86 lines
2.5 KiB
Python
from pydantic_settings import BaseSettings, SettingsConfigDict
|
|
|
|
|
|
class Settings(BaseSettings):
|
|
model_config = SettingsConfigDict(
|
|
env_file=[".env", "../.env"],
|
|
env_file_encoding="utf-8",
|
|
extra="ignore",
|
|
)
|
|
|
|
# Microsoft Graph / App credentials
|
|
TENANT_ID: str = ""
|
|
CLIENT_ID: str = ""
|
|
CLIENT_SECRET: str = ""
|
|
|
|
# MongoDB
|
|
MONGO_URI: str = ""
|
|
DB_NAME: str = "micro_soc"
|
|
|
|
# Periodic fetch
|
|
ENABLE_PERIODIC_FETCH: bool = False
|
|
FETCH_INTERVAL_MINUTES: int = 60
|
|
|
|
# Auth (OIDC/Bearer) settings
|
|
AUTH_ENABLED: bool = False
|
|
AUTH_TENANT_ID: str = ""
|
|
AUTH_CLIENT_ID: str = ""
|
|
AUTH_SCOPE: str = ""
|
|
AUTH_ALLOWED_ROLES: str = ""
|
|
AUTH_ALLOWED_GROUPS: str = ""
|
|
|
|
# Data retention (0 = disabled)
|
|
RETENTION_DAYS: int = 0
|
|
|
|
# CORS
|
|
CORS_ORIGINS: str = "*"
|
|
|
|
# SIEM export
|
|
SIEM_ENABLED: bool = False
|
|
SIEM_WEBHOOK_URL: str = ""
|
|
|
|
# Alerting
|
|
ALERTS_ENABLED: bool = False
|
|
|
|
# LLM / Natural Language Query
|
|
LLM_API_KEY: str = ""
|
|
LLM_BASE_URL: str = "https://api.openai.com/v1"
|
|
LLM_MODEL: str = "gpt-4o-mini"
|
|
LLM_MAX_EVENTS: int = 50
|
|
LLM_TIMEOUT_SECONDS: int = 30
|
|
LLM_API_VERSION: str = "" # e.g. 2025-01-01-preview for Azure OpenAI
|
|
|
|
|
|
_settings = Settings()
|
|
|
|
# Backward-compatible module-level exports
|
|
TENANT_ID = _settings.TENANT_ID
|
|
CLIENT_ID = _settings.CLIENT_ID
|
|
CLIENT_SECRET = _settings.CLIENT_SECRET
|
|
MONGO_URI = _settings.MONGO_URI
|
|
DB_NAME = _settings.DB_NAME
|
|
|
|
ENABLE_PERIODIC_FETCH = _settings.ENABLE_PERIODIC_FETCH
|
|
FETCH_INTERVAL_MINUTES = _settings.FETCH_INTERVAL_MINUTES
|
|
|
|
AUTH_ENABLED = _settings.AUTH_ENABLED
|
|
AUTH_TENANT_ID = _settings.AUTH_TENANT_ID or _settings.TENANT_ID or ""
|
|
AUTH_CLIENT_ID = _settings.AUTH_CLIENT_ID or _settings.CLIENT_ID or ""
|
|
AUTH_SCOPE = _settings.AUTH_SCOPE
|
|
AUTH_ALLOWED_ROLES = {r.strip() for r in _settings.AUTH_ALLOWED_ROLES.split(",") if r.strip()}
|
|
AUTH_ALLOWED_GROUPS = {g.strip() for g in _settings.AUTH_ALLOWED_GROUPS.split(",") if g.strip()}
|
|
|
|
RETENTION_DAYS = _settings.RETENTION_DAYS
|
|
CORS_ORIGINS = [o.strip() for o in _settings.CORS_ORIGINS.split(",") if o.strip()]
|
|
|
|
SIEM_ENABLED = _settings.SIEM_ENABLED
|
|
SIEM_WEBHOOK_URL = _settings.SIEM_WEBHOOK_URL
|
|
ALERTS_ENABLED = _settings.ALERTS_ENABLED
|
|
|
|
LLM_API_KEY = _settings.LLM_API_KEY
|
|
LLM_BASE_URL = _settings.LLM_BASE_URL
|
|
LLM_MODEL = _settings.LLM_MODEL
|
|
LLM_MAX_EVENTS = _settings.LLM_MAX_EVENTS
|
|
LLM_TIMEOUT_SECONDS = _settings.LLM_TIMEOUT_SECONDS
|
|
LLM_API_VERSION = _settings.LLM_API_VERSION
|