Merge pull request #175 from Hikari-Haru/check-index-type

Check index type

models/index_model.py

@@ -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)