MCP Integration

from gaia.agents.base.mcp_agent import MCPAgent
from gaia.agents.base.tools import tool
from typing import List, Dict

class MyMCPAgent(MCPAgent):
    """Agent compatible with MCP protocol."""

    def get_mcp_tool_definitions(self) -> List[Dict]:
        """Return MCP-formatted tool definitions."""
        return [
            {
                "name": "search_database",
                "description": "Search the database for records",
                "inputSchema": {
                    "type": "object",
                    "properties": {
                        "query": {
                            "type": "string",
                            "description": "Search query"
                        },
                        "limit": {
                            "type": "integer",
                            "description": "Max results",
                            "default": 10
                        }
                    },
                    "required": ["query"]
                }
            }
        ]

    def execute_mcp_tool(self, tool_name: str, arguments: Dict) -> Dict:
        """Execute an MCP tool call."""
        if tool_name == "search_database":
            query = arguments["query"]
            limit = arguments.get("limit", 10)
            # Your implementation
            return {"results": [...], "count": 5}

        return {"error": f"Unknown tool: {tool_name}"}

    # Optional: Provide prompts
    def get_mcp_prompts(self) -> List[Dict]:
        return [
            {
                "name": "search_help",
                "description": "Help with database searches",
                "arguments": [
                    {"name": "topic", "description": "What to search for"}
                ]
            }
        ]

    # Optional: Expose resources
    def get_mcp_resources(self) -> List[Dict]:
        return [
            {
                "uri": "database://schema",
                "name": "Database Schema",
                "description": "Current database schema",
                "mimeType": "application/json"
            }
        ]

from gaia.mcp.agent_mcp_server import AgentMCPServer
from my_package.agent import MyMCPAgent

# Create MCP server for your agent
server = AgentMCPServer(
    agent_class=MyMCPAgent,
    name="my-mcp-agent",
    port=8080,
    host="localhost",
    verbose=True,
    agent_params={"custom_param": "value"}
)

# Start server
server.start()

# Now accessible via MCP protocol
# VSCode, Claude Desktop, etc. can connect