diff --git a/dataherald/scripts/delete_and_populate_golden_records.py b/dataherald/scripts/delete_and_populate_golden_records.py index fe7ceeab..f5c4aeb3 100644 --- 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"])],