llama-index

1
from llama_index.legacy.constants import DATA_KEY, TYPE_KEY
2
from llama_index.legacy.data_structs.data_structs import IndexStruct
3
from llama_index.legacy.data_structs.registry import (
4
    INDEX_STRUCT_TYPE_TO_INDEX_STRUCT_CLASS,
5
)
6

7

8
def index_struct_to_json(index_struct: IndexStruct) -> dict:
9
    return {
10
        TYPE_KEY: index_struct.get_type(),
11
        DATA_KEY: index_struct.to_json(),
12
    }
13

14

15
def json_to_index_struct(struct_dict: dict) -> IndexStruct:
16
    type = struct_dict[TYPE_KEY]
17
    data_dict = struct_dict[DATA_KEY]
18
    cls = INDEX_STRUCT_TYPE_TO_INDEX_STRUCT_CLASS[type]
19
    try:
20
        return cls.from_json(data_dict)
21
    except TypeError:
22
        return cls.from_dict(data_dict)
23