final_project2 / src /config.py
dnj0's picture
Update src/config.py
cffffd0 verified
import os
from pathlib import Path
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
OPENAI_MODEL = "gpt-4o-mini"
USE_CACHE = True
CHROMA_DB_PATH = "./chroma_db"
DOCSTORE_PATH = "./docstore"
PROCESSED_FILES_LOG = "./processed_files.txt"
EMBEDDING_MODEL = "sentence-transformers/all-mpnet-base-v2"
EMBEDDING_DIM = 768
MAX_CHUNK_SIZE = 500
CHUNK_OVERLAP = 50
TEMPERATURE = 0.3
MAX_TOKENS = 500
LANGUAGE = "russian"
Path(CHROMA_DB_PATH).mkdir(exist_ok=True)
Path(DOCSTORE_PATH).mkdir(exist_ok=True)
UPLOAD_FOLDER = "./uploaded_pdfs"
Path(UPLOAD_FOLDER).mkdir(exist_ok=True)
MAX_PDF_SIZE_MB = 50
BATCH_SEARCH_RESULTS = 3
CACHE_RESPONSES = True
SUMMARIZE_FIRST = True