From 81a17b897e9fa79001cd55edcef928f27f5bd41f Mon Sep 17 00:00:00 2001 From: dengjiaji Date: Tue, 12 Aug 2025 17:53:02 +0800 Subject: [PATCH] fix bug: trajectory class does not have steps --- .../op/summarizer/comparative_extraction_op.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/experiencemaker/experiencemaker/op/summarizer/comparative_extraction_op.py b/experiencemaker/experiencemaker/op/summarizer/comparative_extraction_op.py index 7f66e427..6d86f527 100644 --- a/experiencemaker/experiencemaker/op/summarizer/comparative_extraction_op.py +++ b/experiencemaker/experiencemaker/op/summarizer/comparative_extraction_op.py @@ -156,17 +156,17 @@ class ComparativeExtractionOp(BaseOp): # Get step sequences success_step_sequences = [] for traj in success_trajectories: - if hasattr(traj, 'segments') and traj.segments: - success_step_sequences.extend(traj.segments) + if hasattr(traj.metadata, 'segments') and traj.metadata["segments"]: + success_step_sequences.extend(traj.metadata["segments"]) else: - success_step_sequences.append(traj.steps) + success_step_sequences.append(traj.messages) failure_step_sequences = [] for traj in failure_trajectories: - if hasattr(traj, 'segments') and traj.segments: - failure_step_sequences.extend(traj.segments) + if hasattr(traj.metadata, 'segments') and traj.metadata["segments"]: + failure_step_sequences.extend(traj.metadata["segments"]) else: - failure_step_sequences.append(traj.steps) + failure_step_sequences.append(traj.messages) # Limit comparison count to avoid computational overload max_sequences = self.op_params.get("max_similarity_sequences", 5)