llama-index

1
"""Index registry."""
2

3
from typing import Dict, Type
4

5
from llama_index.legacy.data_structs.data_structs import (
6
    KG,
7
    EmptyIndexStruct,
8
    IndexDict,
9
    IndexGraph,
10
    IndexList,
11
    IndexStruct,
12
    KeywordTable,
13
    MultiModelIndexDict,
14
)
15
from llama_index.legacy.data_structs.document_summary import IndexDocumentSummary
16
from llama_index.legacy.data_structs.struct_type import IndexStructType
17
from llama_index.legacy.data_structs.table import PandasStructTable, SQLStructTable
18

19
INDEX_STRUCT_TYPE_TO_INDEX_STRUCT_CLASS: Dict[IndexStructType, Type[IndexStruct]] = {
20
    IndexStructType.TREE: IndexGraph,
21
    IndexStructType.LIST: IndexList,
22
    IndexStructType.KEYWORD_TABLE: KeywordTable,
23
    IndexStructType.VECTOR_STORE: IndexDict,
24
    IndexStructType.SQL: SQLStructTable,
25
    IndexStructType.PANDAS: PandasStructTable,
26
    IndexStructType.KG: KG,
27
    IndexStructType.EMPTY: EmptyIndexStruct,
28
    IndexStructType.DOCUMENT_SUMMARY: IndexDocumentSummary,
29
    IndexStructType.MULTIMODAL_VECTOR_STORE: MultiModelIndexDict,
30
}
31