Russian_AI_QA / config.py
Futyn-Maker's picture
Upload folder using huggingface_hub
6b83723 verified
raw
history blame contribute delete
663 Bytes
"""Application configuration loaded from environment variables."""
import os
from pathlib import Path
from dotenv import load_dotenv
load_dotenv()
BASE_DIR = Path(__file__).resolve().parent
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY", "")
EMBEDDING_MODEL = os.getenv("EMBEDDING_MODEL", "gemini-embedding-2-preview")
LLM_MODEL = os.getenv("LLM_MODEL", "gemini-3-flash-preview")
CHROMA_PERSIST_DIR = BASE_DIR / os.getenv("CHROMA_PERSIST_DIR", "chroma_db")
RAW_DATA_DIR = BASE_DIR / os.getenv("RAW_DATA_DIR", "raw_data")
CHROMA_COLLECTION = "rag_documents"
PROMPT_TEMPLATE_PATH = BASE_DIR / "prompt_template.txt"
DEFAULT_TOP_K = 10
DEFAULT_TEMPERATURE = 0.1