mcp-mesh 0.7.21__py3-none-any.whl → 0.8.0b1__py3-none-any.whl

_mcp_mesh/tracing/redis_metadata_publisher.py

@@ -2,93 +2,55 @@
 Redis Trace Publisher
 
 Publishes execution trace data to Redis streams for distributed tracing storage and analysis.
-Based on the session storage pattern from http_wrapper.py.
+Uses Rust core for Redis publishing to share implementation across all language SDKs.
 """
 
 import logging
-import os
 from typing import Any, Optional
 
-logger = logging.getLogger(__name__)
-
-try:
-    import redis
+import mcp_mesh_core
 
-    REDIS_AVAILABLE = True
-except ImportError:
-    REDIS_AVAILABLE = False
-    redis = None
+logger = logging.getLogger(__name__)
 
 
 class RedisTracePublisher:
-    """Non-blocking execution trace publisher to Redis."""
+    """Non-blocking execution trace publisher to Redis via Rust core."""
 
-    def __init__(self, redis_url: Optional[str] = None):
-        # Use existing REDIS_URL pattern from MCP Mesh
-        self.redis_url = redis_url or os.getenv("REDIS_URL", "redis://localhost:6379")
+    def __init__(self):
         self.stream_name = "mesh:trace"
-        self._redis_client = None
         self._available = False
-        self._tracing_enabled = self._is_tracing_enabled()
-
-        if not REDIS_AVAILABLE:
-            logger.warning("redis not available. Execution metadata storage disabled.")
-            self._available = False
+        self._tracing_enabled = mcp_mesh_core.is_tracing_enabled_py()
+
+        if self._tracing_enabled:
+            logger.info("Distributed tracing: enabled")
+            # Initialize Rust core trace publisher (handles Redis connection)
+            self._available = mcp_mesh_core.init_trace_publisher_py()
+            if not self._available:
+                logger.warning("Rust core trace publisher initialization failed")
         else:
-            self._init_redis()
-
-    def _is_tracing_enabled(self) -> bool:
-        """Check if distributed tracing is enabled via environment variable."""
-        from .utils import is_tracing_enabled
-
-        return is_tracing_enabled()
-
-    def _init_redis(self):
-        """Initialize Redis connection with graceful fallback (following session storage pattern)."""
-        if not self._tracing_enabled:
-            self._available = False
-            return
-
-        logger.info("Distributed tracing: enabled")
-
-        if not REDIS_AVAILABLE:
-            self._available = False
-            return
-
-        try:
-            # Use sync Redis client like session storage (no atexit issues)
-            self._redis_client = redis.from_url(self.redis_url, decode_responses=True)
-
-            # Test connection
-            self._redis_client.ping()
-            self._available = True
-        except Exception as e:
-            # Graceful fallback - metadata storage disabled if Redis unavailable
-            self._available = False
+            logger.debug("Distributed tracing: disabled")
 
     def publish_execution_trace(self, trace_data: dict[str, Any]) -> None:
-        """Publish execution trace data to Redis Stream (non-blocking)."""
+        """Publish execution trace data to Redis Stream via Rust core (non-blocking)."""
         if not self._available:
             return  # Silent no-op when Redis unavailable
 
         try:
-            function_name = trace_data.get("function_name", "unknown")
-            trace_id = trace_data.get("trace_id", "no-trace-id")
-
-            # Add timestamp and convert for Redis storage
+            # Convert trace data to strings for Redis storage
             from .utils import add_timestamp_if_missing, convert_for_redis_storage
 
             add_timestamp_if_missing(trace_data)
             redis_trace_data = convert_for_redis_storage(trace_data)
 
-            # Publish to Redis Stream
-            if self._redis_client:
-                message_id = self._redis_client.xadd(self.stream_name, redis_trace_data)
-                logger.debug(f"Published trace for '{function_name}' to Redis stream")
+            # Publish via Rust core
+            mcp_mesh_core.publish_span_py(redis_trace_data)
+            logger.debug(
+                f"Published trace for '{trace_data.get('function_name', 'unknown')}' via Rust core"
+            )
 
         except Exception as e:
             # Non-blocking - never fail agent operations due to trace publishing
-            pass
+            logger.debug(f"Failed to publish trace: {e}")
 
     @property
     def is_available(self) -> bool:
@@ -102,30 +64,13 @@ class RedisTracePublisher:
 
     def get_stats(self) -> dict[str, Any]:
         """Get Redis trace publisher statistics."""
-        stats = {
+        return {
             "redis_available": self._available,
             "tracing_enabled": self._tracing_enabled,
             "stream_name": self.stream_name,
-            "redis_url": (
-                self.redis_url.replace("redis:/", "redis://***")
-                if self.redis_url
-                else None
-            ),
+            "backend": "rust_core",
         }
 
-        if self._available and self._redis_client:
-            try:
-                # Get approximate stream length
-                stream_info = self._redis_client.xinfo_stream(self.stream_name)
-                stats["stream_length"] = stream_info.get("length", 0)
-                stats["stream_last_generated_id"] = stream_info.get(
-                    "last-generated-id", "N/A"
-                )
-            except Exception as e:
-                stats["stream_error"] = str(e)
-
-        return stats
-
 
 # Global instance for reuse
 _trace_publisher: Optional[RedisTracePublisher] = None

_mcp_mesh/tracing/trace_context_helper.py

@@ -67,19 +67,9 @@ class TraceContextHelper:
                     or "",  # Use parent as current (will be overwritten by ExecutionTracer)
                     parent_span=None,  # No grandparent needed at this level
                 )
-            else:
-                # NEW ROOT TRACE: This service is the entry point (no incoming trace headers)
-                from .utils import generate_span_id, generate_trace_id
-
-                root_trace_id = generate_trace_id()
-                root_span_id = generate_span_id()
-
-                # Set context for root trace (no parent_span)
-                TraceContext.set_current(
-                    trace_id=root_trace_id,
-                    span_id=root_span_id,
-                    parent_span=None,  # Root trace has no parent
-                )
+            # else: Don't generate root trace here - let ExecutionTracer do it
+            # This prevents orphan parent_span values when middleware context doesn't
+            # propagate to FastMCP's async context
 
         except Exception as e:
             logger_instance.warning(f"Failed to setup trace context: {e}")

_mcp_mesh/tracing/utils.py

@@ -7,44 +7,58 @@ and maintain consistency.
 
 import json
 import logging
-import os
 import time
 import uuid
-from typing import Any, Optional
+from typing import Any
 
 logger = logging.getLogger(__name__)
 
+# Try to import the Rust core module for tracing
+# Falls back gracefully if not available
+try:
+    import mcp_mesh_core
+
+    _RUST_CORE_AVAILABLE = True
+except ImportError:
+    mcp_mesh_core = None  # type: ignore[assignment]
+    _RUST_CORE_AVAILABLE = False
+    logger.warning(
+        "mcp_mesh_core not available - tracing features will be disabled. "
+        "Build/install mcp-mesh-core for full functionality."
+    )
+
 
 def is_tracing_enabled() -> bool:
-    """Check if distributed tracing is enabled via environment variable.
+    """Check if distributed tracing is enabled via Rust core config resolution.
+
+    Delegates to mcp_mesh_core.is_tracing_enabled_py() for consistent behavior
+    across all language SDKs. Priority: ENV > param > default (false)
 
     Returns:
         True if tracing is enabled, False otherwise
     """
-    return os.getenv("MCP_MESH_DISTRIBUTED_TRACING_ENABLED", "false").lower() in (
-        "true",
-        "1",
-        "yes",
-        "on",
-    )
+    if not _RUST_CORE_AVAILABLE or mcp_mesh_core is None:
+        return False
+
+    return mcp_mesh_core.is_tracing_enabled_py()
 
 
 def generate_span_id() -> str:
-    """Generate a unique span ID for tracing.
+    """Generate a unique span ID for tracing (OpenTelemetry compliant).
 
     Returns:
-        UUID string for span identification
+        16-character hex string (64-bit span ID per OTel spec)
     """
-    return str(uuid.uuid4())
+    return uuid.uuid4().hex[:16]
 
 
 def generate_trace_id() -> str:
-    """Generate a unique trace ID for tracing.
+    """Generate a unique trace ID for tracing (OpenTelemetry compliant).
 
     Returns:
-        UUID string for trace identification
+        32-character hex string (128-bit trace ID per OTel spec)
     """
-    return str(uuid.uuid4())
+    return uuid.uuid4().hex
 
 
 def get_agent_metadata_with_fallback(logger_instance: logging.Logger) -> dict[str, Any]:

_mcp_mesh/utils/fastmcp_schema_extractor.py

@@ -274,7 +274,8 @@ class FastMCPSchemaExtractor:
             return schema
 
         # Get McpMeshAgent parameter names from signature analysis
-        from _mcp_mesh.engine.signature_analyzer import get_mesh_agent_parameter_names
+        from _mcp_mesh.engine.signature_analyzer import \
+            get_mesh_agent_parameter_names
 
         mesh_param_names = get_mesh_agent_parameter_names(function)
 

{mcp_mesh-0.7.21.dist-info → mcp_mesh-0.8.0b1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcp-mesh
-Version: 0.7.21
+Version: 0.8.0b1
 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
@@ -30,6 +30,7 @@ Requires-Dist: fastmcp<3.0.0,>=2.8.0
 Requires-Dist: httpx<1.0.0,>=0.25.0
 Requires-Dist: jinja2>=3.1.0
 Requires-Dist: litellm>=1.30.0
+Requires-Dist: mcp-mesh-core>=0.8.0b1
 Requires-Dist: mcp<2.0.0,>=1.9.0
 Requires-Dist: prometheus-client<1.0.0,>=0.19.0
 Requires-Dist: pydantic<3.0.0,>=2.4.0

mcp_mesh-0.8.0b1.dist-info/RECORD

@@ -0,0 +1,85 @@
+_mcp_mesh/__init__.py,sha256=Hmsj3lCnO9-6rV8hN5SVUNdzKWxdXIsO-WcGCvbt1dw,2721
+_mcp_mesh/reload.py,sha256=5Yll9n0bqxM7pmTjfAaKWg-WT_Vi0YTh0_UNWbCNCIQ,6217
+_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
+_mcp_mesh/engine/decorator_registry.py,sha256=cch2QdQ6bKjHKEGi1XWp1YcLLO3uI2YlxwWBO7Np65E,28229
+_mcp_mesh/engine/dependency_injector.py,sha256=-CeIGvB-zqrGxkRpFE53C6_zEmmBoyBXftvE-H9OyAY,31593
+_mcp_mesh/engine/http_wrapper.py,sha256=Simd6IEsLO2FXQOuf1WEx57SBN6DSr5RzphXnk0asHM,24152
+_mcp_mesh/engine/llm_config.py,sha256=95bOsGWro5E1JGq7oZtEYhVdrzcIJqjht_r5vEdJVz4,2049
+_mcp_mesh/engine/llm_errors.py,sha256=h7BiI14u-jL8vtvBfFbFDDrN7gIw8PQjXIl5AP1SBuA,3276
+_mcp_mesh/engine/mesh_llm_agent.py,sha256=sVh7lPnvixDVJ-p1ONzbeakiEzhsl0HmdmrLPZA2FzQ,34237
+_mcp_mesh/engine/mesh_llm_agent_injector.py,sha256=Y6KkqqUAyP7QsCU2DDX_UIg9RhF4xu9p731jbfMMN-A,28373
+_mcp_mesh/engine/response_parser.py,sha256=g3VNoFJotaLrOAS0pL_OTCrv9t9XQe9Iiz1plsm28bQ,10280
+_mcp_mesh/engine/self_dependency_proxy.py,sha256=OkKt0-B_ADnJlWtHiHItoZCBZ7Su0iz2unEPFfXvrs4,3302
+_mcp_mesh/engine/session_aware_client.py,sha256=QejKag5zYNos5BVffQvNXFMECHFMLNOv78By4e_JzQE,10589
+_mcp_mesh/engine/session_manager.py,sha256=MCr0_fXBaUjXM51WU5EhDkiGvBdfzYQFVNb9DCXXL0A,10418
+_mcp_mesh/engine/signature_analyzer.py,sha256=ftn9XsX0ZHWIaACdjgBVtCuIdqVU_4ST8cvcpzu4HTk,12339
+_mcp_mesh/engine/tool_executor.py,sha256=Bf_9d02EEY9_yHm1p1-5YZ4rY6MPxn4SVpI6-3sm1uo,5456
+_mcp_mesh/engine/tool_schema_builder.py,sha256=SQCxQIrSfdLu9-dLqiFurQLK7dhl0dc0xa0ibaxU-iE,3644
+_mcp_mesh/engine/unified_mcp_proxy.py,sha256=Ee11K5HXuPXvdjqB7fmv0sMEo9ML-SMtQx2EUtYpfrY,37847
+_mcp_mesh/engine/provider_handlers/__init__.py,sha256=8hEc4CheKfXWU3ny4YDktxNxLCWxgfMtyDW9CblPOvs,888
+_mcp_mesh/engine/provider_handlers/base_provider_handler.py,sha256=Lb0U6gAEseU7Ix1eeV4T0WP1ClmeXUz87Nx_iplUYSI,8077
+_mcp_mesh/engine/provider_handlers/claude_handler.py,sha256=iYAmllL5rTWFFTAjbR62Ra9eMWNZjA72a02tppxjgOQ,14343
+_mcp_mesh/engine/provider_handlers/gemini_handler.py,sha256=SrJuhlaoPPHGRQH4jFXR1C_M0FtKeFcvJHERkpIPRNA,6055
+_mcp_mesh/engine/provider_handlers/generic_handler.py,sha256=rAE3QzoB7vz4zbBPPDJTI0MOuJrfhrOdXSyzsob4uVI,5489
+_mcp_mesh/engine/provider_handlers/openai_handler.py,sha256=1WhiHqf72N3584tJDVaBonDmu64tPuuiP12vzPrUNO0,6204
+_mcp_mesh/engine/provider_handlers/provider_handler_registry.py,sha256=klBZW8iX6Jk88TASAUyP8oPo4wvLB3DaTNFHeYvGLiA,5708
+_mcp_mesh/pipeline/__init__.py,sha256=MgPwpwbiD62ND4HXKKNGcnreDk-TvPmQOs5WmjtHQ3M,1263
+_mcp_mesh/pipeline/api_heartbeat/__init__.py,sha256=qGjEgxbGJFSl9Qm3bwu3X5yizAMbN4WpFtIUekDSFuU,690
+_mcp_mesh/pipeline/api_heartbeat/api_lifespan_integration.py,sha256=h0mTmLyPlGDqomSHpbW7S-AZNz1Tyvg1kpy9aeWkQsU,3879
+_mcp_mesh/pipeline/api_heartbeat/rust_api_heartbeat.py,sha256=vrld873jS2zCfJCndldx2XLDjxp_bRY7qq9CNBqZ8wI,15579
+_mcp_mesh/pipeline/api_startup/__init__.py,sha256=eivolkSKot2bJTWP2BV8-RKRT1Zm7SGQYuEUiTxusOQ,577
+_mcp_mesh/pipeline/api_startup/api_pipeline.py,sha256=I9-Q0o2py5oAHZO2DJOeTD1uZo1-Dpn258k5Tr0dv9o,2474
+_mcp_mesh/pipeline/api_startup/api_server_setup.py,sha256=72oCMkCzRfxYrE5sfFJbr57BYJwRSyKxBMISTOHmKyc,14919
+_mcp_mesh/pipeline/api_startup/fastapi_discovery.py,sha256=B-JeZTj2KZ17Sutyw31mpISlBMSF8-5ClV7W5SOPdXg,5606
+_mcp_mesh/pipeline/api_startup/middleware_integration.py,sha256=J7Ux_nJ1VsMqVzl5ApmtlixjUpJV09rtvU4YZ-aiXFk,6241
+_mcp_mesh/pipeline/api_startup/route_collection.py,sha256=WPr4hRPLIWnNIJCoRHZ141ph9tAa_-Pm_j2TiCuWS4k,2002
+_mcp_mesh/pipeline/api_startup/route_integration.py,sha256=qq1AVaWna-CWEXyehyDL3EyeYKgo5aMtei8uBNdvkZ8,12448
+_mcp_mesh/pipeline/mcp_heartbeat/__init__.py,sha256=mhDcSquoHkhRItqgbM8iFfAKC2m7qMW_0smqtUgSl-w,389
+_mcp_mesh/pipeline/mcp_heartbeat/rust_heartbeat.py,sha256=cB7IO0au3097MGpC6JwvYtPt4cVnl_spCkHfJIgD6Ks,25953
+_mcp_mesh/pipeline/mcp_startup/__init__.py,sha256=qy960dnAoHLXMcL_y_rcro9Km2AoCVzC7_CxMwao564,1166
+_mcp_mesh/pipeline/mcp_startup/configuration.py,sha256=OnumIPRVBTne2OEU2VWLZovLKvWcNF9iJVQtlVwuim0,2805
+_mcp_mesh/pipeline/mcp_startup/decorator_collection.py,sha256=RHC6MHtfP9aP0hZ-IJjISZu72e0Pml3LU0qr7dc284w,2294
+_mcp_mesh/pipeline/mcp_startup/fastapiserver_setup.py,sha256=y0xsM6no-yH9OhiYjg0LJN4hTW0vw15iuVK_4DNMxFQ,34032
+_mcp_mesh/pipeline/mcp_startup/fastmcpserver_discovery.py,sha256=Pm24wrSuRGsgeUrHvMPDnNh6RhIZoznnMAUwAkllohk,10661
+_mcp_mesh/pipeline/mcp_startup/heartbeat_loop.py,sha256=4Fgp0_68tlSicyLHkJGB-41Av0jl5sqUeriuu-StNJU,3812
+_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=bo2LeQlIsktbMFfYhX3xlnjAGT9Yj7wG96JYVT5nsyw,25893
+_mcp_mesh/pipeline/mcp_startup/startup_pipeline.py,sha256=56lZzuCo23y3bZkDir05Ip9QlZ7uzt_gbOR32V4tAvo,2350
+_mcp_mesh/pipeline/shared/__init__.py,sha256=s9xmdf6LkoetrVRGd7Zp3NUxcJCW6YZ_yNKzUBcnYys,352
+_mcp_mesh/pipeline/shared/base_step.py,sha256=kyPbNUX79NyGrE_0Q-e-Aek7m1J0TW036njWfv0iZ0I,1080
+_mcp_mesh/pipeline/shared/mesh_pipeline.py,sha256=Lt_Qo46vhIcLR6xcvG0zGZiCvJED1Z1rBoOkZhHgQjk,6496
+_mcp_mesh/pipeline/shared/pipeline_types.py,sha256=iKSgzCoYu3bpIwLViw6lE7T7irEzOm7gpie29lyy7SQ,1504
+_mcp_mesh/shared/__init__.py,sha256=1hDV83TUDjCbgGepMsyf85kzHG1YV25vIvldOITANDo,959
+_mcp_mesh/shared/config_resolver.py,sha256=c6KI4DdpAhIrvuUrw60Of7V4ocA7R6aQB-vkSZSodwc,9708
+_mcp_mesh/shared/content_extractor.py,sha256=culjhieFl_J6EMDv1VFKvS38O3IMhWMs8fHhNuR2rVk,3656
+_mcp_mesh/shared/defaults.py,sha256=DueUHUwHV_FVUwPuuvViJtekyHSpxjqNpoBh9Q7w1mM,4480
+_mcp_mesh/shared/fast_heartbeat_status.py,sha256=OquEsX9ZTbxY1lIsll0Mbb2KDzSJD76sLMOlr3Z73Sc,5188
+_mcp_mesh/shared/fastapi_middleware_manager.py,sha256=LT0Auo2FaCkeSSofFKrZPrYo4XtfYL6ed3CkhcBxSDQ,15513
+_mcp_mesh/shared/health_check_manager.py,sha256=xZjQp-mWdPD64djp06nU5fN-fqi3Q_cIc1PAwMQTRBY,10362
+_mcp_mesh/shared/host_resolver.py,sha256=0BtVSWBDa4ziBB4gieKaf4WJvfqOOGxL32QyE7vy5uE,1617
+_mcp_mesh/shared/logging_config.py,sha256=hf3DCEMDN2WwEaETK721iQkFxX9TENNGmF9H1QhVSls,9076
+_mcp_mesh/shared/server_discovery.py,sha256=Bt6IqU5wHNCjzMiOikMLEI87Id_OMI_F1rsvfzzl0pc,13219
+_mcp_mesh/shared/simple_shutdown.py,sha256=R8jKLasfyx1tuE-UnSUiRHZvDdwONDN9qU97YySnF_I,6634
+_mcp_mesh/shared/sse_parser.py,sha256=1NgnTMr4LQ-tW_cKJYj2oY0B5KDNskNeGlE23LcTOVk,8128
+_mcp_mesh/shared/support_types.py,sha256=k-ICF_UwDkHxQ1D5LwFZrp-UrNb4E5dzw02CRuLW9iI,7264
+_mcp_mesh/tracing/agent_context_helper.py,sha256=BIJ3Kc4Znd6emMAu97aUhSoxSIza3qYUmObLgc9ONiA,4765
+_mcp_mesh/tracing/context.py,sha256=2ozqKEYfx4Qxj64DnbwoVIbMkhNLbaV8BNWtkzAPA7I,2516
+_mcp_mesh/tracing/execution_tracer.py,sha256=dvM6QSbxk5s5rZZsCgVNaXWrr2J4_5I6XAfI1aM0sSA,10268
+_mcp_mesh/tracing/fastapi_tracing_middleware.py,sha256=FXjhA1A1Krk-ngyuOZPc1Ic4Llggv4Wide9OuPmkwCY,6959
+_mcp_mesh/tracing/redis_metadata_publisher.py,sha256=DeFrMt0ZX7k6393dH-xoRS2V5apPR-k80X8ZjrKBHMU,2890
+_mcp_mesh/tracing/trace_context_helper.py,sha256=A0UipvDExePaX-E-4SAp4M8n8uwed9PMo8gibXt1v_Q,6513
+_mcp_mesh/tracing/utils.py,sha256=GWwfvab0tYGr9QAe_zgZjZxgDKTTs0p5Mf8w6WJeWC0,4486
+_mcp_mesh/utils/fastmcp_schema_extractor.py,sha256=NOz3dht21JRKVb_kCTrUhT2MZMmZJz04G6ARcNatVzk,17267
+mesh/__init__.py,sha256=NDQXXD7uuL8Ph48w-Xf7ntEUx5HkB1oZMu3hjGdntTY,3890
+mesh/decorators.py,sha256=sflOWhuJooqIs1tH7DqjOEaGMaVRj4GEDz7R3IQ4fPM,65163
+mesh/helpers.py,sha256=1Y7V6aQvpV8BKfEeeKfjwPJ5g9FjMCzSNifs3se1jkA,12935
+mesh/types.py,sha256=n0MxrBYZJ84xyQWGf_X2ZbVWSAaIcEBkRV7qaCmX6Ac,17008
+mcp_mesh-0.8.0b1.dist-info/METADATA,sha256=uBP_kKN1LJer1-DiU3hwyYvMwO6JJI6BgPjvNZifVwA,5045
+mcp_mesh-0.8.0b1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+mcp_mesh-0.8.0b1.dist-info/licenses/LICENSE,sha256=_EBQHRQThv9FPOLc5eFOUdeeRO0mYwChC7cx60dM1tM,1078
+mcp_mesh-0.8.0b1.dist-info/RECORD,,

mesh/__init__.py

@@ -19,7 +19,8 @@ Use 'import mesh' and then '@mesh.tool()' for consistency with MCP patterns.
 """
 
 from . import decorators
-from .types import LlmMeta, McpMeshAgent, MeshContextModel, MeshLlmAgent, MeshLlmRequest
+from .types import (LlmMeta, McpMeshAgent, MeshContextModel, MeshLlmAgent,
+                    MeshLlmRequest)
 
 # Note: helpers.llm_provider is imported lazily in __getattr__ to avoid
 # initialization timing issues with @mesh.agent auto_run in tests

mesh/decorators.py

@@ -90,9 +90,10 @@ def _start_uvicorn_immediately(http_host: str, http_port: int):
                     """Pure ASGI middleware for trace context and header injection.
 
                     This middleware:
-                    1. Extracts trace context from incoming request headers
-                    2. Sets up trace context for the request lifecycle
-                    3. Injects trace headers into the response (works with SSE)
+                    1. Extracts trace context from incoming request headers AND arguments
+                    2. Strips trace fields (_trace_id, _parent_span) from arguments to avoid validation errors
+                    3. Sets up trace context for the request lifecycle
+                    4. Injects trace headers into the response (works with SSE)
                     """
 
                     def __init__(self, app):
@@ -127,7 +128,7 @@ def _start_uvicorn_immediately(http_host: str, http_port: int):
                                 headers_list, "x-parent-span"
                             )
 
-                            # Setup trace context
+                            # Setup trace context from headers
                             trace_context = {
                                 "trace_id": (
                                     incoming_trace_id if incoming_trace_id else None
@@ -151,6 +152,87 @@ def _start_uvicorn_immediately(http_host: str, http_port: int):
                         except Exception as e:
                             logger.warning(f"Failed to set trace context: {e}")
 
+                        # Create body-modifying receive wrapper to strip trace fields from arguments
+                        # This handles the case where trace context is passed in JSON-RPC arguments
+                        import json as json_module
+
+                        async def receive_with_trace_stripping():
+                            message = await receive()
+                            if message["type"] == "http.request":
+                                body = message.get("body", b"")
+                                if body:
+                                    try:
+                                        payload = json_module.loads(
+                                            body.decode("utf-8")
+                                        )
+                                        if payload.get("method") == "tools/call":
+                                            arguments = payload.get("params", {}).get(
+                                                "arguments", {}
+                                            )
+
+                                            # Extract trace context from arguments if not in headers
+                                            nonlocal trace_id, span_id, parent_span
+                                            if not trace_id and arguments.get(
+                                                "_trace_id"
+                                            ):
+                                                try:
+                                                    from _mcp_mesh.tracing.context import (
+                                                        TraceContext,
+                                                    )
+                                                    from _mcp_mesh.tracing.trace_context_helper import (
+                                                        TraceContextHelper,
+                                                    )
+
+                                                    arg_trace_id = arguments.get(
+                                                        "_trace_id"
+                                                    )
+                                                    arg_parent_span = arguments.get(
+                                                        "_parent_span"
+                                                    )
+                                                    trace_context = {
+                                                        "trace_id": arg_trace_id,
+                                                        "parent_span": arg_parent_span,
+                                                    }
+                                                    TraceContextHelper.setup_request_trace_context(
+                                                        trace_context, logger
+                                                    )
+                                                    current_trace = (
+                                                        TraceContext.get_current()
+                                                    )
+                                                    if current_trace:
+                                                        trace_id = (
+                                                            current_trace.trace_id
+                                                        )
+                                                        span_id = current_trace.span_id
+                                                        parent_span = (
+                                                            current_trace.parent_span
+                                                        )
+                                                except Exception:
+                                                    pass
+
+                                            # Strip trace context fields from arguments
+                                            if (
+                                                "_trace_id" in arguments
+                                                or "_parent_span" in arguments
+                                            ):
+                                                arguments.pop("_trace_id", None)
+                                                arguments.pop("_parent_span", None)
+                                                modified_body = json_module.dumps(
+                                                    payload
+                                                ).encode("utf-8")
+                                                logger.debug(
+                                                    "[TRACE] Stripped trace fields from arguments"
+                                                )
+                                                return {
+                                                    **message,
+                                                    "body": modified_body,
+                                                }
+                                    except Exception as e:
+                                        logger.debug(
+                                            f"[TRACE] Failed to process body: {e}"
+                                        )
+                            return message
+
                         # Wrap send to inject headers before response starts
                         async def send_with_trace_headers(message):
                             if message["type"] == "http.response.start" and trace_id:
@@ -166,7 +248,9 @@ def _start_uvicorn_immediately(http_host: str, http_port: int):
                                 message = {**message, "headers": headers}
                             await send(message)
 
-                        await self.app(scope, receive, send_with_trace_headers)
+                        await self.app(
+                            scope, receive_with_trace_stripping, send_with_trace_headers
+                        )
 
                 app.add_middleware(TraceContextMiddleware)
                 logger.debug(

_mcp_mesh/generated/.openapi-generator/FILES

@@ -1,50 +0,0 @@
-mcp_mesh_registry_client/__init__.py
-mcp_mesh_registry_client/api/__init__.py
-mcp_mesh_registry_client/api/agents_api.py
-mcp_mesh_registry_client/api/health_api.py
-mcp_mesh_registry_client/api/tracing_api.py
-mcp_mesh_registry_client/api_client.py
-mcp_mesh_registry_client/api_response.py
-mcp_mesh_registry_client/configuration.py
-mcp_mesh_registry_client/exceptions.py
-mcp_mesh_registry_client/models/__init__.py
-mcp_mesh_registry_client/models/agent_info.py
-mcp_mesh_registry_client/models/agent_metadata.py
-mcp_mesh_registry_client/models/agent_metadata_dependencies_inner.py
-mcp_mesh_registry_client/models/agent_metadata_dependencies_inner_one_of.py
-mcp_mesh_registry_client/models/agent_registration.py
-mcp_mesh_registry_client/models/agent_registration_metadata.py
-mcp_mesh_registry_client/models/agents_list_response.py
-mcp_mesh_registry_client/models/capability_info.py
-mcp_mesh_registry_client/models/decorator_agent_metadata.py
-mcp_mesh_registry_client/models/decorator_agent_request.py
-mcp_mesh_registry_client/models/decorator_info.py
-mcp_mesh_registry_client/models/dependency_info.py
-mcp_mesh_registry_client/models/dependency_resolution_info.py
-mcp_mesh_registry_client/models/error_response.py
-mcp_mesh_registry_client/models/health_response.py
-mcp_mesh_registry_client/models/heartbeat_request.py
-mcp_mesh_registry_client/models/heartbeat_request_metadata.py
-mcp_mesh_registry_client/models/heartbeat_response.py
-mcp_mesh_registry_client/models/llm_provider.py
-mcp_mesh_registry_client/models/llm_provider_resolution_info.py
-mcp_mesh_registry_client/models/llm_tool_filter.py
-mcp_mesh_registry_client/models/llm_tool_filter_filter_inner.py
-mcp_mesh_registry_client/models/llm_tool_filter_filter_inner_one_of.py
-mcp_mesh_registry_client/models/llm_tool_info.py
-mcp_mesh_registry_client/models/llm_tool_resolution_info.py
-mcp_mesh_registry_client/models/mesh_agent_register_metadata.py
-mcp_mesh_registry_client/models/mesh_agent_registration.py
-mcp_mesh_registry_client/models/mesh_registration_response.py
-mcp_mesh_registry_client/models/mesh_registration_response_dependencies_resolved_value_inner.py
-mcp_mesh_registry_client/models/mesh_tool_dependency_registration.py
-mcp_mesh_registry_client/models/mesh_tool_register_metadata.py
-mcp_mesh_registry_client/models/mesh_tool_registration.py
-mcp_mesh_registry_client/models/registration_response.py
-mcp_mesh_registry_client/models/resolved_llm_provider.py
-mcp_mesh_registry_client/models/rich_dependency.py
-mcp_mesh_registry_client/models/root_response.py
-mcp_mesh_registry_client/models/standardized_dependency.py
-mcp_mesh_registry_client/models/trace_event.py
-mcp_mesh_registry_client/py.typed
-mcp_mesh_registry_client/rest.py

_mcp_mesh/generated/.openapi-generator/VERSION

@@ -1 +0,0 @@
-7.13.0

_mcp_mesh/generated/.openapi-generator-ignore

@@ -1,15 +0,0 @@
-# Skip packaging files - we use this as part of our main package
-pyproject.toml
-setup.py
-setup.cfg
-requirements.txt
-test-requirements.txt
-README.md
-tox.ini
-.travis.yml
-.gitlab-ci.yml
-.github/
-git_push.sh
-.gitignore
-test/
-docs/