Wire up the keyword table v1 to v2 migration (run-llama#1027)

Fixes run-llama#1026
joyxof · Apr 3, 2023 · 3c065e3 · 3c065e3
1 parent cc831a5
commit 3c065e3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/gpt_index/tools/migrate_v1_to_v2.py b/gpt_index/tools/migrate_v1_to_v2.py
@@ -213,6 +213,8 @@ def convert_to_v2_index_struct_and_docstore(
         struct_v2, nodes_v2 = index_dict_to_v2(index_struct)
     elif isinstance(index_struct, KG):
         struct_v2, nodes_v2 = kg_to_v2(index_struct)
+    elif isinstance(index_struct, KeywordTable):
+        struct_v2, nodes_v2 = keyword_table_to_v2(index_struct)
     else:
         raise NotImplementedError(f"Cannot migrate {type(index_struct)} yet.")