Fix notebook

aorwall · Jan 24, 2025 · 60b9f86 · 60b9f86
1 parent c5002a7
commit 60b9f86
Show file tree

Hide file tree

Showing 2 changed files with 1,233 additions and 189 deletions.
diff --git a/moatless/loop.py b/moatless/loop.py
@@ -106,8 +106,8 @@ def run(self):
                         "iteration": len(self.root.get_all_nodes()),
                         "total_cost": total_cost,
                         "action": current_node.action.name if current_node.action else None,
+                        "action_steps": [step.model_dump() for step in current_node.action_steps] if current_node.action_steps else None,
                         "current_node_id": current_node.node_id,
-                        "total_nodes": len(self.root.get_all_nodes()),
                     },
                 )