DH-4677 Fix in migration script when sql_query doesn't have a table (#…

…168)
Dataherald · Sep 18, 2023 · 6773689 · 6773689
1 parent 733eaea
commit 6773689
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dataherald/scripts/migrate_v001_to_v002.py b/dataherald/scripts/migrate_v001_to_v002.py
@@ -47,6 +47,8 @@ def add_db_connection_id(collection_name: str, storage) -> None:
     golden_records = storage.find_all("golden_records")
     for golden_record in golden_records:
         tables = Parser(golden_record["sql_query"]).tables
+        if len(tables) == 0:
+            tables = [""]
         question = golden_record["question"]
         vector_store.add_record(
             documents=question,