mcp-server-motherduck 0.3.1__py3-none-any.whl → 0.3.2__py3-none-any.whl

mcp_server_motherduck/__init__.py

@@ -10,9 +10,16 @@ def main():
         "--db-path",
         help="Path to local DuckDB database file",
     )
+    # This is experimental and will change in the future
+    parser.add_argument(
+        "--result-format",
+        help="Format of the output",
+        default="markdown",
+        choices=["markdown", "duckbox", "text"],
+    )
 
     args = parser.parse_args()
-    asyncio.run(server.main(db_path=args.db_path))
+    asyncio.run(server.main(db_path=args.db_path, result_format=args.result_format))
 
 
 # Optionally expose other important items at package level

mcp_server_motherduck/server.py

@@ -3,22 +3,31 @@ import logging
 import duckdb
 from pydantic import AnyUrl
 from typing import Literal
+import io
+from contextlib import redirect_stdout
 import mcp.server.stdio
 import mcp.types as types
 from mcp.server import NotificationOptions, Server
 from mcp.server.models import InitializationOptions
 from .prompt import PROMPT_TEMPLATE
 
-SERVER_VERSION = "0.3.1"
+
+SERVER_VERSION = "0.3.2"
 
 logger = logging.getLogger("mcp_server_motherduck")
 
 
 class DatabaseClient:
-    def __init__(self, db_path: str = None):
+    def __init__(
+        self,
+        db_path: str = 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()
 
+        self.result_format = result_format
+
     def _initialize_connection(self) -> duckdb.DuckDBPyConnection:
         """Initialize connection to the MotherDuck or DuckDB database"""
 
@@ -59,7 +68,19 @@ class DatabaseClient:
 
     def query(self, query: str) -> str:
         try:
-            return str(self.conn.execute(query).fetchall())
+            if self.result_format == "markdown":
+                # Markdown version of the output
+                return self.conn.execute(query).fetchdf().to_markdown()
+            elif self.result_format == "duckbox":
+                # Duckbox version of the output
+                buffer = io.StringIO()
+                with redirect_stdout(buffer):
+                    self.conn.sql(query).show(max_rows=100, max_col_width=20)
+                return buffer.getvalue()
+            else:
+                # Text version of the output
+                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}")

{mcp_server_motherduck-0.3.1.dist-info → mcp_server_motherduck-0.3.2.dist-info}/METADATA

@@ -1,12 +1,14 @@
 Metadata-Version: 2.4
 Name: mcp-server-motherduck
-Version: 0.3.1
+Version: 0.3.2
 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

mcp_server_motherduck-0.3.2.dist-info/RECORD

@@ -0,0 +1,9 @@
+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=EA8l90R52aXKhU4FuH9x3mnC81srZqg8omO19tYASiQ,7882
+mcp_server_motherduck-0.3.2.dist-info/METADATA,sha256=238sc-UWVhRXcdRDFA4KjSkp7x934JZTekujgYkzCoo,2015
+mcp_server_motherduck-0.3.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mcp_server_motherduck-0.3.2.dist-info/entry_points.txt,sha256=dRTgcvWJn40bz0PVuKPylK6w92cFN32lwunZOgo5j4s,69
+mcp_server_motherduck-0.3.2.dist-info/licenses/LICENSE,sha256=Tj68w9jCiceFKTvZ3jET-008NjhozcQMXpm-fyL9WUI,1067
+mcp_server_motherduck-0.3.2.dist-info/RECORD,,

mcp_server_motherduck-0.3.1.dist-info/RECORD

@@ -1,9 +0,0 @@
-mcp_server_motherduck/__init__.py,sha256=355quIOEeF3hj4vBH-esu6yDKcNHUIrkrZZQLZ6gWLU,460
-mcp_server_motherduck/__main__.py,sha256=zx-Zgf5qrL280vm9L9vM64kBGfdvneJhZcU8pWSYLOw,410
-mcp_server_motherduck/prompt.py,sha256=P7BrmhVXwDkPeSHQ3f25WMP6lpBpN2BxDzYPOQ3fxX8,56699
-mcp_server_motherduck/server.py,sha256=3Ki4T3jiv6viM4-DHKEF5cpweIcVvD2O01lGmuBwOt8,7149
-mcp_server_motherduck-0.3.1.dist-info/METADATA,sha256=3dwytXskQyyNy-wWs4oGVZsXyOO2TtvXWjTK22rlGIw,1955
-mcp_server_motherduck-0.3.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mcp_server_motherduck-0.3.1.dist-info/entry_points.txt,sha256=dRTgcvWJn40bz0PVuKPylK6w92cFN32lwunZOgo5j4s,69
-mcp_server_motherduck-0.3.1.dist-info/licenses/LICENSE,sha256=Tj68w9jCiceFKTvZ3jET-008NjhozcQMXpm-fyL9WUI,1067
-mcp_server_motherduck-0.3.1.dist-info/RECORD,,