Add Smithery platform integration (#131)

* Smithery configuration

* --amend

Author: jonpspri

.pre-commit-config.yaml

@@ -78,6 +78,7 @@ repos:
           - simpleeval>=1.0.3
           - fastmcp>=0.1.0
           - pandera>=0.22.0
+          - smithery>=0.4.2
           - .
         args: ['--config-file', 'pyproject.toml']
         pass_filenames: true

pyproject.toml

@@ -67,6 +67,7 @@ dependencies = [
     "fastapi>=0.117.1",
     "uvicorn>=0.36.0",
     "fastmcp>=2.12.4",
+    "smithery>=0.4.2",
 ]
 
 
@@ -255,6 +256,7 @@ module = [
     "fastparquet",
     "pytest",
     "simpleeval",
+    "smithery",
 ]
 ignore_missing_imports = true
 
@@ -357,6 +359,9 @@ site_author = "Santosh Ray"
 repo_url = "https://github.com/jonpspri/databeak"
 repo_name = "databeak"
 
+[tool.smithery]
+server = "databeak.server:create_server"
+
 [dependency-groups]
 dev = [
     "mdformat>=0.7.22",

smithery.yaml

@@ -1,18 +1,2 @@
 ---
-# Smithery configuration file: https://smithery.ai/docs/build/project-config
-
-startCommand:
-  type: stdio
-  commandFunction:
-    # A JS function that produces the CLI command based on the given config
-    # to start the MCP on stdio.
-    |-
-    (config) => ({command: 'databeak', args: ['--transport', 'stdio']})
-  configSchema:
-    # JSON Schema defining the configuration options for the MCP.
-    type: object
-    default: {}
-    description:
-      No configuration necessary to run the DataBeak MCP server. Supply
-      an empty object.
-  exampleConfig: {}
+runtime: python

src/databeak/__init__.py

@@ -3,6 +3,6 @@
 __author__ = "Jonathan Springer"
 
 from ._version import __version__
-from .server import main, mcp
+from .server import main
 
-__all__ = ["__version__", "main", "mcp"]
+__all__ = ["__version__", "main"]

src/databeak/server.py

@@ -8,6 +8,7 @@
 from pathlib import Path
 
 from fastmcp import FastMCP
+from smithery.decorators import smithery
 
 from databeak._version import __version__
 
@@ -51,30 +52,15 @@ def _load_instructions() -> str:
 # Initialize relaxed JSON schema validation before creating server
 initialize_relaxed_validation()
 
-# Initialize FastMCP server
-mcp = FastMCP("DataBeak", instructions=_load_instructions(), version=__version__)
-
 # All tools have been migrated to specialized servers
 # No direct tool registration needed - using server composition pattern
 
-# Mount specialized servers
-mcp.mount(system_server)
-mcp.mount(io_server)
-mcp.mount(row_operations_server)
-mcp.mount(statistics_server)
-mcp.mount(discovery_server)
-mcp.mount(validation_server)
-mcp.mount(transformation_server)
-mcp.mount(column_server)
-mcp.mount(column_text_server)
-
 
 # ============================================================================
 # PROMPTS
 # ============================================================================
 
 
-@mcp.prompt
 def analyze_csv_prompt(session_id: str, analysis_type: str = "summary") -> str:
     """Generate a prompt to analyze CSV data."""
     return f"""Please analyze the CSV data in session {session_id}.
@@ -89,7 +75,6 @@ def analyze_csv_prompt(session_id: str, analysis_type: str = "summary") -> str:
 """
 
 
-@mcp.prompt
 def data_cleaning_prompt(session_id: str) -> str:
     """Generate a prompt for data cleaning suggestions."""
     return f"""Review the data in session {session_id} and suggest cleaning operations.
@@ -108,6 +93,27 @@ def data_cleaning_prompt(session_id: str) -> str:
 # ============================================================================
 
 
+@smithery.server()
+def create_server() -> FastMCP:
+    """Create and return the FastMCP server instance."""
+    # Initialize FastMCP server
+    mcp = FastMCP("DataBeak", instructions=_load_instructions(), version=__version__)
+    # Mount specialized servers
+    mcp.mount(system_server)
+    mcp.mount(io_server)
+    mcp.mount(row_operations_server)
+    mcp.mount(statistics_server)
+    mcp.mount(discovery_server)
+    mcp.mount(validation_server)
+    mcp.mount(transformation_server)
+    mcp.mount(column_server)
+    mcp.mount(column_text_server)
+
+    mcp.prompt()(analyze_csv_prompt)
+    mcp.prompt()(data_cleaning_prompt)
+    return mcp
+
+
 def main() -> None:
     """Start the DataBeak server."""
     parser = ArgumentParser(description="DataBeak")
@@ -152,7 +158,7 @@ def main() -> None:
         run_args["port"] = args.port
 
     # Run the server
-    mcp.run(**run_args)  # type: ignore[arg-type]
+    create_server().run(**run_args)
 
 
 if __name__ == "__main__":

tests/integration/conftest.py

@@ -71,8 +71,12 @@ async def test_something(databeak_client : Client):
             result = await databeak_client.call_tool("get_session_info", {})
             assert result.is_error is False
     """
-    # Import the server instance
-    from databeak.server import mcp
+    # Create the server instance
+    from databeak.server import create_server
+
+    smithery_server = create_server()
+    # Extract the underlying FastMCP instance from the Smithery wrapper
+    mcp = smithery_server._fastmcp
 
     # Create FastMCP Client with direct server connection
     async with Client(mcp) as client: