File tree Expand file tree Collapse file tree 2 files changed +7
-2
lines changed
Expand file tree Collapse file tree 2 files changed +7
-2
lines changed Original file line number Diff line number Diff line change 44from rag_documents import RagDocumentsAISearch
55from text_2_sql_schema_store import Text2SqlSchemaStoreAISearch
66from text_2_sql_query_cache import Text2SqlQueryCacheAISearch
7+ from text_2_sql_column_value_store import Text2SqlColumnValueStoreAISearch
78import logging
89
910logging .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
Original file line number Diff line number Diff line change 2222from 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
You can’t perform that action at this time.
0 commit comments