mcp-mesh 0.8.0b7__py3-none-any.whl → 0.8.0b9__py3-none-any.whl

_mcp_mesh/__init__.py

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

_mcp_mesh/engine/mesh_llm_agent_injector.py

@@ -161,36 +161,44 @@ class MeshLlmAgentInjector(BaseInjector):
         # Create UnifiedMCPProxy for the provider
         provider_proxy = self._create_provider_proxy(provider_data)
 
-        # Update llm_agents data with provider_proxy and vendor (Phase 2)
-        if function_id in self._llm_agents:
-            self._llm_agents[function_id]["provider_proxy"] = provider_proxy
+        # Update only provider-related fields, preserving tool data if already set.
+        # This avoids race conditions where provider and tools updates can arrive in any order.
+        if function_id not in self._llm_agents:
+            self._llm_agents[function_id] = {}
 
-            # Phase 2: Extract vendor from provider_data for handler selection
-            vendor = provider_data.get("vendor", "unknown")
-            self._llm_agents[function_id]["vendor"] = vendor
+        # Phase 2: Extract vendor from provider_data for handler selection
+        vendor = provider_data.get("vendor", "unknown")
 
+        self._llm_agents[function_id]["provider_proxy"] = provider_proxy
+        self._llm_agents[function_id]["vendor"] = vendor
+
+        logger.info(
+            f"✅ Set provider proxy for '{function_id}': {provider_proxy.function_name} at {provider_proxy.endpoint} (vendor={vendor})"
+        )
+
+        # Re-create and update MeshLlmAgent with new provider (only if tools are also available)
+        # Get the function wrapper from DecoratorRegistry
+        llm_agents = DecoratorRegistry.get_mesh_llm_agents()
+        wrapper = None
+        for agent_func_id, metadata in llm_agents.items():
+            if metadata.function_id == function_id:
+                wrapper = metadata.function
+                break
+
+        # Only update wrapper if we have both tools and provider (tools_metadata indicates tools were processed)
+        if (
+            wrapper
+            and hasattr(wrapper, "_mesh_update_llm_agent")
+            and "tools_metadata" in self._llm_agents[function_id]
+        ):
+            llm_agent = self._create_llm_agent(function_id)
+            wrapper._mesh_update_llm_agent(llm_agent)
             logger.info(
-                f"✅ Set provider proxy for '{function_id}': {provider_proxy.function_name} at {provider_proxy.endpoint} (vendor={vendor})"
+                f"🔄 Updated wrapper with new MeshLlmAgent (with provider) for '{function_id}'"
             )
-
-            # Re-create and update MeshLlmAgent with new provider
-            # Get the function wrapper from DecoratorRegistry
-            llm_agents = DecoratorRegistry.get_mesh_llm_agents()
-            wrapper = None
-            for agent_func_id, metadata in llm_agents.items():
-                if metadata.function_id == function_id:
-                    wrapper = metadata.function
-                    break
-
-            if wrapper and hasattr(wrapper, "_mesh_update_llm_agent"):
-                llm_agent = self._create_llm_agent(function_id)
-                wrapper._mesh_update_llm_agent(llm_agent)
-                logger.info(
-                    f"🔄 Updated wrapper with new MeshLlmAgent (with provider) for '{function_id}'"
-                )
-        else:
-            logger.warning(
-                f"⚠️ Function '{function_id}' not found in _llm_agents, cannot set provider proxy"
+        elif wrapper and hasattr(wrapper, "_mesh_update_llm_agent"):
+            logger.debug(
+                f"⏳ Provider set for '{function_id}', waiting for tools before updating wrapper"
             )
 
     def _create_provider_proxy(self, provider_data: dict[str, Any]) -> UnifiedMCPProxy:
@@ -273,21 +281,23 @@ class MeshLlmAgentInjector(BaseInjector):
                 logger.error(f"❌ Error creating proxy for tool {tool_name}: {e}")
                 # Continue processing other tools
 
-        # Provider proxy will be set separately via process_llm_providers()
-        # (v0.6.1 - providers come from llm_providers field, not dependencies)
-        provider_proxy = None
-
-        # Store LLM agent data with both metadata and proxies
-        # Keep original tool metadata for schema building
-        self._llm_agents[function_id] = {
-            "config": llm_metadata.config,
-            "output_type": llm_metadata.output_type,
-            "param_name": llm_metadata.param_name,
-            "tools_metadata": tools,  # Original metadata for schema building
-            "tools_proxies": tool_proxies_map,  # Proxies for execution
-            "function": llm_metadata.function,
-            "provider_proxy": provider_proxy,  # Provider proxy for mesh delegation
-        }
+        # Update only tool-related fields, preserving provider_proxy if already set.
+        # Provider proxy is managed separately by process_llm_providers().
+        # This avoids race conditions where tools update wipes out provider resolution.
+        if function_id not in self._llm_agents:
+            self._llm_agents[function_id] = {}
+
+        self._llm_agents[function_id].update(
+            {
+                "config": llm_metadata.config,
+                "output_type": llm_metadata.output_type,
+                "param_name": llm_metadata.param_name,
+                "tools_metadata": tools,  # Original metadata for schema building
+                "tools_proxies": tool_proxies_map,  # Proxies for execution
+                "function": llm_metadata.function,
+                # Note: provider_proxy is NOT set here - managed by _process_function_provider
+            }
+        )
 
         logger.info(
             f"✅ Processed {len(tool_proxies_map)} tools for LLM function '{function_id}'"
@@ -431,8 +441,7 @@ class MeshLlmAgentInjector(BaseInjector):
                     if is_template:
                         # Templates enabled - create per-call agent with context
                         # Import signature analyzer for context detection
-                        from .signature_analyzer import \
-                            get_context_parameter_name
+                        from .signature_analyzer import get_context_parameter_name
 
                         # Detect context parameter
                         context_param_name = config_dict.get("context_param")

{mcp_mesh-0.8.0b7.dist-info → mcp_mesh-0.8.0b9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcp-mesh
-Version: 0.8.0b7
+Version: 0.8.0b9
 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,7 +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.0b7
+Requires-Dist: mcp-mesh-core>=0.8.0b9
 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.0b7.dist-info → mcp_mesh-0.8.0b9.dist-info}/RECORD

@@ -1,4 +1,4 @@
-_mcp_mesh/__init__.py,sha256=Ch5NR3kHKmbIo27kH3OATMJfGUj_QdL1zUoKhqpa2Io,2721
+_mcp_mesh/__init__.py,sha256=Pi7UIWRma8F9zTC-gcmeT5ApFCxpepQYRYMvWRVfDtE,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
@@ -10,7 +10,7 @@ _mcp_mesh/engine/http_wrapper.py,sha256=Simd6IEsLO2FXQOuf1WEx57SBN6DSr5RzphXnk0a
 _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=4yp8Lmc-QtBUUirDy1Fmbtl2myybYv6HtnRaaTXcBg0,28372
+_mcp_mesh/engine/mesh_llm_agent_injector.py,sha256=CHfyXWBjRPdsWtV5LTr1TsVPqKu6H0I0xZ2e62MB1p4,28820
 _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
@@ -79,7 +79,7 @@ mesh/__init__.py,sha256=avMnUHkNAK7VgON2OhXkrFB290gr1HErghmTZpOXr-U,4207
 mesh/decorators.py,sha256=3h3tEhKWrlxBDYjP7vVM7iMPy7nx1oVSWVqzugC6WPM,67580
 mesh/helpers.py,sha256=1Y7V6aQvpV8BKfEeeKfjwPJ5g9FjMCzSNifs3se1jkA,12935
 mesh/types.py,sha256=vr0CKyPbP6lHgxj9kh_GMSLo3xkJ66PFPV_opfRb1H4,17772
-mcp_mesh-0.8.0b7.dist-info/METADATA,sha256=qHT5IkNk-JUQKqndyLF0Wjm9Mb_zGMBB2MN-uvgwepI,5040
-mcp_mesh-0.8.0b7.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-mcp_mesh-0.8.0b7.dist-info/licenses/LICENSE,sha256=_EBQHRQThv9FPOLc5eFOUdeeRO0mYwChC7cx60dM1tM,1078
-mcp_mesh-0.8.0b7.dist-info/RECORD,,
+mcp_mesh-0.8.0b9.dist-info/METADATA,sha256=-M8ClhDxH5Y4v4lCg882IS6ebAzgNXTvG-nQrGG_Mzw,5040
+mcp_mesh-0.8.0b9.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+mcp_mesh-0.8.0b9.dist-info/licenses/LICENSE,sha256=_EBQHRQThv9FPOLc5eFOUdeeRO0mYwChC7cx60dM1tM,1078
+mcp_mesh-0.8.0b9.dist-info/RECORD,,