import os
from dotenv import load_dotenv

load_dotenv()


class Config:
    COHERE_API_KEY = os.getenv("COHERE_API_KEY")
    GROQ_API_KEY = os.getenv("GROQ_API_KEY")
    RSS_FEEDS = [
        "http://rss.nytimes.com/services/xml/rss/nyt/Technology.xml",
        "https://feeds.bbci.co.uk/news/technology/rss.xml",
        "https://feeds.a.dj.com/rss/RSSMarketsMain.xml",
        "https://rss.cnn.com/rss/edition.rss"
    ]
    VECTOR_DB_PATH = "data/vector_db.index"
    RAW_NEWS_PATH = "data/raw_news/"
    PROCESSED_NEWS_PATH = "data/processed_news/"
    EMBEDDING_MODEL = "embed-english-v3.0"
    GROQ_MODEL = "llama3-8b-8192"