Skip to content

Add ATIF (Agent Trace Interchange Format) support to AgentRolloutSeedSource #493

@jeremyjordan

Description

@jeremyjordan

Priority Level

Medium (Nice to have)

Is your feature request related to a problem? Please describe.

The existing AgentRolloutSeedSource supports two vendor-specific formats (Claude Code and Codex). ATIF is designed as a portable interchange format that any agent system can emit, making it a natural addition. With ATIF support, users can seed dataset generation from agent traces regardless of which agent framework produced them, without writing a custom handler.

Describe the solution you'd like

Each agent trace is represented as a JSON object. We could either support a directory where each file is a JSON object, or support reading from a .jsonl file where each line is a full agent trace.

High level changes

  1. Add ATIF to AgentRolloutFormat enum
  2. Implement AtifAgentRolloutFormatHandler
  3. Register the handler with BUILTIN_AGENT_ROLLOUT_FORMAT_HANDLERS

Example

from data_designer import DataDesigner, AgentRolloutSeedSource, AgentRolloutFormat

dd = DataDesigner()
config = dd.config_builder()
config.with_seed_dataset(
    AgentRolloutSeedSource(
        format=AgentRolloutFormat.ATIF,
        path="/path/to/traces",
    )
)

Describe alternatives you've considered

No response

Additional context

No response

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions