atlas/home-voice-agent/mcp-server/tools/registry.py

"""
Tool Registry - Manages tool registration and execution.
"""

import logging
import sys
from pathlib import Path
from typing import Any, Dict, List, Optional

# Add parent directory to path for imports
sys.path.insert(0, str(Path(__file__).parent.parent))
from tools.echo import EchoTool
from tools.weather import WeatherTool
from tools.time import (
    GetCurrentTimeTool,
    GetDateTool,
    GetTimezoneInfoTool,
    ConvertTimezoneTool
)
from tools.timers import (
    TimersTool,
    RemindersTool,
    ListTimersTool,
    CancelTimerTool
)
from tools.tasks import (
    AddTaskTool,
    UpdateTaskStatusTool,
    ListTasksTool
)
from tools.notes import (
    CreateNoteTool,
    ReadNoteTool,
    AppendToNoteTool,
    SearchNotesTool,
    ListNotesTool
)
from tools.memory_tools import (
    StoreMemoryTool,
    GetMemoryTool,
    SearchMemoryTool,
    ListMemoryTool
)

logger = logging.getLogger(__name__)


class ToolRegistry:
    """Registry for MCP tools."""
    
    def __init__(self):
        """Initialize tool registry with default tools."""
        self._tools: Dict[str, Any] = {}
        self._register_default_tools()
    
    def _register_default_tools(self):
        """Register default tools."""
        self.register_tool(EchoTool())
        self.register_tool(WeatherTool())
        self.register_tool(GetCurrentTimeTool())
        self.register_tool(GetDateTool())
        self.register_tool(GetTimezoneInfoTool())
        self.register_tool(ConvertTimezoneTool())
        # Timer and reminder tools
        self.register_tool(TimersTool())
        self.register_tool(RemindersTool())
        self.register_tool(ListTimersTool())
        self.register_tool(CancelTimerTool())
        # Task management tools
        self.register_tool(AddTaskTool())
        self.register_tool(UpdateTaskStatusTool())
        self.register_tool(ListTasksTool())
        # Notes and files tools
        self.register_tool(CreateNoteTool())
        self.register_tool(ReadNoteTool())
        self.register_tool(AppendToNoteTool())
        self.register_tool(SearchNotesTool())
        self.register_tool(ListNotesTool())
        # Memory tools
        self.register_tool(StoreMemoryTool())
        self.register_tool(GetMemoryTool())
        self.register_tool(SearchMemoryTool())
        self.register_tool(ListMemoryTool())
        logger.info(f"Registered {len(self._tools)} tools")
    
    def register_tool(self, tool):
        """Register a tool."""
        self._tools[tool.name] = tool
        logger.info(f"Registered tool: {tool.name}")
    
    def list_tools(self) -> List[Dict[str, Any]]:
        """List all registered tools with their schemas."""
        return [tool.get_schema() for tool in self._tools.values()]
    
    def call_tool(self, name: str, arguments: Dict[str, Any]) -> Dict[str, Any]:
        """
        Call a tool by name with arguments.
        
        Returns:
            Dict with 'content' list containing tool results
        """
        if name not in self._tools:
            raise ValueError(f"Tool not found: {name}")
        
        tool = self._tools[name]
        logger.info(f"Calling tool: {name} with arguments: {arguments}")
        
        try:
            result = tool.execute(arguments)
            return {
                "content": [
                    {
                        "type": "text",
                        "text": str(result)
                    }
                ]
            }
        except Exception as e:
            logger.error(f"Tool execution error: {e}", exc_info=True)
            raise