refactor(views): api for freeform views (#41)

deepsense-ai · Jun 3, 2024 · 6d6cdcd · 6d6cdcd
1 parent 8f7a166
commit 6d6cdcd
Show file tree

Hide file tree

Showing 12 changed files with 545 additions and 563 deletions.
diff --git a/src/dbally/views/freeform/text2sql/__init__.py b/src/dbally/views/freeform/text2sql/__init__.py
@@ -1,15 +1,10 @@
-from ._autodiscovery import AutoDiscoveryBuilder, AutoDiscoveryBuilderWithLLM, configure_text2sql_auto_discovery
-from ._config import ColumnConfig, TableConfig, Text2SQLConfig, Text2SQLSimilarityType, Text2SQLTableConfig
-from ._view import BaseText2SQLView
+from .config import ColumnConfig, TableConfig
+from .errors import Text2SQLError
+from .view import BaseText2SQLView
 
 __all__ = [
-    "TableConfig",
-    "ColumnConfig",
-    "Text2SQLConfig",
-    "Text2SQLTableConfig",
-    "Text2SQLSimilarityType",
     "BaseText2SQLView",
-    "configure_text2sql_auto_discovery",
-    "AutoDiscoveryBuilder",
-    "AutoDiscoveryBuilderWithLLM",
+    "ColumnConfig",
+    "TableConfig",
+    "Text2SQLError",
 ]
diff --git a/src/dbally/views/freeform/text2sql/_autodiscovery.py b/src/dbally/views/freeform/text2sql/_autodiscovery.py