diff --git a/tradingagents/graph/discovery_graph.py b/tradingagents/graph/discovery_graph.py index 25dcffa8..34c20632 100644 --- a/tradingagents/graph/discovery_graph.py +++ b/tradingagents/graph/discovery_graph.py @@ -3,7 +3,6 @@ from typing import Any, Callable, Dict, List, Optional from langgraph.graph import END, StateGraph from tradingagents.agents.utils.agent_states import DiscoveryState -from tradingagents.dataflows.discovery import scanners # noqa: F401 # Load scanners to trigger registration from tradingagents.dataflows.discovery.scanner_registry import SCANNER_REGISTRY from tradingagents.dataflows.discovery.utils import PRIORITY_ORDER, Priority, serialize_for_log from tradingagents.tools.executor import execute_tool @@ -66,6 +65,10 @@ class DiscoveryGraph: """ self.config = config or {} + # Load scanner modules to trigger registration + from tradingagents.dataflows.discovery import scanners + _ = scanners # Ensure scanners module is loaded + # Initialize LLMs from tradingagents.utils.llm_factory import create_llms