config.py 1.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  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. # Backward-compatible aliases for older config names.
  11. NEWS_FEED_URL = os.getenv("NEWS_FEED_URL", os.getenv("NEWS_RSS_FEED_URL", "https://breakingthenews.net/news-feed.xml"))
  12. # Optional multi-feed mode: comma-separated feed URLs.
  13. # If set (non-empty), this overrides NEWS_FEED_URL.
  14. NEWS_FEED_URLS = os.getenv("NEWS_FEED_URLS", os.getenv("NEWS_RSS_FEED_URLS", "")).strip()
  15. # Legacy names kept for compatibility.
  16. RSS_FEED_URL = NEWS_FEED_URL
  17. RSS_FEED_URLS = NEWS_FEED_URLS
  18. # Clusters TTL (hours)
  19. CLUSTERS_TTL_HOURS = float(os.getenv("NEWS_CLUSTERS_TTL_HOURS", "24"))
  20. DEFAULT_TOPICS = ["crypto", "macro", "regulation", "ai", "other"]
  21. # Optional LLM enrichment (Groq)
  22. GROQ_API_KEY = os.getenv("GROQ_API_KEY")
  23. GROQ_MODEL = os.getenv("GROQ_MODEL", "llama4-16e")
  24. GROQ_DEBUG = os.getenv("GROQ_DEBUG", "false").lower() == "true"
  25. # Groq enrichment is the default for all incoming news.
  26. # Set GROQ_ENRICH_OTHER_ONLY=true only if you want to restrict it for cost control.
  27. GROQ_ENRICH_OTHER_ONLY = os.getenv("GROQ_ENRICH_OTHER_ONLY", "false").lower() == "true"
  28. # Limit enriched clusters per refresh call.
  29. GROQ_MAX_CLUSTERS_PER_REFRESH = int(os.getenv("GROQ_MAX_CLUSTERS_PER_REFRESH", "20"))
  30. # Background refresh
  31. NEWS_REFRESH_INTERVAL_SECONDS = int(os.getenv("NEWS_REFRESH_INTERVAL_SECONDS", "900"))
  32. NEWS_BACKGROUND_REFRESH_ENABLED = os.getenv("NEWS_BACKGROUND_REFRESH_ENABLED", "true").lower() == "true"
  33. NEWS_BACKGROUND_REFRESH_ON_START = os.getenv("NEWS_BACKGROUND_REFRESH_ON_START", "true").lower() == "true"