kailash 0.1.5__py3-none-any.whl → 0.2.1__py3-none-any.whl

kailash/mcp/__init__.py

@@ -0,0 +1,53 @@
+"""Model Context Protocol (MCP) Service Layer.
+
+This module provides MCP client and server functionality as services for the
+Kailash SDK. MCP capabilities are integrated into LLM agents as built-in
+features rather than standalone nodes, enabling seamless tool discovery,
+resource access, and prompt templating.
+
+Design Philosophy:
+    Provides non-intrusive MCP integration that enhances node capabilities
+    without changing the core workflow execution model. MCP services operate
+    as background capabilities that nodes can leverage transparently.
+
+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
+
+Upstream Dependencies:
+    - Official Anthropic MCP Python SDK for protocol implementation
+    - FastMCP framework for server creation and management
+    - AsyncIO for asynchronous server/client communication
+
+Downstream Consumers:
+    - LLMAgentNode for AI model tool integration
+    - Workflow nodes requiring external tool access
+    - Custom nodes needing MCP server capabilities
+
+Examples:
+    Basic MCP client usage:
+
+    >>> from kailash.mcp 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")
+    >>> @server.tool()
+    ... def calculate(a: int, b: int) -> int:
+    ...     return a + b
+    >>> server.start()
+"""
+
+from .client import MCPClient
+from .server import MCPServer, SimpleMCPServer
+
+__all__ = [
+    "MCPClient",
+    "MCPServer",
+    "SimpleMCPServer",
+]

kailash/mcp/__main__.py

@@ -0,0 +1,13 @@
+"""
+Entry point for running MCP servers as modules.
+
+Usage:
+    python -m kailash.mcp.ai_registry_server
+"""
+
+import asyncio
+
+from .ai_registry_server import main
+
+if __name__ == "__main__":
+    asyncio.run(main())