nvidia-nat 1.4.0a20251103__py3-none-any.whl → 1.4.0a20251105__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -354,7 +354,8 @@ def reinstall_command(workflow_name):
354
354
 
355
355
  @click.command()
356
356
  @click.argument('workflow_name')
357
- def delete_command(workflow_name: str):
357
+ @click.option('-y', '--yes', "yes_flag", is_flag=True, default=False, help='Do not prompt for confirmation.')
358
+ def delete_command(workflow_name: str, yes_flag: bool):
358
359
  """
359
360
  Delete a NAT workflow and uninstall its package.
360
361
 
@@ -362,7 +363,7 @@ def delete_command(workflow_name: str):
362
363
  workflow_name (str): The name of the workflow to delete.
363
364
  """
364
365
  try:
365
- if not click.confirm(f"Are you sure you want to delete the workflow '{workflow_name}'?"):
366
+ if not yes_flag and not click.confirm(f"Are you sure you want to delete the workflow '{workflow_name}'?"):
366
367
  click.echo("Workflow deletion cancelled.")
367
368
  return
368
369
  editable = get_repo_root() is not None
nat/eval/evaluate.py CHANGED
@@ -514,7 +514,7 @@ class EvaluationRun:
514
514
  # Run workflow and evaluate
515
515
  async with WorkflowEvalBuilder.from_config(config=config) as eval_workflow:
516
516
  # Initialize Weave integration
517
- self.weave_eval.initialize_logger(workflow_alias, self.eval_input, config)
517
+ self.weave_eval.initialize_logger(workflow_alias, self.eval_input, config, job_id=job_id)
518
518
 
519
519
  with self.eval_trace_context.evaluation_context():
520
520
  # Run workflow
@@ -82,7 +82,7 @@ class WeaveEvaluationIntegration:
82
82
  """Get the full dataset for Weave."""
83
83
  return [item.full_dataset_entry for item in eval_input.eval_input_items]
84
84
 
85
- def initialize_logger(self, workflow_alias: str, eval_input: EvalInput, config: Any):
85
+ def initialize_logger(self, workflow_alias: str, eval_input: EvalInput, config: Any, job_id: str | None = None):
86
86
  """Initialize the Weave evaluation logger."""
87
87
  if not self.client and not self.initialize_client():
88
88
  # lazy init the client
@@ -92,10 +92,16 @@ class WeaveEvaluationIntegration:
92
92
  weave_dataset = self._get_weave_dataset(eval_input)
93
93
  config_dict = config.model_dump(mode="json")
94
94
  config_dict["name"] = workflow_alias
95
+
96
+ # Include job_id in eval_attributes if provided
97
+ eval_attributes = {}
98
+ if job_id:
99
+ eval_attributes["job_id"] = job_id
100
+
95
101
  self.eval_logger = self.evaluation_logger_cls(model=config_dict,
96
102
  dataset=weave_dataset,
97
103
  name=workflow_alias,
98
- eval_attributes={})
104
+ eval_attributes=eval_attributes)
99
105
  self.pred_loggers = {}
100
106
 
101
107
  # Capture the current evaluation call for context propagation
@@ -136,9 +142,17 @@ class WeaveEvaluationIntegration:
136
142
  coros = []
137
143
  for eval_output_item in eval_output.eval_output_items:
138
144
  if eval_output_item.id in self.pred_loggers:
145
+ # Structure the score as a dict and include reasoning if available
146
+ score_value = {
147
+ "score": eval_output_item.score,
148
+ }
149
+
150
+ if eval_output_item.reasoning is not None:
151
+ score_value["reasoning"] = eval_output_item.reasoning
152
+
139
153
  coros.append(self.pred_loggers[eval_output_item.id].alog_score(
140
154
  scorer=evaluator_name,
141
- score=eval_output_item.score,
155
+ score=score_value,
142
156
  ))
143
157
 
144
158
  # Execute all coroutines concurrently
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: nvidia-nat
3
- Version: 1.4.0a20251103
3
+ Version: 1.4.0a20251105
4
4
  Summary: NVIDIA NeMo Agent toolkit
5
5
  Author: NVIDIA Corporation
6
6
  Maintainer: NVIDIA Corporation
@@ -99,7 +99,7 @@ nat/cli/commands/sizing/calc.py,sha256=3cJHKCbzvV7EwYfLcpfk3_Ki7soAjOaiBcLK-Q6hP
99
99
  nat/cli/commands/sizing/sizing.py,sha256=-Hr9mz_ScEMtBbn6ijvmmWVk0WybLeX0Ryi4qhDiYQU,902
100
100
  nat/cli/commands/workflow/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
101
101
  nat/cli/commands/workflow/workflow.py,sha256=40nIOehOX-4xI-qJqJraBX3XVz3l2VtFsZkMQYd897w,1342
102
- nat/cli/commands/workflow/workflow_commands.py,sha256=hJXsy0vDhL6ITCJaM_xPepXwEom2E6_beTqr0HBldaU,14256
102
+ nat/cli/commands/workflow/workflow_commands.py,sha256=SI2redhG-9tLgWWXcRjkxdFhvKmmzBIqzWHZhRZpAbQ,14399
103
103
  nat/cli/commands/workflow/templates/__init__.py.j2,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
104
104
  nat/cli/commands/workflow/templates/config.yml.j2,sha256=KORGAFt1TW524YxXD2jpm_uTESihUKV5fnSoXQrH1PI,368
105
105
  nat/cli/commands/workflow/templates/pyproject.toml.j2,sha256=lDBC4exHYutXa_skuJj176yMEuZr-DsdzrqQHPZoKpk,1035
@@ -159,7 +159,7 @@ nat/embedder/openai_embedder.py,sha256=To7aCg8UyWPwSoA0MAHanH_MAKFDi3EcZxgLU1xYE
159
159
  nat/embedder/register.py,sha256=TM_LKuSlJr3tEceNVuHfAx_yrCzf1sryD5Ycep5rNGo,883
160
160
  nat/eval/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
161
161
  nat/eval/config.py,sha256=G0LE4JpZaQy3PvERldVATFpQCiDQcVJGUFChgorqzNo,2377
162
- nat/eval/evaluate.py,sha256=XmgSrhu-N11W4WeE7yqZmNLLNM90lU5OyWKuFEhUo2Y,26927
162
+ nat/eval/evaluate.py,sha256=CrJk7HMNH4gzmvrnIvQxIPWeX1w9A3Zv5meat7J6dAg,26942
163
163
  nat/eval/intermediate_step_adapter.py,sha256=mquQfPbq4-Owid2GzSyxtGNXoZ0i8crB6sA49rxnyrU,4483
164
164
  nat/eval/register.py,sha256=Vce8HGsu6KDj7MA_5W2ziQtss1F180ndMjuqGiHxTe8,1358
165
165
  nat/eval/remote_workflow.py,sha256=JAAbD0s753AOjo9baT4OqcB5dVEDmN34jPe0Uk13LcU,6024
@@ -194,7 +194,7 @@ nat/eval/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
194
194
  nat/eval/utils/eval_trace_ctx.py,sha256=hN0YZ0wMOPzh9I-iSav-cGdxY3RWQWoE_tk5BxUf1mc,3264
195
195
  nat/eval/utils/output_uploader.py,sha256=wtAaxvrJrjQPFt8mTLSz31sWDs09KmLkmyelMQOLDws,5667
196
196
  nat/eval/utils/tqdm_position_registry.py,sha256=9CtpCk1wtYCSyieHPaSp8nlZu6EcNUOaUz2RTqfekrA,1286
197
- nat/eval/utils/weave_eval.py,sha256=fma5x9JbWpWrfQbfMHcjMovlRVR0v35yfNt1Avt6Vro,7719
197
+ nat/eval/utils/weave_eval.py,sha256=kmpeSzDtZiHJW_suc1U6MlvPe_9cFIudjhr-C79MVAc,8238
198
198
  nat/experimental/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
199
199
  nat/experimental/decorators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
200
200
  nat/experimental/decorators/experimental_warning_decorator.py,sha256=ADj8fR9jL2FGFDcR8zJZbdvO9Leur8pUEVEUqpfNENY,5757
@@ -475,10 +475,10 @@ nat/utils/reactive/base/observer_base.py,sha256=6BiQfx26EMumotJ3KoVcdmFBYR_fnAss
475
475
  nat/utils/reactive/base/subject_base.py,sha256=UQOxlkZTIeeyYmG5qLtDpNf_63Y7p-doEeUA08_R8ME,2521
476
476
  nat/utils/settings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
477
477
  nat/utils/settings/global_settings.py,sha256=9JaO6pxKT_Pjw6rxJRsRlFCXdVKCl_xUKU2QHZQWWNM,7294
478
- nvidia_nat-1.4.0a20251103.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
479
- nvidia_nat-1.4.0a20251103.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
480
- nvidia_nat-1.4.0a20251103.dist-info/METADATA,sha256=fdOI5luw2_-2PSg7H2XHmAqsJXfIRPPH1iOoK2VDD1s,10248
481
- nvidia_nat-1.4.0a20251103.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
482
- nvidia_nat-1.4.0a20251103.dist-info/entry_points.txt,sha256=4jCqjyETMpyoWbCBf4GalZU8I_wbstpzwQNezdAVbbo,698
483
- nvidia_nat-1.4.0a20251103.dist-info/top_level.txt,sha256=lgJWLkigiVZuZ_O1nxVnD_ziYBwgpE2OStdaCduMEGc,8
484
- nvidia_nat-1.4.0a20251103.dist-info/RECORD,,
478
+ nvidia_nat-1.4.0a20251105.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
479
+ nvidia_nat-1.4.0a20251105.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
480
+ nvidia_nat-1.4.0a20251105.dist-info/METADATA,sha256=et1fcmaVq-vji3uL9r4yjQQm1NQpEwfBZBi-1_fmqV4,10248
481
+ nvidia_nat-1.4.0a20251105.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
482
+ nvidia_nat-1.4.0a20251105.dist-info/entry_points.txt,sha256=4jCqjyETMpyoWbCBf4GalZU8I_wbstpzwQNezdAVbbo,698
483
+ nvidia_nat-1.4.0a20251105.dist-info/top_level.txt,sha256=lgJWLkigiVZuZ_O1nxVnD_ziYBwgpE2OStdaCduMEGc,8
484
+ nvidia_nat-1.4.0a20251105.dist-info/RECORD,,