scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-10-06 02:48:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

fix bug: trajectory class does not have steps

Browse source
This commit is contained in:
dengjiaji 2025-08-12 17:53:02 +08:00
parent dec8b45186
commit 81a17b897e
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
experiencemaker/experiencemaker/op/summarizer/comparative_extraction_op.py
View file

@ -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)