nvidia-nat-mcp 1.4.0a20260109__py3-none-any.whl → 1.4.0a20260125__py3-none-any.whl

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.0a20260109.dist-info → nvidia_nat_mcp-1.4.0a20260125.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-mcp
-Version: 1.4.0a20260109
+Version: 1.4.0a20260125
 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.0a20260109
-Requires-Dist: aiorwlock~=1.5.0
+Requires-Dist: nvidia-nat==v1.4.0a20260125
+Requires-Dist: aiorwlock~=1.5
 Requires-Dist: mcp~=1.25
 Dynamic: license-file
 

{nvidia_nat_mcp-1.4.0a20260109.dist-info → nvidia_nat_mcp-1.4.0a20260125.dist-info}/RECORD

@@ -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.0a20260109.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
-nvidia_nat_mcp-1.4.0a20260109.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-nvidia_nat_mcp-1.4.0a20260109.dist-info/METADATA,sha256=jdQxP0wcXrbE6hEQUvWSC--AE2_rFIunpFkK2z7TArA,2326
-nvidia_nat_mcp-1.4.0a20260109.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_mcp-1.4.0a20260109.dist-info/entry_points.txt,sha256=t5HYRfXR-dyZf6BWRz1u1TqPprKpCWt-WLHKQXfiLLU,231
-nvidia_nat_mcp-1.4.0a20260109.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_mcp-1.4.0a20260109.dist-info/RECORD,,
+nat/plugins/mcp/server/tool_converter.py,sha256=v35IILT5S2jE7ZTYQzm8-4CMu5Xa21BACeRP1MS1vzA,11608
+nvidia_nat_mcp-1.4.0a20260125.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
+nvidia_nat_mcp-1.4.0a20260125.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+nvidia_nat_mcp-1.4.0a20260125.dist-info/METADATA,sha256=i3_5ji_Q4TvgIjZA7_i0L2Spm95EJKxTARJAX7QMKyM,2324
+nvidia_nat_mcp-1.4.0a20260125.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+nvidia_nat_mcp-1.4.0a20260125.dist-info/entry_points.txt,sha256=t5HYRfXR-dyZf6BWRz1u1TqPprKpCWt-WLHKQXfiLLU,231
+nvidia_nat_mcp-1.4.0a20260125.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_mcp-1.4.0a20260125.dist-info/RECORD,,

{nvidia_nat_mcp-1.4.0a20260109.dist-info → nvidia_nat_mcp-1.4.0a20260125.dist-info}/WHEEL

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