mcp-mesh 0.7.15__py3-none-any.whl → 0.7.17__py3-none-any.whl

_mcp_mesh/__init__.py

@@ -31,7 +31,7 @@ from .engine.decorator_registry import (
     get_decorator_stats,
 )
 
-__version__ = "0.7.15"
+__version__ = "0.7.17"
 
 # Store reference to runtime processor if initialized
 _runtime_processor = None

_mcp_mesh/pipeline/mcp_startup/fastapiserver_setup.py

@@ -553,6 +553,7 @@ mcp_mesh_up{{agent="{agent_name}"}} 1
                 port=bind_port,
                 log_level="info",
                 access_log=False,  # Reduce noise
+                ws="websockets-sansio",  # Use modern websockets API (avoids deprecation warnings)
             )
 
             # Create and start server

_mcp_mesh/pipeline/mcp_startup/startup_orchestrator.py

@@ -319,6 +319,7 @@ class DebounceCoordinator:
                 port=bind_port,
                 log_level="info",
                 access_log=False,  # Reduce noise
+                ws="websockets-sansio",  # Use modern websockets API (avoids deprecation warnings)
             )
 
         except KeyboardInterrupt:

_mcp_mesh/reload.py

@@ -0,0 +1,206 @@
+"""
+File watch and reload functionality for MCP Mesh agents.
+
+Provides automatic restart of agent processes when source files change.
+Used by `meshctl start --watch` for development workflows.
+"""
+
+import logging
+import os
+import signal
+import subprocess
+import sys
+import threading
+import time
+from pathlib import Path
+
+logger = logging.getLogger(__name__)
+
+# Debounce delay to batch rapid file changes (seconds)
+DEBOUNCE_DELAY = float(os.getenv("MCP_MESH_RELOAD_DEBOUNCE", "0.5"))
+
+# Delay after stopping process to allow port release (seconds)
+PORT_RELEASE_DELAY = float(os.getenv("MCP_MESH_RELOAD_PORT_DELAY", "0.5"))
+
+
+def get_watch_paths(script_path: str) -> list[Path]:
+    """
+    Determine which paths to watch for changes.
+
+    Watches the directory containing the agent script and all subdirectories.
+    """
+    script_dir = Path(script_path).parent.absolute()
+    return [script_dir]
+
+
+def should_watch_file(path: str) -> bool:
+    """
+    Filter function to determine which files to watch.
+
+    Only watches:
+    - Python files (.py)
+    - Jinja2 templates (.jinja2, .j2)
+    - YAML config files (.yaml, .yml)
+
+    Excludes common non-source directories like __pycache__, .git, .venv, etc.
+    Also excludes hidden files and editor temporary files.
+    """
+    path_obj = Path(path)
+    filename = path_obj.name
+
+    # Skip hidden files and editor temp files (e.g., .!12345!main.py from sed -i)
+    if filename.startswith("."):
+        return False
+
+    # Skip common editor backup/temp files
+    if filename.endswith("~") or filename.endswith(".swp") or filename.endswith(".tmp"):
+        return False
+
+    path_parts = {p.lower() for p in path_obj.parts}
+
+    # Skip common non-source directories (case-insensitive, exact component match)
+    skip_patterns = [
+        "__pycache__",
+        ".git",
+        ".venv",
+        "venv",
+        ".pytest_cache",
+        ".mypy_cache",
+        "node_modules",
+        ".eggs",
+        ".egg-info",
+    ]
+
+    for pattern in skip_patterns:
+        if pattern.lower() in path_parts:
+            return False
+
+    # Only watch specific file types
+    watch_extensions = [".py", ".jinja2", ".j2", ".yaml", ".yml"]
+    return path_obj.suffix.lower() in watch_extensions
+
+
+def terminate_process(process: subprocess.Popen, timeout: int = 3) -> None:
+    """
+    Gracefully terminate a process and all its children, force kill if needed.
+
+    Args:
+        process: The subprocess to terminate
+        timeout: Seconds to wait before force killing
+    """
+    if process is None or process.poll() is not None:
+        return
+
+    pid = process.pid
+
+    # Try graceful termination first - send to process group to include children
+    try:
+        os.killpg(pid, signal.SIGTERM)
+    except (ProcessLookupError, PermissionError):
+        # Process group doesn't exist or we can't signal it, try direct
+        process.terminate()
+
+    try:
+        process.wait(timeout=timeout)
+    except subprocess.TimeoutExpired:
+        logger.warning(
+            f"Process {pid} didn't terminate gracefully, force killing..."
+        )
+        # Force kill the entire process group
+        try:
+            os.killpg(pid, signal.SIGKILL)
+        except (ProcessLookupError, PermissionError):
+            process.kill()
+        process.wait()
+
+
+def run_with_reload(script_path: str) -> None:
+    """
+    Run the agent script with file watching and auto-reload.
+
+    Uses watchfiles to monitor the agent directory for changes.
+    When files change, the agent process is terminated and restarted.
+
+    Args:
+        script_path: Path to the agent's main.py script
+    """
+    try:
+        import watchfiles
+    except ImportError:
+        logger.error(
+            "watchfiles is required for reload mode. "
+            "Install with: pip install watchfiles"
+        )
+        sys.exit(1)
+
+    script_path = os.path.abspath(script_path)
+    watch_paths = get_watch_paths(script_path)
+    python_exec = sys.executable
+
+    logger.info("🔄 Starting agent with file watching enabled")
+    logger.info(f"📁 Watching: {watch_paths[0]}")
+    logger.info("📝 File types: .py, .jinja2, .j2, .yaml, .yml")
+    logger.info("🔃 Agent will restart automatically when files change")
+
+    process = None
+    stop_event = threading.Event()
+
+    def signal_handler(signum, frame):
+        stop_event.set()
+        logger.info("🛑 Stopping agent...")
+        terminate_process(process)
+
+    signal.signal(signal.SIGINT, signal_handler)
+    signal.signal(signal.SIGTERM, signal_handler)
+
+    def start_agent() -> subprocess.Popen:
+        """Start the agent subprocess in a new process group."""
+        return subprocess.Popen(
+            [python_exec, script_path],
+            stdout=sys.stdout,
+            stderr=sys.stderr,
+            stdin=sys.stdin,
+            start_new_session=True,  # Create new process group for clean shutdown
+        )
+
+    # Start initial process
+    process = start_agent()
+    logger.info(f"✅ Agent started (PID: {process.pid})")
+
+    # Watch for file changes
+    try:
+        for changes in watchfiles.watch(
+            *watch_paths,
+            watch_filter=lambda ct, p: should_watch_file(p),
+            debounce=int(DEBOUNCE_DELAY * 1000),  # Convert to milliseconds
+            stop_event=stop_event,
+        ):
+            if stop_event.is_set():
+                break
+
+            # Check if process crashed
+            if process.poll() is not None:
+                logger.warning(f"⚠️ Agent process exited with code {process.returncode}")
+
+            # Log changes
+            for change_type, path in changes:
+                rel_path = os.path.relpath(path, watch_paths[0])
+                logger.info(f"🔄 Detected {change_type.name}: {rel_path}")
+
+            logger.info("🔃 Restarting agent...")
+
+            # Stop current process
+            terminate_process(process)
+
+            # Delay to allow ports to be released
+            time.sleep(PORT_RELEASE_DELAY)
+
+            # Start new process
+            process = start_agent()
+            logger.info(f"✅ Agent restarted (PID: {process.pid})")
+
+    except KeyboardInterrupt:
+        pass
+    finally:
+        terminate_process(process)
+        logger.info("👋 File watcher stopped")

_mcp_mesh/reload_runner.py

@@ -0,0 +1,52 @@
+#!/usr/bin/env python3
+"""
+MCP Mesh Reload Runner - Entry point for file watching mode.
+
+Usage:
+    python -m _mcp_mesh.reload_runner <agent_script.py>
+
+This module is invoked by `meshctl start --watch` to wrap agent execution
+with file watching. When source files change, the agent process is
+automatically restarted.
+
+Environment variables:
+    MCP_MESH_RELOAD_DEBOUNCE: Debounce delay in seconds (default: 0.5)
+    MCP_MESH_RELOAD_PORT_DELAY: Port release delay in seconds (default: 0.5)
+"""
+
+import logging
+import os
+import sys
+
+# Configure logging before imports
+logging.basicConfig(
+    level=logging.INFO,
+    format="%(levelname)-8s %(message)s",
+)
+logger = logging.getLogger(__name__)
+
+
+def main():
+    """Main entry point for reload runner."""
+    if len(sys.argv) < 2:
+        print("Usage: python -m _mcp_mesh.reload_runner <agent_script.py>")
+        sys.exit(1)
+
+    script_path = sys.argv[1]
+
+    if not os.path.exists(script_path):
+        print(f"Error: Script not found: {script_path}")
+        sys.exit(1)
+
+    if not script_path.endswith(".py"):
+        print(f"Error: Expected a Python script (.py), got: {script_path}")
+        sys.exit(1)
+
+    # Import and run with reload
+    from .reload import run_with_reload
+
+    run_with_reload(script_path)
+
+
+if __name__ == "__main__":
+    main()

{mcp_mesh-0.7.15.dist-info → mcp_mesh-0.7.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcp-mesh
-Version: 0.7.15
+Version: 0.7.17
 Summary: Kubernetes-native platform for distributed MCP applications
 Project-URL: Homepage, https://github.com/dhyansraj/mcp-mesh
 Project-URL: Documentation, https://github.com/dhyansraj/mcp-mesh/tree/main/docs

{mcp_mesh-0.7.15.dist-info → mcp_mesh-0.7.17.dist-info}/RECORD

@@ -1,4 +1,6 @@
-_mcp_mesh/__init__.py,sha256=xx8r72WkHqUhfJc2KwE6aOwUNPw5F5OuDJs4znH8B3g,2720
+_mcp_mesh/__init__.py,sha256=QiGicZib2awgedpv_mY7bUW8ymNN5gvbxWgypsdI-7c,2720
+_mcp_mesh/reload.py,sha256=IqZeS7lsFw7bwOzDPE0LJLPkY5nR68BKc8C4srSCX1o,6239
+_mcp_mesh/reload_runner.py,sha256=SgQKzzO2yHfSUBq8s3SpAnovWA0rveimVNaxeLCEo_0,1310
 _mcp_mesh/engine/__init__.py,sha256=U_6Kw3vA_3RiNK0Oln5c5C7WvA9lSONV22wWzfxYHNw,2975
 _mcp_mesh/engine/async_mcp_client.py,sha256=Sz-rXTkb1Mng_f0SpLqLuOdPJ8vZjv3DFy0i8yYOqYk,8792
 _mcp_mesh/engine/base_injector.py,sha256=qzRLZqFP2VvEFagVovkpdldvDmm3VwPHm6tHwV58a2k,5648
@@ -105,13 +107,13 @@ _mcp_mesh/pipeline/mcp_heartbeat/registry_connection.py,sha256=kL7k_yx2zafH3HO4s
 _mcp_mesh/pipeline/mcp_startup/__init__.py,sha256=gS0xNmVx66bkLUMw64olMsN40ZLPH3ymwlLixZ4NuTs,1239
 _mcp_mesh/pipeline/mcp_startup/configuration.py,sha256=6LRLIxrqFMU76qrBb6GjGknUlKPZZ9iqOlxE7F9ZhLs,2808
 _mcp_mesh/pipeline/mcp_startup/decorator_collection.py,sha256=RHC6MHtfP9aP0hZ-IJjISZu72e0Pml3LU0qr7dc284w,2294
-_mcp_mesh/pipeline/mcp_startup/fastapiserver_setup.py,sha256=si_HkHkpBbJDUIciDOA9-tppjnutGVg4K6Qcm6-gCKg,33888
+_mcp_mesh/pipeline/mcp_startup/fastapiserver_setup.py,sha256=SYTMVRtR3qYoWaeIit88DnAyWhAxsG--ZatDtFvTkrc,33987
 _mcp_mesh/pipeline/mcp_startup/fastmcpserver_discovery.py,sha256=Pm24wrSuRGsgeUrHvMPDnNh6RhIZoznnMAUwAkllohk,10661
 _mcp_mesh/pipeline/mcp_startup/heartbeat_loop.py,sha256=v85B0ynomvYu87eIvLe-aSZ7-Iwov2VtM4Fg3PkmrZs,3865
 _mcp_mesh/pipeline/mcp_startup/heartbeat_preparation.py,sha256=sOpzxRc0kYiXwSW9lvv8DSjliT85oZCWPODeJRuiqgg,15635
 _mcp_mesh/pipeline/mcp_startup/lifespan_factory.py,sha256=Hu7IvrhVH9sM7-XQDyWAGA3rgOnNIRyWFBtobkUQ5Es,4404
 _mcp_mesh/pipeline/mcp_startup/server_discovery.py,sha256=VuqqaBE00h6AerPjk-Ab-g51x6jODCbMX4nemLRQIIQ,8375
-_mcp_mesh/pipeline/mcp_startup/startup_orchestrator.py,sha256=dT96P2f6Ci0H8sDe-6_Uro88Y0lZiyjPanN2bPsrTyk,25066
+_mcp_mesh/pipeline/mcp_startup/startup_orchestrator.py,sha256=JuEP4JOCBNG1qhaS2KaT1cFhhAb_NTuF46saJapHjeY,25165
 _mcp_mesh/pipeline/mcp_startup/startup_pipeline.py,sha256=cAAbqioYRswf-P25OpZFX2yL_qN2sl_bVk-kcynFxPw,2347
 _mcp_mesh/pipeline/shared/__init__.py,sha256=s9xmdf6LkoetrVRGd7Zp3NUxcJCW6YZ_yNKzUBcnYys,352
 _mcp_mesh/pipeline/shared/base_step.py,sha256=kyPbNUX79NyGrE_0Q-e-Aek7m1J0TW036njWfv0iZ0I,1080
@@ -141,10 +143,10 @@ _mcp_mesh/tracing/trace_context_helper.py,sha256=3XWVU_cnsqrfjgAGRJaRK0Uvkns363s
 _mcp_mesh/tracing/utils.py,sha256=t9lJuTH7CeuzAiiAaD0WxsJMFJPdzZFR0w6-vyR9f2E,3849
 _mcp_mesh/utils/fastmcp_schema_extractor.py,sha256=M54ffesC-56zl_fNJHj9dZxElDQaWFf1MXdSLCuFStg,17253
 mesh/__init__.py,sha256=Va5XRBWgejQurad7Maz3E-zPY7vu431B2_4sAdCu1zk,3868
-mesh/decorators.py,sha256=v0l_0oAS_UZBQLC1WwhJQTil9AvGMCFL9kYS-N7febk,59627
+mesh/decorators.py,sha256=sFTLngsIo_2uBb-Fvuh9K1S8e2Kzq_70eWT4Z2FA83E,59730
 mesh/helpers.py,sha256=ITua2zdxbXeBLF5qS46A6A1P2GQNbfL_2_BMCKazQ4U,12575
 mesh/types.py,sha256=n0MxrBYZJ84xyQWGf_X2ZbVWSAaIcEBkRV7qaCmX6Ac,17008
-mcp_mesh-0.7.15.dist-info/METADATA,sha256=pV9YaeDfY1G07nCjfD-s24ggiQcKZ3-9zk75uev9VAc,4973
-mcp_mesh-0.7.15.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-mcp_mesh-0.7.15.dist-info/licenses/LICENSE,sha256=_EBQHRQThv9FPOLc5eFOUdeeRO0mYwChC7cx60dM1tM,1078
-mcp_mesh-0.7.15.dist-info/RECORD,,
+mcp_mesh-0.7.17.dist-info/METADATA,sha256=PISE6mS316f6UKCehxZyTLi4cneVYPYW8ybg213Ol38,4973
+mcp_mesh-0.7.17.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+mcp_mesh-0.7.17.dist-info/licenses/LICENSE,sha256=_EBQHRQThv9FPOLc5eFOUdeeRO0mYwChC7cx60dM1tM,1078
+mcp_mesh-0.7.17.dist-info/RECORD,,

mesh/decorators.py

@@ -242,6 +242,7 @@ def _start_uvicorn_immediately(http_host: str, http_port: int):
                     log_level="info",
                     timeout_graceful_shutdown=30,  # Allow time for registry cleanup
                     access_log=False,  # Reduce noise
+                    ws="websockets-sansio",  # Use modern websockets API (avoids deprecation warnings)
                 )
             except Exception as e:
                 logger.error(f"❌ IMMEDIATE UVICORN: Server failed: {e}")