2026-01-19 18:57:16 +08:00
|
|
|
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
|
|
|
|
2026-01-19 18:57:16 +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
|
|
|
|
|
|
|
|
|
2026-01-19 18:57:16 +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]
|
2026-01-19 18:57:16 +08:00
|
|
|
|
|
|
|
|
# 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():
|
2026-01-19 18:57:16 +08:00
|
|
|
try:
|
|
|
|
|
from src.mcp.cache import get_cached_mcp_tools
|
|
|
|
|
|
|
|
|
|
mcp_tools = get_cached_mcp_tools()
|
|
|
|
|
if mcp_tools:
|
2026-01-21 16:14:00 +08:00
|
|
|
logger.info(f"Using {len(mcp_tools)} cached MCP tool(s)")
|
2026-01-19 18:57:16 +08:00
|
|
|
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
|