DH-5337/updating the golden sqls colleciton name

Dataherald · Apr 4, 2024 · 1755cbb · 1755cbb
1 parent 39bef01
commit 1755cbb
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/dataherald/context_store/__init__.py b/dataherald/context_store/__init__.py
@@ -18,7 +18,7 @@ def __init__(self, system: System):
         self.system = system
         self.db = self.system.instance(DB)
         self.golden_sql_collection = os.environ.get(
-            "GOLDEN_SQL_COLLECTION", "dataherald-staging"
+            "GOLDEN_SQL_COLLECTION", "ai-stage"
         )
         self.vector_store = self.system.instance(VectorStore)
 

diff --git a/dataherald/scripts/delete_and_populate_golden_records.py b/dataherald/scripts/delete_and_populate_golden_records.py
@@ -12,7 +12,7 @@
     system.start()
     storage = system.instance(DB)
     golden_sql_collection = os.environ.get(
-        "GOLDEN_SQL_COLLECTION", "dataherald-staging"
+        "GOLDEN_SQL_COLLECTION", "ai-stage"
     )
 
     golden_sqls = storage.find_all("golden_sqls")

diff --git a/dataherald/scripts/migrate_v001_to_v002.py b/dataherald/scripts/migrate_v001_to_v002.py
@@ -36,7 +36,7 @@ def add_db_connection_id(collection_name: str, storage) -> None:
     add_db_connection_id("nl_question", storage)
     # Refresh vector stores
     golden_record_collection = os.environ.get(
-        "GOLDEN_SQL_COLLECTION", "dataherald-staging"
+        "GOLDEN_SQL_COLLECTION", "ai-stage"
     )
     vector_store = system.instance(VectorStore)
     try:

diff --git a/dataherald/scripts/migrate_v006_to_v100.py b/dataherald/scripts/migrate_v006_to_v100.py
@@ -31,7 +31,7 @@ def update_object_id_fields(field_name: str, collection_name: str):
     storage = system.instance(DB)
     # Refresh vector stores
     golden_sql_collection = os.environ.get(
-        "GOLDEN_RECORD_COLLECTION", "dataherald-staging"
+        "GOLDEN_RECORD_COLLECTION", "ai-stage"
     )
     vector_store = system.instance(VectorStore)