sqlsaber 0.3.0__py3-none-any.whl → 0.4.0__py3-none-any.whl

sqlsaber/agents/mcp.py

@@ -0,0 +1,21 @@
+"""Generic SQL agent implementation for MCP tools."""
+
+from typing import AsyncIterator
+from sqlsaber.agents.base import BaseSQLAgent
+from sqlsaber.database.connection import BaseDatabaseConnection
+from sqlsaber.models.events import StreamEvent
+
+
+class MCPSQLAgent(BaseSQLAgent):
+    """MCP SQL Agent for MCP tool operations without LLM-specific logic."""
+
+    def __init__(self, db_connection: BaseDatabaseConnection):
+        super().__init__(db_connection)
+
+    async def query_stream(
+        self, user_query: str, use_history: bool = True
+    ) -> AsyncIterator[StreamEvent]:
+        """Not implemented for generic agent as it's only used for tool operations."""
+        raise NotImplementedError(
+            "MCPSQLAgent does not support query streaming. Use specific agent implementations for conversation."
+        )

sqlsaber/database/schema.py

@@ -201,8 +201,8 @@ class PostgreSQLSchemaIntrospector(BaseSchemaIntrospector):
                     t.table_type,
                     COALESCE(ts.approximate_row_count, 0) as row_count
                 FROM information_schema.tables t
-                LEFT JOIN table_stats ts 
-                    ON t.table_schema = ts.schemaname 
+                LEFT JOIN table_stats ts
+                    ON t.table_schema = ts.schemaname
                     AND t.table_name = ts.tablename
                 WHERE t.table_schema NOT IN ('pg_catalog', 'information_schema')
                 ORDER BY t.table_schema, t.table_name;

sqlsaber/mcp/__init__.py

@@ -0,0 +1,5 @@
+"""MCP (Model Context Protocol) server implementation for SQLSaber."""
+
+from .mcp import mcp
+
+__all__ = ["mcp"]

sqlsaber/mcp/mcp.py

@@ -0,0 +1,138 @@
+"""FastMCP server implementation for SQLSaber."""
+
+import json
+from typing import Optional
+
+from fastmcp import FastMCP
+
+from sqlsaber.agents.mcp import MCPSQLAgent
+from sqlsaber.config.database import DatabaseConfigManager
+from sqlsaber.database.connection import DatabaseConnection
+
+INSTRUCTIONS = """
+This server provides helpful resources and tools that will help you address users queries on their database.
+
+- Get all databases using `get_databases()`
+- Call `list_tables()` to get a list of all tables in the database with row counts. Use this first to discover available tables.
+- Call `introspect_schema()` to introspect database schema to understand table structures.
+- Call `execute_sql()` to execute SQL queries against the database and retrieve results.
+
+Guidelines:
+- Use list_tables first, then introspect_schema for specific tables only
+- Use table patterns like 'sample%' or '%experiment%' to filter related tables
+- Use proper JOIN syntax and avoid cartesian products
+- Include appropriate WHERE clauses to limit results
+- Handle errors gracefully and suggest fixes
+"""
+
+# Create the FastMCP server instance
+mcp = FastMCP(name="SQL Assistant", instructions=INSTRUCTIONS)
+
+# Initialize the database config manager
+config_manager = DatabaseConfigManager()
+
+
+async def _create_agent_for_database(database_name: str) -> Optional[MCPSQLAgent]:
+    """Create a MCPSQLAgent for the specified database."""
+    try:
+        # Look up configured database connection
+        db_config = config_manager.get_database(database_name)
+        if not db_config:
+            return None
+        connection_string = db_config.to_connection_string()
+
+        # Create database connection
+        db_conn = DatabaseConnection(connection_string)
+
+        # Create and return the agent
+        agent = MCPSQLAgent(db_conn)
+        return agent
+
+    except Exception:
+        return None
+
+
+@mcp.tool
+def get_databases() -> dict:
+    """List all configured databases with their types."""
+    databases = []
+    for db_config in config_manager.list_databases():
+        databases.append(
+            {
+                "name": db_config.name,
+                "type": db_config.type,
+                "database": db_config.database,
+                "host": db_config.host,
+                "port": db_config.port,
+                "is_default": db_config.name == config_manager.get_default_name(),
+            }
+        )
+
+    return {"databases": databases, "count": len(databases)}
+
+
+@mcp.tool
+async def list_tables(database: str) -> str:
+    """
+    Get a list of all tables in the database with row counts. Use this first to discover available tables.
+    """
+    try:
+        agent = await _create_agent_for_database(database)
+        if not agent:
+            return json.dumps(
+                {"error": f"Database '{database}' not found or could not connect"}
+            )
+
+        result = await agent.list_tables()
+        await agent.db.close()
+        return result
+
+    except Exception as e:
+        return json.dumps({"error": f"Error listing tables: {str(e)}"})
+
+
+@mcp.tool
+async def introspect_schema(database: str, table_pattern: Optional[str] = None) -> str:
+    """
+    Introspect database schema to understand table structures. Use optional pattern to filter tables (e.g., 'public.users', 'user%', '%order%').
+    """
+    try:
+        agent = await _create_agent_for_database(database)
+        if not agent:
+            return json.dumps(
+                {"error": f"Database '{database}' not found or could not connect"}
+            )
+
+        result = await agent.introspect_schema(table_pattern)
+        await agent.db.close()
+        return result
+
+    except Exception as e:
+        return json.dumps({"error": f"Error introspecting schema: {str(e)}"})
+
+
+@mcp.tool
+async def execute_sql(database: str, query: str, limit: Optional[int] = 100) -> str:
+    """Execute a SQL query against the specified database."""
+    try:
+        agent = await _create_agent_for_database(database)
+        if not agent:
+            return json.dumps(
+                {"error": f"Database '{database}' not found or could not connect"}
+            )
+
+        result = await agent.execute_sql(query, limit)
+        await agent.db.close()
+        return result
+
+    except Exception as e:
+        return json.dumps({"error": f"Error executing SQL: {str(e)}"})
+
+
+def main():
+    """Entry point for the MCP server console script."""
+    mcp.run()
+
+
+if __name__ == "__main__":
+    main()

{sqlsaber-0.3.0.dist-info → sqlsaber-0.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlsaber
-Version: 0.3.0
+Version: 0.4.0
 Summary: SQLSaber - Agentic SQL assistant like Claude Code
 License-File: LICENSE
 Requires-Python: >=3.12
@@ -8,6 +8,7 @@ Requires-Dist: aiomysql>=0.2.0
 Requires-Dist: aiosqlite>=0.21.0
 Requires-Dist: anthropic>=0.54.0
 Requires-Dist: asyncpg>=0.30.0
+Requires-Dist: fastmcp>=2.9.0
 Requires-Dist: httpx>=0.28.1
 Requires-Dist: keyring>=25.6.0
 Requires-Dist: pandas>=2.0.0
@@ -34,6 +35,7 @@ Ask your questions in natural language and it will gather the right context and
 - 💬 Interactive REPL mode
 - 🎨 Beautiful formatted output with syntax highlighting
 - 🗄️ Support for PostgreSQL, SQLite, and MySQL
+- 🔌 MCP (Model Context Protocol) server support
 
 ## Installation
 
@@ -140,6 +142,43 @@ saber query "show me the distribution of customer ages"
 saber query "which products had the highest sales growth last quarter?"
 ```
 
+## MCP Server Integration
+
+SQLSaber includes an MCP (Model Context Protocol) server that allows AI agents like Claude Code to directly leverage tools available in SQLSaber.
+
+### Starting the MCP Server
+
+Run the MCP server using uvx:
+
+```bash
+uvx saber-mcp
+```
+
+### Configuring MCP Clients
+
+#### Claude Code
+
+Add SQLSaber as an MCP server in Claude Code:
+
+```bash
+claude mcp add -- uvx saber-mcp
+```
+
+#### Other MCP Clients
+
+For other MCP clients, configure them to run the command: `uvx saber-mcp`
+
+### Available MCP Tools
+
+Once connected, the MCP client will have access to these tools:
+
+- `get_databases()` - Lists all configured databases
+- `list_tables(database)` - Get all tables in a database with row counts
+- `introspect_schema(database, table_pattern?)` - Get detailed schema information
+- `execute_sql(database, query, limit?)` - Execute SQL queries (read-only)
+
+The MCP server uses your existing SQLSaber database configurations, so make sure to set up your databases using `saber db add` first.
+
 ## How It Works
 
 SQLSaber uses an intelligent three-step process optimized for minimal token usage:

{sqlsaber-0.3.0.dist-info → sqlsaber-0.4.0.dist-info}/RECORD

@@ -3,6 +3,7 @@ sqlsaber/__main__.py,sha256=RIHxWeWh2QvLfah-2OkhI5IJxojWfy4fXpMnVEJYvxw,78
 sqlsaber/agents/__init__.py,sha256=LWeSeEUE4BhkyAYFF3TE-fx8TtLud3oyEtyB8ojFJgo,167
 sqlsaber/agents/anthropic.py,sha256=xAjKeQSnaut-P5VBeBISbQeqdP41epDjX6MJb2ZUXWg,14060
 sqlsaber/agents/base.py,sha256=IuVyCaA7VsA92odfQS2_lYNzwIZwPxK55mL_xRewgwQ,6943
+sqlsaber/agents/mcp.py,sha256=FKtXgDrPZ2-xqUYCw2baI5JzrWekXaC5fjkYW1_Mg50,827
 sqlsaber/agents/streaming.py,sha256=_EO390-FHUrL1fRCNfibtE9QuJz3LGQygbwG3CB2ViY,533
 sqlsaber/cli/__init__.py,sha256=qVSLVJLLJYzoC6aj6y9MFrzZvAwc4_OgxU9DlkQnZ4M,86
 sqlsaber/cli/commands.py,sha256=h418lgh_Xp7XEQ1xvjcDyplC2JON0-y98QMaDm6o29k,4919
@@ -18,15 +19,17 @@ sqlsaber/config/database.py,sha256=vKFOxPjVakjQhj1uoLcfzhS9ZFr6Z2F5b4MmYALQZoA,1
 sqlsaber/config/settings.py,sha256=zjQ7nS3ybcCb88Ea0tmwJox5-q0ettChZw89ZqRVpX8,3975
 sqlsaber/database/__init__.py,sha256=a_gtKRJnZVO8-fEZI7g3Z8YnGa6Nio-5Y50PgVp07ss,176
 sqlsaber/database/connection.py,sha256=s8GSFZebB8be8sVUr-N0x88-20YfkfljJFRyfoB1gH0,15154
-sqlsaber/database/schema.py,sha256=S3uPSXcrU3swoOPPjbXSxo1_fZ8vV3ELuPoV2GRy1ZI,27953
+sqlsaber/database/schema.py,sha256=9QoH-gADzWlepq-tGz3nPU3miSUU0koWmpDaoWvz8Q0,27951
+sqlsaber/mcp/__init__.py,sha256=COdWq7wauPBp5Ew8tfZItFzbcLDSEkHBJSMhxzy8C9c,112
+sqlsaber/mcp/mcp.py,sha256=ACm1P1TnicjOptQgeLNhXg5xgZf4MYq2kqdfVdj6wh0,4477
 sqlsaber/memory/__init__.py,sha256=GiWkU6f6YYVV0EvvXDmFWe_CxarmDCql05t70MkTEWs,63
 sqlsaber/memory/manager.py,sha256=ML2NEO5Z4Aw36sEI9eOvWVnjl-qT2VOTojViJAj7Seo,2777
 sqlsaber/memory/storage.py,sha256=DvZBsSPaAfk_DqrNEn86uMD-TQsWUI6rQLfNw6PSCB8,5788
 sqlsaber/models/__init__.py,sha256=RJ7p3WtuSwwpFQ1Iw4_DHV2zzCtHqIzsjJzxv8kUjUE,287
 sqlsaber/models/events.py,sha256=55m41tDwMsFxnKKA5_VLJz8iV-V4Sq3LDfta4VoutJI,737
 sqlsaber/models/types.py,sha256=3U_30n91EB3IglBTHipwiW4MqmmaA2qfshfraMZyPps,896
-sqlsaber-0.3.0.dist-info/METADATA,sha256=YKw06cBqX0bNZK98kJfYAES_c8VPXOhvOtIKONt1xFk,3982
-sqlsaber-0.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-sqlsaber-0.3.0.dist-info/entry_points.txt,sha256=POwcsEskUp7xQQWabrAi6Eawz4qc5eBlB3KzAiBq-Y0,124
-sqlsaber-0.3.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sqlsaber-0.3.0.dist-info/RECORD,,
+sqlsaber-0.4.0.dist-info/METADATA,sha256=CL1mNjOLrc6VDJqE2dSrCXO5OJz9gTMxYNoYq6jtzYE,5071
+sqlsaber-0.4.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+sqlsaber-0.4.0.dist-info/entry_points.txt,sha256=jmFo96Ylm0zIKXJBwhv_P5wQ7SXP9qdaBcnTp8iCEe8,195
+sqlsaber-0.4.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+sqlsaber-0.4.0.dist-info/RECORD,,

{sqlsaber-0.3.0.dist-info → sqlsaber-0.4.0.dist-info}/entry_points.txt

@@ -1,4 +1,6 @@
 [console_scripts]
 saber = sqlsaber.cli.commands:main
+saber-mcp = sqlsaber.mcp.mcp:main
 sql = sqlsaber.cli.commands:main
 sqlsaber = sqlsaber.cli.commands:main
+sqlsaber-mcp = sqlsaber.mcp.mcp:main