diff --git a/app/agent/__init__.py b/app/agent/__init__.py
index e6c78ae7..e36822d5 100644
--- a/app/agent/__init__.py
+++ b/app/agent/__init__.py
@@ -5,7 +5,7 @@ from typing import Dict, List
 
 from langchain.agents import create_agent
 from langchain.agents.middleware import (
-    SummarizationMiddleware,
+    SummarizationMiddleware, LLMToolSelectorMiddleware,
 )
 from langchain_core.messages import (
     HumanMessage,
@@ -92,6 +92,11 @@ class MoviePilotAgent:
 
             # 中间件
             middlewares = [
+                # 工具选择
+                LLMToolSelectorMiddleware(
+                    model=llm,
+                    max_tools=20
+                ),
                 # 记忆管理
                 MemoryMiddleware(
                     sources=[str(settings.CONFIG_PATH / "agent" / "MEMORY.md")]
diff --git a/requirements.in b/requirements.in
index 432f39c8..c8c6baa2 100644
--- a/requirements.in
+++ b/requirements.in
@@ -89,6 +89,7 @@ langchain-openai~=1.1.11
 langchain-google-genai~=4.2.1
 langchain-deepseek~=1.0.1
 langchain-experimental~=0.4.1
+langgraph~=1.1.3
 openai~=2.29.0
 ddgs~=9.10.0
 websocket-client~=1.8.0