File size: 522 Bytes
640b9e7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import json

def load_embeddings(filename):
    '''
    Load embeddings from specified json file

    Args:
        filename: Name of the json file to load embeddings from

    Returns:
        List of embeddings loaded from the file
    '''
    print("Loading embeddings...")
    with open(filename, 'r') as file:
        embeddings = json.load(file)
    return embeddings

DOC_CHUNK_SIZE = 1000
DOC_CHUNK_OVERLAP = 40
DOC_DIRECTORY = './'
EMBEDDING_FILE = 'embeddings.json'
BATCH_SIZE = 100
INDEX_NAME = "project-falcon"