Skip to content

Commit 0dfe95b

Browse files
committed
Try deploy script
1 parent 1678189 commit 0dfe95b

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

deploy_ai_search/deploy.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
from rag_documents import RagDocumentsAISearch
55
from text_2_sql_schema_store import Text2SqlSchemaStoreAISearch
66
from text_2_sql_query_cache import Text2SqlQueryCacheAISearch
7+
from text_2_sql_column_value_store import Text2SqlColumnValueStoreAISearch
78
import logging
89

910
logging.basicConfig(level=logging.INFO)
@@ -33,6 +34,11 @@ def deploy_config(arguments: argparse.Namespace):
3334
single_query_cache_file=arguments.single_query_cache_file,
3435
enable_query_cache_indexer=arguments.enable_query_cache_indexer,
3536
)
37+
elif arguments.index_type == "text_2_sql_column_value_store":
38+
index_config = Text2SqlColumnValueStoreAISearch(
39+
suffix=arguments.suffix,
40+
rebuild=arguments.rebuild,
41+
)
3642
else:
3743
raise ValueError("Invalid Indexer Type")
3844

deploy_ai_search/text_2_sql_column_value_store.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,14 +22,13 @@
2222
from text_2_sql_core.utils.database import DatabaseEngine
2323

2424

25-
class Text2SqlSchemaStoreAISearch(AISearch):
25+
class Text2SqlColumnValueStoreAISearch(AISearch):
2626
"""This class is used to deploy the sql index."""
2727

2828
def __init__(
2929
self,
3030
suffix: str | None = None,
3131
rebuild: bool | None = False,
32-
single_data_dictionary_file: bool | None = False,
3332
):
3433
"""Initialize the Text2SqlAISearch class. This class implements the deployment of the sql index.
3534

0 commit comments

Comments
 (0)