mcp-server-motherduck 0.3.3__py3-none-any.whl → 0.3.4__py3-none-any.whl

mcp_server_motherduck/__init__.py

@@ -1,10 +1,15 @@
 from . import server
 import asyncio
 import argparse
+import logging
+
+logger = logging.getLogger("mcp_server_motherduck")
+logging.basicConfig(level=logging.INFO, format="[%(name)s] %(levelname)s - %(message)s")
 
 
 def main():
     """Main entry point for the package."""
+
     parser = argparse.ArgumentParser(description="MotherDuck MCP Server")
     parser.add_argument(
         "--db-path",
@@ -19,6 +24,10 @@ def main():
     )
 
     args = parser.parse_args()
+    logger.info("🦆 MotherDuck MCP Server v" + server.SERVER_VERSION)
+    logger.info("Ready to execute SQL queries via DuckDB/MotherDuck")
+    logger.info("Waiting for client connection...\n")
+
     asyncio.run(server.main(db_path=args.db_path, result_format=args.result_format))
 
 

mcp_server_motherduck/server.py

@@ -12,7 +12,7 @@ from mcp.server.models import InitializationOptions
 from .prompt import PROMPT_TEMPLATE
 
 
-SERVER_VERSION = "0.3.3"
+SERVER_VERSION = "0.3.4"
 
 logger = logging.getLogger("mcp_server_motherduck")
 
@@ -20,26 +20,31 @@ logger = logging.getLogger("mcp_server_motherduck")
 class DatabaseClient:
     def __init__(
         self,
-        db_path: str = None,
+        db_path: str | None = None,
         result_format: Literal["markdown", "duckbox", "text"] = "markdown",
     ):
         self.db_path, self.db_type = self._resolve_db_path_type(db_path)
-        self.conn = self._initialize_connection()
+        logger.info(f"Database client initialized in `{self.db_type}` mode")
 
+        self.conn = self._initialize_connection()
         self.result_format = result_format
 
     def _initialize_connection(self) -> duckdb.DuckDBPyConnection:
         """Initialize connection to the MotherDuck or DuckDB database"""
 
-        logger.info(f"Connecting to {self.db_type} database: `{self.db_path}`")
+        logger.info(f"🔌 Connecting to {self.db_type} database: {self.db_path}")
 
-        return duckdb.connect(
+        conn = duckdb.connect(
             self.db_path,
             config={"custom_user_agent": f"mcp-server-motherduck/{SERVER_VERSION}"},
         )
 
+        logger.info(f"✅ Successfully connected to {self.db_type} database")
+
+        return conn
+
     def _resolve_db_path_type(
-        self, db_path: str = None
+        self, db_path: str | None = None
     ) -> tuple[str, Literal["duckdb", "motherduck"]]:
         """Resolve and validate the database path"""
         # Use MotherDuck if token is available and no path specified
@@ -70,7 +75,12 @@ class DatabaseClient:
         try:
             if self.result_format == "markdown":
                 # Markdown version of the output
-                return self.conn.execute(query).fetchdf().to_markdown()
+                logger.info(
+                    f"🔍 Executing query: {query[:60]}{'...' if len(query) > 60 else ''}"
+                )
+                result = self.conn.execute(query).fetchdf().to_markdown()
+                logger.info("✅ Query executed successfully")
+                return result
             elif self.result_format == "duckbox":
                 # Duckbox version of the output
                 buffer = io.StringIO()
@@ -82,10 +92,9 @@ class DatabaseClient:
                 return str(self.conn.execute(query).fetchall())
 
         except Exception as e:
-            logger.error(f"Database error executing query: {e}")
-            raise ValueError(f"Error executing query: {e}")
+            raise ValueError(f"❌ Error executing query: {e}")
 
-    def mcp_config(self) -> str:
+    def mcp_config(self) -> dict[str, str]:
         """Used for debugging purposes to show the current MCP config"""
         return {
             "current_working_directory": os.getcwd(),
@@ -94,7 +103,9 @@ class DatabaseClient:
         }
 
 
-async def main(db_path: str, result_format: Literal["markdown", "duckbox", "text"] = "markdown"):
+async def main(
+    db_path: str, result_format: Literal["markdown", "duckbox", "text"] = "markdown"
+):
     logger.info(f"Starting MotherDuck MCP Server with DB path: {db_path}")
     server = Server("mcp-server-motherduck")
     db_client = DatabaseClient(db_path=db_path, result_format=result_format)
@@ -194,9 +205,15 @@ async def main(db_path: str, result_format: Literal["markdown", "duckbox", "text
         logger.info(f"Calling tool: {name}::{arguments}")
         try:
             if name == "query":
+                if arguments is None:
+                    return [
+                        types.TextContent(type="text", text="Error: No query provided")
+                    ]
                 tool_response = db_client.query(arguments["query"])
                 return [types.TextContent(type="text", text=str(tool_response))]
 
+            return [types.TextContent(type="text", text=f"Unsupported tool: {name}")]
+
         except Exception as e:
             logger.error(f"Error executing tool {name}: {e}")
             raise ValueError(f"Error executing tool {name}: {str(e)}")
@@ -215,3 +232,7 @@ async def main(db_path: str, result_format: Literal["markdown", "duckbox", "text
                 ),
             ),
         )
+
+        # This will only be reached when the server is shutting down
+        logger.info("\n🦆 MotherDuck MCP Server shutting down...")
+        logger.info(f"Database connection to {db_client.db_path} closed.")

mcp_server_motherduck-0.3.4.dist-info/METADATA

@@ -0,0 +1,164 @@
+Metadata-Version: 2.4
+Name: mcp-server-motherduck
+Version: 0.3.4
+Summary: A MCP server for MotherDuck and local DuckDB
+Author-email: tdoehmen <till@motherduck.com>
+License-File: LICENSE
+Requires-Python: >=3.10
+Requires-Dist: duckdb>=1.2.1
+Requires-Dist: mcp>=1.3.0
+Requires-Dist: pandas>=2.0.0
+Requires-Dist: pydantic-settings>=2.8.1
+Requires-Dist: tabulate>=0.9.0
+Description-Content-Type: text/markdown
+
+# MotherDuck MCP Server
+
+An MCP server implementation that integrates MotherDuck and local DuckDB, providing SQL analytics capabilities to Claude.
+
+## Features
+
+- **Hybrid execution**: query data from both cloud-based MotherDuck and local DuckDB
+- **Cloud storage integration**: access data stored in Amazon S3 or other cloud storage thanks to MotherDuck's integrations
+- **Data sharing**: create and share databases
+- **SQL analytics**: use DuckDB's SQL dialect to query any size of data directly from Claude
+- **Serverless architecture**: run analytics without needing to configure instances or clusters
+
+## Components
+
+### Prompts
+
+The server provides one prompt:
+
+- `duckdb-motherduck-initial-prompt`: A prompt to initialize a connection to DuckDB or MotherDuck and start working with it
+
+### Tools
+
+The server offers one tool:
+
+- `query`: Execute a SQL query on the MotherDuck/DuckDB database
+  - **Inputs**:
+    - `query` (string, required): The SQL query to execute
+
+All interactions with both DuckDB and MotherDuck are done through writing SQL queries.
+
+## Getting Started
+
+### Prerequisites
+
+- A MotherDuck account (sign up at [motherduck.com](https://motherduck.com))
+- A MotherDuck access token
+- `uvx` installed, you can install it using `pip install uvx` or `brew install uvx`
+
+If you plan to use MotherDuck MCP with Claude Desktop, you will also need Claude Desktop installed.
+
+### Setting up your MotherDuck token
+
+1. Sign up for a [MotherDuck account](https://app.motherduck.com/?auth_flow=signup)
+2. Generate an access token via the [MotherDuck UI](https://app.motherduck.com/settings/tokens?auth_flow=signup)
+3. Store the token securely for use in the configuration
+
+### Usage with Claude Desktop
+
+1. Install Claude Desktop from [claude.ai/download](https://claude.ai/download) if you haven't already
+
+2. Open the Claude Desktop configuration file:
+
+- To quickly access it or create it the first time, open the Claude Desktop app, select Settings, and click on the "Developer" tab, finally click on the "Edit Config" button.
+- Add the following configuration to your `claude_desktop_config.json`:
+
+```json
+"mcpServers": {
+  "mcp-server-motherduck": {
+    "command": "uvx",
+    "args": [
+      "mcp-server-motherduck"
+    ],
+    "env": {
+      "motherduck_token": "YOUR_MOTHERDUCK_TOKEN_HERE",
+      "HOME": "YOUR_HOME_FOLDER_PATH"
+    }
+  }
+}
+```
+
+**Important Notes**:
+
+- Replace `YOUR_MOTHERDUCK_TOKEN_HERE` with your actual MotherDuck token
+- Replace `YOUR_HOME_FOLDER_PATH` with the path to your home directory (needed by DuckDB for file operations). For example, on macOS, it would be `/Users/your_username`
+- The `HOME` environment variable is required for DuckDB to function properly.
+
+## Example Queries
+
+Once configured, you can ask Claude to run queries like:
+
+- "Create a new database and table in MotherDuck"
+- "Query data from my local CSV file"
+- "Join data from my local DuckDB database with a table in MotherDuck"
+- "Analyze data stored in Amazon S3"
+
+## Testing
+
+The server is designed to be run by tools like Claude Desktop, but you can start it manually for testing purposes. When testing the server manually, you can specify which database to connect to using the `--db-path` parameter:
+
+1. **Default MotherDuck database**:
+
+   - To connect to the default MotherDuck database, you will need to export the `motherduck_token` environment variable.
+
+   ```bash
+   export motherduck_token=<your_motherduck_token>
+   uvx mcp-server-motherduck --db-path md:
+   ```
+
+   - Alternatively, you can pass the token directly:
+
+   ```bash
+   motherduck_token=<your_motherduck_token> uvx mcp-server-motherduck --db-path md:
+   ```
+
+2. **Specific MotherDuck database**:
+
+   ```bash
+   uvx mcp-server-motherduck --db-path md:your_database_name
+   ```
+
+3. **Local DuckDB database**:
+
+   ```bash
+   uvx mcp-server-motherduck --db-path /path/to/your/local.db
+   ```
+
+4. **In-memory database** (default if no path and no token):
+
+   ```bash
+   uvx mcp-server-motherduck
+   ```
+
+If you don't specify a database path but have set the `motherduck_token` environment variable, the server will automatically connect to the default MotherDuck database (`md:`).
+
+## Running in SSE mode
+
+The server could also be run ing SSE mode using `supergateway` by running the following command:
+
+```bash
+motherduck_token=<your_motherduck_token> HOME=<your_home_folder_path> npx -y supergateway --stdio "uvx mcp-server-motherduck"
+```
+
+And you can point your clients such as Claude Desktop, Cursor to this endpoint.
+
+### Environment Variables
+
+The server uses the following environment variables:
+
+- `motherduck_token`: Your MotherDuck authentication token
+- `HOME`: Directory used by DuckDB for file operations
+
+## Troubleshooting
+
+- If you encounter connection issues, verify your MotherDuck token is correct
+- For local file access problems, ensure the `HOME` environment variable is set correctly
+- Check that the `uvx` command is available in your PATH
+
+## License
+
+This MCP server is licensed under the MIT License. This means you are free to use, modify, and distribute the software, subject to the terms and conditions of the MIT License. For more details, please see the LICENSE file in the project repository.

mcp_server_motherduck-0.3.4.dist-info/RECORD

@@ -0,0 +1,8 @@
+mcp_server_motherduck/__init__.py,sha256=fvkO3-wgyUVAdIzy-As4pl3_LYHFMgO9STVTHAyTyKA,1078
+mcp_server_motherduck/prompt.py,sha256=P7BrmhVXwDkPeSHQ3f25WMP6lpBpN2BxDzYPOQ3fxX8,56699
+mcp_server_motherduck/server.py,sha256=N7ySKj43BtMRGE_9HGhWr8_3wGSsjxKD9xIJJv7W4Eg,8841
+mcp_server_motherduck-0.3.4.dist-info/METADATA,sha256=jhdryCCXMVJdKKi0o7E6v9NiJBakx_d9HRECRn21mks,5623
+mcp_server_motherduck-0.3.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mcp_server_motherduck-0.3.4.dist-info/entry_points.txt,sha256=dRTgcvWJn40bz0PVuKPylK6w92cFN32lwunZOgo5j4s,69
+mcp_server_motherduck-0.3.4.dist-info/licenses/LICENSE,sha256=Tj68w9jCiceFKTvZ3jET-008NjhozcQMXpm-fyL9WUI,1067
+mcp_server_motherduck-0.3.4.dist-info/RECORD,,

mcp_server_motherduck/__main__.py

@@ -1,19 +0,0 @@
-from . import server
-import asyncio
-import argparse
-
-
-def main():
-    """Main entry point for the package."""
-    parser = argparse.ArgumentParser(description="MotherDuck MCP Server")
-    parser.add_argument(
-        "--db-path",
-        help="Path to local DuckDB database file",
-    )
-
-    args = parser.parse_args()
-    asyncio.run(server.main(db_path=args.db_path))
-
-
-if __name__ == "__main__":
-    main()

mcp_server_motherduck-0.3.3.dist-info/METADATA

@@ -1,64 +0,0 @@
-Metadata-Version: 2.4
-Name: mcp-server-motherduck
-Version: 0.3.3
-Summary: A MCP server for MotherDuck and local DuckDB
-Author-email: tdoehmen <till@motherduck.com>
-License-File: LICENSE
-Requires-Python: >=3.10
-Requires-Dist: duckdb>=1.1.3
-Requires-Dist: mcp>=1.0.0
-Requires-Dist: pandas>=2.0.0
-Requires-Dist: tabulate>=0.9.0
-Description-Content-Type: text/markdown
-
-# mcp-server-motherduck MCP server
-
-An [MCP server](https://modelcontextprotocol.io/introduction) for MotherDuck and local DuckDB.
-
-## Components
-
-### Prompts
-
-The server provides one prompt:
-
-- duckdb-motherduck-prompt: A prompt to initialize a connection to duckdb or motherduck and start working with it
-
-### Tools
-
-The server offers one tool:
-
-- query: Execute a query on the MotherDuck (DuckDB) database
-  - Takes "query" as required string arguments
-
-This is because all the interactions with both the DuckDB and MotherDuck are done through writing SQL queries.
-
-## Usage with Claude Desktop
-
-Add the snippet below to your Claude Desktop config and make sure to set the HOME var to your home folder (needed by DuckDB).
-
-When using MotherDuck, you also need to set a [MotherDuck token](https://motherduck.com/docs/key-tasks/authenticating-and-connecting-to-motherduck/authenticating-to-motherduck/#storing-the-access-token-as-an-environment-variable) env var.
-
-On MacOS: `~/Library/Application\ Support/Claude/claude_desktop_config.json`
-
-On Windows: `%APPDATA%/Claude/claude_desktop_config.json`
-
-### Servers Configuration
-
-```
-"mcpServers": {
-  "mcp-server-motherduck": {
-    "command": "uvx",
-    "args": [
-      "mcp-server-motherduck"
-    ],
-    "env": {
-      "motherduck_token": "<your-motherduck-token>",
-      "HOME": "<your-home-folder-for-project-files>"
-    }
-  }
-}
-```
-
-## License
-
-This MCP server is licensed under the MIT License. This means you are free to use, modify, and distribute the software, subject to the terms and conditions of the MIT License. For more details, please see the LICENSE file in the project repository.

mcp_server_motherduck-0.3.3.dist-info/RECORD

@@ -1,9 +0,0 @@
-mcp_server_motherduck/__init__.py,sha256=0BPvfyJm7xnVEIZD7JLiuHuR0FFhf_Xa9xt7ieTZHWQ,723
-mcp_server_motherduck/__main__.py,sha256=zx-Zgf5qrL280vm9L9vM64kBGfdvneJhZcU8pWSYLOw,410
-mcp_server_motherduck/prompt.py,sha256=P7BrmhVXwDkPeSHQ3f25WMP6lpBpN2BxDzYPOQ3fxX8,56699
-mcp_server_motherduck/server.py,sha256=ROYQvC2fRkLRyWj_SW8NYC1omlms6lnlhdLLMBn_Aiw,7979
-mcp_server_motherduck-0.3.3.dist-info/METADATA,sha256=CTtfZg5sIqoPmAIPk49lYg6xE-lE1fLOfDsf754AGWU,2015
-mcp_server_motherduck-0.3.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mcp_server_motherduck-0.3.3.dist-info/entry_points.txt,sha256=dRTgcvWJn40bz0PVuKPylK6w92cFN32lwunZOgo5j4s,69
-mcp_server_motherduck-0.3.3.dist-info/licenses/LICENSE,sha256=Tj68w9jCiceFKTvZ3jET-008NjhozcQMXpm-fyL9WUI,1067
-mcp_server_motherduck-0.3.3.dist-info/RECORD,,