charieswei
/
TradingAgents
mirror of https://github.com/TauricResearch/TradingAgents.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat(027-checkpoint-resume-contrib): clear checkpoint on successful completion

This commit is contained in:
Clayton Brown 2026-04-20 20:26:08 +10:00
parent c960a8804b
commit be78acbfa9
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tradingagents/graph/trading_graph.py
View File

@ -33,7 +33,7 @@ from tradingagents.agents.utils.agent_utils import (
get_global_news 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 .conditional_logic import ConditionalLogic
from .setup import GraphSetup from .setup import GraphSetup
from .propagation import Propagator from .propagation import Propagator
@ -247,6 +247,10 @@ class TradingAgentsGraph:
# Log state # Log state
self._log_state(trade_date, final_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 decision and processed signal
return final_state, self.process_signal(final_state["final_trade_decision"]) return final_state, self.process_signal(final_state["final_trade_decision"])