hud-python 0.4.1__py3-none-any.whl → 0.4.3__py3-none-any.whl

hud/cli/runner.py

@@ -1,160 +1,160 @@
-"""Run Docker images as MCP servers."""
-
-from __future__ import annotations
-
-import asyncio
-import subprocess
-import sys
-
-import click
-from fastmcp import FastMCP
-
-from hud.utils.design import HUDDesign
-
-
-def run_stdio_server(image: str, docker_args: list[str], verbose: bool) -> None:
-    """Run Docker image as stdio MCP server (direct passthrough)."""
-    design = HUDDesign(stderr=True)  # Use stderr for stdio mode
-
-    # Build docker command
-    docker_cmd = ["docker", "run", "--rm", "-i", *docker_args, image]
-
-    if verbose:
-        design.info(f"🐳 Running: {' '.join(docker_cmd)}")
-
-    # Run docker directly with stdio passthrough
-    try:
-        result = subprocess.run(docker_cmd, stdin=sys.stdin)  # noqa: S603
-        sys.exit(result.returncode)
-    except KeyboardInterrupt:
-        design.info("\n👋 Shutting down...")
-        sys.exit(0)
-    except Exception as e:
-        design.error(f"Error: {e}")
-        sys.exit(1)
-
-
-async def run_http_server(image: str, docker_args: list[str], port: int, verbose: bool) -> None:
-    """Run Docker image as HTTP MCP server (proxy mode)."""
-    from .utils import find_free_port
-
-    design = HUDDesign()
-
-    # Find available port
-    actual_port = find_free_port(port)
-    if actual_port is None:
-        design.error(f"No available ports found starting from {port}")
-        return
-
-    if actual_port != port:
-        design.warning(f"Port {port} in use, using port {actual_port} instead")
-
-    # Generate container name
-    container_name = f"run-{image.replace(':', '-').replace('/', '-')}"
-
-    # Remove any existing container with the same name
-    try:
-        subprocess.run(  # noqa: ASYNC221, S603
-            ["docker", "rm", "-f", container_name],  # noqa: S607
-            stdout=subprocess.DEVNULL,
-            stderr=subprocess.DEVNULL,
-            check=False,  # Don't raise error if container doesn't exist
-        )
-    except Exception:
-        click.echo(f"Failed to remove existing container {container_name}", err=True)
-
-    # Build docker command for stdio container
-    docker_cmd = (
-        *[
-            "docker",
-            "run",
-            "--rm",
-            "-i",
-            "--name",
-            container_name,
-        ],
-        docker_args,
-        [image],
-    )
-
-    # Create MCP config for stdio transport
-    config = {
-        "mcpServers": {
-            "default": {
-                "command": docker_cmd[0],
-                "args": docker_cmd[1:] if len(docker_cmd) > 1 else [],
-                # transport defaults to stdio
-            }
-        }
-    }
-
-    # Set up logging suppression
-    import logging
-    import os
-
-    os.environ["FASTMCP_DISABLE_BANNER"] = "1"
-
-    if not verbose:
-        logging.getLogger("fastmcp").setLevel(logging.ERROR)
-        logging.getLogger("mcp").setLevel(logging.ERROR)
-        logging.getLogger("uvicorn").setLevel(logging.ERROR)
-        logging.getLogger("uvicorn.access").setLevel(logging.ERROR)
-        logging.getLogger("uvicorn.error").setLevel(logging.ERROR)
-
-        import warnings
-
-        warnings.filterwarnings("ignore", category=DeprecationWarning)
-
-    # Create HTTP proxy
-    proxy = FastMCP.as_proxy(config, name=f"HUD Run - {image}")
-
-    # Show header
-    design.info("")  # Empty line
-    design.header("HUD MCP Server", icon="🌐")
-
-    # Show configuration
-    design.section_title("Server Information")
-    design.info(f"Port: {actual_port}")
-    design.info(f"URL: http://localhost:{actual_port}/mcp")
-    design.info(f"Container: {container_name}")
-    design.info("")
-    design.progress_message("Press Ctrl+C to stop")
-
-    try:
-        await proxy.run_async(
-            transport="http",
-            host="0.0.0.0",  # noqa: S104
-            port=actual_port,
-            path="/mcp",
-            log_level="error" if not verbose else "info",
-            show_banner=False,
-        )
-    except KeyboardInterrupt:
-        design.info("\n👋 Shutting down...")
-
-
-def run_mcp_server(
-    image: str, docker_args: list[str], transport: str, port: int, verbose: bool
-) -> None:
-    """Run Docker image as MCP server with specified transport."""
-    if transport == "stdio":
-        run_stdio_server(image, docker_args, verbose)
-    elif transport == "http":
-        try:
-            asyncio.run(run_http_server(image, docker_args, port, verbose))
-        except Exception as e:
-            # Suppress the graceful shutdown errors
-            if not any(
-                x in str(e)
-                for x in [
-                    "timeout graceful shutdown exceeded",
-                    "Cancel 0 running task(s)",
-                    "Application shutdown complete",
-                ]
-            ):
-                design = HUDDesign()
-                design.error(f"Unexpected error: {e}")
-    else:
-        design = HUDDesign()
-        design.error(f"Unknown transport: {transport}")
-        sys.exit(1)
+"""Run Docker images as MCP servers."""
+
+from __future__ import annotations
+
+import asyncio
+import subprocess
+import sys
+
+import click
+from fastmcp import FastMCP
+
+from hud.utils.design import HUDDesign
+
+
+def run_stdio_server(image: str, docker_args: list[str], verbose: bool) -> None:
+    """Run Docker image as stdio MCP server (direct passthrough)."""
+    design = HUDDesign(stderr=True)  # Use stderr for stdio mode
+
+    # Build docker command
+    docker_cmd = ["docker", "run", "--rm", "-i", *docker_args, image]
+
+    if verbose:
+        design.info(f"🐳 Running: {' '.join(docker_cmd)}")
+
+    # Run docker directly with stdio passthrough
+    try:
+        result = subprocess.run(docker_cmd, stdin=sys.stdin)  # noqa: S603
+        sys.exit(result.returncode)
+    except KeyboardInterrupt:
+        design.info("\n👋 Shutting down...")
+        sys.exit(0)
+    except Exception as e:
+        design.error(f"Error: {e}")
+        sys.exit(1)
+
+
+async def run_http_server(image: str, docker_args: list[str], port: int, verbose: bool) -> None:
+    """Run Docker image as HTTP MCP server (proxy mode)."""
+    from .utils import find_free_port
+
+    design = HUDDesign()
+
+    # Find available port
+    actual_port = find_free_port(port)
+    if actual_port is None:
+        design.error(f"No available ports found starting from {port}")
+        return
+
+    if actual_port != port:
+        design.warning(f"Port {port} in use, using port {actual_port} instead")
+
+    # Generate container name
+    container_name = f"run-{image.replace(':', '-').replace('/', '-')}"
+
+    # Remove any existing container with the same name
+    try:
+        subprocess.run(  # noqa: ASYNC221, S603
+            ["docker", "rm", "-f", container_name],  # noqa: S607
+            stdout=subprocess.DEVNULL,
+            stderr=subprocess.DEVNULL,
+            check=False,  # Don't raise error if container doesn't exist
+        )
+    except Exception:
+        click.echo(f"Failed to remove existing container {container_name}", err=True)
+
+    # Build docker command for stdio container
+    docker_cmd = (
+        *[
+            "docker",
+            "run",
+            "--rm",
+            "-i",
+            "--name",
+            container_name,
+        ],
+        docker_args,
+        [image],
+    )
+
+    # Create MCP config for stdio transport
+    config = {
+        "mcpServers": {
+            "default": {
+                "command": docker_cmd[0],
+                "args": docker_cmd[1:] if len(docker_cmd) > 1 else [],
+                # transport defaults to stdio
+            }
+        }
+    }
+
+    # Set up logging suppression
+    import logging
+    import os
+
+    os.environ["FASTMCP_DISABLE_BANNER"] = "1"
+
+    if not verbose:
+        logging.getLogger("fastmcp").setLevel(logging.ERROR)
+        logging.getLogger("mcp").setLevel(logging.ERROR)
+        logging.getLogger("uvicorn").setLevel(logging.ERROR)
+        logging.getLogger("uvicorn.access").setLevel(logging.ERROR)
+        logging.getLogger("uvicorn.error").setLevel(logging.ERROR)
+
+        import warnings
+
+        warnings.filterwarnings("ignore", category=DeprecationWarning)
+
+    # Create HTTP proxy
+    proxy = FastMCP.as_proxy(config, name=f"HUD Run - {image}")
+
+    # Show header
+    design.info("")  # Empty line
+    design.header("HUD MCP Server", icon="🌐")
+
+    # Show configuration
+    design.section_title("Server Information")
+    design.info(f"Port: {actual_port}")
+    design.info(f"URL: http://localhost:{actual_port}/mcp")
+    design.info(f"Container: {container_name}")
+    design.info("")
+    design.progress_message("Press Ctrl+C to stop")
+
+    try:
+        await proxy.run_async(
+            transport="http",
+            host="0.0.0.0",  # noqa: S104
+            port=actual_port,
+            path="/mcp",
+            log_level="error" if not verbose else "info",
+            show_banner=False,
+        )
+    except KeyboardInterrupt:
+        design.info("\n👋 Shutting down...")
+
+
+def run_mcp_server(
+    image: str, docker_args: list[str], transport: str, port: int, verbose: bool
+) -> None:
+    """Run Docker image as MCP server with specified transport."""
+    if transport == "stdio":
+        run_stdio_server(image, docker_args, verbose)
+    elif transport == "http":
+        try:
+            asyncio.run(run_http_server(image, docker_args, port, verbose))
+        except Exception as e:
+            # Suppress the graceful shutdown errors
+            if not any(
+                x in str(e)
+                for x in [
+                    "timeout graceful shutdown exceeded",
+                    "Cancel 0 running task(s)",
+                    "Application shutdown complete",
+                ]
+            ):
+                design = HUDDesign()
+                design.error(f"Unexpected error: {e}")
+    else:
+        design = HUDDesign()
+        design.error(f"Unknown transport: {transport}")
+        sys.exit(1)

hud/cli/tests/__init__.py

@@ -1,3 +1,3 @@
-"""Tests for HUD CLI module."""
-
-from __future__ import annotations
+"""Tests for HUD CLI module."""
+
+from __future__ import annotations