fix: save history in state variale with Event update

Author: shangkunwang01

MaxKernel/auto_agent/subagents/pipeline_agent.py

@@ -79,6 +79,7 @@ async def _run_async_impl(
       self._clear_iteration_metrics(ctx)
 
       if self._should_end_at_step(ctx, iteration, "plan"):
+        yield self._create_history_event(ctx)
         iteration += 1
         continue
 
@@ -87,6 +88,7 @@ async def _run_async_impl(
       async for event in self.implement_agent.run_async(ctx):
         yield event
       if self._should_end_at_step(ctx, iteration, "implement"):
+        yield self._create_history_event(ctx)
         iteration += 1
         continue
 
@@ -106,10 +108,12 @@ async def _run_async_impl(
         self._save_iteration_files_and_snapshot(
           ctx, iteration, step_name="validate"
         )
+        yield self._create_history_event(ctx)
         iteration += 1
         continue
 
       if self._should_end_at_step(ctx, iteration, "validate"):
+        yield self._create_history_event(ctx)
         iteration += 1
         continue
 
@@ -146,10 +150,12 @@ async def _run_async_impl(
         self._save_iteration_files_and_snapshot(
           ctx, iteration, step_name="test_run"
         )
+        yield self._create_history_event(ctx)
         iteration += 1
         continue
 
       if self._should_end_at_step(ctx, iteration, "test_run"):
+        yield self._create_history_event(ctx)
         iteration += 1
         continue
 
@@ -158,6 +164,7 @@ async def _run_async_impl(
       async for event in self.autotune_agent.run_async(ctx):
         yield event
       if self._should_end_at_step(ctx, iteration, "autotune"):
+        yield self._create_history_event(ctx)
         iteration += 1
         continue
 
@@ -166,17 +173,13 @@ async def _run_async_impl(
       async for event in self.profile_agent.run_async(ctx):
         yield event
       if self._should_end_at_step(ctx, iteration, "profile"):
+        yield self._create_history_event(ctx)
         iteration += 1
         continue
 
       self._save_iteration_files_and_snapshot(ctx, iteration)
 
-      yield Event(
-        author=self.name,
-        actions=EventActions(
-          state_delta={"history": ctx.session.state.get("history", [])}
-        ),
-      )
+      yield self._create_history_event(ctx)
 
       # Step 7: Check if improvement is needed
       # needs_improvement = ctx.session.state.get("needs_improvement", False)
@@ -210,6 +213,14 @@ async def _run_async_impl(
       ),
     )
 
+  def _create_history_event(self, ctx: InvocationContext) -> Event:
+    return Event(
+      author=self.name,
+      actions=EventActions(
+        state_delta={"history": ctx.session.state.get("history", [])}
+      ),
+    )
+
   def _should_end_at_step(
     self, ctx: InvocationContext, iteration: int, step_name: str
   ) -> bool: