llama-index

1
"""Set of constants."""
2

3
DEFAULT_TEMPERATURE = 0.1
4
DEFAULT_CONTEXT_WINDOW = 3900  # tokens
5
DEFAULT_NUM_OUTPUTS = 256  # tokens
6
DEFAULT_NUM_INPUT_FILES = 10  # files
7

8
DEFAULT_EMBED_BATCH_SIZE = 10
9

10
DEFAULT_CHUNK_SIZE = 1024  # tokens
11
DEFAULT_CHUNK_OVERLAP = 20  # tokens
12
DEFAULT_SIMILARITY_TOP_K = 2
13
DEFAULT_IMAGE_SIMILARITY_TOP_K = 2
14

15
# NOTE: for text-embedding-ada-002
16
DEFAULT_EMBEDDING_DIM = 1536
17

18
# context window size for llm predictor
19
COHERE_CONTEXT_WINDOW = 2048
20
AI21_J2_CONTEXT_WINDOW = 8192
21

22

23
TYPE_KEY = "__type__"
24
DATA_KEY = "__data__"
25
VECTOR_STORE_KEY = "vector_store"
26
IMAGE_STORE_KEY = "image_store"
27
GRAPH_STORE_KEY = "graph_store"
28
INDEX_STORE_KEY = "index_store"
29
DOC_STORE_KEY = "doc_store"
30