import os
from dotenv import load_dotenv

load_dotenv()


class Config:
    # Cohere
    COHERE_API_KEY = os.getenv("COHERE_API_KEY")
    EMBED_MODEL = "embed-english-v3.0"
    RERANK_MODEL = "rerank-english-v3.0"

    # Groq
    GROQ_API_KEY = os.getenv("GROQ_API_KEY")
    GROQ_MODEL = "llama3-70b-8192"

    # Claude
    CLAUDE_API_KEY = os.getenv("CLAUDE_API_KEY")
    CLAUDE_MODEL = "claude-3-5-sonnet-20241022"

    # Vector Store
    VECTOR_STORE_TYPE = "pinecone"
    PINECONE_API_KEY = os.getenv("PINECONE_API_KEY")
    PINECONE_INDEX = "scp-docs"
    PINECONE_ENV = "gcp-starter"
    EMBEDDING_DIMENSION = 1024

    # Document Processing
    MAX_DOC_SIZE = 10 * 1024 * 1024  # 10MB
    ALLOWED_EXTENSIONS = {'.pdf', '.docx', '.txt'}

    # Paths
    UPLOAD_FOLDER = "documents/"