config.py 1.5 KB

123456789101112131415161718192021222324252627282930313233343536
  1. import os
  2. from pathlib import Path
  3. from dotenv import load_dotenv
  4. # Load .env from project folder so Groq/debug flags are available under uvicorn/nohup.
  5. _HERE = Path(__file__).resolve().parent.parent
  6. load_dotenv(_HERE / ".env")
  7. DATA_DIR = Path(os.getenv("NEWS_MCP_DATA_DIR", Path(__file__).resolve().parent / "data"))
  8. DATA_DIR.mkdir(parents=True, exist_ok=True)
  9. DB_PATH = Path(os.getenv("NEWS_MCP_DB_PATH", str(DATA_DIR / "news.sqlite")))
  10. RSS_FEED_URL = os.getenv("NEWS_RSS_FEED_URL", "https://breakingthenews.net/news-feed.xml")
  11. # Clusters TTL (hours)
  12. CLUSTERS_TTL_HOURS = float(os.getenv("NEWS_CLUSTERS_TTL_HOURS", "24"))
  13. DEFAULT_TOPICS = ["crypto", "macro", "regulation", "ai", "other"]
  14. # Optional LLM enrichment (Groq)
  15. GROQ_API_KEY = os.getenv("GROQ_API_KEY")
  16. GROQ_MODEL = os.getenv("GROQ_MODEL", "llama4-16e")
  17. GROQ_DEBUG = os.getenv("GROQ_DEBUG", "false").lower() == "true"
  18. # Cost control: only enrich clusters whose heuristic topic is "other" by default.
  19. GROQ_ENRICH_OTHER_ONLY = os.getenv("GROQ_ENRICH_OTHER_ONLY", "true").lower() == "true"
  20. # Limit enriched clusters per refresh call.
  21. GROQ_MAX_CLUSTERS_PER_REFRESH = int(os.getenv("GROQ_MAX_CLUSTERS_PER_REFRESH", "20"))
  22. # Background refresh
  23. NEWS_REFRESH_INTERVAL_SECONDS = int(os.getenv("NEWS_REFRESH_INTERVAL_SECONDS", "900"))
  24. NEWS_BACKGROUND_REFRESH_ENABLED = os.getenv("NEWS_BACKGROUND_REFRESH_ENABLED", "true").lower() == "true"
  25. NEWS_BACKGROUND_REFRESH_ON_START = os.getenv("NEWS_BACKGROUND_REFRESH_ON_START", "true").lower() == "true"