| 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" |
|
|
