nvidia-nat 1.3.0a20251001__py3-none-any.whl → 1.3.0a20251002__py3-none-any.whl

nat/agent/prompt_optimizer/register.py

@@ -51,7 +51,7 @@ async def prompt_optimizer_function(config: PromptOptimizerConfig, builder: Buil
         from .prompt import mutator_prompt
     except ImportError as exc:
         raise ImportError("langchain-core is not installed. Please install it to use MultiLLMPlanner.\n"
-                          "This error can be resolve by installing nvidia-nat[langchain]") from exc
+                          "This error can be resolve by installing \"nvidia-nat[langchain]\".") from exc
 
     llm = await builder.get_llm(config.optimizer_llm, wrapper_type=LLMFrameworkEnum.LANGCHAIN)
 
@@ -111,7 +111,7 @@ async def prompt_recombiner_function(config: PromptRecombinerConfig, builder: Bu
         from langchain_core.prompts import PromptTemplate
     except ImportError as exc:
         raise ImportError("langchain-core is not installed. Please install it to use MultiLLMPlanner.\n"
-                          "This error can be resolve by installing nvidia-nat[langchain].") from exc
+                          "This error can be resolve by installing \"nvidia-nat[langchain]\".") from exc
 
     llm = await builder.get_llm(config.optimizer_llm, wrapper_type=LLMFrameworkEnum.LANGCHAIN)
 

nat/cli/commands/mcp/mcp.py

@@ -297,7 +297,7 @@ async def list_tools_via_function_group(
             if fn is not None:
                 tools.append(to_tool_entry(full, fn))
         else:
-            for full, fn in (await fns).items():
+            for full, fn in fns.items():
                 tools.append(to_tool_entry(full, fn))
 
     return tools

nat/cli/main.py

@@ -30,6 +30,9 @@ def run_cli():
     import os
     import sys
 
+    # Suppress warnings from transformers
+    os.environ["TRANSFORMERS_VERBOSITY"] = "error"
+
     parent_dir = os.path.dirname(os.path.dirname(os.path.dirname(__file__)))
 
     if (parent_dir not in sys.path):

nat/profiler/decorators/framework_wrapper.py

@@ -123,7 +123,7 @@ def set_framework_profiler_handler(
                 except ImportError as e:
                     logger.warning(
                         "ADK profiler not available. " +
-                        "Install NAT with ADK extras: pip install 'nvidia-nat[adk]'. Error: %s",
+                        "Install NAT with ADK extras: pip install \"nvidia-nat[adk]\". Error: %s",
                         e)
                 else:
                     handler = ADKProfilerHandler()

nat/profiler/forecasting/models/linear_model.py

@@ -36,7 +36,7 @@ class LinearModel(ForecastingBaseModel):
         except ImportError:
             logger.error(
                 "scikit-learn is not installed. Please install scikit-learn to use the LinearModel "
-                "profiling model or install `nvidia-nat[profiler]` to install all necessary profiling packages.")
+                "profiling model or install \"nvidia-nat[profiler]\" to install all necessary profiling packages.")
 
             raise
 

nat/profiler/forecasting/models/random_forest_regressor.py

@@ -36,7 +36,7 @@ class RandomForestModel(ForecastingBaseModel):
         except ImportError:
             logger.error(
                 "scikit-learn is not installed. Please install scikit-learn to use the RandomForest "
-                "profiling model or install `nvidia-nat[profiler]` to install all necessary profiling packages.")
+                "profiling model or install \"nvidia-nat[profiler]\" to install all necessary profiling packages.")
 
             raise
 

nat/profiler/inference_optimization/bottleneck_analysis/nested_stack_analysis.py

@@ -304,7 +304,7 @@ def save_gantt_chart(all_nodes: list[CallNode], output_path: str) -> None:
         import matplotlib.pyplot as plt
     except ImportError:
         logger.error("matplotlib is not installed. Please install matplotlib to use generate plots for the profiler "
-                     "or install `nvidia-nat[profiler]` to install all necessary profiling packages.")
+                     "or install \"nvidia-nat[profiler]\" to install all necessary profiling packages.")
 
         raise
 

nat/profiler/inference_optimization/experimental/prefix_span_analysis.py

@@ -212,7 +212,7 @@ def run_prefixspan(sequences_map: dict[int, list[PrefixCallNode]],
         from prefixspan import PrefixSpan
     except ImportError:
         logger.error("prefixspan is not installed. Please install prefixspan to run the prefix analysis in the "
-                     "profiler or install `nvidia-nat[profiler]` to install all necessary profiling packages.")
+                     "profiler or install \"nvidia-nat[profiler]\" to install all necessary profiling packages.")
 
         raise
 

{nvidia_nat-1.3.0a20251001.dist-info → nvidia_nat-1.3.0a20251002.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat
-Version: 1.3.0a20251001
+Version: 1.3.0a20251002
 Summary: NVIDIA NeMo Agent toolkit
 Author: NVIDIA Corporation
 Maintainer: NVIDIA Corporation

{nvidia_nat-1.3.0a20251001.dist-info → nvidia_nat-1.3.0a20251002.dist-info}/RECORD

@@ -5,7 +5,7 @@ nat/agent/dual_node.py,sha256=pfvXa1iLKtrNBHsh-tM5RWRmVe7QkyYhQNanOfWdJJs,2569
 nat/agent/register.py,sha256=rPhHDyqRBIKyon3HqhOmpAjqPP18Or6wDQb0wRUBIjQ,1032
 nat/agent/prompt_optimizer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nat/agent/prompt_optimizer/prompt.py,sha256=2I1cGZH73Q1pwxA_V-9b1sFCdQ1PLFxliaRtrIkVVVU,3042
-nat/agent/prompt_optimizer/register.py,sha256=qV9Qkd4XDfwVYSzZOyI9RlRzK26jukaB51RriYgWd6k,6790
+nat/agent/prompt_optimizer/register.py,sha256=s1WZzHYivU1LwH3iZZ3YSEiOfIurzwxqvUGrfG2_rz4,6799
 nat/agent/react_agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nat/agent/react_agent/agent.py,sha256=sWrg9WrglTKQQyG3EcjNm2JTEchCPEo9li-Po7TJKss,21294
 nat/agent/react_agent/output_parser.py,sha256=m7K6wRwtckBBpAHqOf3BZ9mqZLwrP13Kxz5fvNxbyZE,4219
@@ -59,7 +59,7 @@ nat/builder/workflow.py,sha256=bHrxK-VFsxUTw2wZgkWcCttpCMDeWNGPfmIGEW_bjZo,6908
 nat/builder/workflow_builder.py,sha256=Fdf2eIYrRg1ovLkySzzgh5C2PNLV7QzQIN7pLHSr6zI,56339
 nat/cli/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
 nat/cli/entrypoint.py,sha256=7NHq3Grb_qzpJzzbb0YtTsWd4YhA6y9ljHwsOQPQAIs,4928
-nat/cli/main.py,sha256=KzUSDB1TGzHRP_Fnq5XJRopx-i9S2oycgQrXrf2vDkE,2164
+nat/cli/main.py,sha256=LZMKvoHYR926mghMjVpfLiI2qraqtrhMY9hvuAQCRWk,2258
 nat/cli/register_workflow.py,sha256=DOQQgUWB_NO9k7nlkP4cAx_RX03Cndk032K-kqyuvEs,23285
 nat/cli/type_registry.py,sha256=HbPU-7lzSHZ4aepJ3qOgfnl5LzK6-KHwcerhFpWw6mU,48839
 nat/cli/cli_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -83,7 +83,7 @@ nat/cli/commands/info/info.py,sha256=BGqshIEDpNRH9hM-06k-Gq-QX-qNddPICSWCN-ReC-g
 nat/cli/commands/info/list_channels.py,sha256=K97TE6wtikgImY-wAbFNi0HHUGtkvIFd2woaG06VkT0,1277
 nat/cli/commands/info/list_components.py,sha256=QlAJVONBA77xW8Lx6Autw5NTAZNy_VrJGr1GL9MfnHM,4532
 nat/cli/commands/mcp/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
-nat/cli/commands/mcp/mcp.py,sha256=pwpnWJS1UztkTZYQCZjG9T7IPnk5Je9tA63KmhpYbn8,43860
+nat/cli/commands/mcp/mcp.py,sha256=xpnPnLOuqHrDkZk-9oK96Ub9TNRidr_0rP8BuedxO4U,43852
 nat/cli/commands/object_store/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
 nat/cli/commands/object_store/object_store.py,sha256=_ivB-R30a-66fNy-fUzi58HQ0Ay0gYsGz7T1xXoRa3Y,8576
 nat/cli/commands/registry/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
@@ -343,15 +343,15 @@ nat/profiler/callbacks/llama_index_callback_handler.py,sha256=xzYve07uMlSTDjj929
 nat/profiler/callbacks/semantic_kernel_callback_handler.py,sha256=BknzhQNB-MDMhR4QC9JScCp-zXq7KZ33SFb7X0MiTaI,11087
 nat/profiler/callbacks/token_usage_base_model.py,sha256=X0b_AbBgVQAAbgbDMim-3S3XdQ7PaPs9qMUACvMAe5o,1104
 nat/profiler/decorators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nat/profiler/decorators/framework_wrapper.py,sha256=lKX-nMkwKc4Pcb3PiA_w6f0Zfa_aC8BQsDB1pWwJTjI,7680
+nat/profiler/decorators/framework_wrapper.py,sha256=qlQ_o6fzsV6jUBbHYFuGjX_Aq-NDZ23l9l3ViDKxy5g,7682
 nat/profiler/decorators/function_tracking.py,sha256=-ai_4djCbNwMan5SiTq3MVMBrcKoUWyxzviAV-Eh4xg,16771
 nat/profiler/forecasting/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nat/profiler/forecasting/config.py,sha256=5BhMa8csuPCjEnTaNQjo_2IoO7esh1ch02MoSWkvwPw,791
 nat/profiler/forecasting/model_trainer.py,sha256=6Ci2KN4sNj1V3yduHXlA0tn50yHUmM2VcbRRS2L96OA,2456
 nat/profiler/forecasting/models/__init__.py,sha256=pLpWi7u1UrguKIYD-BYu8IExvJLX_U2cuW3Ifp3zCOY,937
 nat/profiler/forecasting/models/forecasting_base_model.py,sha256=PP6PObogPsfvzVGo6trfcDZ5b-VutqGVjdeOMAZB2W8,1235
-nat/profiler/forecasting/models/linear_model.py,sha256=YSf0tT4yLtEHW5-V02EFUks__MRiTzt0OxuwJOaJ_vc,7008
-nat/profiler/forecasting/models/random_forest_regressor.py,sha256=fiom3CGzsy0A-8gBG0FoEfj1T4GywaQjmDf9db1fskw,9548
+nat/profiler/forecasting/models/linear_model.py,sha256=3Jyw238v0j4MkR5cqPazYP-gjw6bMUS6fS6xZlRndJs,7010
+nat/profiler/forecasting/models/random_forest_regressor.py,sha256=3Lgguy5CFKoPCJyrS7f8ImeV6SCLAGajfnZCIAmBs8Y,9550
 nat/profiler/inference_optimization/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nat/profiler/inference_optimization/data_models.py,sha256=gJzlzKIkUbPPA4QyhQIXtE3hzuxdKFpB5ywECJ4T2Vg,11779
 nat/profiler/inference_optimization/llm_metrics.py,sha256=kC8LUonJiXjQ5ogH8D7GTEYk44ypwncRJrN0ByRWz6k,9482
@@ -359,11 +359,11 @@ nat/profiler/inference_optimization/prompt_caching.py,sha256=n3kfgWibCjkXbp6_Qo8
 nat/profiler/inference_optimization/token_uniqueness.py,sha256=zCPPwaBj1LOp6b9xKg9rJ80nbx1pgiRbo2PID_N8d2c,4536
 nat/profiler/inference_optimization/workflow_runtimes.py,sha256=RXCsLoWkaLi32fi8kqajhgVMIedtzljUUZ8LA-xJDEU,2664
 nat/profiler/inference_optimization/bottleneck_analysis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nat/profiler/inference_optimization/bottleneck_analysis/nested_stack_analysis.py,sha256=J8gKaCe_F9PtHB051ZzqE-gyD5NMXqd2pZ9pnOqHHzI,16739
+nat/profiler/inference_optimization/bottleneck_analysis/nested_stack_analysis.py,sha256=ouXHdlzqGD1hHLCXE7fNmUmUJqk8-Kw33XcHYg9ACdU,16741
 nat/profiler/inference_optimization/bottleneck_analysis/simple_stack_analysis.py,sha256=DBbOxgKAYsGMIkFnbHGNPDdZQllmXBKW0Nkiso05NzI,11096
 nat/profiler/inference_optimization/experimental/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nat/profiler/inference_optimization/experimental/concurrency_spike_analysis.py,sha256=3hFMDbfW7MKxFIhO7y35qtP5wPtLAA2ZqltlnI-OsVc,16953
-nat/profiler/inference_optimization/experimental/prefix_span_analysis.py,sha256=-e5pEW4CQXYm0c5JrI-d0p-yZzvHlUzZYM8GOYA4hEU,16662
+nat/profiler/inference_optimization/experimental/prefix_span_analysis.py,sha256=mTC93u3e78nFDP36TnkNse8_HwNQKn3Qw4ksMycRAA8,16664
 nat/profiler/parameter_optimization/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nat/profiler/parameter_optimization/optimizable_utils.py,sha256=93Pl8A14Zq_f3XsxSH-yFnEJ6B7W5hp7doPnPoLlRB4,3714
 nat/profiler/parameter_optimization/optimizer_runtime.py,sha256=rXmCOq81o7ZorQOUYociVjuO3NO9CIjFBbwql2u_4H4,2715
@@ -469,10 +469,10 @@ nat/utils/reactive/base/observer_base.py,sha256=6BiQfx26EMumotJ3KoVcdmFBYR_fnAss
 nat/utils/reactive/base/subject_base.py,sha256=UQOxlkZTIeeyYmG5qLtDpNf_63Y7p-doEeUA08_R8ME,2521
 nat/utils/settings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nat/utils/settings/global_settings.py,sha256=9JaO6pxKT_Pjw6rxJRsRlFCXdVKCl_xUKU2QHZQWWNM,7294
-nvidia_nat-1.3.0a20251001.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
-nvidia_nat-1.3.0a20251001.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-nvidia_nat-1.3.0a20251001.dist-info/METADATA,sha256=Fuc_ccANJdpLGWxdHQ0F2EIU3KPPI7KgJjjXi2WTzfo,22945
-nvidia_nat-1.3.0a20251001.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat-1.3.0a20251001.dist-info/entry_points.txt,sha256=4jCqjyETMpyoWbCBf4GalZU8I_wbstpzwQNezdAVbbo,698
-nvidia_nat-1.3.0a20251001.dist-info/top_level.txt,sha256=lgJWLkigiVZuZ_O1nxVnD_ziYBwgpE2OStdaCduMEGc,8
-nvidia_nat-1.3.0a20251001.dist-info/RECORD,,
+nvidia_nat-1.3.0a20251002.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
+nvidia_nat-1.3.0a20251002.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+nvidia_nat-1.3.0a20251002.dist-info/METADATA,sha256=_QPmMUXzMuyjKCSmGZGiaBfJRTukJucRV3vGqa9yO50,22945
+nvidia_nat-1.3.0a20251002.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat-1.3.0a20251002.dist-info/entry_points.txt,sha256=4jCqjyETMpyoWbCBf4GalZU8I_wbstpzwQNezdAVbbo,698
+nvidia_nat-1.3.0a20251002.dist-info/top_level.txt,sha256=lgJWLkigiVZuZ_O1nxVnD_ziYBwgpE2OStdaCduMEGc,8
+nvidia_nat-1.3.0a20251002.dist-info/RECORD,,