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
)
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"])