mirror of
https://github.com/hanzoai/mcp.git
synced 2026-11-02 09:31:12 +00:00
zeekay
6db5cf2d7e
- Add docker-publish.yml workflow for ghcr.io - Add production-ready Dockerfile (if missing) - Enable automated builds on push to main - Support for semantic versioning tags
86 lines
2.4 KiB
Python
86 lines
2.4 KiB
Python
"""Jupyter notebook tools package for Hanzo AI.
|
|
|
|
This package provides tools for working with Jupyter notebooks (.ipynb files),
|
|
including reading and editing notebook cells.
|
|
"""
|
|
|
|
from mcp.server import FastMCP
|
|
|
|
from hanzo_mcp.tools.common.base import BaseTool, ToolRegistry
|
|
from hanzo_mcp.tools.common.permissions import PermissionManager
|
|
from hanzo_mcp.tools.jupyter.jupyter import JupyterTool
|
|
|
|
# Export all tool classes
|
|
__all__ = [
|
|
"JupyterTool",
|
|
"get_jupyter_tools",
|
|
"register_jupyter_tools",
|
|
]
|
|
|
|
|
|
def get_read_only_jupyter_tools(
|
|
permission_manager: PermissionManager,
|
|
) -> list[BaseTool]:
|
|
"""Create instances of read only Jupyter notebook tools.
|
|
|
|
Args:
|
|
permission_manager: Permission manager for access control
|
|
|
|
Returns:
|
|
List of Jupyter notebook tool instances
|
|
"""
|
|
return [] # Tool handles both read and write
|
|
|
|
|
|
def get_jupyter_tools(permission_manager: PermissionManager) -> list[BaseTool]:
|
|
"""Create instances of all Jupyter notebook tools.
|
|
|
|
Args:
|
|
permission_manager: Permission manager for access control
|
|
|
|
Returns:
|
|
List of Jupyter notebook tool instances
|
|
"""
|
|
return [
|
|
JupyterTool(permission_manager),
|
|
]
|
|
|
|
|
|
def register_jupyter_tools(
|
|
mcp_server: FastMCP,
|
|
permission_manager: PermissionManager,
|
|
enabled_tools: dict[str, bool] | None = None,
|
|
) -> list[BaseTool]:
|
|
"""Register Jupyter notebook tools with the MCP server.
|
|
|
|
Args:
|
|
mcp_server: The FastMCP server instance
|
|
permission_manager: Permission manager for access control
|
|
enabled_tools: Dictionary of individual tool enable states (default: None)
|
|
|
|
Returns:
|
|
List of registered tools
|
|
"""
|
|
# Define tool mapping
|
|
tool_classes = {
|
|
"jupyter": JupyterTool,
|
|
# Legacy names for backward compatibility
|
|
"notebook_read": JupyterTool,
|
|
"notebook_edit": JupyterTool,
|
|
}
|
|
|
|
tools = []
|
|
|
|
if enabled_tools:
|
|
# Use individual tool configuration
|
|
for tool_name, enabled in enabled_tools.items():
|
|
if enabled and tool_name in tool_classes:
|
|
tool_class = tool_classes[tool_name]
|
|
tools.append(tool_class(permission_manager))
|
|
else:
|
|
# Use all tools (backward compatibility)
|
|
tools = get_jupyter_tools(permission_manager)
|
|
|
|
ToolRegistry.register_tools(mcp_server, tools)
|
|
return tools
|