Agent Trajectory Observability Format (ATOF)

1
from nemo_relay import plugin
2
from nemo_relay.observability import AtofConfig, ComponentSpec, ObservabilityConfig
3
4
config = plugin.PluginConfig(
5
    components=[
6
        ComponentSpec(
7
            ObservabilityConfig(
8
                atof=AtofConfig(
9
                    enabled=True,
10
                    output_directory="logs",
11
                    filename="events.jsonl",
12
                    mode="overwrite",
13
                )
14
            )
15
        )
16
    ]
17
)
18
19
report = plugin.validate(config)
20
if any(diagnostic["level"] == "error" for diagnostic in report["diagnostics"]):
21
    raise RuntimeError(report["diagnostics"])
22
23
await plugin.initialize(config)
24
try:
25
    # Run instrumented application work here.
26
    pass
27
finally:
28
    plugin.clear()

1
from nemo_relay import AtofExporter, AtofExporterConfig, AtofExporterMode
2
3
config = AtofExporterConfig()
4
config.output_directory = "logs"
5
config.filename = "events.jsonl"
6
config.mode = AtofExporterMode.Overwrite
7
8
exporter = AtofExporter(config)
9
exporter.register("atof-exporter")
10
11
# Run instrumented application work here.
12
13
exporter.force_flush()
14
exporter.deregister("atof-exporter")
15
exporter.shutdown()