Merge pull request #177 from Intugle/fix/schema-fix

Fix/schema fix

Author: shiva-intugle

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "intugle"
-version = "1.2.1"
+version = "1.2.3"
 authors = [
     { name="Intugle", email="hello@intugle.ai" },
 ]

src/intugle/adapters/types/databricks/databricks.py

@@ -115,7 +115,7 @@ def connect(self):
                     print("Found active Spark session. Using it for execution.")
                     params = DatabricksNotebookConfig.model_validate(connection_parameters_dict)
                     self.catalog = params.catalog
-                    self._schema = params.schema
+                    self._schema = params.schema_
                     return
             except (AttributeError, TypeError):
                 self.spark = None
@@ -129,7 +129,7 @@ def connect(self):
             print("No active Spark session found. Creating a new SQL connector connection.")
             params = DatabricksSQLConnectorConfig.model_validate(connection_parameters_dict)
             self.catalog = params.catalog
-            self._schema = params.schema
+            self._schema = params.schema_
             self.connection = sql.connect(
                 server_hostname=params.host, http_path=params.http_path, access_token=params.token
             )

src/intugle/adapters/types/duckdb/models.py

@@ -1,6 +1,8 @@
+from typing import Literal
+
 from intugle.common.schema import SchemaBase
 
 
 class DuckdbConfig(SchemaBase): 
     path: str
-    type: str
+    type: Literal["csv", "parquet", "excel", "table"]

src/intugle/adapters/types/mysql/mysql.py

@@ -93,7 +93,7 @@ def connect(self):
             database=params.database,
         )
 
-    def _get_connection(self) -> pymysql.connections.Connection:
+    def _get_connection(self):
         """Ensures the connection is active and returns it."""
         if self.connection is None:
             self.connect()

src/intugle/adapters/types/postgres/postgres.py

@@ -137,7 +137,7 @@ async def _connect_async(self):
 
         params = PostgresConnectionConfig.model_validate(connection_parameters_dict)
         self._database = params.database
-        self._schema = params.schema
+        self._schema = params.schema_
         self.connection = await asyncpg.connect(
             user=params.user,
             password=params.password,

src/intugle/adapters/types/snowflake/snowflake.py

@@ -95,9 +95,9 @@ def connect(self):
                 )
 
             connection_parameters = SnowflakeConnectionConfig.model_validate(connection_parameters_dict)
-            self.session = Session.builder.configs(connection_parameters.model_dump()).create()
+            self.session = Session.builder.configs(connection_parameters.model_dump(by_alias=True)).create()
             self._database = connection_parameters.database
-            self._schema = connection_parameters.schema
+            self._schema = connection_parameters.schema_
 
     @staticmethod
     def check_data(data: Any) -> SnowflakeConfig:

src/intugle/adapters/types/sqlserver/sqlserver.py

@@ -98,7 +98,7 @@ def connect(self):
 
         params = SQLServerConnectionConfig.model_validate(connection_parameters_dict)
         self._database = params.database
-        self._schema = params.schema
+        self._schema = params.schema_
 
         conn_str = (
             f"SERVER={params.host},{params.port};"

src/intugle/link_predictor/__init__.py

@@ -0,0 +1,2 @@
+
+