Spaces:
Runtime error
Runtime error
import os | |
from dotenv import load_dotenv | |
from pathlib import Path | |
# env_path = Path(__file__).resolve().parent.parent / '.env' | |
# load_dotenv(dotenv_path=env_path, override=True) | |
load_dotenv(override=True) | |
class Config: | |
API_KEY = os.getenv('OPENAI_API_KEY') | |
MODEL_ID = os.getenv('MODEL_ID') | |
MODEL_BASENAME = os.getenv('MODEL_BASENAME') | |
COLLECTION_NAME = os.getenv('COLLECTION_NAME') | |
PERSIST_DIRECTORY = os.path.join(os.path.dirname(__file__),'vector_store') | |
os.makedirs(PERSIST_DIRECTORY, exist_ok=True) | |
UPLOAD_DIR = os.path.join(os.path.dirname(__file__),'uploads') | |
os.makedirs(UPLOAD_DIR, exist_ok=True) | |
LOG_DIR = os.path.join(os.path.dirname(__file__),'log_dir') | |
os.makedirs(LOG_DIR, exist_ok=True) | |
MODELS_PATH = os.path.join(os.path.dirname(__file__),'models') | |
CACHE_DIR = os.path.join(os.path.dirname(__file__),'models') | |
# os.makedirs(CACHE_DIR, exist_ok=True) | |
# MODELS_PATH = '/models' | |
MODEL_NAME = "sentence-transformers/all-mpnet-base-v2" | |
MODEL_KWARGS = {'device': 'cpu'} | |
ENCODE_KWARGS = {'normalize_embeddings': False} | |
CHUNK_SIZE = 1024 | |
CHUNK_OVERLAP = 200 |