From ec266cb5c25c06b8586f0082ebf4288699bd5cd0 Mon Sep 17 00:00:00 2001 From: reopio Date: Thu, 16 Oct 2025 07:12:11 +0000 Subject: [PATCH] Add support for vllm --- tradingagents/graph/trading_graph.py | 17 ++++++----------- 1 file changed, 6 insertions(+), 11 deletions(-) diff --git a/tradingagents/graph/trading_graph.py b/tradingagents/graph/trading_graph.py index 6f1f8cf1..0a74aad9 100644 --- a/tradingagents/graph/trading_graph.py +++ b/tradingagents/graph/trading_graph.py @@ -99,17 +99,12 @@ class TradingAgentsGraph: ] ), ).ask() - self.bull_memory = FinancialSituationMemory("bull_memory", self.config) - self.bear_memory = FinancialSituationMemory("bear_memory", self.config) - self.trader_memory = FinancialSituationMemory("trader_memory", self.config) - self.invest_judge_memory = FinancialSituationMemory("invest_judge_memory", self.config) - self.risk_manager_memory = FinancialSituationMemory("risk_manager_memory", self.config) - else: - self.bull_memory = FinancialSituationMemory("bull_memory", self.config) - self.bear_memory = FinancialSituationMemory("bear_memory", self.config) - self.trader_memory = FinancialSituationMemory("trader_memory", self.config) - self.invest_judge_memory = FinancialSituationMemory("invest_judge_memory", self.config) - self.risk_manager_memory = FinancialSituationMemory("risk_manager_memory", self.config) + + self.bull_memory = FinancialSituationMemory("bull_memory", self.config) + self.bear_memory = FinancialSituationMemory("bear_memory", self.config) + self.trader_memory = FinancialSituationMemory("trader_memory", self.config) + self.invest_judge_memory = FinancialSituationMemory("invest_judge_memory", self.config) + self.risk_manager_memory = FinancialSituationMemory("risk_manager_memory", self.config) # Create tool nodes self.tool_nodes = self._create_tool_nodes()