src/config.py

import os
from pathlib import Path

from dotenv import load_dotenv

load_dotenv()

ROOT_DIR = Path(__file__).resolve().parent.parent
DEFAULT_DOC_DIR = ROOT_DIR / "my_docs"
CRAWLER_CACHE_PATH = ROOT_DIR / "crawler_docs.json"
PERSIST_ROOT = ROOT_DIR / "chroma_db"
PERSIST_ROOT.mkdir(parents=True, exist_ok=True)

HF_TOKEN = os.getenv("HF_TOKEN", "")
HF_MODEL = os.getenv("HF_MODEL", "deepseek-ai/DeepSeek-V4-Pro:novita")
HF_TASK = os.getenv("HF_TASK", "conversational")
OLLAMA_MODEL = os.getenv("OLLAMA_MODEL", "llama3")

DOC_DIR = Path(os.getenv("DOC_DIR", str(DEFAULT_DOC_DIR))).resolve()
CRAWLER_OUTPUT_PATH = Path(os.getenv("CRAWLER_OUTPUT_PATH", str(CRAWLER_CACHE_PATH))).resolve()