holoviz-mcp 0.0.1a0__py3-none-any.whl → 0.0.1a2__py3-none-any.whl

holoviz_mcp/server.py

@@ -0,0 +1,77 @@
+"""HoloViz MCP Server.
+
+This MCP server provides comprehensive tools, resources and prompts for working with the HoloViz ecosystem,
+including Panel and hvPlot following best practices.
+
+The server is composed of multiple sub-servers that provide various functionalities:
+
+- Documentation: Search and access HoloViz documentation as context
+- hvPlot: Tools, resources and prompts for using hvPlot to develop quick, interactive plots in Python
+- Panel: Tools, resources and prompts for using Panel Material UI
+"""
+
+import asyncio
+import logging
+import os
+
+from fastmcp import FastMCP
+
+from holoviz_mcp.config.loader import get_config
+from holoviz_mcp.docs_mcp.server import mcp as docs_mcp
+from holoviz_mcp.hvplot_mcp.server import mcp as hvplot_mcp
+from holoviz_mcp.panel_mcp.server import mcp as panel_mcp
+
+logger = logging.getLogger(__name__)
+
+mcp: FastMCP = FastMCP(
+    name="holoviz",
+    instructions="""
+    [his MCP server provides comprehensive tools, resources and prompts for working with the HoloViz ecosystem following best practices.
+
+    HoloViz provides a set of core Python packages that make visualization easier, more accurate, and more powerful:
+
+    - [Panel](https://panel.holoviz.org): for making apps and dashboards for your plots from any supported plotting library.
+    - [hvPlot](https://hvplot.holoviz.org): to quickly generate interactive plots from your data.
+    - [HoloViews](https://holoviews.org): to help you make all of your data instantly visualizable.
+    - [GeoViews](https://geoviews.org): to extend HoloViews for geographic data.
+    - [Datashader](https://datashader.org): for rendering even the largest datasets.
+    - [Lumen](https://lumen.holoviz.org): to build data-driven dashboards from a simple YAML specification that's well suited to modern AI tools like LLMs.
+    - [Param](https://param.holoviz.org): to create declarative user-configurable objects.
+    - [Colorcet](https://colorcet.holoviz.org): for perceptually uniform colormaps.
+
+    The server is composed of multiple sub-servers that provide various functionalities:
+
+    - Documentation: Search and access HoloViz documentation and reference guides
+    - Panel: Tools, resources and prompts for using Panel and Panel Extension packages
+    - hvPlot: Tools, resources and prompts for using hvPlot to develop quick, interactive plots in Python
+    """,
+)
+
+
+async def setup_composed_server() -> None:
+    """Set up the composed server by importing all sub-servers with prefixes.
+
+    This uses static composition (import_server), which copies components
+    from sub-servers into the main server with appropriate prefixes.
+    """
+    await mcp.import_server(docs_mcp, prefix="docs")
+    await mcp.import_server(hvplot_mcp, prefix="hvplot")
+    await mcp.import_server(panel_mcp, prefix="panel")
+
+
+def main() -> None:
+    """Set up and run the composed MCP server."""
+    pid = f"Process ID: {os.getpid()}"
+    print(pid)  # noqa: T201
+
+    async def setup_and_run() -> None:
+        await setup_composed_server()
+        config = get_config()
+        await mcp.run_async(transport=config.server.transport)
+
+    asyncio.run(setup_and_run())
+
+
+if __name__ == "__main__":
+    # Run the composed MCP server
+    main()

holoviz_mcp/shared/__init__.py

@@ -0,0 +1 @@
+"""Shared features for the holoviz-mcp project."""

holoviz_mcp/shared/extract_tools.py

@@ -0,0 +1,74 @@
+#!/usr/bin/env python3
+"""Extract tools from the MCP server for documentation."""
+
+import asyncio
+import logging
+
+from holoviz_mcp.server import mcp
+from holoviz_mcp.server import setup_composed_server
+
+# Configure logging
+logging.basicConfig(level=logging.INFO, format="%(message)s")
+logger = logging.getLogger(__name__)
+
+
+async def extract_tools():
+    """Extract available tools from the HoloViz MCP server and return as structured data."""
+    await setup_composed_server()
+    tools_dict = await mcp.get_tools()
+
+    # Group tools by category
+    docs_tools = []
+    panel_tools = []
+    utility_tools = []
+
+    for tool_name, tool_info in tools_dict.items():
+        tool_data = {"name": tool_name, "description": getattr(tool_info, "description", "No description available"), "parameters": []}
+
+        # Get input schema
+        input_schema = getattr(tool_info, "inputSchema", None)
+        if input_schema and hasattr(input_schema, "get") and "properties" in input_schema:
+            for param_name, param_info in input_schema["properties"].items():
+                required = param_name in input_schema.get("required", [])
+                param_type = param_info.get("type", "unknown")
+                desc = param_info.get("description", "No description")
+                tool_data["parameters"].append({"name": param_name, "type": param_type, "required": required, "description": desc})
+
+        # Categorize tools
+        if any(x in tool_name for x in ["docs", "get_best_practices", "get_reference_guide", "get_document", "update_docs"]) or (
+            tool_name == "search" and "component" not in str(tool_info)
+        ):
+            docs_tools.append(tool_data)
+        elif any(x in tool_name for x in ["component", "packages"]) or "search" in tool_name:
+            panel_tools.append(tool_data)
+        else:
+            utility_tools.append(tool_data)
+
+    return {"panel_tools": panel_tools, "docs_tools": docs_tools, "utility_tools": utility_tools}
+
+
+async def main():
+    """Extract and print available tools from the HoloViz MCP server."""
+    tools_data = await extract_tools()
+    logger.info("## 🛠️ Available Tools")
+    logger.info("")
+
+    def print_tools(tools_list, category_name):
+        if not tools_list:
+            return
+        logger.info("<details>")
+        logger.info(f"<summary><b>{category_name}</b></summary>")
+        logger.info("")
+        for tool_data in tools_list:
+            logger.info(f"- **{tool_data['name']}**: {tool_data['description']}")
+            logger.info("")
+        logger.info("</details>")
+        logger.info("")
+
+    print_tools(tools_data["panel_tools"], "Panel Components")
+    print_tools(tools_data["docs_tools"], "Documentation")
+    print_tools(tools_data["utility_tools"], "Utilities")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())