File size: 683 Bytes
cffffd0
ecab17a
 
 
 
cffffd0
 
ecab17a
 
 
 
 
 
 
 
cffffd0
 
 
 
ecab17a
 
 
 
 
 
 
 
f84a554
ecab17a
cffffd0
 
 
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

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