sunholo 0.143.16__py3-none-any.whl → 0.144.0__py3-none-any.whl

sunholo/mcp/cli.py

@@ -1,338 +1,22 @@
-import os
-import asyncio
-from typing import Any, Sequence
-from functools import lru_cache
-import subprocess
-
-try:
-    from mcp.server import Server
-    from mcp.server.stdio import stdio_server
-    from mcp.types import (
-        Resource,
-        Tool,
-        TextContent,
-        ImageContent,
-        EmbeddedResource,
-    )
-
-    from rich import print
-    from ..cli.sun_rich import console
-except ImportError:
-    Server = None
-    console = None
-
-from pydantic import AnyUrl
-
-# Configure logging
-from ..custom_logging import setup_logging
-logger = setup_logging("sunholo-mcp")
-
-class SunholoMCPServer:
-    def __init__(self):
-        logger.info("Initializing Sunholo MCP Server")
-
-        if Server is None:
-            raise ImportError("SunholoMCPServer requires `sunholo[anthropic]` to be installed")
-        
-        self.server = Server("sunholo-mcp-server")
-        self.server.onerror = self.handle_error
-
-        self.setup_handlers()
-
-    def handle_error(self, error: Exception):
-        """Handle server errors"""
-        logger.error(f"MCP Server error: {error}", exc_info=True)
-
-    def setup_handlers(self):
-        """Set up all the MCP protocol handlers"""
-        self.setup_resource_handlers()
-        self.setup_tool_handlers()
-
-    def setup_resource_handlers(self):
-        """Configure resource-related handlers"""
-        
-        @self.server.list_resources()
-        async def list_resources() -> list[Resource]:
-            """List available Sunholo resources"""
-            return [
-                Resource(
-                    uri="sunholo://vacs/list",
-                    name="Available Sunholo VACs",
-                    mimeType="application/json",
-                    description="List of available Virtual Agent Computers"
-                )
-            ]
-
-        @self.server.read_resource()
-        async def read_resource(uri: AnyUrl) -> str:
-            """Read Sunholo resources based on URI"""
-            logger.info(f"{uri} available")
-            if str(uri) == "sunholo://vacs/list":
-                try:
-                    # Execute sunholo vac list command
-                    result = subprocess.run(
-                        ["sunholo", "vac", "list"],
-                        capture_output=True,
-                        text=True
-                    )
-                    return result.stdout
-                except subprocess.CalledProcessError as e:
-                    raise RuntimeError(f"Failed to list VACs: {str(e)}")
-                except Exception as e:
-                    raise RuntimeError(f"Error accessing Sunholo: {str(e)}")
-
-            raise ValueError(f"Unknown resource: {uri}")
-
-    def setup_tool_handlers(self):
-        """Configure tool-related handlers"""
-
-        @self.server.list_tools()
-        async def list_tools() -> list[Tool]:
-            """List available Sunholo tools"""
-            return [
-                Tool(
-                    name="chat_with_vac",
-                    description="Chat with a specific Sunholo VAC",
-                    inputSchema={
-                        "type": "object",
-                        "properties": {
-                            "vac_name": {
-                                "type": "string",
-                                "description": "Name of the VAC to chat with"
-                            },
-                            "message": {
-                                "type": "string",
-                                "description": "Message to send to the VAC"
-                            },
-                        },
-                        "required": ["vac_name", "message"]
-                    }
-                ),
-                Tool(
-                    name="list_configs",
-                    description="List Sunholo configurations",
-                    inputSchema={
-                        "type": "object",
-                        "properties": {
-                            "kind": {
-                                "type": "string",
-                                "description": "Filter configurations by kind e.g. vacConfig"
-                            },
-                            "vac": {
-                                "type": "string", 
-                                "description": "Filter configurations by VAC name"
-                            },
-                            "validate": {
-                                "type": "boolean",
-                                "description": "Validate the configuration files"
-                            }
-                        }
-                    }
-                ),
-                Tool(
-                    name="embed_content",
-                    description="Embed content in a VAC's vector store",
-                    inputSchema={
-                        "type": "object",
-                        "properties": {
-                            "vac_name": {
-                                "type": "string",
-                                "description": "Name of the VAC to embed content for"
-                            },
-                            "content": {
-                                "type": "string",
-                                "description": "Content to embed"
-                            },
-                            "local_chunks": {
-                                "type": "boolean",
-                                "description": "Whether to process chunks locally",
-                                "default": False
-                            }
-                        },
-                        "required": ["vac_name", "content"]
-                    }
-                )
-            ]
-
-        @self.server.call_tool()
-        async def call_tool(
-            name: str,
-            arguments: Any
-        ) -> Sequence[TextContent | ImageContent | EmbeddedResource]:
-            """Handle tool calls for Sunholo interactions"""
-            
-            if name == "chat_with_vac":
-                if not isinstance(arguments, dict):
-                    raise ValueError("Invalid arguments format")
-
-                vac_name = arguments.get("vac_name")
-                message = arguments.get("message")
-
-                if not vac_name or not message:
-                    raise ValueError("Missing required arguments")
-
-                try:
-                    cmd = ["sunholo", "vac", "chat", vac_name, message]
-                    cmd.append("--headless")
-
-                    result = subprocess.run(
-                        cmd,
-                        capture_output=True,
-                        text=True
-                    )
-
-                    return [
-                        TextContent(
-                            type="text",
-                            text=result.stdout
-                        )
-                    ]
-                except subprocess.CalledProcessError as e:
-                    return [
-                        TextContent(
-                            type="text",
-                            text=f"Error chatting with VAC: {e.stderr}"
-                        )
-                    ]
-
-            elif name == "embed_content":
-                if not isinstance(arguments, dict):
-                    raise ValueError("Invalid arguments format")
-
-                vac_name = arguments.get("vac_name")
-                content = arguments.get("content")
-                local_chunks = arguments.get("local_chunks", False)
-
-                if not vac_name or not content:
-                    raise ValueError("Missing required arguments")
-
-                try:
-                    cmd = ["sunholo", "embed", vac_name, content]
-                    if local_chunks:
-                        cmd.append("--local-chunks")
-
-                    result = subprocess.run(
-                        cmd,
-                        capture_output=True,
-                        text=True
-                    )
-
-                    return [
-                        TextContent(
-                            type="text",
-                            text=result.stdout
-                        )
-                    ]
-                except subprocess.CalledProcessError as e:
-                    return [
-                        TextContent(
-                            type="text",
-                            text=f"Error embedding content: {e.stderr}"
-                        )
-                    ]
-            elif name == "list_configs":
-
-                # Build command
-                cmd = ["sunholo", "list-configs"]
-                
-                if arguments.get("kind"):
-                    cmd.extend(["--kind", arguments["kind"]])
-                if arguments.get("vac"):
-                    cmd.extend(["--vac", arguments["vac"]])
-                if arguments.get("validate"):
-                    cmd.append("--validate")
-
-                # Execute sunholo command
-                try:
-                    result = subprocess.run(cmd, capture_output=True, text=True)
-                    return [TextContent(type="text", text=result.stdout)]
-                except subprocess.CalledProcessError as e:
-                    return [TextContent(type="text", text=f"Error: {e.stderr}")]
-
-
-            raise ValueError(f"Unknown tool: {name}")
-
-    async def run(self):
-        """Run the MCP server"""
-        async with stdio_server() as (read_stream, write_stream):
-            await self.server.run(
-                read_stream,
-                write_stream,
-                self.server.create_initialization_options()
-            )
-
-def cli_mcp(args):
-    """CLI handler for the MCP server command"""
-    try:
-
-        if not os.getenv("VAC_CONFIG_FOLDER"):
-            raise ValueError("sunholo configuration folder must be present in config/ or via VAC_CONFIG_FOLDER")
-        
-        # Create and run the MCP server
-        server = SunholoMCPServer()
-        msg = {"message": "Starting Sunholo MCP server..."}
-
-        logger.info(msg)
-        console.print(msg)
-        
-        asyncio.run(server.run())
-
-    except Exception as e:
-        logger.error(f"Error running MCP server: {str(e)}")
-        raise
-
-def cli_mcp_bridge(args):
-    """CLI handler for the MCP bridge command"""
-    try:
-        from .stdio_http_bridge import run_bridge
-        
-        http_url = args.url
-        logger.info(f"Starting MCP stdio-to-HTTP bridge for {http_url}")
-        
-        if console:
-            console.print(f"Starting MCP bridge to {http_url}...")
-        
-        asyncio.run(run_bridge(http_url))
-        
-    except KeyboardInterrupt:
-        logger.info("Bridge stopped by user")
-    except Exception as e:
-        logger.error(f"Error running MCP bridge: {str(e)}")
-        raise
-
-def setup_mcp_subparser(subparsers):
-    """
-    Sets up an argparse subparser for the 'mcp' command 3.
-
-    By default will use configurations within the folder specified by 'VAC_CONFIG_FOLDER'
-
-    Example command:
-    ```bash
-    sunholo mcp
-    ```
-    """
-    mcp_parser = subparsers.add_parser('mcp', 
-                                      help='MCP (Model Context Protocol) commands')
-    
-    # Create subcommands for mcp
-    mcp_subparsers = mcp_parser.add_subparsers(title='mcp commands',
-                                               description='MCP subcommands',
-                                               help='MCP subcommands',
-                                               dest='mcp_command')
-    
-    # mcp server command (default)
-    server_parser = mcp_subparsers.add_parser('server',
-                                             help='Start an Anthropic MCP server that wraps sunholo functionality')
-    server_parser.set_defaults(func=cli_mcp)
-    
-    # mcp bridge command
-    bridge_parser = mcp_subparsers.add_parser('bridge',
-                                             help='Start a stdio-to-HTTP bridge for MCP servers')
-    bridge_parser.add_argument('url',
-                                  nargs='?',
-                                  default='http://127.0.0.1:1956/mcp',
-                                  help='HTTP URL of the MCP server (default: http://127.0.0.1:1956/mcp)')
-    bridge_parser.set_defaults(func=cli_mcp_bridge)
-    
-    # Set default behavior when no subcommand is provided
-    mcp_parser.set_defaults(func=cli_mcp)
+#   Copyright [2024] [Holosun ApS]
+#
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+
+"""
+CLI for MCP server commands using FastMCP.
+"""
+
+# Import the FastMCP implementation
+from .cli_fastmcp import cli_mcp, cli_mcp_bridge, setup_mcp_subparser
+
+__all__ = ['cli_mcp', 'cli_mcp_bridge', 'setup_mcp_subparser']

sunholo/mcp/cli_fastmcp.py

@@ -0,0 +1,201 @@
+import os
+import asyncio
+import subprocess
+from typing import List, Optional
+
+from fastmcp import FastMCP
+
+# Configure logging
+from ..custom_logging import setup_logging
+logger = setup_logging("sunholo-mcp")
+
+# Initialize FastMCP server
+mcp = FastMCP("sunholo-mcp-server")
+
+@mcp.resource("sunholo://vacs/list")
+async def list_vacs() -> str:
+    """
+    List of available Virtual Agent Computers.
+    
+    Returns the list of configured VACs in the system.
+    """
+    try:
+        result = subprocess.run(
+            ["sunholo", "vac", "list"],
+            capture_output=True,
+            text=True,
+            check=True
+        )
+        return result.stdout
+    except subprocess.CalledProcessError as e:
+        raise RuntimeError(f"Failed to list VACs: {str(e)}")
+    except Exception as e:
+        raise RuntimeError(f"Error accessing Sunholo: {str(e)}")
+
+@mcp.tool
+async def chat_with_vac(vac_name: str, message: str, headless: bool = True) -> str:
+    """
+    Chat with a specific Sunholo VAC.
+    
+    Args:
+        vac_name: Name of the VAC to chat with
+        message: Message to send to the VAC
+        headless: Run in headless mode (default: True)
+    
+    Returns:
+        Response from the VAC
+    """
+    try:
+        cmd = ["sunholo", "vac", "chat", vac_name, message]
+        if headless:
+            cmd.append("--headless")
+        
+        result = subprocess.run(
+            cmd,
+            capture_output=True,
+            text=True,
+            check=True
+        )
+        return result.stdout
+    except subprocess.CalledProcessError as e:
+        return f"Error chatting with VAC: {e.stderr}"
+
+@mcp.tool
+async def list_configs(
+    kind: Optional[str] = None,
+    vac: Optional[str] = None,
+    validate: bool = False
+) -> str:
+    """
+    List Sunholo configurations.
+    
+    Args:
+        kind: Filter configurations by kind (e.g., 'vacConfig')
+        vac: Filter configurations by VAC name
+        validate: Validate the configuration files
+    
+    Returns:
+        Configuration listing output
+    """
+    cmd = ["sunholo", "list-configs"]
+    
+    if kind:
+        cmd.extend(["--kind", kind])
+    if vac:
+        cmd.extend(["--vac", vac])
+    if validate:
+        cmd.append("--validate")
+    
+    try:
+        result = subprocess.run(cmd, capture_output=True, text=True, check=True)
+        return result.stdout
+    except subprocess.CalledProcessError as e:
+        return f"Error: {e.stderr}"
+
+@mcp.tool
+async def embed_content(
+    vac_name: str,
+    content: str,
+    local_chunks: bool = False
+) -> str:
+    """
+    Embed content in a VAC's vector store.
+    
+    Args:
+        vac_name: Name of the VAC to embed content for
+        content: Content to embed
+        local_chunks: Whether to process chunks locally
+    
+    Returns:
+        Embedding operation result
+    """
+    try:
+        cmd = ["sunholo", "embed", vac_name, content]
+        if local_chunks:
+            cmd.append("--local-chunks")
+        
+        result = subprocess.run(
+            cmd,
+            capture_output=True,
+            text=True,
+            check=True
+        )
+        return result.stdout
+    except subprocess.CalledProcessError as e:
+        return f"Error embedding content: {e.stderr}"
+
+def cli_mcp(args):
+    """CLI handler for the MCP server command using FastMCP."""
+    try:
+        if not os.getenv("VAC_CONFIG_FOLDER"):
+            raise ValueError("sunholo configuration folder must be present in config/ or via VAC_CONFIG_FOLDER")
+        
+        logger.info("Starting Sunholo MCP server with FastMCP...")
+        
+        # Run the FastMCP server in stdio mode (default for Claude Desktop)
+        mcp.run()
+        
+    except Exception as e:
+        logger.error(f"Error running MCP server: {str(e)}")
+        raise
+
+def cli_mcp_bridge(args):
+    """
+    CLI handler for the MCP bridge command.
+    FastMCP handles HTTP transport natively, so this can be simplified.
+    """
+    try:
+        http_url = args.url
+        
+        # Parse port from URL if provided
+        import urllib.parse
+        parsed = urllib.parse.urlparse(http_url)
+        port = parsed.port or 8000
+        
+        logger.info(f"Starting FastMCP HTTP server on port {port}")
+        
+        # Run FastMCP in HTTP mode
+        mcp.run(transport="http", port=port)
+        
+    except KeyboardInterrupt:
+        logger.info("Server stopped by user")
+    except Exception as e:
+        logger.error(f"Error running MCP server: {str(e)}")
+        raise
+
+def setup_mcp_subparser(subparsers):
+    """
+    Sets up an argparse subparser for the 'mcp' command.
+    
+    By default will use configurations within the folder specified by 'VAC_CONFIG_FOLDER'
+    
+    Example command:
+    ```bash
+    sunholo mcp
+    ```
+    """
+    mcp_parser = subparsers.add_parser('mcp', 
+                                      help='MCP (Model Context Protocol) commands')
+    
+    # Create subcommands for mcp
+    mcp_subparsers = mcp_parser.add_subparsers(title='mcp commands',
+                                               description='MCP subcommands',
+                                               help='MCP subcommands',
+                                               dest='mcp_command')
+    
+    # mcp server command (default)
+    server_parser = mcp_subparsers.add_parser('server',
+                                             help='Start an Anthropic MCP server that wraps sunholo functionality')
+    server_parser.set_defaults(func=cli_mcp)
+    
+    # mcp bridge command - now simplified with FastMCP
+    bridge_parser = mcp_subparsers.add_parser('bridge',
+                                             help='Start an HTTP MCP server (FastMCP handles transport)')
+    bridge_parser.add_argument('url',
+                                  nargs='?',
+                                  default='http://127.0.0.1:8000/mcp',
+                                  help='HTTP URL for the MCP server (default: http://127.0.0.1:8000/mcp)')
+    bridge_parser.set_defaults(func=cli_mcp_bridge)
+    
+    # Set default behavior when no subcommand is provided
+    mcp_parser.set_defaults(func=cli_mcp)