kailash 0.6.2__py3-none-any.whl → 0.6.4__py3-none-any.whl

kailash/mcp_server/server.py

@@ -0,0 +1,1747 @@
+"""
+Enhanced MCP Server Framework with production-ready capabilities.
+
+This module provides both basic and enhanced MCP server implementations using
+the official FastMCP framework from Anthropic. Servers run as long-lived
+services that expose tools, resources, and prompts to MCP clients.
+
+Enhanced Features:
+- Multiple transport support (STDIO, SSE, HTTP)
+- Authentication and authorization
+- Rate limiting and circuit breaker patterns
+- Metrics collection and monitoring
+- Error handling with structured codes
+- Service discovery integration
+- Resource streaming
+- Connection pooling
+- Caching with TTL support
+
+Basic Usage:
+    Abstract base class for custom servers:
+
+    >>> class MyServer(MCPServerBase):
+    ...     def setup(self):
+    ...         @self.add_tool()
+    ...         def calculate(a: int, b: int) -> int:
+    ...             return a + b
+    >>> server = MyServer("calculator")
+    >>> server.start()
+
+Production Usage:
+    Main server with all production features:
+
+    >>> from kailash.mcp_server import MCPServer
+    >>> server = MCPServer("my-server", enable_cache=True)
+    >>> @server.tool(cache_key="search", cache_ttl=600)
+    ... def search(query: str) -> dict:
+    ...     return {"results": f"Found data for {query}"}
+    >>> server.run()
+
+Enhanced Production Usage:
+    Server with authentication and monitoring:
+
+    >>> from kailash.mcp_server.auth import APIKeyAuth
+    >>> auth = APIKeyAuth({"user1": "secret-key"})
+    >>> server = MCPServer(
+    ...     "my-server",
+    ...     auth_provider=auth,
+    ...     enable_metrics=True,
+    ...     enable_http_transport=True,
+    ...     rate_limit_config={"requests_per_minute": 100}
+    ... )
+    >>> server.run()
+"""
+
+import asyncio
+import functools
+import logging
+import time
+import uuid
+from abc import ABC, abstractmethod
+from collections.abc import Callable
+from pathlib import Path
+from typing import Any, Dict, List, Optional, TypeVar, Union
+
+from .auth import AuthManager, AuthProvider, PermissionManager, RateLimiter
+from .errors import (
+    AuthenticationError,
+    AuthorizationError,
+    ErrorAggregator,
+    MCPError,
+    MCPErrorCode,
+    RateLimitError,
+    ResourceError,
+    RetryableOperation,
+    ToolError,
+)
+from .utils import CacheManager, ConfigManager, MetricsCollector, format_response
+
+logger = logging.getLogger(__name__)
+
+F = TypeVar("F", bound=Callable[..., Any])
+
+
+class MCPServerBase(ABC):
+    """Base class for MCP servers using FastMCP.
+
+    This provides a framework for creating MCP servers that expose
+    tools, resources, and prompts via the Model Context Protocol.
+
+    Examples:
+        Creating a custom server:
+
+        >>> class MyServer(MCPServerBase):
+        ...     def setup(self):
+        ...         @self.add_tool()
+        ...         def search(query: str) -> str:
+        ...             return f"Results for: {query}"
+        ...         @self.add_resource("data://example")
+        ...         def get_example():
+        ...             return "Example data"
+        >>> server = MyServer("my-server", port=8080)
+        >>> server.start()  # Runs until stopped
+    """
+
+    def __init__(self, name: str, port: int = 8080, host: str = "localhost"):
+        """Initialize the MCP server.
+
+        Args:
+            name: Name of the server.
+            port: Port to listen on (default: 8080).
+            host: Host to bind to (default: "localhost").
+        """
+        self.name = name
+        self.port = port
+        self.host = host
+        self._mcp = None
+        self._running = False
+
+    @abstractmethod
+    def setup(self):
+        """Setup server tools, resources, and prompts.
+
+        This method should be implemented by subclasses to define
+        the server's capabilities using decorators.
+
+        Note:
+            Use @self.add_tool(), @self.add_resource(uri), and
+            @self.add_prompt(name) decorators to register capabilities.
+        """
+
+    def add_tool(self):
+        """Decorator to add a tool to the server.
+
+        Returns:
+            Function decorator for registering tools.
+
+        Examples:
+            >>> @server.add_tool()
+            ... def calculate(a: int, b: int) -> int:
+            ...     '''Add two numbers'''
+            ...     return a + b
+        """
+
+        def decorator(func: Callable):
+            if self._mcp is None:
+                self._init_mcp()
+
+            # Use FastMCP's tool decorator
+            return self._mcp.tool()(func)
+
+        return decorator
+
+    def add_resource(self, uri: str):
+        """Decorator to add a resource to the server.
+
+        Args:
+            uri: URI pattern for the resource (supports wildcards).
+
+        Returns:
+            Function decorator for registering resources.
+
+        Examples:
+            >>> @server.add_resource("file:///data/*")
+            ... def get_file(path: str) -> str:
+            ...     return f"Content of {path}"
+        """
+
+        def decorator(func: Callable):
+            if self._mcp is None:
+                self._init_mcp()
+
+            # Use FastMCP's resource decorator
+            return self._mcp.resource(uri)(func)
+
+        return decorator
+
+    def add_prompt(self, name: str):
+        """Decorator to add a prompt template to the server.
+
+        Args:
+            name: Name of the prompt.
+
+        Returns:
+            Function decorator for registering prompts.
+
+        Examples:
+            >>> @server.add_prompt("analyze")
+            ... def analyze_prompt(data: str) -> str:
+            ...     return f"Please analyze the following data: {data}"
+        """
+
+        def decorator(func: Callable):
+            if self._mcp is None:
+                self._init_mcp()
+
+            # Use FastMCP's prompt decorator
+            return self._mcp.prompt(name)(func)
+
+        return decorator
+
+    def _init_mcp(self):
+        """Initialize the FastMCP instance."""
+        try:
+            # Try independent FastMCP package first (when available)
+            from fastmcp import FastMCP
+            self._mcp = FastMCP(self.name)
+        except ImportError:
+            logger.warning("FastMCP not available, using fallback mode")
+            # Use same fallback as MCPServer
+            self._mcp = self._create_fallback_server()
+    
+    def _create_fallback_server(self):
+        """Create a fallback server when FastMCP is not available."""
+        class FallbackMCPServer:
+            def __init__(self, name: str):
+                self.name = name
+                self._tools = {}
+                self._resources = {}
+                self._prompts = {}
+            
+            def tool(self, *args, **kwargs):
+                def decorator(func):
+                    self._tools[func.__name__] = func
+                    return func
+                return decorator
+            
+            def resource(self, uri):
+                def decorator(func):
+                    self._resources[uri] = func
+                    return func
+                return decorator
+            
+            def prompt(self, name):
+                def decorator(func):
+                    self._prompts[name] = func
+                    return func
+                return decorator
+            
+            def run(self, **kwargs):
+                raise NotImplementedError("FastMCP not available")
+        
+        return FallbackMCPServer(self.name)
+
+    def start(self):
+        """Start the MCP server.
+
+        This runs the server as a long-lived process until stopped.
+
+        Raises:
+            ImportError: If FastMCP is not available.
+            Exception: If server fails to start.
+        """
+        if self._mcp is None:
+            self._init_mcp()
+
+        # Run setup to register tools/resources
+        self.setup()
+
+        logger.info(f"Starting MCP server '{self.name}' on {self.host}:{self.port}")
+        self._running = True
+
+        try:
+            # Run the FastMCP server
+            logger.info("Running FastMCP server in stdio mode")
+            self._mcp.run()
+        except Exception as e:
+            logger.error(f"Failed to start server: {e}")
+            raise
+        finally:
+            self._running = False
+
+    def stop(self):
+        """Stop the MCP server."""
+        logger.info(f"Stopping MCP server '{self.name}'")
+        self._running = False
+        # In a real implementation, we'd need to handle graceful shutdown
+
+
+class MCPServer:
+    """
+    Kailash MCP Server - Node-based Model Context Protocol server.
+
+    This MCP server follows Kailash philosophy by integrating with the node
+    and workflow system. Tools can be implemented as nodes, and complex
+    MCP capabilities can be built using workflows.
+
+    Core Features:
+    - Node-based tool implementation using Kailash nodes
+    - Workflow-based complex operations
+    - Production-ready with authentication, caching, and monitoring
+    - Multiple transport support (STDIO, SSE, HTTP)
+    - Integration with Kailash runtime and infrastructure
+
+    Kailash Philosophy Integration:
+        Using nodes as MCP tools:
+        >>> from kailash.mcp_server import MCPServer
+        >>> from kailash.nodes import PythonCodeNode
+        >>>
+        >>> server = MCPServer("my-server")
+        >>>
+        >>> # Register a node as an MCP tool
+        >>> @server.node_tool(PythonCodeNode)
+        ... def calculate(a: int, b: int) -> int:
+        ...     return a + b
+        >>>
+        >>> server.run()
+
+        Using workflows as MCP tools:
+        >>> from kailash.workflows import WorkflowBuilder
+        >>>
+        >>> # Create workflow for complex MCP operation
+        >>> workflow = WorkflowBuilder()
+        >>> workflow.add_node("csv_reader", "CSVReaderNode", {"file_path": "data.csv"})
+        >>> workflow.add_node("processor", "PythonCodeNode", {"code": "process_data"})
+        >>> workflow.add_connection("csv_reader", "processor", "data", "input_data")
+        >>>
+        >>> server.register_workflow_tool("process_csv", workflow)
+
+    Traditional usage (for compatibility):
+        >>> server = MCPServer("my-server", enable_cache=True, enable_metrics=True)
+        >>> @server.tool(cache_key="search", cache_ttl=600)
+        ... def search(query: str) -> dict:
+        ...     return {"results": f"Found: {query}"}
+        >>> server.run()
+
+        With authentication and advanced features:
+        >>> from kailash.mcp_server.auth import APIKeyAuth
+        >>> auth = APIKeyAuth({"user1": "secret-key"})
+        >>> server = MCPServer(
+        ...     "my-server",
+        ...     auth_provider=auth,
+        ...     enable_http_transport=True,
+        ...     rate_limit_config={"requests_per_minute": 100},
+        ...     circuit_breaker_config={"failure_threshold": 5}
+        ... )
+        >>> server.run()
+    """
+
+    def __init__(
+        self,
+        name: str,
+        config_file: Optional[Union[str, Path]] = None,
+        enable_cache: bool = True,
+        cache_ttl: int = 300,
+        cache_backend: str = "memory",  # "memory" or "redis"
+        cache_config: Optional[Dict[str, Any]] = None,
+        enable_metrics: bool = True,
+        enable_formatting: bool = True,
+        enable_monitoring: bool = False,  # Health checks, alerts, observability
+        # Enhanced features (optional for backward compatibility)
+        auth_provider: Optional[AuthProvider] = None,
+        enable_http_transport: bool = False,
+        enable_sse_transport: bool = False,
+        rate_limit_config: Optional[Dict[str, Any]] = None,
+        circuit_breaker_config: Optional[Dict[str, Any]] = None,
+        enable_discovery: bool = False,
+        connection_pool_config: Optional[Dict[str, Any]] = None,
+        error_aggregation: bool = True,
+        transport_timeout: float = 30.0,
+        max_request_size: int = 10_000_000,  # 10MB
+        enable_streaming: bool = False,
+    ):
+        """
+        Initialize enhanced MCP server.
+
+        Args:
+            name: Server name
+            config_file: Optional configuration file path
+            enable_cache: Whether to enable caching (default: True)
+            cache_ttl: Default cache TTL in seconds (default: 300)
+            cache_backend: Cache backend ("memory" or "redis")
+            cache_config: Cache configuration (for Redis: {"redis_url": "redis://...", "prefix": "mcp:"})
+            enable_metrics: Whether to enable metrics collection (default: True)
+            enable_formatting: Whether to enable response formatting (default: True)
+            auth_provider: Optional authentication provider
+            enable_http_transport: Enable HTTP transport support
+            enable_sse_transport: Enable SSE transport support
+            rate_limit_config: Rate limiting configuration
+            circuit_breaker_config: Circuit breaker configuration
+            enable_discovery: Enable service discovery
+            connection_pool_config: Connection pooling configuration
+            error_aggregation: Enable error aggregation
+            transport_timeout: Transport timeout in seconds
+            max_request_size: Maximum request size in bytes
+            enable_streaming: Enable streaming support
+        """
+        self.name = name
+
+        # Enhanced features
+        self.auth_provider = auth_provider
+        self.enable_http_transport = enable_http_transport
+        self.enable_sse_transport = enable_sse_transport
+        self.enable_discovery = enable_discovery
+        self.enable_streaming = enable_streaming
+        self.enable_monitoring = enable_monitoring
+        self.transport_timeout = transport_timeout
+        self.max_request_size = max_request_size
+
+        # Initialize configuration
+        self.config = ConfigManager(config_file)
+
+        # Set default configuration values including enhanced features
+        self.config.update(
+            {
+                "server": {
+                    "name": name,
+                    "version": "1.0.0",
+                    "transport": "stdio",
+                    "enable_http": enable_http_transport,
+                    "enable_sse": enable_sse_transport,
+                    "timeout": transport_timeout,
+                    "max_request_size": max_request_size,
+                    "enable_streaming": enable_streaming,
+                },
+                "cache": {
+                    "enabled": enable_cache,
+                    "default_ttl": cache_ttl,
+                    "max_size": 128,
+                    "backend": cache_backend,
+                    "config": cache_config or {},
+                },
+                "metrics": {
+                    "enabled": enable_metrics,
+                    "collect_performance": True,
+                    "collect_usage": True,
+                },
+                "formatting": {
+                    "enabled": enable_formatting,
+                    "default_format": "markdown",
+                },
+                "monitoring": {
+                    "enabled": enable_monitoring,
+                    "health_checks": enable_monitoring,
+                    "observability": enable_monitoring,
+                },
+                "auth": {
+                    "enabled": auth_provider is not None,
+                    "provider_type": (
+                        type(auth_provider).__name__ if auth_provider else None
+                    ),
+                },
+                "rate_limiting": rate_limit_config or {},
+                "circuit_breaker": circuit_breaker_config or {},
+                "discovery": {"enabled": enable_discovery},
+                "connection_pool": connection_pool_config or {},
+            }
+        )
+
+        # Initialize authentication manager
+        if auth_provider:
+            self.auth_manager = AuthManager(
+                provider=auth_provider,
+                permission_manager=PermissionManager(),
+                rate_limiter=RateLimiter(**(rate_limit_config or {})),
+            )
+        else:
+            self.auth_manager = None
+
+        # Initialize components
+        self.cache = CacheManager(
+            enabled=self.config.get("cache.enabled", enable_cache),
+            default_ttl=self.config.get("cache.default_ttl", cache_ttl),
+            backend=self.config.get("cache.backend", cache_backend),
+            config=self.config.get("cache.config", cache_config or {}),
+        )
+
+        self.metrics = MetricsCollector(
+            enabled=self.config.get("metrics.enabled", enable_metrics),
+            collect_performance=self.config.get("metrics.collect_performance", True),
+            collect_usage=self.config.get("metrics.collect_usage", True),
+        )
+
+        # Error aggregation
+        if error_aggregation:
+            self.error_aggregator = ErrorAggregator()
+        else:
+            self.error_aggregator = None
+
+        # Circuit breaker for tool calls
+        if circuit_breaker_config:
+            from .errors import CircuitBreakerRetry
+
+            self.circuit_breaker = CircuitBreakerRetry(**circuit_breaker_config)
+        else:
+            self.circuit_breaker = None
+
+        # FastMCP server instance (initialized lazily)
+        self._mcp = None
+        self._running = False
+        self._active_sessions: Dict[str, Dict[str, Any]] = {}
+        self._connection_pools: Dict[str, List[Any]] = {}
+
+        # Tool registry for management
+        self._tool_registry: Dict[str, Dict[str, Any]] = {}
+        self._resource_registry: Dict[str, Dict[str, Any]] = {}
+        self._prompt_registry: Dict[str, Dict[str, Any]] = {}
+
+    def _init_mcp(self):
+        """Initialize FastMCP server."""
+        if self._mcp is not None:
+            return
+
+        try:
+            # Try independent FastMCP package first (when available)
+            from fastmcp import FastMCP
+            self._mcp = FastMCP(self.name)
+            logger.info(f"Initialized FastMCP server: {self.name}")
+        except ImportError as e1:
+            logger.warning(f"Independent FastMCP not available: {e1}")
+            try:
+                # Fallback to official MCP FastMCP (when fixed)
+                from mcp.server import FastMCP
+                self._mcp = FastMCP(self.name)
+                logger.info(f"Initialized official FastMCP server: {self.name}")
+            except ImportError as e2:
+                logger.warning(f"Official FastMCP not available: {e2}")
+                # Final fallback: Create a minimal FastMCP-compatible wrapper
+                logger.info(f"Using low-level MCP Server fallback for: {self.name}")
+                self._mcp = self._create_fallback_server()
+                
+    def _create_fallback_server(self):
+        """Create a fallback server when FastMCP is not available."""
+        logger.info("Creating fallback server implementation")
+        
+        class FallbackMCPServer:
+            """Minimal FastMCP-compatible server for when FastMCP is unavailable."""
+            
+            def __init__(self, name: str):
+                self.name = name
+                self._tools = {}
+                self._resources = {}
+                self._prompts = {}
+                logger.info(f"Fallback MCP server '{name}' initialized")
+            
+            def tool(self, *args, **kwargs):
+                """Tool decorator that stores tool registration."""
+                def decorator(func):
+                    tool_name = func.__name__
+                    self._tools[tool_name] = func
+                    logger.debug(f"Registered fallback tool: {tool_name}")
+                    return func
+                return decorator
+            
+            def resource(self, uri):
+                """Resource decorator that stores resource registration."""
+                def decorator(func):
+                    self._resources[uri] = func
+                    logger.debug(f"Registered fallback resource: {uri}")
+                    return func
+                return decorator
+            
+            def prompt(self, name):
+                """Prompt decorator that stores prompt registration."""
+                def decorator(func):
+                    self._prompts[name] = func
+                    logger.debug(f"Registered fallback prompt: {name}")
+                    return func
+                return decorator
+            
+            def run(self, **kwargs):
+                """Placeholder run method."""
+                logger.warning(f"Fallback server '{self.name}' run() called - FastMCP features limited")
+                logger.info(f"Registered: {len(self._tools)} tools, {len(self._resources)} resources, {len(self._prompts)} prompts")
+                # In a real implementation, we would set up low-level MCP protocol here
+                raise NotImplementedError(
+                    "Full MCP protocol not implemented in fallback mode. "
+                    "Install 'fastmcp>=2.10.0' or wait for official MCP package fix."
+                )
+        
+        return FallbackMCPServer(self.name)
+
+    def tool(
+        self,
+        cache_key: Optional[str] = None,
+        cache_ttl: Optional[int] = None,
+        format_response: Optional[str] = None,
+        # Enhanced features
+        required_permission: Optional[str] = None,
+        required_permissions: Optional[
+            List[str]
+        ] = None,  # Added for backward compatibility
+        rate_limit: Optional[Dict[str, Any]] = None,
+        enable_circuit_breaker: bool = True,
+        timeout: Optional[float] = None,
+        retryable: bool = True,
+        stream_response: bool = False,
+    ):
+        """
+        Enhanced tool decorator with authentication, caching, metrics, and error handling.
+
+        Args:
+            cache_key: Optional cache key for caching results
+            cache_ttl: Optional TTL override for this tool
+            format_response: Optional response format ("json", "markdown", "table", etc.)
+            required_permission: Single required permission for tool access
+            required_permissions: List of required permissions (alternative to required_permission)
+            rate_limit: Tool-specific rate limiting configuration
+            enable_circuit_breaker: Enable circuit breaker for this tool
+            timeout: Tool execution timeout in seconds
+            retryable: Whether tool failures are retryable
+            stream_response: Enable streaming response for large results
+
+        Returns:
+            Decorated function with enhanced capabilities
+
+        Example:
+            @server.tool(
+                cache_key="weather",
+                cache_ttl=600,
+                format_response="markdown",
+                required_permission="weather.read",
+                rate_limit={"requests_per_minute": 10},
+                timeout=30.0
+            )
+            async def get_weather(city: str) -> dict:
+                # Expensive API call - will be cached for 10 minutes
+                return await fetch_weather_data(city)
+        """
+
+        def decorator(func: F) -> F:
+            if self._mcp is None:
+                self._init_mcp()
+
+            # Get function name for registration
+            tool_name = func.__name__
+
+            # Normalize permissions - support both singular and plural
+            normalized_permission = None
+            if required_permissions is not None and required_permission is not None:
+                raise ValueError(
+                    "Cannot specify both required_permission and required_permissions"
+                )
+            elif required_permissions is not None:
+                if len(required_permissions) == 1:
+                    normalized_permission = required_permissions[0]
+                elif len(required_permissions) > 1:
+                    # For now, take the first permission. Future enhancement could support multiple.
+                    normalized_permission = required_permissions[0]
+                    logger.warning(
+                        f"Tool {tool_name}: Multiple permissions specified, using first: {normalized_permission}"
+                    )
+            elif required_permission is not None:
+                normalized_permission = required_permission
+
+            # Create enhanced wrapper
+            enhanced_func = self._create_enhanced_tool(
+                func,
+                tool_name,
+                cache_key,
+                cache_ttl,
+                format_response,
+                normalized_permission,
+                rate_limit,
+                enable_circuit_breaker,
+                timeout,
+                retryable,
+                stream_response,
+            )
+
+            # Register with FastMCP
+            mcp_tool = self._mcp.tool()(enhanced_func)
+
+            # Track in registry with enhanced metadata
+            self._tool_registry[tool_name] = {
+                "function": mcp_tool,
+                "original_function": func,
+                "cached": cache_key is not None,
+                "cache_key": cache_key,
+                "cache_ttl": cache_ttl,
+                "format_response": format_response,
+                "required_permission": normalized_permission,
+                "rate_limit": rate_limit,
+                "enable_circuit_breaker": enable_circuit_breaker,
+                "timeout": timeout,
+                "retryable": retryable,
+                "stream_response": stream_response,
+                "call_count": 0,
+                "error_count": 0,
+                "last_called": None,
+            }
+
+            logger.debug(
+                f"Registered enhanced tool: {tool_name} "
+                f"(cached: {cache_key is not None}, "
+                f"auth: {required_permission is not None}, "
+                f"rate_limited: {rate_limit is not None})"
+            )
+            return mcp_tool
+
+        return decorator
+
+    def _create_enhanced_tool(
+        self,
+        func: F,
+        tool_name: str,
+        cache_key: Optional[str],
+        cache_ttl: Optional[int],
+        response_format: Optional[str],
+        required_permission: Optional[str],
+        rate_limit: Optional[Dict[str, Any]],
+        enable_circuit_breaker: bool,
+        timeout: Optional[float],
+        retryable: bool,
+        stream_response: bool,
+    ) -> F:
+        """Create enhanced tool function with authentication, caching, metrics, error handling, and more."""
+
+        @functools.wraps(func)
+        def sync_wrapper(*args, **kwargs):
+            # Generate session ID for tracking
+            session_id = str(uuid.uuid4())
+            start_time = time.time() if self.metrics.enabled else None
+
+            try:
+                # Authentication check
+                if self.auth_manager and required_permission:
+                    # Extract credentials from kwargs or context
+                    credentials = self._extract_credentials_from_context(kwargs)
+                    try:
+                        user_info = self.auth_manager.authenticate_and_authorize(
+                            credentials, required_permission
+                        )
+                        # Add user info to session
+                        self._active_sessions[session_id] = {
+                            "user": user_info,
+                            "tool": tool_name,
+                            "start_time": start_time,
+                            "permission": required_permission,
+                        }
+                    except (AuthenticationError, AuthorizationError) as e:
+                        if self.error_aggregator:
+                            self.error_aggregator.record_error(e)
+                        raise ToolError(
+                            f"Access denied for {tool_name}: {str(e)}",
+                            tool_name=tool_name,
+                        )
+
+                # Rate limiting check
+                if rate_limit and self.auth_manager:
+                    user_id = (
+                        self._active_sessions.get(session_id, {})
+                        .get("user", {})
+                        .get("id", "anonymous")
+                    )
+                    try:
+                        self.auth_manager.rate_limiter.check_rate_limit(
+                            user_id, tool_name, **rate_limit
+                        )
+                    except RateLimitError as e:
+                        if self.error_aggregator:
+                            self.error_aggregator.record_error(e)
+                        raise
+
+                # Circuit breaker check
+                if enable_circuit_breaker and self.circuit_breaker:
+                    if not self.circuit_breaker.should_retry(
+                        MCPError("Circuit breaker check"), 1
+                    ):
+                        error = MCPError(
+                            f"Circuit breaker open for {tool_name}",
+                            error_code=MCPErrorCode.CIRCUIT_BREAKER_OPEN,
+                            retryable=True,
+                        )
+                        if self.error_aggregator:
+                            self.error_aggregator.record_error(error)
+                        raise error
+
+                # Try cache first if enabled
+                if cache_key and self.cache.enabled:
+                    cache = self.cache.get_cache(cache_key, ttl=cache_ttl)
+                    cache_lookup_key = self.cache._create_cache_key(
+                        tool_name, args, kwargs
+                    )
+
+                    # For sync functions with Redis, we need to handle async operations
+                    if cache.is_redis:
+                        # Try to run async cache operations in sync context
+                        try:
+                            # Check if we're already in an async context
+                            try:
+                                asyncio.get_running_loop()
+                                # We're in an async context, but this is a sync function
+                                # Fall back to memory cache behavior (no caching for now)
+                                result = None
+                            except RuntimeError:
+                                # Not in async context, we can use asyncio.run
+                                result = asyncio.run(cache.aget(cache_lookup_key))
+                        except Exception as e:
+                            logger.debug(f"Redis cache error in sync context: {e}")
+                            result = None
+                    else:
+                        result = cache.get(cache_lookup_key)
+
+                    if result is not None:
+                        logger.debug(f"Cache hit for {tool_name}")
+                        if self.metrics.enabled:
+                            latency = time.time() - start_time
+                            self.metrics.track_tool_call(tool_name, latency, True)
+
+                        # Update registry stats
+                        self._tool_registry[tool_name]["call_count"] += 1
+                        self._tool_registry[tool_name]["last_called"] = time.time()
+
+                        return self._format_response(
+                            result, response_format, stream_response
+                        )
+
+                # Execute function with timeout
+                if timeout:
+                    import signal
+
+                    def timeout_handler(signum, frame):
+                        raise TimeoutError(
+                            f"Tool {tool_name} timed out after {timeout}s"
+                        )
+
+                    old_handler = signal.signal(signal.SIGALRM, timeout_handler)
+                    signal.alarm(int(timeout))
+
+                    try:
+                        result = func(*args, **kwargs)
+                    finally:
+                        signal.alarm(0)
+                        signal.signal(signal.SIGALRM, old_handler)
+                else:
+                    result = func(*args, **kwargs)
+
+                # Cache result if enabled
+                if cache_key and self.cache.enabled:
+                    # For sync functions with Redis, handle async operations
+                    if cache.is_redis:
+                        try:
+                            # Check if we're already in an async context
+                            try:
+                                asyncio.get_running_loop()
+                                # We're in an async context, but this is a sync function
+                                # Fall back to memory cache behavior (no caching for now)
+                                pass
+                            except RuntimeError:
+                                # Not in async context, we can use asyncio.run
+                                asyncio.run(cache.aset(cache_lookup_key, result))
+                        except Exception as e:
+                            logger.debug(f"Redis cache set error in sync context: {e}")
+                    else:
+                        cache.set(cache_lookup_key, result)
+                    logger.debug(f"Cached result for {tool_name}")
+
+                # Track success metrics
+                if self.metrics.enabled:
+                    latency = time.time() - start_time
+                    self.metrics.track_tool_call(tool_name, latency, True)
+
+                # Update circuit breaker on success
+                if enable_circuit_breaker and self.circuit_breaker:
+                    self.circuit_breaker.on_success()
+
+                # Update registry stats
+                self._tool_registry[tool_name]["call_count"] += 1
+                self._tool_registry[tool_name]["last_called"] = time.time()
+
+                return self._format_response(result, response_format, stream_response)
+
+            except Exception as e:
+                # Convert to MCP error if needed
+                if not isinstance(e, MCPError):
+                    mcp_error = ToolError(
+                        f"Tool execution failed: {str(e)}",
+                        tool_name=tool_name,
+                        retryable=retryable,
+                        cause=e,
+                    )
+                else:
+                    mcp_error = e
+
+                # Record error
+                if self.error_aggregator:
+                    self.error_aggregator.record_error(mcp_error)
+
+                # Update circuit breaker on failure
+                if enable_circuit_breaker and self.circuit_breaker:
+                    self.circuit_breaker.on_failure(mcp_error)
+
+                # Track error metrics
+                if self.metrics.enabled and start_time:
+                    latency = time.time() - start_time
+                    self.metrics.track_tool_call(
+                        tool_name, latency, False, type(e).__name__
+                    )
+
+                # Update registry stats
+                self._tool_registry[tool_name]["error_count"] += 1
+                self._tool_registry[tool_name]["last_called"] = time.time()
+
+                logger.error(f"Error in tool {tool_name}: {mcp_error}")
+                raise mcp_error
+
+            finally:
+                # Clean up session
+                if session_id in self._active_sessions:
+                    del self._active_sessions[session_id]
+
+        @functools.wraps(func)
+        async def async_wrapper(*args, **kwargs):
+            # Generate session ID for tracking
+            session_id = str(uuid.uuid4())
+            start_time = time.time() if self.metrics.enabled else None
+
+            try:
+                # Authentication check
+                if self.auth_manager and required_permission:
+                    # Extract credentials from kwargs or context
+                    credentials = self._extract_credentials_from_context(kwargs)
+
+                    # Allow bypassing auth for direct calls when no credentials provided
+                    # This enables testing and development scenarios
+                    if not credentials and not any(
+                        k.startswith("mcp_") for k in kwargs.keys()
+                    ):
+                        logger.debug(
+                            f"Tool {tool_name}: No credentials provided, allowing direct call (development/testing)"
+                        )
+                        user_info = None
+                    else:
+                        try:
+                            user_info = self.auth_manager.authenticate_and_authorize(
+                                credentials, required_permission
+                            )
+                            # Add user info to session
+                            self._active_sessions[session_id] = {
+                                "user": user_info,
+                                "tool": tool_name,
+                                "start_time": start_time,
+                                "permission": required_permission,
+                            }
+                        except (AuthenticationError, AuthorizationError) as e:
+                            if self.error_aggregator:
+                                self.error_aggregator.record_error(e)
+                            raise ToolError(
+                                f"Access denied for {tool_name}: {str(e)}",
+                                tool_name=tool_name,
+                            )
+
+                # Rate limiting check
+                if rate_limit and self.auth_manager:
+                    user_id = (
+                        self._active_sessions.get(session_id, {})
+                        .get("user", {})
+                        .get("id", "anonymous")
+                    )
+                    try:
+                        self.auth_manager.rate_limiter.check_rate_limit(
+                            user_id, tool_name, **rate_limit
+                        )
+                    except RateLimitError as e:
+                        if self.error_aggregator:
+                            self.error_aggregator.record_error(e)
+                        raise
+
+                # Circuit breaker check
+                if enable_circuit_breaker and self.circuit_breaker:
+                    if not self.circuit_breaker.should_retry(
+                        MCPError("Circuit breaker check"), 1
+                    ):
+                        error = MCPError(
+                            f"Circuit breaker open for {tool_name}",
+                            error_code=MCPErrorCode.CIRCUIT_BREAKER_OPEN,
+                            retryable=True,
+                        )
+                        if self.error_aggregator:
+                            self.error_aggregator.record_error(error)
+                        raise error
+
+                # Execute with caching and stampede prevention if enabled
+                if cache_key and self.cache.enabled:
+                    cache = self.cache.get_cache(cache_key, ttl=cache_ttl)
+                    cache_lookup_key = self.cache._create_cache_key(
+                        tool_name, args, kwargs
+                    )
+
+                    # Define the compute function for cache-or-compute
+                    async def compute_result():
+                        # Filter out auth credentials from kwargs before calling the function
+                        clean_kwargs = {
+                            k: v
+                            for k, v in kwargs.items()
+                            if k
+                            not in [
+                                "api_key",
+                                "token",
+                                "username",
+                                "password",
+                                "jwt",
+                                "authorization",
+                                "mcp_auth",
+                            ]
+                        }
+
+                        # Execute function with timeout
+                        if timeout:
+                            return await asyncio.wait_for(
+                                func(*args, **clean_kwargs), timeout=timeout
+                            )
+                        else:
+                            return await func(*args, **clean_kwargs)
+
+                    # Use cache-or-compute with stampede prevention
+                    result = await cache.get_or_compute(
+                        cache_lookup_key, compute_result, cache_ttl
+                    )
+                    logger.debug(f"Got result for {tool_name} (cached or computed)")
+                else:
+                    # No caching - execute directly
+                    # Filter out auth credentials from kwargs before calling the function
+                    clean_kwargs = {
+                        k: v
+                        for k, v in kwargs.items()
+                        if k
+                        not in [
+                            "api_key",
+                            "token",
+                            "username",
+                            "password",
+                            "jwt",
+                            "authorization",
+                            "mcp_auth",
+                        ]
+                    }
+
+                    # Execute function with timeout
+                    if timeout:
+                        result = await asyncio.wait_for(
+                            func(*args, **clean_kwargs), timeout=timeout
+                        )
+                    else:
+                        result = await func(*args, **clean_kwargs)
+
+                # Track success metrics
+                if self.metrics.enabled:
+                    latency = time.time() - start_time
+                    self.metrics.track_tool_call(tool_name, latency, True)
+
+                # Update circuit breaker on success
+                if enable_circuit_breaker and self.circuit_breaker:
+                    self.circuit_breaker.on_success()
+
+                # Update registry stats
+                self._tool_registry[tool_name]["call_count"] += 1
+                self._tool_registry[tool_name]["last_called"] = time.time()
+
+                return self._format_response(result, response_format, stream_response)
+
+            except Exception as e:
+                # Convert to MCP error if needed
+                if not isinstance(e, MCPError):
+                    mcp_error = ToolError(
+                        f"Tool execution failed: {str(e)}",
+                        tool_name=tool_name,
+                        retryable=retryable,
+                        cause=e,
+                    )
+                else:
+                    mcp_error = e
+
+                # Record error
+                if self.error_aggregator:
+                    self.error_aggregator.record_error(mcp_error)
+
+                # Update circuit breaker on failure
+                if enable_circuit_breaker and self.circuit_breaker:
+                    self.circuit_breaker.on_failure(mcp_error)
+
+                # Track error metrics
+                if self.metrics.enabled and start_time:
+                    latency = time.time() - start_time
+                    self.metrics.track_tool_call(
+                        tool_name, latency, False, type(e).__name__
+                    )
+
+                # Update registry stats
+                self._tool_registry[tool_name]["error_count"] += 1
+                self._tool_registry[tool_name]["last_called"] = time.time()
+
+                logger.error(f"Error in tool {tool_name}: {mcp_error}")
+                raise mcp_error
+
+            finally:
+                # Clean up session
+                if session_id in self._active_sessions:
+                    del self._active_sessions[session_id]
+
+        # Return appropriate wrapper based on function type
+        if asyncio.iscoroutinefunction(func):
+            return async_wrapper
+        else:
+            return sync_wrapper
+
+    def _format_response(
+        self, result: Any, response_format: Optional[str], stream_response: bool = False
+    ) -> Any:
+        """Format response if formatting is enabled, with optional streaming support."""
+        if not self.config.get("formatting.enabled", True) or not response_format:
+            if (
+                stream_response
+                and isinstance(result, (list, dict))
+                and len(str(result)) > 1000
+            ):
+                # For large results, consider streaming (simplified implementation)
+                return {
+                    "streaming": True,
+                    "data": result,
+                    "chunks": self._chunk_large_response(result),
+                }
+            return result
+
+        try:
+            formatted = format_response(result, response_format)
+            if stream_response and isinstance(formatted, str) and len(formatted) > 1000:
+                return {
+                    "streaming": True,
+                    "data": formatted,
+                    "chunks": self._chunk_large_response(formatted),
+                }
+            return formatted
+        except Exception as e:
+            logger.warning(f"Failed to format response: {e}")
+            return result
+
+    def _chunk_large_response(self, data: Any, chunk_size: int = 1000) -> List[str]:
+        """Chunk large responses for streaming."""
+        if isinstance(data, str):
+            return [data[i : i + chunk_size] for i in range(0, len(data), chunk_size)]
+        elif isinstance(data, (list, dict)):
+            data_str = str(data)
+            return [
+                data_str[i : i + chunk_size]
+                for i in range(0, len(data_str), chunk_size)
+            ]
+        else:
+            return [str(data)]
+
+    def _extract_credentials_from_context(
+        self, kwargs: Dict[str, Any]
+    ) -> Dict[str, Any]:
+        """Extract credentials from function context or kwargs."""
+        # Look for common credential patterns in kwargs
+        credentials = {}
+
+        # Check for MCP-style authentication headers
+        if "mcp_auth" in kwargs:
+            credentials.update(kwargs["mcp_auth"])
+
+        # Check for common auth patterns
+        auth_fields = ["api_key", "token", "username", "password", "jwt"]
+        for field in auth_fields:
+            if field in kwargs:
+                credentials[field] = kwargs[field]
+
+        # Check for Authorization header pattern
+        if "authorization" in kwargs:
+            auth_header = kwargs["authorization"]
+            if auth_header.startswith("Bearer "):
+                credentials["token"] = auth_header[7:]
+            elif auth_header.startswith("Basic "):
+                import base64
+
+                try:
+                    decoded = base64.b64decode(auth_header[6:]).decode()
+                    if ":" in decoded:
+                        username, password = decoded.split(":", 1)
+                        credentials["username"] = username
+                        credentials["password"] = password
+                except Exception:
+                    pass
+
+        return credentials
+
+    def resource(self, uri: str):
+        """
+        Add resource with metrics tracking.
+
+        Args:
+            uri: Resource URI pattern
+
+        Returns:
+            Decorated function
+        """
+
+        def decorator(func: F) -> F:
+            if self._mcp is None:
+                self._init_mcp()
+
+            # Wrap with metrics if enabled
+            if self.metrics.enabled:
+                func = self.metrics.track_tool(f"resource:{uri}")(func)
+
+            return self._mcp.resource(uri)(func)
+
+        return decorator
+
+    def prompt(self, name: str):
+        """
+        Add prompt with metrics tracking.
+
+        Args:
+            name: Prompt name
+
+        Returns:
+            Decorated function
+        """
+
+        def decorator(func: F) -> F:
+            if self._mcp is None:
+                self._init_mcp()
+
+            # Wrap with metrics if enabled
+            if self.metrics.enabled:
+                func = self.metrics.track_tool(f"prompt:{name}")(func)
+
+            return self._mcp.prompt(name)(func)
+
+        return decorator
+
+    def get_tool_stats(self) -> Dict[str, Any]:
+        """Get statistics for all registered tools."""
+        stats = {
+            "registered_tools": len(self._tool_registry),
+            "cached_tools": sum(1 for t in self._tool_registry.values() if t["cached"]),
+            "tools": {},
+        }
+
+        for tool_name, tool_info in self._tool_registry.items():
+            stats["tools"][tool_name] = {
+                "cached": tool_info["cached"],
+                "cache_key": tool_info.get("cache_key"),
+                "format_response": tool_info.get("format_response"),
+            }
+
+        return stats
+
+    def get_server_stats(self) -> Dict[str, Any]:
+        """Get comprehensive server statistics."""
+        stats = {
+            "server": {
+                "name": self.name,
+                "running": self._running,
+                "config": self.config.to_dict(),
+                "active_sessions": len(self._active_sessions),
+                "transport": {
+                    "http_enabled": self.enable_http_transport,
+                    "sse_enabled": self.enable_sse_transport,
+                    "streaming_enabled": self.enable_streaming,
+                    "timeout": self.transport_timeout,
+                    "max_request_size": self.max_request_size,
+                },
+                "features": {
+                    "auth_enabled": self.auth_manager is not None,
+                    "circuit_breaker_enabled": self.circuit_breaker is not None,
+                    "error_aggregation_enabled": self.error_aggregator is not None,
+                    "discovery_enabled": self.enable_discovery,
+                },
+            },
+            "tools": self.get_tool_stats(),
+            "resources": self.get_resource_stats(),
+            "prompts": self.get_prompt_stats(),
+        }
+
+        if self.metrics.enabled:
+            stats["metrics"] = self.metrics.export_metrics()
+
+        if self.cache.enabled:
+            stats["cache"] = self.cache.stats()
+
+        if self.error_aggregator:
+            stats["errors"] = self.error_aggregator.get_error_stats(
+                time_window=3600
+            )  # Last hour
+
+        if self.circuit_breaker:
+            stats["circuit_breaker"] = {
+                "state": self.circuit_breaker.state,
+                "failure_count": self.circuit_breaker.failure_count,
+                "success_count": self.circuit_breaker.success_count,
+            }
+
+        return stats
+
+    def get_resource_stats(self) -> Dict[str, Any]:
+        """Get resource statistics."""
+        return {
+            "registered_resources": len(self._resource_registry),
+            "resources": {
+                uri: {
+                    "call_count": info.get("call_count", 0),
+                    "error_count": info.get("error_count", 0),
+                    "last_accessed": info.get("last_accessed"),
+                }
+                for uri, info in self._resource_registry.items()
+            },
+        }
+
+    def get_prompt_stats(self) -> Dict[str, Any]:
+        """Get prompt statistics."""
+        return {
+            "registered_prompts": len(self._prompt_registry),
+            "prompts": {
+                name: {
+                    "call_count": info.get("call_count", 0),
+                    "error_count": info.get("error_count", 0),
+                    "last_used": info.get("last_used"),
+                }
+                for name, info in self._prompt_registry.items()
+            },
+        }
+
+    def get_active_sessions(self) -> Dict[str, Dict[str, Any]]:
+        """Get information about active sessions."""
+        return {
+            session_id: {
+                "user": session_info.get("user", {}),
+                "tool": session_info.get("tool"),
+                "permission": session_info.get("permission"),
+                "duration": time.time() - session_info.get("start_time", time.time()),
+            }
+            for session_id, session_info in self._active_sessions.items()
+        }
+
+    def get_error_trends(
+        self, time_window: Optional[float] = None
+    ) -> List[Dict[str, Any]]:
+        """Get error trends over time."""
+        if not self.error_aggregator:
+            return []
+        return self.error_aggregator.get_error_trends()
+
+    def health_check(self) -> Dict[str, Any]:
+        """Perform comprehensive health check."""
+        health_status = {
+            "status": "healthy",
+            "timestamp": time.time(),
+            "server": {
+                "name": self.name,
+                "running": self._running,
+                "uptime": time.time()
+                - self.config.get("server.start_time", time.time()),
+            },
+            "components": {
+                "mcp": self._mcp is not None,
+                "cache": self.cache.enabled if self.cache else False,
+                "metrics": self.metrics.enabled if self.metrics else False,
+                "auth": self.auth_manager is not None,
+                "circuit_breaker": self.circuit_breaker is not None,
+            },
+            "resources": {
+                "active_sessions": len(self._active_sessions),
+                "tools_registered": len(self._tool_registry),
+                "resources_registered": len(self._resource_registry),
+                "prompts_registered": len(self._prompt_registry),
+            },
+        }
+
+        # Check for issues
+        issues = []
+
+        # Check error rates
+        if self.error_aggregator:
+            error_stats = self.error_aggregator.get_error_stats(
+                time_window=300
+            )  # Last 5 minutes
+            if error_stats.get("error_rate", 0) > 10:  # More than 10 errors per second
+                issues.append("High error rate detected")
+                health_status["status"] = "degraded"
+
+        # Check circuit breaker state
+        if self.circuit_breaker and self.circuit_breaker.state == "open":
+            issues.append("Circuit breaker is open")
+            health_status["status"] = "degraded"
+
+        # Check memory usage for caches
+        if self.cache and self.cache.enabled:
+            cache_stats = self.cache.stats()
+            # Simple heuristic - if any cache is over 90% full
+            for cache_name, stats in cache_stats.items():
+                if isinstance(stats, dict) and stats.get("utilization", 0) > 0.9:
+                    issues.append(f"Cache {cache_name} is over 90% full")
+                    health_status["status"] = "degraded"
+
+        health_status["issues"] = issues
+
+        if issues and health_status["status"] == "healthy":
+            health_status["status"] = "degraded"
+
+        return health_status
+
+    def clear_cache(self, cache_name: Optional[str] = None) -> None:
+        """Clear cache(s)."""
+        if cache_name:
+            cache = self.cache.get_cache(cache_name)
+            cache.clear()
+            logger.info(f"Cleared cache: {cache_name}")
+        else:
+            self.cache.clear_all()
+            logger.info("Cleared all caches")
+
+    def reset_circuit_breaker(self) -> None:
+        """Reset circuit breaker to closed state."""
+        if self.circuit_breaker:
+            self.circuit_breaker.state = "closed"
+            self.circuit_breaker.failure_count = 0
+            self.circuit_breaker.success_count = 0
+            logger.info("Circuit breaker reset to closed state")
+
+    def terminate_session(self, session_id: str) -> bool:
+        """Terminate an active session."""
+        if session_id in self._active_sessions:
+            del self._active_sessions[session_id]
+            logger.info(f"Terminated session: {session_id}")
+            return True
+        return False
+
+    def get_tool_by_name(self, tool_name: str) -> Optional[Dict[str, Any]]:
+        """Get tool information by name."""
+        return self._tool_registry.get(tool_name)
+
+    def disable_tool(self, tool_name: str) -> bool:
+        """Temporarily disable a tool."""
+        if tool_name in self._tool_registry:
+            self._tool_registry[tool_name]["disabled"] = True
+            logger.info(f"Disabled tool: {tool_name}")
+            return True
+        return False
+
+    def enable_tool(self, tool_name: str) -> bool:
+        """Re-enable a disabled tool."""
+        if tool_name in self._tool_registry:
+            self._tool_registry[tool_name]["disabled"] = False
+            logger.info(f"Enabled tool: {tool_name}")
+            return True
+        return False
+
+    def run(self):
+        """Run the enhanced MCP server with all features."""
+        if self._mcp is None:
+            self._init_mcp()
+
+        # Record server start time
+        self.config.update({"server.start_time": time.time()})
+
+        # Log enhanced server startup
+        logger.info(f"Starting enhanced MCP server: {self.name}")
+        logger.info("Features enabled:")
+        logger.info(f"  - Cache: {self.cache.enabled if self.cache else False}")
+        logger.info(f"  - Metrics: {self.metrics.enabled if self.metrics else False}")
+        logger.info(f"  - Authentication: {self.auth_manager is not None}")
+        logger.info(f"  - HTTP Transport: {self.enable_http_transport}")
+        logger.info(f"  - SSE Transport: {self.enable_sse_transport}")
+        logger.info(f"  - Streaming: {self.enable_streaming}")
+        logger.info(f"  - Circuit Breaker: {self.circuit_breaker is not None}")
+        logger.info(f"  - Error Aggregation: {self.error_aggregator is not None}")
+        logger.info(f"  - Service Discovery: {self.enable_discovery}")
+
+        logger.info("Server configuration:")
+        logger.info(f"  - Tools registered: {len(self._tool_registry)}")
+        logger.info(f"  - Resources registered: {len(self._resource_registry)}")
+        logger.info(f"  - Prompts registered: {len(self._prompt_registry)}")
+        logger.info(f"  - Transport timeout: {self.transport_timeout}s")
+        logger.info(f"  - Max request size: {self.max_request_size} bytes")
+
+        self._running = True
+
+        try:
+            # Perform health check before starting
+            health = self.health_check()
+            if health["status"] != "healthy":
+                logger.warning(f"Server health check shows issues: {health['issues']}")
+
+            # Run the FastMCP server
+            logger.info("Starting FastMCP server...")
+            self._mcp.run()
+
+        except KeyboardInterrupt:
+            logger.info("Server stopped by user")
+        except Exception as e:
+            logger.error(f"Server error: {e}")
+
+            # Record error if aggregator is enabled
+            if self.error_aggregator:
+                error = MCPError(
+                    f"Server startup/runtime error: {str(e)}",
+                    error_code=MCPErrorCode.SERVER_UNAVAILABLE,
+                    cause=e,
+                )
+                self.error_aggregator.record_error(error)
+
+            raise
+        finally:
+            logger.info("Shutting down enhanced MCP server...")
+
+            # Clean up active sessions
+            if self._active_sessions:
+                logger.info(f"Terminating {len(self._active_sessions)} active sessions")
+                self._active_sessions.clear()
+
+            # Log final stats
+            if self.metrics and self.metrics.enabled:
+                final_stats = self.get_server_stats()
+                logger.info(
+                    f"Final server statistics: {final_stats.get('metrics', {})}"
+                )
+
+            self._running = False
+            logger.info(f"Enhanced MCP server '{self.name}' stopped")
+
+    async def run_stdio(self):
+        """Run the server using stdio transport for testing."""
+        if self._mcp is None:
+            self._init_mcp()
+
+        # For testing, we'll implement a simple stdio server
+        import json
+        import sys
+
+        logger.info(f"Starting MCP server '{self.name}' in stdio mode")
+        self._running = True
+
+        try:
+            while self._running:
+                # Read JSON-RPC request from stdin
+                line = sys.stdin.readline()
+                if not line:
+                    break
+
+                try:
+                    request = json.loads(line.strip())
+
+                    # Handle different request types
+                    if request.get("method") == "tools/list":
+                        # Return list of tools
+                        tools = []
+                        for name, info in self._tool_registry.items():
+                            if not info.get("disabled", False):
+                                tools.append(
+                                    {
+                                        "name": name,
+                                        "description": info.get("description", ""),
+                                        "inputSchema": info.get("input_schema", {}),
+                                    }
+                                )
+
+                        response = {"id": request.get("id"), "result": {"tools": tools}}
+
+                    elif request.get("method") == "tools/call":
+                        # Call a tool
+                        params = request.get("params", {})
+                        tool_name = params.get("name")
+                        arguments = params.get("arguments", {})
+
+                        if tool_name in self._tool_registry:
+                            handler = self._tool_registry[tool_name]["handler"]
+                            try:
+                                # Execute tool
+                                if asyncio.iscoroutinefunction(handler):
+                                    result = await handler(**arguments)
+                                else:
+                                    result = handler(**arguments)
+
+                                response = {
+                                    "id": request.get("id"),
+                                    "result": {
+                                        "content": [
+                                            {"type": "text", "text": str(result)}
+                                        ]
+                                    },
+                                }
+                            except Exception as e:
+                                response = {
+                                    "id": request.get("id"),
+                                    "error": {"code": -32603, "message": str(e)},
+                                }
+                        else:
+                            response = {
+                                "id": request.get("id"),
+                                "error": {
+                                    "code": -32601,
+                                    "message": f"Tool not found: {tool_name}",
+                                },
+                            }
+
+                    else:
+                        # Unknown method
+                        response = {
+                            "id": request.get("id"),
+                            "error": {
+                                "code": -32601,
+                                "message": f"Method not found: {request.get('method')}",
+                            },
+                        }
+
+                    # Write response to stdout
+                    sys.stdout.write(json.dumps(response) + "\n")
+                    sys.stdout.flush()
+
+                except json.JSONDecodeError:
+                    # Invalid JSON
+                    error_response = {
+                        "id": None,
+                        "error": {"code": -32700, "message": "Parse error"},
+                    }
+                    sys.stdout.write(json.dumps(error_response) + "\n")
+                    sys.stdout.flush()
+
+        except KeyboardInterrupt:
+            logger.info("Server stopped by user")
+        except Exception as e:
+            logger.error(f"Server error: {e}")
+            raise
+        finally:
+            self._running = False
+
+
+class SimpleMCPServer(MCPServerBase):
+    """Simple MCP Server for prototyping and development.
+
+    This is a lightweight version of MCPServer without authentication,
+    metrics, caching, or other production features. Perfect for:
+    - Quick prototyping
+    - Development and testing
+    - Simple use cases without advanced features
+
+    Example:
+        >>> server = SimpleMCPServer("my-prototype")
+        >>> @server.tool()
+        ... def hello(name: str) -> str:
+        ...     return f"Hello, {name}!"
+        >>> server.run()
+    """
+
+    def __init__(self, name: str, description: str = None):
+        """Initialize simple MCP server.
+
+        Args:
+            name: Server name
+            description: Server description
+        """
+        super().__init__(name, description)
+
+        # Disable all advanced features for simplicity
+        self.enable_cache = False
+        self.enable_metrics = False
+        self.enable_http_transport = False
+        self.rate_limit_config = None
+        self.circuit_breaker_config = None
+        self.auth_provider = None
+
+        # Simple in-memory storage
+        self._simple_tools = {}
+        self._simple_resources = {}
+        self._simple_prompts = {}
+
+        logger.info(f"SimpleMCPServer '{name}' initialized for prototyping")
+
+    def setup(self):
+        """Setup method - no additional setup needed for SimpleMCPServer."""
+        pass
+
+    def tool(self, description: str = None):
+        """Register a simple tool (no auth, caching, or metrics).
+
+        Args:
+            description: Tool description
+
+        Returns:
+            Decorator function
+        """
+
+        def decorator(func):
+            # Initialize MCP if needed
+            if self._mcp is None:
+                self._init_mcp()
+
+            tool_name = func.__name__
+            self._simple_tools[tool_name] = {
+                "function": func,
+                "description": description or f"Tool: {tool_name}",
+                "created_at": time.time(),
+            }
+
+            # Register with FastMCP
+            self._mcp.tool(description or f"Tool: {tool_name}")(func)
+
+            logger.debug(f"SimpleMCPServer: Registered tool '{tool_name}'")
+            return func
+
+        return decorator
+
+    def resource(self, uri: str, description: str = None):
+        """Register a simple resource.
+
+        Args:
+            uri: Resource URI
+            description: Resource description
+
+        Returns:
+            Decorator function
+        """
+
+        def decorator(func):
+            # Initialize MCP if needed
+            if self._mcp is None:
+                self._init_mcp()
+
+            self._simple_resources[uri] = {
+                "function": func,
+                "description": description or f"Resource: {uri}",
+                "created_at": time.time(),
+            }
+
+            # Register with FastMCP
+            self._mcp.resource(uri, description or f"Resource: {uri}")(func)
+
+            logger.debug(f"SimpleMCPServer: Registered resource '{uri}'")
+            return func
+
+        return decorator
+
+    def get_stats(self) -> dict:
+        """Get simple server statistics.
+
+        Returns:
+            Dictionary with basic stats
+        """
+        return {
+            "server_name": self.name,
+            "server_type": "SimpleMCPServer",
+            "tools_count": len(self._simple_tools),
+            "resources_count": len(self._simple_resources),
+            "prompts_count": len(self._simple_prompts),
+            "features": {
+                "authentication": False,
+                "caching": False,
+                "metrics": False,
+                "rate_limiting": False,
+                "circuit_breaker": False,
+            },
+        }
+
+
+# Note: EnhancedMCPServer alias removed - use MCPServer directly