db-connect-mcp 0.1.0__tar.gz → 0.1.2__tar.gz

{db_connect_mcp-0.1.0 → db_connect_mcp-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: db-connect-mcp
-Version: 0.1.0
+Version: 0.1.2
 Summary: Multi-database MCP server for PostgreSQL, MySQL, and ClickHouse
 Project-URL: Homepage, https://github.com/yugui923/db-connect-mcp
 Project-URL: Repository, https://github.com/yugui923/db-connect-mcp

{db_connect_mcp-0.1.0 → db_connect_mcp-0.1.2}/pyproject.toml

@@ -4,14 +4,12 @@ build-backend = "hatchling.build"
 
 [project]
 name = "db-connect-mcp"
-version = "0.1.0"
+version = "0.1.2"
 description = "Multi-database MCP server for PostgreSQL, MySQL, and ClickHouse"
 readme = "README.md"
 requires-python = ">=3.13"
-license = {text = "MIT"}
-authors = [
-    {name = "Yuri Gui", email = "yugui923@users.noreply.github.com"}
-]
+license = { text = "MIT" }
+authors = [{ name = "Yuri Gui", email = "yugui923@users.noreply.github.com" }]
 keywords = [
     "mcp",
     "database",
@@ -20,7 +18,7 @@ keywords = [
     "clickhouse",
     "claude",
     "analysis",
-    "data-exploration"
+    "data-exploration",
 ]
 classifiers = [
     "Development Status :: 4 - Beta",
@@ -67,8 +65,7 @@ dev = [
 only-include = ["src/db_connect_mcp"]
 
 [tool.hatch.build.targets.wheel]
-only-include = ["src/db_connect_mcp"]
-sources = ["src"]
+packages = ["src/db_connect_mcp"]
 
 [tool.pytest.ini_options]
 asyncio_mode = "auto"

{db_connect_mcp-0.1.0 → db_connect_mcp-0.1.2}/src/db_connect_mcp/adapters/base.py

@@ -1,7 +1,7 @@
 """Base adapter abstract class for database-specific implementations."""
 
 from abc import ABC, abstractmethod
-from typing import Any, Optional
+from typing import Any, Optional, Union, TYPE_CHECKING
 
 from sqlalchemy.ext.asyncio import AsyncConnection
 
@@ -10,6 +10,12 @@ from db_connect_mcp.models.database import SchemaInfo
 from db_connect_mcp.models.statistics import ColumnStats, Distribution
 from db_connect_mcp.models.table import TableInfo
 
+if TYPE_CHECKING:
+    from db_connect_mcp.core.connection import AsyncConnectionWrapper
+
+# Type alias for connection types
+ConnectionType = Union[AsyncConnection, "AsyncConnectionWrapper"]
+
 
 class BaseAdapter(ABC):
     """Base adapter defining database-specific interface."""
@@ -22,7 +28,7 @@ class BaseAdapter(ABC):
 
     @abstractmethod
     async def enrich_schema_info(
-        self, conn: AsyncConnection, schema_info: SchemaInfo
+        self, conn: ConnectionType, schema_info: SchemaInfo
     ) -> SchemaInfo:
         """
         Enrich schema info with database-specific metadata.
@@ -38,7 +44,7 @@ class BaseAdapter(ABC):
 
     @abstractmethod
     async def enrich_table_info(
-        self, conn: AsyncConnection, table_info: TableInfo
+        self, conn: ConnectionType, table_info: TableInfo
     ) -> TableInfo:
         """
         Enrich table info with database-specific metadata.
@@ -55,7 +61,7 @@ class BaseAdapter(ABC):
     @abstractmethod
     async def get_column_statistics(
         self,
-        conn: AsyncConnection,
+        conn: ConnectionType,
         table_name: str,
         column_name: str,
         schema: Optional[str],
@@ -77,7 +83,7 @@ class BaseAdapter(ABC):
     @abstractmethod
     async def get_value_distribution(
         self,
-        conn: AsyncConnection,
+        conn: ConnectionType,
         table_name: str,
         column_name: str,
         schema: Optional[str],