Files
deer-flow/backend/src/tools/tools.py

48 lines
1.6 KiB
Python
Raw Normal View History

import logging
2026-01-14 07:19:43 +08:00
from langchain.tools import BaseTool
from src.config import get_app_config
from src.reflection import resolve_variable
2026-01-18 19:55:36 +08:00
from src.tools.builtins import ask_clarification_tool, present_file_tool
2026-01-16 21:48:00 +08:00
logger = logging.getLogger(__name__)
2026-01-16 21:48:00 +08:00
BUILTIN_TOOLS = [
present_file_tool,
2026-01-18 19:55:36 +08:00
ask_clarification_tool,
2026-01-16 21:48:00 +08:00
]
2026-01-14 07:19:43 +08:00
def get_available_tools(groups: list[str] | None = None, include_mcp: bool = True) -> list[BaseTool]:
"""Get all available tools from config.
Note: MCP tools should be initialized at application startup using
`initialize_mcp_tools()` from src.mcp module.
Args:
groups: Optional list of tool groups to filter by.
include_mcp: Whether to include tools from MCP servers (default: True).
Returns:
List of available tools.
"""
2026-01-14 07:19:43 +08:00
config = get_app_config()
2026-01-16 21:48:00 +08:00
loaded_tools = [resolve_variable(tool.use, BaseTool) for tool in config.tools if groups is None or tool.group in groups]
# Get cached MCP tools if enabled
mcp_tools = []
2026-01-20 13:57:36 +08:00
if include_mcp and config.extensions and config.extensions.get_enabled_mcp_servers():
try:
from src.mcp.cache import get_cached_mcp_tools
mcp_tools = get_cached_mcp_tools()
if mcp_tools:
logger.info(f"Using {len(mcp_tools)} cached MCP tool(s)")
except ImportError:
logger.warning("MCP module not available. Install 'langchain-mcp-adapters' package to enable MCP tools.")
except Exception as e:
logger.error(f"Failed to get cached MCP tools: {e}")
return loaded_tools + BUILTIN_TOOLS + mcp_tools