import os
from dotenv import load_dotenv

# Load environment variables from .env file
load_dotenv()

# API Keys
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")

# Model Configuration
EMBEDDING_MODEL = "text-embedding-3-small"
CHAT_MODEL = "gpt-4o"

# Retrieval Configuration
TOP_K = 200
EMBEDDING_BATCH_SIZE = 20

# File Paths
DATA_DIR = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(__file__))), "data")
EMBEDDINGS_DIR = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(__file__))), "embeddings")

# FAISS Index Configuration
FAISS_INDEX_PATH = os.path.join(EMBEDDINGS_DIR, "faiss_index.index")
DOC_CHUNKS_PATH = os.path.join(DATA_DIR, "doc_chunks.pkl")
EMBEDDINGS_PATH = os.path.join(EMBEDDINGS_DIR, "embeddings.pkl")