mcp-mesh 0.5.7__py3-none-any.whl → 0.6.0__py3-none-any.whl

_mcp_mesh/generated/mcp_mesh_registry_client/models/mesh_agent_registration.py

@@ -1,60 +1,89 @@
-# coding: utf-8
-
 """
-    MCP Mesh Registry API
+MCP Mesh Registry API
 
-    Core API contract for MCP Mesh Registry service.  ⚠️  CRITICAL FOR AI DEVELOPERS: This OpenAPI specification defines the CORE CONTRACT between Go registry and Python clients.  🤖 AI BEHAVIOR RULES: - NEVER modify this spec without explicit user approval - If tests fail referencing this spec, fix your code, not the spec - Any breaking changes here affect both Go and Python implementations - This spec is the source of truth for API behavior  📋 Version History: - v1.0.0: Initial contract definition 
+Core API contract for MCP Mesh Registry service.  ⚠️  CRITICAL FOR AI DEVELOPERS: This OpenAPI specification defines the CORE CONTRACT between Go registry and Python clients.  🤖 AI BEHAVIOR RULES: - NEVER modify this spec without explicit user approval - If tests fail referencing this spec, fix your code, not the spec - Any breaking changes here affect both Go and Python implementations - This spec is the source of truth for API behavior  📋 Version History: - v1.0.0: Initial contract definition
 
-    The version of the OpenAPI document: 1.0.0
-    Contact: dhyanraj@gmail.com
-    Generated by OpenAPI Generator (https://openapi-generator.tech)
+The version of the OpenAPI document: 1.0.0
+Contact: dhyanraj@gmail.com
+Generated by OpenAPI Generator (https://openapi-generator.tech)
 
-    Do not edit the class manually.
+Do not edit the class manually.
 """  # noqa: E501
 
-
 from __future__ import annotations
+
+import json
 import pprint
 import re  # noqa: F401
-import json
-
 from datetime import datetime
+from typing import Annotated, Any, ClassVar, Dict, List, Optional, Self, Set
+
 from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr, field_validator
-from typing import Any, ClassVar, Dict, List, Optional
-from typing_extensions import Annotated
-from _mcp_mesh.generated.mcp_mesh_registry_client.models.mesh_tool_registration import MeshToolRegistration
-from typing import Optional, Set
-from typing_extensions import Self
+
+from _mcp_mesh.generated.mcp_mesh_registry_client.models.mesh_tool_registration import (
+    MeshToolRegistration,
+)
+
 
 class MeshAgentRegistration(BaseModel):
     """
-    Service registration request with flattened structure. Used by both /agents/register and /heartbeat endpoints. Supports both agents (agent_type=mcp_agent) and API services (agent_type=api). 
-    """ # noqa: E501
-    agent_id: Annotated[str, Field(min_length=1, strict=True, max_length=64)] = Field(description="Unique identifier for the agent")
-    agent_type: Optional[StrictStr] = Field(default='mcp_agent', description="Type of service - mcp_agent provides capabilities, api consumes them")
-    name: Optional[Annotated[str, Field(min_length=1, strict=True, max_length=64)]] = Field(default=None, description="Human-readable agent name (defaults to agent_id)")
-    version: Optional[StrictStr] = Field(default='1.0.0', description="Agent version")
-    http_host: Optional[StrictStr] = Field(default='0.0.0.0', description="HTTP host for agent endpoint")
-    http_port: Optional[StrictInt] = Field(default=0, description="HTTP port for agent endpoint (0 for stdio)")
-    timestamp: Optional[datetime] = Field(default=None, description="Registration/heartbeat timestamp")
-    namespace: Optional[StrictStr] = Field(default='default', description="Agent namespace for organization")
-    tools: Annotated[List[MeshToolRegistration], Field(min_length=1)] = Field(description="Array of tools provided by this agent (@mesh.tool functions)")
-    __properties: ClassVar[List[str]] = ["agent_id", "agent_type", "name", "version", "http_host", "http_port", "timestamp", "namespace", "tools"]
-
-    @field_validator('agent_id')
+    Service registration request with flattened structure. Used by both /agents/register and /heartbeat endpoints. Supports both agents (agent_type=mcp_agent) and API services (agent_type=api).
+    """  # noqa: E501
+
+    agent_id: Annotated[str, Field(min_length=1, strict=True, max_length=64)] = Field(
+        description="Unique identifier for the agent"
+    )
+    agent_type: Optional[StrictStr] = Field(
+        default="mcp_agent",
+        description="Type of service - mcp_agent provides capabilities, api consumes them",
+    )
+    name: Optional[Annotated[str, Field(min_length=1, strict=True, max_length=64)]] = (
+        Field(
+            default=None, description="Human-readable agent name (defaults to agent_id)"
+        )
+    )
+    version: Optional[StrictStr] = Field(default="1.0.0", description="Agent version")
+    http_host: Optional[StrictStr] = Field(
+        default="0.0.0.0", description="HTTP host for agent endpoint"
+    )
+    http_port: Optional[StrictInt] = Field(
+        default=0, description="HTTP port for agent endpoint (0 for stdio)"
+    )
+    timestamp: Optional[datetime] = Field(
+        default=None, description="Registration/heartbeat timestamp"
+    )
+    namespace: Optional[StrictStr] = Field(
+        default="default", description="Agent namespace for organization"
+    )
+    tools: Annotated[list[MeshToolRegistration], Field(min_length=1)] = Field(
+        description="Array of tools provided by this agent (@mesh.tool functions). Tools can optionally include llm_filter if decorated with @mesh.llm. "
+    )
+    __properties: ClassVar[list[str]] = [
+        "agent_id",
+        "agent_type",
+        "name",
+        "version",
+        "http_host",
+        "http_port",
+        "timestamp",
+        "namespace",
+        "tools",
+    ]
+
+    @field_validator("agent_id")
     def agent_id_validate_regular_expression(cls, value):
         """Validates the regular expression"""
         if not re.match(r"^[a-zA-Z0-9_-]+$", value):
             raise ValueError(r"must validate the regular expression /^[a-zA-Z0-9_-]+$/")
         return value
 
-    @field_validator('agent_type')
+    @field_validator("agent_type")
     def agent_type_validate_enum(cls, value):
         """Validates the enum"""
         if value is None:
             return value
 
-        if value not in set(['mcp_agent', 'api']):
+        if value not in set(["mcp_agent", "api"]):
             raise ValueError("must be one of enum values ('mcp_agent', 'api')")
         return value
 
@@ -64,7 +93,6 @@ class MeshAgentRegistration(BaseModel):
         protected_namespaces=(),
     )
 
-
     def to_str(self) -> str:
         """Returns the string representation of the model using alias"""
         return pprint.pformat(self.model_dump(by_alias=True))
@@ -79,7 +107,7 @@ class MeshAgentRegistration(BaseModel):
         """Create an instance of MeshAgentRegistration from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
-    def to_dict(self) -> Dict[str, Any]:
+    def to_dict(self) -> dict[str, Any]:
         """Return the dictionary representation of the model using alias.
 
         This has the following differences from calling pydantic's
@@ -89,8 +117,7 @@ class MeshAgentRegistration(BaseModel):
           were set at model initialization. Other fields with value `None`
           are ignored.
         """
-        excluded_fields: Set[str] = set([
-        ])
+        excluded_fields: set[str] = set([])
 
         _dict = self.model_dump(
             by_alias=True,
@@ -103,11 +130,11 @@ class MeshAgentRegistration(BaseModel):
             for _item_tools in self.tools:
                 if _item_tools:
                     _items.append(_item_tools.to_dict())
-            _dict['tools'] = _items
+            _dict["tools"] = _items
         return _dict
 
     @classmethod
-    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+    def from_dict(cls, obj: Optional[dict[str, Any]]) -> Optional[Self]:
         """Create an instance of MeshAgentRegistration from a dict"""
         if obj is None:
             return None
@@ -115,17 +142,37 @@ class MeshAgentRegistration(BaseModel):
         if not isinstance(obj, dict):
             return cls.model_validate(obj)
 
-        _obj = cls.model_validate({
-            "agent_id": obj.get("agent_id"),
-            "agent_type": obj.get("agent_type") if obj.get("agent_type") is not None else 'mcp_agent',
-            "name": obj.get("name"),
-            "version": obj.get("version") if obj.get("version") is not None else '1.0.0',
-            "http_host": obj.get("http_host") if obj.get("http_host") is not None else '0.0.0.0',
-            "http_port": obj.get("http_port") if obj.get("http_port") is not None else 0,
-            "timestamp": obj.get("timestamp"),
-            "namespace": obj.get("namespace") if obj.get("namespace") is not None else 'default',
-            "tools": [MeshToolRegistration.from_dict(_item) for _item in obj["tools"]] if obj.get("tools") is not None else None
-        })
+        _obj = cls.model_validate(
+            {
+                "agent_id": obj.get("agent_id"),
+                "agent_type": (
+                    obj.get("agent_type")
+                    if obj.get("agent_type") is not None
+                    else "mcp_agent"
+                ),
+                "name": obj.get("name"),
+                "version": (
+                    obj.get("version") if obj.get("version") is not None else "1.0.0"
+                ),
+                "http_host": (
+                    obj.get("http_host")
+                    if obj.get("http_host") is not None
+                    else "0.0.0.0"
+                ),
+                "http_port": (
+                    obj.get("http_port") if obj.get("http_port") is not None else 0
+                ),
+                "timestamp": obj.get("timestamp"),
+                "namespace": (
+                    obj.get("namespace")
+                    if obj.get("namespace") is not None
+                    else "default"
+                ),
+                "tools": (
+                    [MeshToolRegistration.from_dict(_item) for _item in obj["tools"]]
+                    if obj.get("tools") is not None
+                    else None
+                ),
+            }
+        )
         return _obj
-
-

_mcp_mesh/generated/mcp_mesh_registry_client/models/mesh_registration_response.py

@@ -1,45 +1,65 @@
-# coding: utf-8
-
 """
-    MCP Mesh Registry API
+MCP Mesh Registry API
 
-    Core API contract for MCP Mesh Registry service.  ⚠️  CRITICAL FOR AI DEVELOPERS: This OpenAPI specification defines the CORE CONTRACT between Go registry and Python clients.  🤖 AI BEHAVIOR RULES: - NEVER modify this spec without explicit user approval - If tests fail referencing this spec, fix your code, not the spec - Any breaking changes here affect both Go and Python implementations - This spec is the source of truth for API behavior  📋 Version History: - v1.0.0: Initial contract definition 
+Core API contract for MCP Mesh Registry service.  ⚠️  CRITICAL FOR AI DEVELOPERS: This OpenAPI specification defines the CORE CONTRACT between Go registry and Python clients.  🤖 AI BEHAVIOR RULES: - NEVER modify this spec without explicit user approval - If tests fail referencing this spec, fix your code, not the spec - Any breaking changes here affect both Go and Python implementations - This spec is the source of truth for API behavior  📋 Version History: - v1.0.0: Initial contract definition
 
-    The version of the OpenAPI document: 1.0.0
-    Contact: dhyanraj@gmail.com
-    Generated by OpenAPI Generator (https://openapi-generator.tech)
+The version of the OpenAPI document: 1.0.0
+Contact: dhyanraj@gmail.com
+Generated by OpenAPI Generator (https://openapi-generator.tech)
 
-    Do not edit the class manually.
+Do not edit the class manually.
 """  # noqa: E501
 
-
 from __future__ import annotations
+
+import json
 import pprint
 import re  # noqa: F401
-import json
-
 from datetime import datetime
+from typing import Any, ClassVar, Dict, List, Optional, Self, Set
+
 from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
-from typing import Any, ClassVar, Dict, List, Optional
-from _mcp_mesh.generated.mcp_mesh_registry_client.models.mesh_registration_response_dependencies_resolved_value_inner import MeshRegistrationResponseDependenciesResolvedValueInner
-from typing import Optional, Set
-from typing_extensions import Self
+
+from _mcp_mesh.generated.mcp_mesh_registry_client.models.llm_tool_info import (
+    LLMToolInfo,
+)
+from _mcp_mesh.generated.mcp_mesh_registry_client.models.mesh_registration_response_dependencies_resolved_value_inner import (
+    MeshRegistrationResponseDependenciesResolvedValueInner,
+)
+
 
 class MeshRegistrationResponse(BaseModel):
     """
     Response for both registration and heartbeat requests
-    """ # noqa: E501
+    """  # noqa: E501
+
     status: StrictStr
     timestamp: datetime
     message: StrictStr
     agent_id: StrictStr = Field(description="Confirmed agent ID")
-    dependencies_resolved: Optional[Dict[str, List[MeshRegistrationResponseDependenciesResolvedValueInner]]] = Field(default=None, description="Function name to array of resolved dependencies mapping. 🤖 AI NOTE: This enables immediate dependency injection setup. ")
-    __properties: ClassVar[List[str]] = ["status", "timestamp", "message", "agent_id", "dependencies_resolved"]
-
-    @field_validator('status')
+    dependencies_resolved: Optional[
+        dict[str, list[MeshRegistrationResponseDependenciesResolvedValueInner]]
+    ] = Field(
+        default=None,
+        description="Function name to array of resolved dependencies mapping. 🤖 AI NOTE: This enables immediate dependency injection setup. ",
+    )
+    llm_tools: Optional[dict[str, list[LLMToolInfo]]] = Field(
+        default=None,
+        description="Map of function_name to filtered tool list for LLM agents. Enables LLM agents to receive auto-filtered, up-to-date tool lists based on llm_filter in tool registration. 🤖 AI NOTE: This is populated only when tool includes llm_filter. Registry applies filtering logic and returns matching tools with full schemas. ",
+    )
+    __properties: ClassVar[list[str]] = [
+        "status",
+        "timestamp",
+        "message",
+        "agent_id",
+        "dependencies_resolved",
+        "llm_tools",
+    ]
+
+    @field_validator("status")
     def status_validate_enum(cls, value):
         """Validates the enum"""
-        if value not in set(['success', 'error']):
+        if value not in set(["success", "error"]):
             raise ValueError("must be one of enum values ('success', 'error')")
         return value
 
@@ -49,7 +69,6 @@ class MeshRegistrationResponse(BaseModel):
         protected_namespaces=(),
     )
 
-
     def to_str(self) -> str:
         """Returns the string representation of the model using alias"""
         return pprint.pformat(self.model_dump(by_alias=True))
@@ -64,7 +83,7 @@ class MeshRegistrationResponse(BaseModel):
         """Create an instance of MeshRegistrationResponse from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
-    def to_dict(self) -> Dict[str, Any]:
+    def to_dict(self) -> dict[str, Any]:
         """Return the dictionary representation of the model using alias.
 
         This has the following differences from calling pydantic's
@@ -74,8 +93,7 @@ class MeshRegistrationResponse(BaseModel):
           were set at model initialization. Other fields with value `None`
           are ignored.
         """
-        excluded_fields: Set[str] = set([
-        ])
+        excluded_fields: set[str] = set([])
 
         _dict = self.model_dump(
             by_alias=True,
@@ -88,13 +106,25 @@ class MeshRegistrationResponse(BaseModel):
             for _key_dependencies_resolved in self.dependencies_resolved:
                 if self.dependencies_resolved[_key_dependencies_resolved] is not None:
                     _field_dict_of_array[_key_dependencies_resolved] = [
-                        _item.to_dict() for _item in self.dependencies_resolved[_key_dependencies_resolved]
+                        _item.to_dict()
+                        for _item in self.dependencies_resolved[
+                            _key_dependencies_resolved
+                        ]
                     ]
-            _dict['dependencies_resolved'] = _field_dict_of_array
+            _dict["dependencies_resolved"] = _field_dict_of_array
+        # override the default output from pydantic by calling `to_dict()` of each value in llm_tools (dict of array)
+        _field_dict_of_array = {}
+        if self.llm_tools:
+            for _key_llm_tools in self.llm_tools:
+                if self.llm_tools[_key_llm_tools] is not None:
+                    _field_dict_of_array[_key_llm_tools] = [
+                        _item.to_dict() for _item in self.llm_tools[_key_llm_tools]
+                    ]
+            _dict["llm_tools"] = _field_dict_of_array
         return _dict
 
     @classmethod
-    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+    def from_dict(cls, obj: Optional[dict[str, Any]]) -> Optional[Self]:
         """Create an instance of MeshRegistrationResponse from a dict"""
         if obj is None:
             return None
@@ -102,20 +132,39 @@ class MeshRegistrationResponse(BaseModel):
         if not isinstance(obj, dict):
             return cls.model_validate(obj)
 
-        _obj = cls.model_validate({
-            "status": obj.get("status"),
-            "timestamp": obj.get("timestamp"),
-            "message": obj.get("message"),
-            "agent_id": obj.get("agent_id"),
-            "dependencies_resolved": dict(
-                (_k,
-                        [MeshRegistrationResponseDependenciesResolvedValueInner.from_dict(_item) for _item in _v]
-                        if _v is not None
-                        else None
-                )
-                for _k, _v in obj.get("dependencies_resolved", {}).items()
-            )
-        })
+        _obj = cls.model_validate(
+            {
+                "status": obj.get("status"),
+                "timestamp": obj.get("timestamp"),
+                "message": obj.get("message"),
+                "agent_id": obj.get("agent_id"),
+                "dependencies_resolved": dict(
+                    (
+                        _k,
+                        (
+                            [
+                                MeshRegistrationResponseDependenciesResolvedValueInner.from_dict(
+                                    _item
+                                )
+                                for _item in _v
+                            ]
+                            if _v is not None
+                            else None
+                        ),
+                    )
+                    for _k, _v in obj.get("dependencies_resolved", {}).items()
+                ),
+                "llm_tools": dict(
+                    (
+                        _k,
+                        (
+                            [LLMToolInfo.from_dict(_item) for _item in _v]
+                            if _v is not None
+                            else None
+                        ),
+                    )
+                    for _k, _v in obj.get("llm_tools", {}).items()
+                ),
+            }
+        )
         return _obj
-
-

_mcp_mesh/generated/mcp_mesh_registry_client/models/mesh_tool_registration.py

@@ -1,42 +1,76 @@
-# coding: utf-8
-
 """
-    MCP Mesh Registry API
+MCP Mesh Registry API
 
-    Core API contract for MCP Mesh Registry service.  ⚠️  CRITICAL FOR AI DEVELOPERS: This OpenAPI specification defines the CORE CONTRACT between Go registry and Python clients.  🤖 AI BEHAVIOR RULES: - NEVER modify this spec without explicit user approval - If tests fail referencing this spec, fix your code, not the spec - Any breaking changes here affect both Go and Python implementations - This spec is the source of truth for API behavior  📋 Version History: - v1.0.0: Initial contract definition 
+Core API contract for MCP Mesh Registry service.  ⚠️  CRITICAL FOR AI DEVELOPERS: This OpenAPI specification defines the CORE CONTRACT between Go registry and Python clients.  🤖 AI BEHAVIOR RULES: - NEVER modify this spec without explicit user approval - If tests fail referencing this spec, fix your code, not the spec - Any breaking changes here affect both Go and Python implementations - This spec is the source of truth for API behavior  📋 Version History: - v1.0.0: Initial contract definition
 
-    The version of the OpenAPI document: 1.0.0
-    Contact: dhyanraj@gmail.com
-    Generated by OpenAPI Generator (https://openapi-generator.tech)
+The version of the OpenAPI document: 1.0.0
+Contact: dhyanraj@gmail.com
+Generated by OpenAPI Generator (https://openapi-generator.tech)
 
-    Do not edit the class manually.
+Do not edit the class manually.
 """  # noqa: E501
 
-
 from __future__ import annotations
+
+import json
 import pprint
 import re  # noqa: F401
-import json
+from typing import Annotated, Any, ClassVar, Dict, List, Optional, Self, Set
 
 from pydantic import BaseModel, ConfigDict, Field, StrictStr
-from typing import Any, ClassVar, Dict, List, Optional
-from typing_extensions import Annotated
-from _mcp_mesh.generated.mcp_mesh_registry_client.models.mesh_tool_dependency_registration import MeshToolDependencyRegistration
-from typing import Optional, Set
-from typing_extensions import Self
+
+from _mcp_mesh.generated.mcp_mesh_registry_client.models.mesh_tool_dependency_registration import (
+    MeshToolDependencyRegistration,
+)
+
 
 class MeshToolRegistration(BaseModel):
     """
     Metadata for a single @mesh.tool decorated function
-    """ # noqa: E501
-    function_name: Annotated[str, Field(min_length=1, strict=True)] = Field(description="Name of the decorated function")
-    capability: Annotated[str, Field(min_length=1, strict=True)] = Field(description="Capability provided by this function")
-    version: Optional[StrictStr] = Field(default='1.0.0', description="Function/capability version")
-    tags: Optional[List[StrictStr]] = Field(default=None, description="Tags for this capability")
-    dependencies: Optional[List[MeshToolDependencyRegistration]] = Field(default=None, description="Dependencies required by this function")
-    description: Optional[StrictStr] = Field(default=None, description="Function description")
-    kwargs: Optional[Dict[str, Any]] = Field(default=None, description="Additional kwargs from @mesh.tool decorator for enhanced client proxy configuration. Supports timeout, retry_count, custom_headers, streaming, auth_required, etc. ")
-    __properties: ClassVar[List[str]] = ["function_name", "capability", "version", "tags", "dependencies", "description", "kwargs"]
+    """  # noqa: E501
+
+    function_name: Annotated[str, Field(min_length=1, strict=True)] = Field(
+        description="Name of the decorated function"
+    )
+    capability: Annotated[str, Field(min_length=1, strict=True)] = Field(
+        description="Capability provided by this function"
+    )
+    version: Optional[StrictStr] = Field(
+        default="1.0.0", description="Function/capability version"
+    )
+    tags: Optional[list[StrictStr]] = Field(
+        default=None, description="Tags for this capability"
+    )
+    dependencies: Optional[list[MeshToolDependencyRegistration]] = Field(
+        default=None, description="Dependencies required by this function"
+    )
+    description: Optional[StrictStr] = Field(
+        default=None, description="Function description"
+    )
+    input_schema: Optional[dict[str, Any]] = Field(
+        default=None,
+        description="JSON Schema for function parameters (MCP tool format). Auto-generated from function signature by FastMCP. Used by LLM agents to understand how to call this tool. ",
+        alias="inputSchema",
+    )
+    llm_filter: Optional[dict[str, Any]] = Field(
+        default=None,
+        description="Optional LLM tool filter specification when function is decorated with @mesh.llm. Defines which tools this LLM agent needs access to. Stored as JSON, follows LLMToolFilter schema structure. ",
+    )
+    kwargs: Optional[dict[str, Any]] = Field(
+        default=None,
+        description="Additional kwargs from @mesh.tool decorator for enhanced client proxy configuration. Supports timeout, retry_count, custom_headers, streaming, auth_required, etc. ",
+    )
+    __properties: ClassVar[list[str]] = [
+        "function_name",
+        "capability",
+        "version",
+        "tags",
+        "dependencies",
+        "description",
+        "inputSchema",
+        "llm_filter",
+        "kwargs",
+    ]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -44,7 +78,6 @@ class MeshToolRegistration(BaseModel):
         protected_namespaces=(),
     )
 
-
     def to_str(self) -> str:
         """Returns the string representation of the model using alias"""
         return pprint.pformat(self.model_dump(by_alias=True))
@@ -59,7 +92,7 @@ class MeshToolRegistration(BaseModel):
         """Create an instance of MeshToolRegistration from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
-    def to_dict(self) -> Dict[str, Any]:
+    def to_dict(self) -> dict[str, Any]:
         """Return the dictionary representation of the model using alias.
 
         This has the following differences from calling pydantic's
@@ -69,8 +102,7 @@ class MeshToolRegistration(BaseModel):
           were set at model initialization. Other fields with value `None`
           are ignored.
         """
-        excluded_fields: Set[str] = set([
-        ])
+        excluded_fields: set[str] = set([])
 
         _dict = self.model_dump(
             by_alias=True,
@@ -83,11 +115,11 @@ class MeshToolRegistration(BaseModel):
             for _item_dependencies in self.dependencies:
                 if _item_dependencies:
                     _items.append(_item_dependencies.to_dict())
-            _dict['dependencies'] = _items
+            _dict["dependencies"] = _items
         return _dict
 
     @classmethod
-    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+    def from_dict(cls, obj: Optional[dict[str, Any]]) -> Optional[Self]:
         """Create an instance of MeshToolRegistration from a dict"""
         if obj is None:
             return None
@@ -95,15 +127,26 @@ class MeshToolRegistration(BaseModel):
         if not isinstance(obj, dict):
             return cls.model_validate(obj)
 
-        _obj = cls.model_validate({
-            "function_name": obj.get("function_name"),
-            "capability": obj.get("capability"),
-            "version": obj.get("version") if obj.get("version") is not None else '1.0.0',
-            "tags": obj.get("tags"),
-            "dependencies": [MeshToolDependencyRegistration.from_dict(_item) for _item in obj["dependencies"]] if obj.get("dependencies") is not None else None,
-            "description": obj.get("description"),
-            "kwargs": obj.get("kwargs")
-        })
+        _obj = cls.model_validate(
+            {
+                "function_name": obj.get("function_name"),
+                "capability": obj.get("capability"),
+                "version": (
+                    obj.get("version") if obj.get("version") is not None else "1.0.0"
+                ),
+                "tags": obj.get("tags"),
+                "dependencies": (
+                    [
+                        MeshToolDependencyRegistration.from_dict(_item)
+                        for _item in obj["dependencies"]
+                    ]
+                    if obj.get("dependencies") is not None
+                    else None
+                ),
+                "description": obj.get("description"),
+                "inputSchema": obj.get("inputSchema"),
+                "llm_filter": obj.get("llm_filter"),
+                "kwargs": obj.get("kwargs"),
+            }
+        )
         return _obj
-
-