Struct TrajectoryCollector 

pub struct TrajectoryCollector {
    pub trajectory: Trajectory,
    /* private fields */
}

TrajectoryCollector — collects trajectory data during an agent run.

Attach to AgentRunner to capture all steps incrementally.

Fields

trajectory: Trajectory

Implementations

impl TrajectoryCollector

pub fn new(session_id: impl Into<String>, model: impl Into<String>) -> Self

pub fn start_step(&mut self, action: &str, action_args: &str)

Start tracking a new tool call step.

pub fn complete_step(&mut self, observation: &str, success: bool)

Complete the current step with an observation.

pub fn fail_step(&mut self, error: &str)

Mark the current step as failed.

pub fn set_thought(&mut self, thought: &str)

Attach a thought to the current step.

pub fn set_response(&mut self, response: &str)

Attach a final response to the current step.

pub fn record_message( &mut self, role: &str, content: &str, tool_call_id: Option<&str>, tool_name: Option<&str>, )

Record a message to the trajectory.

pub fn save(&self, path: &Path) -> Result<()>

Finish and save to file.

pub fn finalize(&mut self, success: bool) -> String

Mark the overall trajectory as successful and return JSON.

pub fn to_json(&self) -> Result<String>

Serialize the trajectory to JSON.