mcp-server-motherduck 0.6.4__py3-none-any.whl → 0.7.1__py3-none-any.whl

mcp_server_motherduck/configs.py

@@ -1,6 +1,6 @@
 from typing import Any
 
-SERVER_VERSION = "0.6.4"
+SERVER_VERSION = "0.7.1"
 
 SERVER_LOCALHOST = "127.0.0.1"
 

mcp_server_motherduck/database.py

@@ -36,6 +36,10 @@ class DatabaseClient:
 
         logger.info(f"🔌 Connecting to {self.db_type} database")
 
+        # S3 databases don't support read-only mode
+        if self.db_type == "s3" and self._read_only:
+            raise ValueError("Read-only mode is not supported for S3 databases")
+
         if self.db_type == "duckdb" and self._read_only:
             # check that we can connect, issue a `select 1` and then close + return None
             try:
@@ -53,6 +57,66 @@ class DatabaseClient:
                 logger.error(f"❌ Read-only check failed: {e}")
                 raise
 
+        # Check if this is an S3 path
+        if self.db_type == "s3":
+            # For S3, we need to create an in-memory connection and attach the S3 database
+            conn = duckdb.connect(':memory:')
+            
+            # Install and load the httpfs extension for S3 support
+            import io
+            from contextlib import redirect_stdout, redirect_stderr
+            
+            null_file = io.StringIO()
+            with redirect_stdout(null_file), redirect_stderr(null_file):
+                try:
+                    conn.execute("INSTALL httpfs;")
+                except:
+                    pass  # Extension might already be installed
+                conn.execute("LOAD httpfs;")
+            
+            # Configure S3 credentials from environment variables using CREATE SECRET
+            aws_access_key = os.environ.get('AWS_ACCESS_KEY_ID')
+            aws_secret_key = os.environ.get('AWS_SECRET_ACCESS_KEY')
+            aws_region = os.environ.get('AWS_DEFAULT_REGION', 'us-east-1')
+            
+            
+            if aws_access_key and aws_secret_key:
+                # Use CREATE SECRET for better credential management
+                conn.execute(f"""
+                    CREATE SECRET IF NOT EXISTS s3_secret (
+                        TYPE S3,
+                        KEY_ID '{aws_access_key}',
+                        SECRET '{aws_secret_key}',
+                        REGION '{aws_region}'
+                    );
+                """)
+            
+            # Attach the S3 database
+            try:
+                # For S3, we always attach as READ_ONLY since S3 storage is typically read-only
+                # Even when not in read_only mode, we attach as READ_ONLY for S3
+                conn.execute(f"ATTACH '{self.db_path}' AS s3db (READ_ONLY);")
+                # Use the attached database
+                conn.execute("USE s3db;")
+                logger.info(f"✅ Successfully connected to {self.db_type} database (attached as read-only)")
+            except Exception as e:
+                logger.error(f"Failed to attach S3 database: {e}")
+                # If the database doesn't exist and we're not in read-only mode, try to create it
+                if "database does not exist" in str(e) and not self._read_only:
+                    logger.info("S3 database doesn't exist, attempting to create it...")
+                    try:
+                        # Create a new database at the S3 location
+                        conn.execute(f"ATTACH '{self.db_path}' AS s3db;")
+                        conn.execute("USE s3db;")
+                        logger.info(f"✅ Created new S3 database at {self.db_path}")
+                    except Exception as create_error:
+                        logger.error(f"Failed to create S3 database: {create_error}")
+                        raise
+                else:
+                    raise
+                
+            return conn
+
         conn = duckdb.connect(
             self.db_path,
             config={"custom_user_agent": f"mcp-server-motherduck/{SERVER_VERSION}"},
@@ -65,8 +129,12 @@ class DatabaseClient:
 
     def _resolve_db_path_type(
         self, db_path: str, motherduck_token: str | None = None, saas_mode: bool = False
-    ) -> tuple[str, Literal["duckdb", "motherduck"]]:
+    ) -> tuple[str, Literal["duckdb", "motherduck", "s3"]]:
         """Resolve and validate the database path"""
+        # Handle S3 paths
+        if db_path.startswith("s3://"):
+            return db_path, "s3"
+        
         # Handle MotherDuck paths
         if db_path.startswith("md:"):
             if motherduck_token:
@@ -102,7 +170,7 @@ class DatabaseClient:
 
     def _execute(self, query: str) -> str:
         if self.conn is None:
-            # open short lived readonly connection, run query, close connection, return result
+            # open short lived readonly connection for local DuckDB, run query, close connection, return result
             conn = duckdb.connect(
                 self.db_path,
                 config={"custom_user_agent": f"mcp-server-motherduck/{SERVER_VERSION}"},

{mcp_server_motherduck-0.6.4.dist-info → mcp_server_motherduck-0.7.1.dist-info}/METADATA

@@ -1,13 +1,13 @@
 Metadata-Version: 2.4
 Name: mcp-server-motherduck
-Version: 0.6.4
+Version: 0.7.1
 Summary: A MCP server for MotherDuck and local DuckDB
 Author-email: tdoehmen <till@motherduck.com>
 License-File: LICENSE
 Requires-Python: >=3.10
 Requires-Dist: anyio>=4.8.0
 Requires-Dist: click>=8.1.8
-Requires-Dist: duckdb==1.3.1
+Requires-Dist: duckdb==1.4.1
 Requires-Dist: mcp>=1.9.4
 Requires-Dist: pytz>=2025.2
 Requires-Dist: starlette>=0.46.1
@@ -19,7 +19,7 @@ Description-Content-Type: text/markdown
 
 An MCP server implementation that interacts with DuckDB and MotherDuck databases, providing SQL analytics capabilities to AI Assistants and IDEs.
 
-[![Install MCP Server](https://cursor.com/deeplink/mcp-install-dark.svg)](cursor://anysphere.cursor-deeplink/mcp/install?name=DuckDB&config=eyJjb21tYW5kIjoidXZ4IG1jcC1zZXJ2ZXItbW90aGVyZHVjayAtLWRiLXBhdGggOm1lbW9yeToiLCJlbnYiOnsibW90aGVyZHVja190b2tlbiI6IiJ9fQ%3D%3D)
+[<img src="https://cursor.com/deeplink/mcp-install-dark.svg" alt="Install in Cursor">](https://cursor.com/en/install-mcp?name=DuckDB&config=eyJjb21tYW5kIjoidXZ4IG1jcC1zZXJ2ZXItbW90aGVyZHVjayAtLWRiLXBhdGggOm1lbW9yeToiLCJlbnYiOnsibW90aGVyZHVja190b2tlbiI6IiJ9fQ%3D%3D)
 
 ## Resources
 - [Close the Loop: Faster Data Pipelines with MCP, DuckDB & AI (Blogpost)](https://motherduck.com/blog/faster-data-pipelines-with-mcp-duckdb-ai/)
@@ -59,7 +59,7 @@ The MCP server supports the following parameters:
 |-----------|------|---------|----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
 | `--transport` | Choice | `stdio` | Transport type. Options: `stdio`, `sse`, `stream`                                                                                                                                                                                                              |
 | `--port` | Integer | `8000` | Port to listen on for sse and stream transport mode                                                                                                                                                                                                            |
-| `--db-path` | String | `md:` | Path to local DuckDB database file or MotherDuck database                                                                                                                                                                                                      |
+| `--db-path` | String | `md:` | Path to local DuckDB database file, MotherDuck database, or S3 URL (e.g., `s3://bucket/path/to/db.duckdb`)                                                                                                                                                     |
 | `--motherduck-token` | String | `None` | Access token to use for MotherDuck database connections (uses `motherduck_token` env var by default)                                                                                                                                                           |
 | `--read-only` | Flag | `False` | Flag for connecting to DuckDB or MotherDuck in read-only mode. For DuckDB it uses short-lived connections to enable concurrent access                                                                                                                          |
 | `--home-dir` | String | `None` | Home directory for DuckDB (uses `HOME` env var by default)                                                                                                                                                                                                     |
@@ -354,6 +354,37 @@ modeling data within duckdb and then an MCP client (Windsurf/Cline/Claude/Cursor
 was used for exploring the database. The short lived connections allow each tool
 to run and then release their connection, allowing the next tool to connect.
 
+## Connect to DuckDB on S3
+
+You can connect to DuckDB databases stored on Amazon S3 by providing an S3 URL as the database path. The server will automatically configure the necessary S3 credentials from your environment variables.
+
+```json
+{
+  "mcpServers": {
+    "mcp-server-motherduck": {
+      "command": "uvx",
+      "args": [
+        "mcp-server-motherduck",
+        "--db-path",
+        "s3://your-bucket/path/to/database.duckdb"
+      ],
+      "env": {
+        "AWS_ACCESS_KEY_ID": "<your_key>",
+        "AWS_SECRET_ACCESS_KEY": "<your_secret>",
+        "AWS_DEFAULT_REGION": "<your_region>"
+      }
+    }
+  }
+}
+```
+
+
+**Note**: For S3 connections:
+- AWS credentials must be provided via environment variables (`AWS_ACCESS_KEY_ID`, `AWS_SECRET_ACCESS_KEY`, and optionally `AWS_DEFAULT_REGION`)
+- The S3 database is attached to an in-memory DuckDB instance
+- The httpfs extension is automatically installed and configured for S3 access
+- Both read and write operations are supported
+
 ## Example Queries
 
 Once configured, you can e.g. ask Claude to run queries like:

mcp_server_motherduck-0.7.1.dist-info/RECORD

@@ -0,0 +1,10 @@
+mcp_server_motherduck/__init__.py,sha256=vogjO-bwHYZFut85eeiVmRepamcWHk8bImri_kBKZMU,6026
+mcp_server_motherduck/configs.py,sha256=uhfR1Tt6jaB5cG4XEHkxcbEJ6dmpBaWNL6a1GXoFCSE,771
+mcp_server_motherduck/database.py,sha256=-n1Z0UXsi5Y3nO1_df1CeHiSEe644gJNkJ7r_pdWytw,7761
+mcp_server_motherduck/prompt.py,sha256=P7BrmhVXwDkPeSHQ3f25WMP6lpBpN2BxDzYPOQ3fxX8,56699
+mcp_server_motherduck/server.py,sha256=U1LM2oQ36gO_pAZuez9HV_u8YDWdER8tQIdDbiXfzx0,5232
+mcp_server_motherduck-0.7.1.dist-info/METADATA,sha256=fBp9w2oJ-C4Qj2v8aAYiHHFHHx59w1Pm0fqUe_9EQAM,17820
+mcp_server_motherduck-0.7.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mcp_server_motherduck-0.7.1.dist-info/entry_points.txt,sha256=dRTgcvWJn40bz0PVuKPylK6w92cFN32lwunZOgo5j4s,69
+mcp_server_motherduck-0.7.1.dist-info/licenses/LICENSE,sha256=Tj68w9jCiceFKTvZ3jET-008NjhozcQMXpm-fyL9WUI,1067
+mcp_server_motherduck-0.7.1.dist-info/RECORD,,

mcp_server_motherduck-0.6.4.dist-info/RECORD

@@ -1,10 +0,0 @@
-mcp_server_motherduck/__init__.py,sha256=vogjO-bwHYZFut85eeiVmRepamcWHk8bImri_kBKZMU,6026
-mcp_server_motherduck/configs.py,sha256=-9cNtMBsNn0OK-0pxRWxMoTy4VbfUdCMKQG_BPMmIu4,771
-mcp_server_motherduck/database.py,sha256=3qkeBIpf08WV6pz9rf52b_MipzaHALurFugMx-XARI8,4513
-mcp_server_motherduck/prompt.py,sha256=P7BrmhVXwDkPeSHQ3f25WMP6lpBpN2BxDzYPOQ3fxX8,56699
-mcp_server_motherduck/server.py,sha256=U1LM2oQ36gO_pAZuez9HV_u8YDWdER8tQIdDbiXfzx0,5232
-mcp_server_motherduck-0.6.4.dist-info/METADATA,sha256=McZsr6b8gFcH4ju85Ck2eC7-Z0cS06V_fOyecihfVIA,16831
-mcp_server_motherduck-0.6.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mcp_server_motherduck-0.6.4.dist-info/entry_points.txt,sha256=dRTgcvWJn40bz0PVuKPylK6w92cFN32lwunZOgo5j4s,69
-mcp_server_motherduck-0.6.4.dist-info/licenses/LICENSE,sha256=Tj68w9jCiceFKTvZ3jET-008NjhozcQMXpm-fyL9WUI,1067
-mcp_server_motherduck-0.6.4.dist-info/RECORD,,