bugfix annoy_search

c7nw3r · Aug 30, 2023 · 59e4437 · 59e4437
1 parent 90c56d8
commit 59e4437
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/localsearch/searcher/annoy_search.py b/localsearch/searcher/annoy_search.py
@@ -107,7 +107,8 @@ def _rebuild(self):
         folder = self.config.raw_data_dir if self.config.raw_data_dir else self.path.replace(".ann", "")
         for path in os.listdir(folder):
             if path.endswith(".json"):
-                idx = int(path.replace(".json", ""))
+                idx = path.replace(".json", "")
+                idx = read_json(f"{folder}/id_str/{idx}.json")["id"]
                 try:
                     vector = self.index.get_item_vector(idx)
                     new_index.add_item(idx, vector)