From f806d4d1e74468831e37bc3a1367a2cba5d4ec1d Mon Sep 17 00:00:00 2001 From: Greyson LaLonde Date: Fri, 10 Apr 2026 08:02:20 +0800 Subject: [PATCH] fix: preserve overridden task agent messages during invalidation --- lib/crewai/src/crewai/cli/checkpoint_tui.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/lib/crewai/src/crewai/cli/checkpoint_tui.py b/lib/crewai/src/crewai/cli/checkpoint_tui.py index c3f1b8b6e..d50ac753f 100644 --- a/lib/crewai/src/crewai/cli/checkpoint_tui.py +++ b/lib/crewai/src/crewai/cli/checkpoint_tui.py @@ -565,9 +565,14 @@ async def _run_checkpoint_tui_async(location: str) -> None: break # Invalidate all subsequent tasks so they re-run with # the modified context instead of using cached results + overridden_agent = crew.tasks[task_idx].agent for subsequent in crew.tasks[task_idx + 1 :]: subsequent.output = None - if subsequent.agent and subsequent.agent.agent_executor: + if ( + subsequent.agent + and subsequent.agent is not overridden_agent + and subsequent.agent.agent_executor + ): subsequent.agent.agent_executor._resuming = False subsequent.agent.agent_executor.messages = [] click.echo()