Update imports

Author: BenConstable9

text_2_sql/autogen/src/autogen_text_2_sql/autogen_text_2_sql.py

@@ -11,11 +11,11 @@
 
 
 class AutoGenText2Sql:
-    def __init__(self, target_engine: str, engine_specific_rules: str, **kwargs: dict):
+    def __init__(self, engine_specific_rules: str, **kwargs: dict):
         self.use_query_cache = False
         self.pre_run_query_cache = False
 
-        self.target_engine = target_engine
+        self.target_engine = os.environ["Text2Sql__DatabaseEngine"].upper()
         self.engine_specific_rules = engine_specific_rules
 
         self.kwargs = kwargs

text_2_sql/text_2_sql_core/src/text_2_sql_core/data_dictionary/databricks_data_dictionary_creator.py

@@ -1,6 +1,10 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
-from data_dictionary_creator import DataDictionaryCreator, EntityItem, ColumnItem
+from text_2_sql_core.data_dictionary.data_dictionary_creator import (
+    DataDictionaryCreator,
+    EntityItem,
+    ColumnItem,
+)
 import asyncio
 import os
 from text_2_sql_core.utils.database import DatabaseEngine

text_2_sql/text_2_sql_core/src/text_2_sql_core/data_dictionary/snowflake_data_dictionary_creator.py

@@ -1,6 +1,9 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
-from data_dictionary_creator import DataDictionaryCreator, EntityItem
+from text_2_sql_core.data_dictionary.data_dictionary_creator import (
+    DataDictionaryCreator,
+    EntityItem,
+)
 import asyncio
 import os
 

text_2_sql/text_2_sql_core/src/text_2_sql_core/data_dictionary/tsql_data_dictionary_creator.py

@@ -1,6 +1,9 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
-from data_dictionary_creator import DataDictionaryCreator, EntityItem
+from text_2_sql_core.data_dictionary.data_dictionary_creator import (
+    DataDictionaryCreator,
+    EntityItem,
+)
 import asyncio
 import os
 from text_2_sql_core.utils.database import DatabaseEngine