Update cmbagent/hand_offs.py

cmbagent/hand_offs.py

@@ -98,16 +98,16 @@ def register_all_hand_offs(cmbagent_instance):
     # plan_recorder: conditional routing based on feedback_left
     # If feedback_left == 0, planning is complete -> go to terminator
     # Otherwise, continue to plan_reviewer for feedback
-    agents['plan_recorder'].agent.handoffs.add_after_works([
-        OnContextCondition(
-            target=AgentTarget(agents['terminator'].agent),
-            condition=ExpressionContextCondition(ContextExpression("${feedback_left} == 0")),
+    agents['plan_recorder'].agent.handoffs.set_after_work(
+    AgentTarget(agents['plan_reviewer'].agent)
+).add_after_work(
+    OnContextCondition(
+        target=AgentTarget(agents['terminator'].agent),
+        condition=ExpressionContextCondition(
+            ContextExpression("${feedback_left} == 0")
         ),
-        OnContextCondition(
-            target=AgentTarget(agents['plan_reviewer'].agent),
-            condition=None,  # Default fallback
-        ),
-    ])
+    )
+)
 
     # Response formatters that route differently based on mode
     mode_dependent_formatters = ['camb_response_formatter']