From c960a8804b086fbaf7f8aacb211cc2e9ad654ad2 Mon Sep 17 00:00:00 2001 From: Clayton Brown Date: Mon, 20 Apr 2026 20:25:25 +1000 Subject: [PATCH] feat(027-checkpoint-resume-contrib): use thread_id(ticker, date) for deterministic checkpoint keying --- tradingagents/graph/trading_graph.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tradingagents/graph/trading_graph.py b/tradingagents/graph/trading_graph.py index e8a970b5..64ec56c1 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 +from .checkpointer import get_checkpointer, thread_id from .conditional_logic import ConditionalLogic from .setup import GraphSetup from .propagation import Propagator @@ -221,6 +221,11 @@ class TradingAgentsGraph: ) args = self.propagator.get_graph_args() + # Inject thread_id so same ticker+date resumes, different date starts fresh + if self.config.get("checkpoint_enabled"): + tid = thread_id(company_name, str(trade_date)) + args.setdefault("config", {}).setdefault("configurable", {})["thread_id"] = tid + if self.debug: # Debug mode with tracing trace = []