Skip to content
Draft
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 13 additions & 9 deletions extensions/cli/src/stream/handleToolCalls.ts
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,8 @@ import { filterExcludedTools } from "../permissions/index.js";
import {
getServiceSync,
MCPServiceState,
MCPTool,
SERVICE_NAMES,
serviceContainer,
services,
} from "../services/index.js";
import type { ToolPermissionServiceState } from "../services/ToolPermissionService.js";
Expand Down Expand Up @@ -180,17 +180,21 @@ export async function getAllTools() {
const allBuiltinTools = getAllBuiltinTools();
const builtinToolNames = allBuiltinTools.map((tool) => tool.name);

let mcpTools: MCPTool[] = [];
let mcpToolNames: string[] = [];
const mcpServiceResult = getServiceSync<MCPServiceState>(SERVICE_NAMES.MCP);
if (mcpServiceResult.state === "ready") {
mcpTools = mcpServiceResult?.value?.tools ?? [];
mcpToolNames = mcpTools.map((t) => t.name);
let mcpServiceState: MCPServiceState | null = null;

// MCP loading is lazy in TUI mode
if (services.toolPermissions.isHeadless()) {
mcpServiceState = await serviceContainer.get(SERVICE_NAMES.MCP);
} else {
// MCP is lazy
// throw new Error("MCP Service not initialized");
const mcpServiceResult = getServiceSync<MCPServiceState>(SERVICE_NAMES.MCP);
if (mcpServiceResult.state === "ready") {
mcpServiceState = mcpServiceResult?.value;
}
}

const mcpTools = mcpServiceState?.tools ?? [];
const mcpToolNames = mcpTools.map((t) => t.name);

const allToolNames = [...builtinToolNames, ...mcpToolNames];

// Check if the ToolPermissionService is ready
Expand Down
Loading