Function wrapMCPTools

src/node/services/mcpServerManager.ts:136–182

(
  tools: Record<string, Tool>,
  options?: { onActivity?: () => void; onClosed?: () => void }
)

 * This ensures image content is properly converted to media type.
 */
export function wrapMCPTools(
  tools: Record<string, Tool>,
  options?: { onActivity?: () => void; onClosed?: () => void }
): Record<string, Tool> {
  const { onActivity, onClosed } = options ?? {};
  const wrapped: Record<string, Tool> = {};
  for (const [toolName, tool] of Object.entries(tools)) {
    // Only wrap tools that have an execute function
    if (!tool.execute) {
      wrapped[toolName] = tool;
      continue;
    }

    const originalExecute = tool.execute;
    wrapped[toolName] = {
      ...tool,
      execute: async (args: Parameters<typeof originalExecute>[0], context) => {
        // Mark the MCP server set as active *before* execution, so failed tool
        // calls (including closed-client races) still count as activity.
        onActivity?.();

        try {
          const abortSignal =
            context && typeof context === "object" && "abortSignal" in context
              ? (context as { abortSignal?: AbortSignal }).abortSignal
              : undefined;

          const result: unknown = await runMCPToolWithDeadline(
            () => Promise.resolve(originalExecute(args, context)) as Promise<unknown>,
            { toolName, timeoutMs: MCP_TOOL_CALL_TIMEOUT_MS, signal: abortSignal }
          );
          return transformMCPResult(result as MCPCallToolResult);
        } catch (error) {
          if (shouldRecycleClientAfterToolError(error)) {
            try {
              onClosed?.();
            } catch {
              // Swallow — original tool error takes priority.
            }
          }
          throw error;
        }
      },
    };
  }
  return wrapped;
}

type ResolvedHeaders = Record<string, string> | undefined;