File size: 680 Bytes
1625bb7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
# 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 |