From be78acbfa9fd892b6ccf482ba75b007d70eed70b Mon Sep 17 00:00:00 2001 From: Clayton Brown Date: Mon, 20 Apr 2026 20:26:08 +1000 Subject: [PATCH] feat(027-checkpoint-resume-contrib): clear checkpoint on successful completion --- tradingagents/graph/trading_graph.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tradingagents/graph/trading_graph.py b/tradingagents/graph/trading_graph.py index 64ec56c1..560ec842 100644 --- a/tradingagents/graph/trading_graph.py +++ b/tradingagents/graph/trading_graph.py @@ -33,7 +33,7 @@ from tradingagents.agents.utils.agent_utils import ( get_global_news ) -from .checkpointer import get_checkpointer, thread_id +from .checkpointer import clear_checkpoint, get_checkpointer, thread_id from .conditional_logic import ConditionalLogic from .setup import GraphSetup from .propagation import Propagator @@ -247,6 +247,10 @@ class TradingAgentsGraph: # Log state self._log_state(trade_date, final_state) + # Clear checkpoint on successful completion to avoid stale state + if self.config.get("checkpoint_enabled"): + clear_checkpoint(self.config["data_cache_dir"], company_name, str(trade_date)) + # Return decision and processed signal return final_state, self.process_signal(final_state["final_trade_decision"])