nvidia-nat 1.3.0rc2__py3-none-any.whl → 1.4.0a20251008__py3-none-any.whl

nat/experimental/test_time_compute/functions/execute_score_select_function.py

@@ -46,7 +46,7 @@ async def execute_score_select_function(config: ExecuteScoreSelectFunctionConfig
 
     from pydantic import BaseModel
 
-    executable_fn: Function = builder.get_function(name=config.augmented_fn)
+    executable_fn: Function = await builder.get_function(name=config.augmented_fn)
 
     if config.scorer:
         scorer = await builder.get_ttc_strategy(strategy_name=config.scorer,

nat/experimental/test_time_compute/functions/ttc_tool_wrapper_function.py

@@ -98,8 +98,8 @@ async def register_ttc_tool_wrapper_function(
 
         augmented_function_desc = config.tool_description
 
-    fn_input_schema: BaseModel = augmented_function.input_schema
-    fn_output_schema: BaseModel = augmented_function.single_output_schema
+    fn_input_schema: type[BaseModel] = augmented_function.input_schema
+    fn_output_schema: type[BaseModel] | type[None] = augmented_function.single_output_schema
 
     runnable_llm = input_llm.with_structured_output(schema=fn_input_schema)
 

{nvidia_nat-1.3.0rc2.dist-info → nvidia_nat-1.4.0a20251008.dist-info}/METADATA

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

{nvidia_nat-1.3.0rc2.dist-info → nvidia_nat-1.4.0a20251008.dist-info}/RECORD

@@ -202,10 +202,10 @@ nat/experimental/test_time_compute/editing/iterative_plan_refinement_editor.py,s
 nat/experimental/test_time_compute/editing/llm_as_a_judge_editor.py,sha256=QAu46omTvXJOqWzN_Xeeuhf25R0WxHqMjAdXcB9IxqQ,8539
 nat/experimental/test_time_compute/editing/motivation_aware_summarization.py,sha256=S7s_BigZ1ru-3lP7c58Zpt5Nxavi8Ko-A1cmH7KkX3s,4503
 nat/experimental/test_time_compute/functions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nat/experimental/test_time_compute/functions/execute_score_select_function.py,sha256=OHiQ01a6g3OEd-K1WZynuQspmrvBpgKKxxqO-UbM4z0,4550
+nat/experimental/test_time_compute/functions/execute_score_select_function.py,sha256=K1sQPkhvGyWIJxJXJ_ZGH1LqQK2v8cCKUpx0RvG22PA,4556
 nat/experimental/test_time_compute/functions/plan_select_execute_function.py,sha256=ZkxFwF8SF6y88qa9ZqBsie--bnbsHpWC72ky-ttBzV0,10067
 nat/experimental/test_time_compute/functions/ttc_tool_orchestration_function.py,sha256=GMPibwNI6o2ljQkDUsD7C-wFm8qRYTx_B7PA6IBFrFI,8446
-nat/experimental/test_time_compute/functions/ttc_tool_wrapper_function.py,sha256=TMHLz31RgPG3rt5ilfkES-xqJSl-f8qc_-ZU9bYFQyk,6913
+nat/experimental/test_time_compute/functions/ttc_tool_wrapper_function.py,sha256=Iahv7-IznuuKdjlJre6aKYPf2tZEnuwWswe2yYyDYWg,6938
 nat/experimental/test_time_compute/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nat/experimental/test_time_compute/models/editor_config.py,sha256=wFHcd4GjWy4fxuXrYnC0gJ0ZszsAjziohK8vS7cc8EA,7153
 nat/experimental/test_time_compute/models/scoring_config.py,sha256=zi93HQrtY1FiRuZgDOdlDZql6T-1W6SteybSUMXjr8U,5500
@@ -470,10 +470,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.0rc2.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
-nvidia_nat-1.3.0rc2.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-nvidia_nat-1.3.0rc2.dist-info/METADATA,sha256=xvXF8y1Su3huZJlBr45f6m_FkBp0Co_CVZpZjKr_PZs,22821
-nvidia_nat-1.3.0rc2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat-1.3.0rc2.dist-info/entry_points.txt,sha256=4jCqjyETMpyoWbCBf4GalZU8I_wbstpzwQNezdAVbbo,698
-nvidia_nat-1.3.0rc2.dist-info/top_level.txt,sha256=lgJWLkigiVZuZ_O1nxVnD_ziYBwgpE2OStdaCduMEGc,8
-nvidia_nat-1.3.0rc2.dist-info/RECORD,,
+nvidia_nat-1.4.0a20251008.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
+nvidia_nat-1.4.0a20251008.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+nvidia_nat-1.4.0a20251008.dist-info/METADATA,sha256=rMyOzBSSp9zJsVVBrJCSIlsUhvnc0tiKDNX4d1H0i2o,22827
+nvidia_nat-1.4.0a20251008.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat-1.4.0a20251008.dist-info/entry_points.txt,sha256=4jCqjyETMpyoWbCBf4GalZU8I_wbstpzwQNezdAVbbo,698
+nvidia_nat-1.4.0a20251008.dist-info/top_level.txt,sha256=lgJWLkigiVZuZ_O1nxVnD_ziYBwgpE2OStdaCduMEGc,8
+nvidia_nat-1.4.0a20251008.dist-info/RECORD,,