Check index type

Rene Teigen 1 year ago
parent c80434aa15
commit c2af9763f2

@ -4,6 +4,7 @@ import random
import tempfile
import traceback
import asyncio
import json
from collections import defaultdict
import aiohttp
@ -239,7 +240,13 @@ class Index_handler:
return index
def index_load_file(self, file_path) -> [GPTSimpleVectorIndex, ComposableGraph]:
if "composed_deep" in str(file_path):
with open(file_path, "r", encoding="utf8") as f:
file_contents = f.read()
index_dict = json.loads(file_contents)
doc_id = index_dict['index_struct_id']
doc_type = index_dict['docstore']['docs'][doc_id]['__type__']
f.close()
if doc_type == "tree":
index = GPTTreeIndex.load_from_disk(file_path)
else:
index = GPTSimpleVectorIndex.load_from_disk(file_path)

Loading…
Cancel
Save