from pydantic_settings import BaseSettings
import os
from pathlib import Path

class Settings(BaseSettings):
    MODEL_NAME: str = "mistralai/Mistral-7B-Instruct-v0.2"

    # Model parameters
    MODEL_MAX_TOKENS: int = 512
    MODEL_TEMPERATURE: float = 0.7

    # Retriever parameters
    RETRIEVER_K_BEFORE_RERANK: int = 80
    RETRIEVER_K_AFTER_RERANK: int = 4
    RETRIEVER_K_CONSTANT: int = 60

    HF_TOKEN: str

    class Config:
        env_file = ".env"

settings = Settings()