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

kailash/__init__.py

@@ -33,7 +33,7 @@ except ImportError:
 # For backward compatibility
 WorkflowGraph = Workflow
 
-__version__ = "0.6.2"
+__version__ = "0.6.3"
 
 __all__ = [
     # Core workflow components

kailash/{mcp → mcp_server}/__init__.py

@@ -12,8 +12,8 @@ Design Philosophy:
 
 Key Components:
     - MCPClient: Connects to MCP servers for tool and resource access
-    - MCPServer: Base framework for creating custom MCP servers
-    - SimpleMCPServer: Quick server creation for basic use cases
+    - MCPServer: Main production-ready server with all features
+    - MCPServerBase: Abstract base class for custom server implementations
 
 Upstream Dependencies:
     - Official Anthropic MCP Python SDK for protocol implementation
@@ -28,26 +28,31 @@ Downstream Consumers:
 Examples:
     Basic MCP client usage:
 
-    >>> from kailash.mcp import MCPClient
+    >>> from kailash.mcp_server import MCPClient
     >>> client = MCPClient()
     >>> tools = await client.discover_tools(server_config)
     >>> result = await client.call_tool(server_config, "search", {"query": "AI"})
 
     Simple MCP server creation:
 
-    >>> from kailash.mcp import SimpleMCPServer
-    >>> server = SimpleMCPServer("my-tools")
+    >>> from kailash.mcp_server import MCPServer
+    >>> server = MCPServer("my-tools")
     >>> @server.tool()
     ... def calculate(a: int, b: int) -> int:
     ...     return a + b
-    >>> server.start()
+    >>> server.run()
 """
 
 from .client import MCPClient
-from .server_enhanced import MCPServer, SimpleMCPServer
+
+# For backward compatibility
+from .server import EnhancedMCPServer, MCPServer, MCPServerBase, SimpleMCPServer
 
 __all__ = [
     "MCPClient",
     "MCPServer",
+    "MCPServerBase",
+    # Backward compatibility
     "SimpleMCPServer",
+    "EnhancedMCPServer",
 ]

kailash/{mcp → mcp_server}/ai_registry_server.py

@@ -5,7 +5,7 @@ AI Registry MCP Server using Anthropic's Official MCP Python SDK.
 This creates a real MCP server that exposes AI Registry tools following
 the actual Model Context Protocol specification.
 
-Run as: python -m kailash.mcp.ai_registry_server
+Run as: python -m kailash.mcp_server.ai_registry_server
 """
 
 import asyncio
@@ -708,5 +708,5 @@ if __name__ == "__main__":
 
 # For module execution
 def run_server():
-    """Entry point for python -m kailash.mcp.ai_registry_server"""
+    """Entry point for python -m kailash.mcp_server.ai_registry_server"""
     asyncio.run(main())

kailash/{mcp/server_enhanced.py → mcp_server/server.py}

@@ -1,16 +1,39 @@
 """
-Enhanced MCP Server with production-ready capabilities.
-
-This module provides an enhanced MCP server that includes caching, configuration,
-metrics, and other production features by default, while maintaining compatibility
-with the official Anthropic FastMCP framework.
+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.
+
+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()
 """
 
 import asyncio
 import functools
 import logging
+from abc import ABC, abstractmethod
+from collections.abc import Callable
 from pathlib import Path
-from typing import Any, Callable, Dict, Optional, TypeVar, Union
+from typing import Any, Dict, Optional, TypeVar, Union
 
 from .utils import CacheManager, ConfigManager, MetricsCollector, format_response
 
@@ -19,18 +42,199 @@ 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:
+            from mcp.server import FastMCP
+
+            self._mcp = FastMCP(self.name)
+        except ImportError:
+            logger.error(
+                "FastMCP not available. Install with: pip install 'mcp[server]'"
+            )
+            raise
+
+    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 EnhancedMCPServer:
     """
-    Production-ready MCP server with enhanced capabilities.
+    Production-ready MCP server (available as SimpleMCPServer).
 
-    Features included by default:
-    - Caching with TTL support
+    This is the main concrete MCP server implementation with all production
+    features available. Features can be enabled/disabled as needed.
+
+    Features available:
+    - Caching with TTL support (enable_cache=True)
+    - Metrics collection and monitoring (enable_metrics=True)
+    - Response formatting utilities (enable_formatting=True)
     - Hierarchical configuration management
-    - Metrics collection and monitoring
-    - Response formatting utilities
     - Error handling and logging
 
-    All features can be disabled if not needed.
+    Examples:
+        Basic usage (recommended):
+        >>> from kailash.mcp_server import MCPServer
+        >>> server = MCPServer("my-server")
+        >>> @server.tool()
+        ... def search(query: str) -> dict:
+        ...     return {"results": f"Found: {query}"}
+        >>> server.run()
+
+        With production features enabled:
+        >>> 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()
     """
 
     def __init__(
@@ -104,15 +308,21 @@ class EnhancedMCPServer:
             return
 
         try:
-            from mcp.server.fastmcp import FastMCP
+            # Now we can safely import from external mcp.server (no namespace collision)
+            from mcp.server import FastMCP
 
             self._mcp = FastMCP(self.name)
             logger.info(f"Initialized FastMCP server: {self.name}")
-        except ImportError:
+        except ImportError as e:
+            logger.error(
+                f"FastMCP import failed with: {e}. Details: {type(e).__name__}"
+            )
             logger.error(
                 "FastMCP not available. Install with: pip install 'mcp[server]'"
             )
-            raise
+            raise ImportError(
+                "FastMCP not available. Install with: pip install 'mcp[server]'"
+            ) from e
 
     def tool(
         self,
@@ -413,37 +623,10 @@ class EnhancedMCPServer:
             self._running = False
 
 
-# For backward compatibility, make EnhancedMCPServer the default MCPServer
+# Clean public API design:
+# - MCPServerBase: Abstract base for custom implementations (e.g., AIRegistryServer)
+# - MCPServer: Main concrete server with all production features
+# - SimpleMCPServer: Alias for backward compatibility
+# - EnhancedMCPServer: Alias for backward compatibility
 MCPServer = EnhancedMCPServer
-
-
-class SimpleMCPServer(EnhancedMCPServer):
-    """
-    Simplified MCP server with minimal configuration.
-
-    This inherits all enhanced capabilities but disables some features
-    by default for simpler use cases.
-    """
-
-    def __init__(self, name: str, description: str = ""):
-        """
-        Initialize simple MCP server.
-
-        Args:
-            name: Server name
-            description: Server description
-        """
-        # Initialize with some features disabled for simplicity
-        super().__init__(
-            name=name,
-            enable_cache=False,  # Disable cache by default
-            enable_metrics=False,  # Disable metrics by default
-            enable_formatting=True,  # Keep formatting for better output
-        )
-
-        self.description = description
-
-        # Update config for simple use
-        self.config.update(
-            {"server": {"name": name, "description": description, "version": "1.0.0"}}
-        )
+SimpleMCPServer = EnhancedMCPServer

kailash/{mcp → mcp_server}/servers/ai_registry.py

@@ -9,10 +9,10 @@ import os
 from pathlib import Path
 from typing import Any
 
-from kailash.mcp.server import MCPServer
+from kailash.mcp_server.server import MCPServerBase
 
 
-class AIRegistryServer(MCPServer):
+class AIRegistryServer(MCPServerBase):
     """MCP server for AI use case registry.
 
     Provides tools and resources for exploring AI use cases from

kailash/{mcp → mcp_server}/utils/__init__.py

@@ -10,12 +10,7 @@ This module provides production-ready utilities for MCP servers including:
 
 from .cache import CacheManager, LRUCache, cached_query
 from .config import ConfigManager
-from .formatters import (
-    format_response,
-    json_formatter,
-    markdown_formatter,
-    search_formatter,
-)
+from .formatters import format_response, json_formatter, markdown_formatter, search_formatter
 from .metrics import MetricsCollector
 
 __all__ = [

kailash/middleware/mcp/client_integration.py

@@ -20,7 +20,7 @@ from kailash.workflow.builder import WorkflowBuilder
 
 # Import existing Kailash MCP components
 try:
-    from kailash.mcp import MCPClient
+    from kailash.mcp_server import MCPClient
 
     _KAILASH_MCP_AVAILABLE = True
 except ImportError:

kailash/middleware/mcp/enhanced_server.py

@@ -24,8 +24,8 @@ from kailash.workflow.builder import WorkflowBuilder
 
 # Import existing Kailash MCP components
 try:
-    from kailash.mcp import MCPServer, SimpleMCPServer
-    from kailash.mcp.utils import CacheManager, ConfigManager, MetricsCollector
+    from kailash.mcp_server import MCPServer, SimpleMCPServer
+    from kailash.mcp_server.utils import CacheManager, ConfigManager, MetricsCollector
 
     _KAILASH_MCP_AVAILABLE = True
 except ImportError:

kailash/nodes/ai/iterative_llm_agent.py

@@ -457,7 +457,7 @@ class IterativeLLMAgentNode(LLMAgentNode):
         try:
             # Ensure MCP client is initialized
             if not hasattr(self, "_mcp_client"):
-                from kailash.mcp import MCPClient
+                from kailash.mcp_server import MCPClient
 
                 self._mcp_client = MCPClient()
 

kailash/nodes/ai/llm_agent.py

@@ -985,7 +985,7 @@ class LLMAgentNode(Node):
                 import asyncio
                 from datetime import datetime
 
-                from kailash.mcp import MCPClient
+                from kailash.mcp_server import MCPClient
 
                 # Initialize MCP client if not already done
                 if not hasattr(self, "_mcp_client"):
@@ -1182,7 +1182,7 @@ class LLMAgentNode(Node):
 
         if use_real_mcp:
             try:
-                from kailash.mcp import MCPClient
+                from kailash.mcp_server import MCPClient
 
                 # Initialize MCP client if not already done
                 if not hasattr(self, "_mcp_client"):
@@ -1767,7 +1767,7 @@ class LLMAgentNode(Node):
         server_config = mcp_tool.get("function", {}).get("mcp_server_config", {})
 
         try:
-            from kailash.mcp import MCPClient
+            from kailash.mcp_server import MCPClient
 
             # Initialize MCP client if not already done
             if not hasattr(self, "_mcp_client"):

{kailash-0.6.2.dist-info → kailash-0.6.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kailash
-Version: 0.6.2
+Version: 0.6.3
 Summary: Python SDK for the Kailash container-node architecture
 Home-page: https://github.com/integrum/kailash-python-sdk
 Author: Integrum
@@ -21,7 +21,7 @@ Requires-Dist: matplotlib>=3.5
 Requires-Dist: pyyaml>=6.0
 Requires-Dist: click>=8.0
 Requires-Dist: pytest>=8.3.5
-Requires-Dist: mcp[cli]>=1.9.2
+Requires-Dist: mcp[cli,server]>=1.9.2
 Requires-Dist: pandas>=2.2.3
 Requires-Dist: numpy>=2.2.5
 Requires-Dist: scipy>=1.15.3

{kailash-0.6.2.dist-info → kailash-0.6.3.dist-info}/RECORD

@@ -1,4 +1,4 @@
-kailash/__init__.py,sha256=ZV1hjURaVjSCN-pI1JcNDbpBxyaYv5DYgnhgupZRzSc,1724
+kailash/__init__.py,sha256=8W7LrVttYlpEMHu8_N4a5CilyUWMORrc8rnCbaAJ8HU,1724
 kailash/__main__.py,sha256=vr7TVE5o16V6LsTmRFKG6RDKUXHpIWYdZ6Dok2HkHnI,198
 kailash/access_control.py,sha256=2ctdRFeSeu-d7DU04Aovxh6Rt_4t3IyQfkKEjTeQiMM,25519
 kailash/access_control_abac.py,sha256=FPfa_8PuDP3AxTjdWfiH3ntwWO8NodA0py9W8SE5dno,30263
@@ -43,18 +43,17 @@ kailash/gateway/api.py,sha256=xpK8PIamsqQPpKAJwacyV7RA_Snjv2pc_0ljnnU9Oy4,9534
 kailash/gateway/enhanced_gateway.py,sha256=IlN1XV01FQrF4rGcq_z9LE4uUHAAAQoVsRNToXENen0,13399
 kailash/gateway/resource_resolver.py,sha256=IC1dceiKfjfUWToYCIBcrUapuR3LlDG6RJ4o7haLY10,7746
 kailash/gateway/security.py,sha256=kf4Quf6u7dqhs80fQQ982eHbRb4weDKG0DaYNeKntT4,7557
-kailash/mcp/__init__.py,sha256=jQHP7EVT126QXmi0TqR1mU3QNrUeEB4oIC4sD4B2a8c,1813
-kailash/mcp/ai_registry_server.py,sha256=9pOzJnJFpxJnZPfLo2QvVZ5yvAq5IRqzXPbQL1rL1ns,28104
-kailash/mcp/client.py,sha256=sTouSiuiu1nbahMXSWcP8-mr1k7cqdBCzSxm8G7le-s,16058
-kailash/mcp/client_new.py,sha256=YU671JvAM0uvuX0uhGZCIKI8co3fqz0cs6HqLZ59Xyo,10285
-kailash/mcp/server.py,sha256=aWU0DHj89FN_eEgH6aVjA05WjthugrXMHdPlgwJ6kZg,8246
-kailash/mcp/server_enhanced.py,sha256=lRIDSNs0c8urMq_SURwi7eBhTWKa-rq2FAB8xZf9CwI,14724
-kailash/mcp/servers/ai_registry.py,sha256=7k17ld0DUQYL476N5EbiNpPy8D0HqPRhrl8Wk1ajjj8,9992
-kailash/mcp/utils/__init__.py,sha256=R20N-iiKXUPxc9MOh6vPO1vIfkPmwhEQ5KNFgGd4xSs,771
-kailash/mcp/utils/cache.py,sha256=dLEseovPaXL4lRzMSw7tqd3tJHwnWfhdZ-HKGyScJXI,8414
-kailash/mcp/utils/config.py,sha256=DyZxgdy3vqI5pwhQ_E-42mhueVGNHiuOtTUOrM9HC_U,8124
-kailash/mcp/utils/formatters.py,sha256=D-2j1nvmprApiUI13HWY-L2_WPSAcJDtVdHcshAuOdo,9740
-kailash/mcp/utils/metrics.py,sha256=MNUjWGQyq1EGdeqzAKCCZJNgcWHOyaYAV8MlS2cb-4k,13754
+kailash/mcp_server/__init__.py,sha256=ETEeaRTqj2xeX07MyaRgcPiaomdLoJ24iNmRbRjJQok,1953
+kailash/mcp_server/ai_registry_server.py,sha256=BkkZcohRyGlSbiUkpRTaIA6M_yJUgtgtMda-dvbT86s,28118
+kailash/mcp_server/client.py,sha256=sTouSiuiu1nbahMXSWcP8-mr1k7cqdBCzSxm8G7le-s,16058
+kailash/mcp_server/client_new.py,sha256=YU671JvAM0uvuX0uhGZCIKI8co3fqz0cs6HqLZ59Xyo,10285
+kailash/mcp_server/server.py,sha256=z-72J5MKq98_j7zwo0HJMbMI9bp5yuNpyfIzD9N23yw,20600
+kailash/mcp_server/servers/ai_registry.py,sha256=IdF_keUuJlMsvjLjSAykxxbm46K4qA7eCj7T-lYSrzk,10007
+kailash/mcp_server/utils/__init__.py,sha256=VjGFpvugTYUxoNgtwVEKkmrkVS3iZ_kIbP4jR1PRM6E,750
+kailash/mcp_server/utils/cache.py,sha256=dLEseovPaXL4lRzMSw7tqd3tJHwnWfhdZ-HKGyScJXI,8414
+kailash/mcp_server/utils/config.py,sha256=DyZxgdy3vqI5pwhQ_E-42mhueVGNHiuOtTUOrM9HC_U,8124
+kailash/mcp_server/utils/formatters.py,sha256=D-2j1nvmprApiUI13HWY-L2_WPSAcJDtVdHcshAuOdo,9740
+kailash/mcp_server/utils/metrics.py,sha256=MNUjWGQyq1EGdeqzAKCCZJNgcWHOyaYAV8MlS2cb-4k,13754
 kailash/middleware/__init__.py,sha256=ZGo0qujL-qWn82nIrojY96N1rMPTWFKHumW6CGGpb4Y,10409
 kailash/middleware/auth/__init__.py,sha256=VkKM8H-zVFx2PLGL7kyxE2IfSiV1HiwveSysbmxMcg0,2077
 kailash/middleware/auth/access_control.py,sha256=2FwQjp_fZm2fg-V_CIgAN69GGL9YwyZwsbQis0hbW7I,14848
@@ -87,8 +86,8 @@ kailash/middleware/gateway/durable_gateway.py,sha256=EsIgMNxS_no2W40AXDyE7FmVdnG
 kailash/middleware/gateway/durable_request.py,sha256=SCnp-bF0tQX9oahr9reqcZjJ_YhyJkeYYl-un9rJ6lo,15437
 kailash/middleware/gateway/event_store.py,sha256=A3Kh2MhVVPbXWvjeo550SqEGPiJYyspAfu6Gv7UZzo4,16131
 kailash/middleware/mcp/__init__.py,sha256=EdZB8zOMSBEEmudRzs8ksz9QZJYWQMEx7Tm1MOwIWnI,922
-kailash/middleware/mcp/client_integration.py,sha256=opzhB5TUts_ND8gARXh93nKCc1u4kwo6SqNMMWqMcSU,18258
-kailash/middleware/mcp/enhanced_server.py,sha256=XUjQt0KyRX207FVYAFenkYHa_K8FWWKquROgXQWkoOQ,18453
+kailash/middleware/mcp/client_integration.py,sha256=dY1RmX-g5E6JzUFuWxk7viuOYIh8bMwoUSvHQMVEsYk,18265
+kailash/middleware/mcp/enhanced_server.py,sha256=vVd7WE95c-5861VtKfmTZ91iUWFRw-aKEKwHlu6edQo,18467
 kailash/nodes/__init__.py,sha256=E6CEp1ooq4GgFhKtwVAczOhPt5N3x-AVQ-R0n3_IFyA,936
 kailash/nodes/base.py,sha256=Fu9c2I5k_Ke192y4fj2NVhf-Y_I0nPr0sDE1zMSRCJY,55417
 kailash/nodes/base_async.py,sha256=mpntaeFMbUYLIyTvjsb221mXckx_H2dGX2LhxeKhhfA,6569
@@ -112,8 +111,8 @@ kailash/nodes/ai/agents.py,sha256=CRA3cdapQjpuvOniXUh6ZVWAlRxUIepVw1BROW6QzdY,20
 kailash/nodes/ai/ai_providers.py,sha256=XeDIaYH7PrX8frWCVV6CYFNQXFjHX_61T3eR1yullqs,65355
 kailash/nodes/ai/embedding_generator.py,sha256=rsos3B6oWrgGTMIbwSWIBzGH9kq3SFVD_-bEDrujBRs,31860
 kailash/nodes/ai/intelligent_agent_orchestrator.py,sha256=xw44C-CkcNH3SVmEJ49o4oNV3o4ZqjLE9aLpggwoIXs,83021
-kailash/nodes/ai/iterative_llm_agent.py,sha256=pv54W_YDfDPDl6DJf0ul9_rs2mL2kE_C59sSAJ4CRn8,52884
-kailash/nodes/ai/llm_agent.py,sha256=-E95jRYZMJzZz7Y4jh4HYHmx7Hoid5ZvZqcEP-OsHqE,82177
+kailash/nodes/ai/iterative_llm_agent.py,sha256=14SxTRGGGYRqNC0SFMaKAFplUcsQXAVzXKcjri_0oWM,52891
+kailash/nodes/ai/llm_agent.py,sha256=0N5NU2HLwbh2_zFxrgwq2FlkVS6fifpEcMSI0BGX7ZQ,82198
 kailash/nodes/ai/models.py,sha256=wsEeUTuegy87mnLtKgSTg7ggCXvC1n3MsL-iZ4qujHs,16393
 kailash/nodes/ai/self_organizing.py,sha256=M7yCLkN4I1JCNU7PWuwrqwQSlaG9MJVxYIR44TV52MM,62877
 kailash/nodes/ai/vision_utils.py,sha256=OHD9cVH_mq0WpJyQkNTj_mpipIVWfSV_bF9eA6CdyeA,4166
@@ -271,9 +270,9 @@ kailash/workflow/state.py,sha256=UTZxs5-Ona6uvBhx1__i6-RX8gB4qazkBIWE7uyRmWQ,760
 kailash/workflow/templates.py,sha256=98EN5H4fO9b4xeczk20Hu5L8hNcAuRQNGayT6vnZYCw,48540
 kailash/workflow/validation.py,sha256=JIbIajWVIaWHSvWtgZ4WUVJaBaUOCz5B9cyTwM--dL4,33060
 kailash/workflow/visualization.py,sha256=ICMWCWqh5fOQ7eJygbvu2PMWHxe-H5_0epwdZuz8cMw,19737
-kailash-0.6.2.dist-info/licenses/LICENSE,sha256=Axe6g7bTrJkToK9h9j2SpRUKKNaDZDCo2lQ2zPxCE6s,1065
-kailash-0.6.2.dist-info/METADATA,sha256=lksSZ5ngxdJDTik-awxRlqV87v3BoZe82PPS6biRnww,25925
-kailash-0.6.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-kailash-0.6.2.dist-info/entry_points.txt,sha256=M_q3b8PG5W4XbhSgESzIJjh3_4OBKtZFYFsOdkr2vO4,45
-kailash-0.6.2.dist-info/top_level.txt,sha256=z7GzH2mxl66498pVf5HKwo5wwfPtt9Aq95uZUpH6JV0,8
-kailash-0.6.2.dist-info/RECORD,,
+kailash-0.6.3.dist-info/licenses/LICENSE,sha256=Axe6g7bTrJkToK9h9j2SpRUKKNaDZDCo2lQ2zPxCE6s,1065
+kailash-0.6.3.dist-info/METADATA,sha256=PH0D-ORbBHT7eJjuIEMbrcr1CWNlQpY36Ncy4jGJ-wg,25932
+kailash-0.6.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+kailash-0.6.3.dist-info/entry_points.txt,sha256=M_q3b8PG5W4XbhSgESzIJjh3_4OBKtZFYFsOdkr2vO4,45
+kailash-0.6.3.dist-info/top_level.txt,sha256=z7GzH2mxl66498pVf5HKwo5wwfPtt9Aq95uZUpH6JV0,8
+kailash-0.6.3.dist-info/RECORD,,

kailash/mcp/server.py

@@ -1,292 +0,0 @@
-"""MCP Server Framework using official Anthropic SDK.
-
-This module provides a comprehensive framework for creating MCP servers using
-the official FastMCP framework from Anthropic. Servers run as long-lived
-services that expose tools, resources, and prompts to MCP clients, enabling
-dynamic capability extension for AI workflows.
-
-Note:
-    This module requires the FastMCP framework to be installed.
-    Install with: pip install 'mcp[server]'
-
-Examples:
-    Basic server with tools:
-
-    >>> from kailash.mcp.server import MCPServer
-    >>> class MyServer(MCPServer):
-    ...     def setup(self):
-    ...         @self.add_tool()
-    ...         def calculate(a: int, b: int) -> int:
-    ...             return a + b
-    >>> server = MyServer("calculator", port=8080)
-    >>> server.start()
-
-    Quick server creation:
-
-    >>> from kailash.mcp.server import SimpleMCPServer
-    >>> server = SimpleMCPServer("my-tools")
-    >>> @server.tool()
-    ... def search(query: str) -> list:
-    ...     return [f"Result for {query}"]
-    >>> server.start()
-"""
-
-import logging
-from abc import ABC, abstractmethod
-from collections.abc import Callable
-
-logger = logging.getLogger(__name__)
-
-
-class MCPServer(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(MCPServer):
-        ...     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:
-            from mcp.server.fastmcp import FastMCP
-
-            self._mcp = FastMCP(self.name)
-        except ImportError:
-            logger.error(
-                "FastMCP not available. Install with: pip install 'mcp[server]'"
-            )
-            raise
-
-    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 SimpleMCPServer(MCPServer):
-    """Simple MCP server for basic use cases.
-
-    This provides an easy way to create MCP servers without subclassing.
-
-    Examples:
-        >>> server = SimpleMCPServer("my-server")
-        >>> @server.tool()
-        ... def add(a: int, b: int) -> int:
-        ...     return a + b
-        >>> server.start()
-    """
-
-    def __init__(self, name: str, port: int = 8080, host: str = "localhost"):
-        """Initialize the simple MCP server.
-
-        Args:
-            name: Name of the server.
-            port: Port to listen on (default: 8080).
-            host: Host to bind to (default: "localhost").
-        """
-        super().__init__(name, port, host)
-        self._tools = []
-        self._resources = []
-        self._prompts = []
-
-    def tool(self):
-        """Decorator to add a tool.
-
-        Returns:
-            Function decorator for registering tools.
-        """
-
-        def decorator(func):
-            self._tools.append(func)
-            return func
-
-        return decorator
-
-    def resource(self, uri: str):
-        """Decorator to add a resource.
-
-        Args:
-            uri: URI pattern for the resource.
-
-        Returns:
-            Function decorator for registering resources.
-        """
-
-        def decorator(func):
-            self._resources.append((uri, func))
-            return func
-
-        return decorator
-
-    def prompt(self, name: str):
-        """Decorator to add a prompt.
-
-        Args:
-            name: Name of the prompt.
-
-        Returns:
-            Function decorator for registering prompts.
-        """
-
-        def decorator(func):
-            self._prompts.append((name, func))
-            return func
-
-        return decorator
-
-    def setup(self):
-        """Setup the server with registered components.
-
-        Registers all tools, resources, and prompts that were decorated
-        before calling start().
-        """
-        # Register all tools
-        for tool_func in self._tools:
-            self.add_tool()(tool_func)
-
-        # Register all resources
-        for uri, resource_func in self._resources:
-            self.add_resource(uri)(resource_func)
-
-        # Register all prompts
-        for name, prompt_func in self._prompts:
-            self.add_prompt(name)(prompt_func)