File size: 1,140 Bytes
abb6f94
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
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)

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