From 1755cbbe0579aac8b2b5de4eeb8100f775605b5b Mon Sep 17 00:00:00 2001 From: mohammadrezapourreza Date: Thu, 4 Apr 2024 09:40:52 -0400 Subject: [PATCH] DH-5337/updating the golden sqls colleciton name --- dataherald/context_store/__init__.py | 2 +- dataherald/scripts/delete_and_populate_golden_records.py | 2 +- dataherald/scripts/migrate_v001_to_v002.py | 2 +- dataherald/scripts/migrate_v006_to_v100.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/dataherald/context_store/__init__.py b/dataherald/context_store/__init__.py index 2ae52dbc..19d85a03 100644 --- 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 index cc6bad8e..a1578097 100644 --- 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 index a45ce309..b6dd504b 100644 --- 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 index 6f15e115..1e6b8ba3 100644 --- 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)