diff --git a/src/main.py b/src/main.py
index 101968f..499b153 100644
--- a/src/main.py
+++ b/src/main.py
@@ -17,6 +17,7 @@ from src.daemon.ticker import Ticker
 from src.daemon.spawner import AgentSpawner
 from src.daemon.dispatcher import Dispatcher
 from src.daemon.counter import ActiveAgentCounter
+from src.daemon.router import AgentRouter, AgentProfile, LLMDriver
 from src.utils import get_data_root
 
 logger = logging.getLogger("moziplus-v2")