"""
MCP (Model Context Protocol) base classes.
This module provides the base classes for MCP protocol implementation.
"""
from abc import ABC, abstractmethod
from typing import Dict, Any, List, Optional
from .resource_types import (
Resource,
ResourceType,
Tool,
Prompt,
ServerInfo,
ClientInfo,
ResourceTemplate,
GetPromptResult
)
class Server(ABC):
"""
Base class for MCP servers.
Provides the core functionality and interface for MCP protocol implementation.
"""
def __init__(self, name: str, version: str, protocolVersion: str = "1.0"):
"""
Initialize the MCP server.
Args:
name: The name of the server
version: The version of the server
protocolVersion: The version of the MCP protocol to use
"""
self.name = name
self.version = version
self.protocolVersion = protocolVersion
self._running = False
self._shutdown_requested = False
@property
@abstractmethod
def capabilities(self) -> Dict[str, Any]:
"""
Get the server's capabilities.
Must be implemented by subclasses.
"""
pass
@abstractmethod
async def initialize(self, client_info: ClientInfo) -> ServerInfo:
"""
Initialize the server with client information.
Must be implemented by subclasses.
Args:
client_info: Information about the client
Returns:
ServerInfo object with server capabilities
"""
pass
@abstractmethod
async def get_resource(self, uri: str) -> Resource:
"""
Get a resource by URI.
Must be implemented by subclasses.
Args:
uri: The URI of the resource to get
Returns:
The requested resource
"""
pass
@abstractmethod
async def list_resources(self, template: Optional[ResourceTemplate] = None) -> List[Resource]:
"""
List available resources.
Must be implemented by subclasses.
Args:
template: Optional template to filter resources
Returns:
List of matching resources
"""
pass
@abstractmethod
async def list_tools(self) -> List[Tool]:
"""
List available tools.
Must be implemented by subclasses.
Returns:
List of available tools
"""
pass
@abstractmethod
async def list_prompts(self) -> List[Prompt]:
"""
List available prompts.
Must be implemented by subclasses.
Returns:
List of available prompts
"""
pass
@abstractmethod
async def get_prompt(self, name: str, args: Dict[str, Any]) -> GetPromptResult:
"""
Get a prompt by name.
Must be implemented by subclasses.
Args:
name: The name of the prompt
args: Arguments for the prompt
Returns:
The prompt result
"""
pass
async def run(self):
"""
Run the server.
This method should be implemented by subclasses to handle the specific
protocol implementation (stdio, SSE, etc.).
"""
pass
async def stop(self):
"""
Stop the server gracefully.
"""
pass
