"""Advanced AI tools registration for Enterprise AI capabilities.
Registers AI processing, smart suggestions, autonomous agents, and other
advanced AI-powered tools for the Keyboard Maestro MCP server.
"""
import logging
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from fastmcp import FastMCP
logger = logging.getLogger(__name__)
def register_advanced_ai_tools(mcp: "FastMCP") -> None:
"""Register advanced AI MCP tools for enterprise automation.
Includes:
- km_ai_processing: AI/ML model integration
- km_smart_suggestions: AI-powered automation suggestions
- km_autonomous_agent: Self-managing automation agents
- km_audit_system: Advanced audit logging and compliance
- km_enterprise_sync: Enterprise system integration
- km_cloud_connector: Multi-cloud platform integration
Args:
mcp: FastMCP server instance
"""
try:
# Import tools from server.tools package
from ..server.tools import km_autonomous_agent
# Note: Other tools like km_ai_processing, km_smart_suggestions, etc.
# would be imported here when they are properly implemented in server/tools/
# Register autonomous agent tools
mcp.tool()(km_autonomous_agent)
logger.info("Registered km_autonomous_agent tool")
# TODO: Register these when they're properly moved to server/tools/
# mcp.tool()(km_ai_processing)
# logger.info("Registered km_ai_processing tool")
# mcp.tool()(km_smart_suggestions)
# logger.info("Registered km_smart_suggestions tool")
# mcp.tool()(km_audit_system)
# logger.info("Registered km_audit_system tool")
# mcp.tool()(km_enterprise_sync)
# logger.info("Registered km_enterprise_sync tool")
# mcp.tool()(km_cloud_connector)
# logger.info("Registered km_cloud_connector tool")
logger.info("Successfully registered advanced AI tools")
except Exception as e:
logger.error(f"Error registering advanced AI tools: {e}")
raise
