import os
from pathlib import Path
from dotenv import load_dotenv

load_dotenv(Path(__file__).parent.parent / ".env")

NVIDIA_API_KEY = os.environ["NVIDIA_API_KEY"]
LLM_BASE_URL = "https://integrate.api.nvidia.com/v1"
LLM_MODEL = os.getenv("LLM_MODEL", "meta/llama-3.1-70b-instruct")
EMBED_MODEL = os.getenv("EMBED_MODEL", "nvidia/llama-nemotron-embed-1b-v2")

_data_dir = Path(__file__).parent / "data"
CHROMA_PERSIST_DIR = os.getenv("CHROMA_PERSIST_DIR", str(_data_dir / "chroma"))
DATA_DIR = _data_dir
RAW_DIR = _data_dir / "raw"
PARSED_DIR = _data_dir / "parsed"
CURATED_DIR = _data_dir / "curated"
CHROMA_COLLECTION = "epadoms_laws"