config.py

# config.py
import os
from dotenv import load_dotenv

load_dotenv()

# API keys
API_KEY = os.getenv("API_KEY")
BASE_URL = os.getenv("BASE_URL")
STREAM_API_KEY = os.getenv("STREAM_API_KEY")
STREAM_BASE_URL = os.getenv("STREAM_BASE_URL")
STREAM_MODEL = os.getenv("STREAM_MODEL")
DEFAULT_MODEL = os.getenv("DEFAULT_MODEL")
OPENAI_MODEL = os.getenv("OPENAI_MODEL", "gpt-4o-mini")

# Knowledge base configuration
EMBEDDING_MODEL = os.getenv("EMBEDDING_MODEL", "BAAI/bge-m3")
RERANK_MODEL = os.getenv("RERANK_MODEL", "BAAI/bge-reranker-v2-m3")

# System configuration
UPLOAD_FOLDER = "uploads"
AGENTS_FOLDER = "agents"
MAX_CONTENT_LENGTH = 16 * 1024 * 1024  # 16MB