[DH-4996] Fix script to populate vector store

Dataherald · Nov 13, 2023 · 451ae16 · 451ae16
1 parent f37743f
commit 451ae16
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dataherald/scripts/delete_and_populate_golden_records.py b/dataherald/scripts/delete_and_populate_golden_records.py
@@ -29,12 +29,12 @@
         question = golden_record["question"]
         vector_store.add_record(
             documents=question,
-            db_connection_id=golden_record["db_connection_id"],
+            db_connection_id=str(golden_record["db_connection_id"]),
             collection=golden_record_collection,
             metadata=[
                 {
                     "tables_used": tables[0],
-                    "db_connection_id": golden_record["db_connection_id"],
+                    "db_connection_id": str(golden_record["db_connection_id"]),
                 }
             ],
             ids=[str(golden_record["_id"])],