nat.atif.scripts.generate_atif_trajectory#

Run a NeMo Agent Toolkit workflow and export the trajectory as ATIF JSON.

This script loads any NAT workflow from a YAML config, executes it with a given input, captures all IntermediateStep events, converts them to an ATIF v1.6 trajectory, and writes the result to a JSON file.

Prerequisites:

  • The workflow’s package must be installed (for example, pip install -e examples/getting_started/simple_calculator).

  • An appropriate API key must be set (for example, NVIDIA_API_KEY).

Usage (from repo root):
python -m nat.atif.scripts.generate_atif_trajectory

–config examples/getting_started/simple_calculator/src/nat_simple_calculator/configs/config.yml –input “What is 7 * 8?” -o atif_output.json

Attributes#

logger

Functions#

run_and_capture_atif(→ dict)

Load a workflow, run it, and return the ATIF trajectory as a dict.

main(→ None)

Module Contents#

logger#
async run_and_capture_atif(
config_file: pathlib.Path,
question: str,
session_id: str | None = None,
) → dict#

Load a workflow, run it, and return the ATIF trajectory as a dict.

main() → None#