diff --git a/tradingagents/graph/trading_graph.py b/tradingagents/graph/trading_graph.py index fe9b9f6c..0fe3239c 100644 --- a/tradingagents/graph/trading_graph.py +++ b/tradingagents/graph/trading_graph.py @@ -96,11 +96,7 @@ class TradingAgentsGraph: trust_remote_code=True, tensor_parallel_size=1, ) - self.quick_thinking_llm = VLLM( - model=self.config["quick_think_llm"], - trust_remote_code=True, - tensor_parallel_size=1, - ) + self.quick_thinking_llm = self.deep_thinking_llm else: raise ValueError(f"Unsupported LLM provider: {self.config['llm_provider']}")