nvidia-nat-mcp 1.4.0a20260107__py3-none-any.whl → 1.4.0a20260128__py3-none-any.whl

nat/plugins/mcp/cli/commands.py

@@ -24,6 +24,7 @@ from typing import cast
 import click
 from pydantic import BaseModel
 
+from nat.builder.function import FunctionGroup
 from nat.cli.commands.start import start_command
 
 logger = logging.getLogger(__name__)
@@ -310,8 +311,9 @@ async def list_tools_via_function_group(
         fns = await group.get_accessible_functions()
 
         def to_tool_entry(full_name: str, fn_obj) -> dict[str, str | None]:
-            # full_name like "mcp_client.<tool>"
-            name = full_name.split(".", 1)[1] if "." in full_name else full_name
+            # full_name like "mcp_client__<tool>"
+            sep = FunctionGroup.SEPARATOR
+            name = full_name.split(sep, 1)[1] if sep in full_name else full_name
             schema = getattr(fn_obj, 'input_schema', None)
             if schema is None:
                 schema_str = None
@@ -327,10 +329,12 @@ async def list_tools_via_function_group(
             return {"name": name, "description": getattr(fn_obj, 'description', ''), "input_schema": schema_str}
 
         if tool_name:
-            full = f"mcp_client.{tool_name}"
+            full = f"mcp_client{FunctionGroup.SEPARATOR}{tool_name}"
             fn = fns.get(full)
             if fn is not None:
                 tools.append(to_tool_entry(full, fn))
+            else:
+                logger.debug(f"Tool '{tool_name}' not found. Available: {list(fns.keys())}")
         else:
             for full, fn in fns.items():
                 tools.append(to_tool_entry(full, fn))
@@ -908,10 +912,10 @@ async def call_tool_and_print(command: str | None,
 
         group = await builder.add_function_group("mcp_client", group_cfg)
         fns = await group.get_accessible_functions()
-        full = f"mcp_client.{tool_name}"
+        full = f"mcp_client{FunctionGroup.SEPARATOR}{tool_name}"
         fn = fns.get(full)
         if fn is None:
-            raise RuntimeError(f"Tool '{tool_name}' not found")
+            raise RuntimeError(f"Tool '{tool_name}' not found. Available: {list(fns.keys())}")
         # The group exposes a Function that we can invoke with kwargs
         result = await fn.acall_invoke(**(tool_args or {}))
         # Ensure string output for terminal

nat/plugins/mcp/server/front_end_plugin_worker.py

@@ -184,7 +184,11 @@ class MCPFrontEndPluginWorkerBase(ABC):
 
         # Register each function with MCP, passing SessionManager for observability
         for function_name, session_manager in session_managers.items():
-            register_function_with_mcp(mcp, function_name, session_manager, self.memory_profiler)
+            register_function_with_mcp(mcp,
+                                       function_name,
+                                       session_manager,
+                                       self.memory_profiler,
+                                       function=functions.get(function_name))
 
         # Add a simple fallback function if no functions were found
         if not session_managers:

nat/plugins/mcp/server/tool_converter.py

@@ -257,7 +257,8 @@ def get_function_description(function: FunctionBase) -> str:
 def register_function_with_mcp(mcp: FastMCP,
                                function_name: str,
                                session_manager: 'SessionManager',
-                               memory_profiler: 'MemoryProfiler | None' = None) -> None:
+                               memory_profiler: 'MemoryProfiler | None' = None,
+                               function: FunctionBase | None = None) -> None:
     """Register a NAT Function as an MCP tool using SessionManager.
 
     Each function is wrapped in a SessionManager
@@ -274,12 +275,15 @@ def register_function_with_mcp(mcp: FastMCP,
     # Get the workflow from the session manager
     workflow = session_manager.workflow
 
-    # Get the input schema from the workflow
-    input_schema = workflow.input_schema
+    # Prefer the function's schema/description when available, fall back to workflow
+    target_function = function or workflow
+
+    # Get the input schema from the most specific object available
+    input_schema = getattr(target_function, "input_schema", workflow.input_schema)
     logger.info("Function %s has input schema: %s", function_name, input_schema)
 
     # Get function description
-    function_description = get_function_description(workflow)
+    function_description = get_function_description(target_function)
 
     # Create and register the wrapper function with MCP
     wrapper_func = create_function_wrapper(function_name, session_manager, input_schema, memory_profiler)

{nvidia_nat_mcp-1.4.0a20260107.dist-info → nvidia_nat_mcp-1.4.0a20260128.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-mcp
-Version: 1.4.0a20260107
+Version: 1.4.0a20260128
 Summary: Subpackage for MCP client integration in NeMo Agent toolkit
 Author: NVIDIA Corporation
 Maintainer: NVIDIA Corporation
@@ -16,8 +16,8 @@ Requires-Python: <3.14,>=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE-3rd-party.txt
 License-File: LICENSE.md
-Requires-Dist: nvidia-nat==v1.4.0a20260107
-Requires-Dist: aiorwlock~=1.5.0
+Requires-Dist: nvidia-nat==v1.4.0a20260128
+Requires-Dist: aiorwlock~=1.5
 Requires-Dist: mcp~=1.25
 Dynamic: license-file
 

{nvidia_nat_mcp-1.4.0a20260107.dist-info → nvidia_nat_mcp-1.4.0a20260128.dist-info}/RECORD

@@ -15,7 +15,7 @@ nat/plugins/mcp/auth/service_account/provider.py,sha256=Gsb44cy-5Ap_1BcG3Ny9fc94
 nat/plugins/mcp/auth/service_account/provider_config.py,sha256=rrzHAo1qmK1yI5CNaadNTVsmh5qtYF1UrYz2Q1nLF2E,5332
 nat/plugins/mcp/auth/service_account/token_client.py,sha256=bGAjq_vG8oo8ZoQfRJjEgigGmF0hgF0umaUOWv0_ZOY,5986
 nat/plugins/mcp/cli/__init__.py,sha256=GLpBQw8r1Nj38vIOluSHsTlLINI24tZ_JJbcWWksBJE,709
-nat/plugins/mcp/cli/commands.py,sha256=cBdNXslsihsHOXbS5MoDbabbEB4milfRHLGOTEodDDg,46602
+nat/plugins/mcp/cli/commands.py,sha256=1ukbLA9V3tMenFKJ5PksEnaPKuJkyITnH9SfuFr7sLE,46882
 nat/plugins/mcp/client/__init__.py,sha256=SFdgClH-7zCAAcujmdvtfiqSz3Xe1KVRX7fviW-45Zg,714
 nat/plugins/mcp/client/client_base.py,sha256=jWNwb20qWIUuIJg7Ldp_WgiJlQmYin4YvVSJ9_ECP5c,26831
 nat/plugins/mcp/client/client_config.py,sha256=qrBjXCKHpMoS2A5i9DoXkN7hdYnGy77FBcZDGZlyb4w,7179
@@ -23,15 +23,15 @@ nat/plugins/mcp/client/client_impl.py,sha256=dygcLnhnWnMh7u45aLvYbGEP1uv02uSb886
 nat/plugins/mcp/server/__init__.py,sha256=Ioza5tZX7qlOicklSVbhcXO3r7EjRLENvV5AtBSusAQ,723
 nat/plugins/mcp/server/front_end_config.py,sha256=yFmEMKcz_ANJnef4Zu4dwGtnfNZ0uzVdc_dYYCLmgCg,5749
 nat/plugins/mcp/server/front_end_plugin.py,sha256=WAqgYFHAswUK1WKk_PIUo6Lyf0pvJLelT3esHhmIHFM,6896
-nat/plugins/mcp/server/front_end_plugin_worker.py,sha256=OBk1sMjH_1coBXj3dYXNWyoK4wofVQERt3_0gLscP4I,17992
+nat/plugins/mcp/server/front_end_plugin_worker.py,sha256=2_PQ938IbYKFKDsmDw6rcgBWHZdbxZetUopU4Ik1FDQ,18187
 nat/plugins/mcp/server/introspection_token_verifier.py,sha256=oyhZkCTo6u3w7m469DicF57BNmhTPLcMYtWuHvcCsRg,2840
 nat/plugins/mcp/server/memory_profiler.py,sha256=KjJw4ARSmEFult2MorFySiucog0bD8ClI9ZbkJhIwns,12816
 nat/plugins/mcp/server/register_frontend.py,sha256=gfHfQaQ4NhkZkDSrFur1bxbG93kbAGBtumm4eKYOzKE,1178
-nat/plugins/mcp/server/tool_converter.py,sha256=Y-3Cr8jNrRjQFFFmmILIeUI6kO3C_CUpCZ2k9NJnJdk,11339
-nvidia_nat_mcp-1.4.0a20260107.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
-nvidia_nat_mcp-1.4.0a20260107.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-nvidia_nat_mcp-1.4.0a20260107.dist-info/METADATA,sha256=9otJsZHNUMeKwR5lTUW4cTfwSZ9VKrIcODC38JNm2Mc,2326
-nvidia_nat_mcp-1.4.0a20260107.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_mcp-1.4.0a20260107.dist-info/entry_points.txt,sha256=t5HYRfXR-dyZf6BWRz1u1TqPprKpCWt-WLHKQXfiLLU,231
-nvidia_nat_mcp-1.4.0a20260107.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_mcp-1.4.0a20260107.dist-info/RECORD,,
+nat/plugins/mcp/server/tool_converter.py,sha256=v35IILT5S2jE7ZTYQzm8-4CMu5Xa21BACeRP1MS1vzA,11608
+nvidia_nat_mcp-1.4.0a20260128.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
+nvidia_nat_mcp-1.4.0a20260128.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+nvidia_nat_mcp-1.4.0a20260128.dist-info/METADATA,sha256=dxVfJTfdSV1FteEA8baIBUjx70l7Da7WSv1H9Oi6tuY,2324
+nvidia_nat_mcp-1.4.0a20260128.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+nvidia_nat_mcp-1.4.0a20260128.dist-info/entry_points.txt,sha256=t5HYRfXR-dyZf6BWRz1u1TqPprKpCWt-WLHKQXfiLLU,231
+nvidia_nat_mcp-1.4.0a20260128.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_mcp-1.4.0a20260128.dist-info/RECORD,,

{nvidia_nat_mcp-1.4.0a20260107.dist-info → nvidia_nat_mcp-1.4.0a20260128.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any