51 lines
2.8 KiB
Python
51 lines
2.8 KiB
Python
"""Environment-backed Fiscal Hub configuration."""
|
|
|
|
from dataclasses import dataclass
|
|
import os
|
|
from urllib.parse import quote
|
|
|
|
|
|
def _database_url() -> str:
|
|
configured = os.getenv("DATABASE_URL")
|
|
if configured:
|
|
return configured
|
|
password = quote(os.getenv("POSTGRES_PASSWORD", "gti"), safe="")
|
|
return f"postgresql+psycopg://gti:{password}@postgres:5432/gti_fiscal"
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class Settings:
|
|
database_url: str = _database_url()
|
|
hub_port: int = int(os.getenv("HUB_PORT", "8000"))
|
|
# AGENT_DEV_TOKEN auto-bootstraps an agent from a fixed dev token on every request.
|
|
# Leave unset in any environment other than local development — production Agents
|
|
# must be created via scripts/register_agent.py, whose token is never derived from
|
|
# this setting.
|
|
agent_dev_token: str = os.getenv("AGENT_DEV_TOKEN", "")
|
|
environment: str = os.getenv("APP_ENV", "development")
|
|
log_level: str = os.getenv("LOG_LEVEL", "INFO")
|
|
base_url: str = os.getenv("BASE_URL", "http://localhost:8000")
|
|
# An Agent is considered ONLINE if a heartbeat arrived within this many seconds.
|
|
agent_online_threshold_seconds: int = int(os.getenv("AGENT_ONLINE_THRESHOLD_SECONDS", "120"))
|
|
ncm_dataset_url: str = os.getenv("NCM_DATASET_URL", "https://portalunico.siscomex.gov.br/classif/api/publico/nomenclatura/download/json")
|
|
cest_dataset_url: str = os.getenv("CEST_DATASET_URL", "https://www.confaz.fazenda.gov.br/legislacao/convenios/2018/CV142_18")
|
|
dataset_storage_path: str = os.getenv("DATASET_STORAGE_PATH", "./data/datasets")
|
|
http_timeout_seconds: float = float(os.getenv("HTTP_TIMEOUT_SECONDS", "30"))
|
|
# ---- AI semantic layer V1 ----
|
|
# Credentials live ONLY in the Hub environment; the Windows Agent never sees them.
|
|
ai_enabled: bool = os.getenv("AI_ENABLED", "0") == "1"
|
|
ai_provider: str = os.getenv("AI_PROVIDER", "baseline") # baseline | openai | ... (AISemanticProvider)
|
|
ai_model: str = os.getenv("AI_MODEL", "baseline-deterministic-v1")
|
|
ai_max_batch_size: int = int(os.getenv("AI_MAX_BATCH_SIZE", "20"))
|
|
ai_max_run_cost_usd: float = float(os.getenv("AI_MAX_RUN_COST_USD", "5.0"))
|
|
ai_min_confidence: float = float(os.getenv("AI_MIN_CONFIDENCE", "0.6"))
|
|
ai_price_in_per_m: float | None = float(os.getenv("AI_PRICE_IN_PER_M")) if os.getenv("AI_PRICE_IN_PER_M") else None
|
|
ai_price_out_per_m: float | None = float(os.getenv("AI_PRICE_OUT_PER_M")) if os.getenv("AI_PRICE_OUT_PER_M") else None
|
|
# Real-LLM evaluation (OpenAI-compatible). The key lives ONLY in this environment.
|
|
# Z.ai general endpoint (NOT the /api/coding/paas/v4 coding-plan endpoint).
|
|
ai_api_key: str = os.getenv("AI_API_KEY", "")
|
|
ai_base_url: str = os.getenv("AI_BASE_URL", "https://api.z.ai/api/paas/v4")
|
|
|
|
|
|
settings = Settings()
|