From 03dd44c642b5e1c17aad05c00d9a040eea6411c9 Mon Sep 17 00:00:00 2001 From: Ed_ Date: Wed, 24 Jun 2026 17:08:53 -0400 Subject: [PATCH] refactor(ai_client): use mcp_tool_specs.tool_names() (3 sites) --- src/ai_client.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/ai_client.py b/src/ai_client.py index 013f8960..cdd88a95 100644 --- a/src/ai_client.py +++ b/src/ai_client.py @@ -39,6 +39,7 @@ from typing import Optional, Callable, Any, List, Union, cast, Iterable from src import project_manager from src import file_cache from src import mcp_client +from src import mcp_tool_specs from src import mma_prompts from src import performance_monitor from src import project_manager @@ -557,7 +558,7 @@ def _set_tool_preset_result(preset_name: Optional[str]) -> Result[None]: if preset_name in presets: preset = presets[preset_name] _active_tool_preset = preset - new_tools = {name: False for name in mcp_client.TOOL_NAMES} + new_tools = {name: False for name in mcp_tool_specs.tool_names()} new_tools[TOOL_NAME] = False for cat in preset.categories.values(): for tool in cat: @@ -579,7 +580,7 @@ def set_tool_preset(preset_name: Optional[str]) -> None: _tool_approval_modes = {} if not preset_name or preset_name == "None": # Enable all tools if no preset - _agent_tools = {name: True for name in mcp_client.TOOL_NAMES} + _agent_tools = {name: True for name in mcp_tool_specs.tool_names()} _agent_tools[TOOL_NAME] = True _active_tool_preset = None else: @@ -1009,7 +1010,7 @@ async def _execute_single_tool_call_async( tool_executed = True if not tool_executed: - is_native = name in mcp_client.TOOL_NAMES + is_native = name in mcp_tool_specs.tool_names() ext_tools = mcp_client.get_external_mcp_manager().get_all_tools() is_external = name in ext_tools if name and (is_native or is_external):