@ai-sdk/harness-acp 1.0.1 → 1.0.2

package/dist/index.d.ts

@@ -1,6 +1,104 @@
 import { HarnessV1PermissionMode, HarnessV1 } from '@ai-sdk/harness';
 import { ToolSet } from '@ai-sdk/provider-utils';
 
+type ACPMetadata = Record<string, unknown>;
+type ACPRole = 'assistant' | 'user';
+type ACPAnnotations = {
+    audience?: Array<ACPRole> | null;
+    lastModified?: string | null;
+    priority?: number | null;
+    _meta?: ACPMetadata | null;
+};
+type ACPTextContent = {
+    annotations?: ACPAnnotations | null;
+    text: string;
+    _meta?: ACPMetadata | null;
+};
+type ACPImageContent = {
+    annotations?: ACPAnnotations | null;
+    data: string;
+    mimeType: string;
+    uri?: string | null;
+    _meta?: ACPMetadata | null;
+};
+type ACPAudioContent = {
+    annotations?: ACPAnnotations | null;
+    data: string;
+    mimeType: string;
+    _meta?: ACPMetadata | null;
+};
+type ACPResourceLink = {
+    annotations?: ACPAnnotations | null;
+    description?: string | null;
+    mimeType?: string | null;
+    name: string;
+    size?: number | null;
+    title?: string | null;
+    uri: string;
+    _meta?: ACPMetadata | null;
+};
+type ACPTextResourceContents = {
+    mimeType?: string | null;
+    text: string;
+    uri: string;
+    _meta?: ACPMetadata | null;
+};
+type ACPBlobResourceContents = {
+    blob: string;
+    mimeType?: string | null;
+    uri: string;
+    _meta?: ACPMetadata | null;
+};
+type ACPEmbeddedResource = {
+    annotations?: ACPAnnotations | null;
+    resource: ACPTextResourceContents | ACPBlobResourceContents;
+    _meta?: ACPMetadata | null;
+};
+type ACPContentBlock = (ACPTextContent & {
+    type: 'text';
+}) | (ACPImageContent & {
+    type: 'image';
+}) | (ACPAudioContent & {
+    type: 'audio';
+}) | (ACPResourceLink & {
+    type: 'resource_link';
+}) | (ACPEmbeddedResource & {
+    type: 'resource';
+});
+type ACPToolCallContent = {
+    type: 'content';
+    content: ACPContentBlock;
+    _meta?: ACPMetadata | null;
+} | {
+    type: 'diff';
+    path: string;
+    oldText?: string | null;
+    newText: string;
+    _meta?: ACPMetadata | null;
+} | {
+    type: 'terminal';
+    terminalId: string;
+    _meta?: ACPMetadata | null;
+};
+type ACPToolKind = 'read' | 'edit' | 'delete' | 'move' | 'search' | 'execute' | 'think' | 'fetch' | 'switch_mode' | 'other';
+type ACPToolCallStatus = 'pending' | 'in_progress' | 'completed' | 'failed';
+type ACPToolCallLocation = {
+    path: string;
+    line?: number | null;
+    _meta?: ACPMetadata | null;
+};
+type ACPToolCall = {
+    toolCallId: string;
+    title: string;
+    kind?: ACPToolKind;
+    status?: ACPToolCallStatus;
+    content?: Array<ACPToolCallContent>;
+    locations?: Array<ACPToolCallLocation>;
+    rawInput?: unknown;
+    rawOutput?: unknown;
+    _meta?: ACPMetadata | null;
+};
+
 type ACPSerializablePrimitive = string | number | boolean | null;
 type ACPSerializableValue = ACPSerializablePrimitive | ReadonlyArray<ACPSerializableValue> | {
     readonly [key: string]: ACPSerializableValue;
@@ -53,6 +151,8 @@ type ACPPermissionModeMapping = Readonly<Record<HarnessV1PermissionMode, ACPPerm
 type ACPV1Settings = {
     readonly version?: 'v1';
     readonly harnessId: string;
+    readonly mcpServers?: Record<string, unknown>;
+    readonly isMcpToolCall?: (toolCall: ACPToolCall) => boolean;
     readonly auth?: ACPProviderAuthenticationMode;
     readonly source: ACPSource;
     readonly executable: string;
@@ -70,6 +170,11 @@ type ACPV1Settings = {
     readonly session?: {
         readonly meta?: Readonly<Record<string, ACPSerializableValue>>;
     };
+    /**
+     * Creates the authentication token used by the sandbox bridge. Defaults to
+     * a random 32-byte hexadecimal token.
+     */
+    readonly mintBridgeToken?: (sandboxId: string) => string;
 };
 
 type ACPAuthOptions = ACPProviderAuthenticationMode;
@@ -80,6 +185,12 @@ type ACPClientApp = {
 
 type ACPHarnessSettings<TBuiltinTools extends ToolSet = {}> = {
     readonly builtinTools?: TBuiltinTools;
+    /**
+     * MCP server definitions keyed by server name. Each definition uses the
+     * underlying runtime's native MCP server configuration format.
+     */
+    readonly mcpServers?: Record<string, unknown>;
+    readonly isMcpToolCall?: (toolCall: ACPToolCall) => boolean;
     readonly port?: number;
     readonly startupTimeoutMs?: number;
     readonly clientApp?: ACPClientApp;
@@ -96,9 +207,10 @@ type ACPHarnessSettings<TBuiltinTools extends ToolSet = {}> = {
     readonly modelId?: ACPV1Settings['modelId'];
     readonly permissionModeMapping?: ACPV1Settings['permissionModeMapping'];
     readonly session?: ACPV1Settings['session'];
+    readonly mintBridgeToken?: ACPV1Settings['mintBridgeToken'];
 };
 declare function createACP<TBuiltinTools extends ToolSet = {}>(settings: ACPHarnessSettings<TBuiltinTools>): HarnessV1<TBuiltinTools>;
 
 declare const VERSION: string;
 
-export { type ACPAuthOptions, type ACPAuthentication, type ACPClientApp, type ACPHarnessSettings, type ACPNpmLockedSource, type ACPNpmSimpleSource, type ACPPermissionModeMapping, type ACPPermissionModeTarget, type ACPProfileValue, type ACPProviderAuthentication, type ACPProviderAuthenticationMode, type ACPSerializablePrimitive, type ACPSerializableValue, type ACPSource, type ACPV1Settings, type ACPValueSource, VERSION, createACP };
+export { type ACPAnnotations, type ACPAudioContent, type ACPAuthOptions, type ACPAuthentication, type ACPBlobResourceContents, type ACPClientApp, type ACPContentBlock, type ACPEmbeddedResource, type ACPHarnessSettings, type ACPImageContent, type ACPMetadata, type ACPNpmLockedSource, type ACPNpmSimpleSource, type ACPPermissionModeMapping, type ACPPermissionModeTarget, type ACPProfileValue, type ACPProviderAuthentication, type ACPProviderAuthenticationMode, type ACPResourceLink, type ACPRole, type ACPSerializablePrimitive, type ACPSerializableValue, type ACPSource, type ACPTextContent, type ACPTextResourceContents, type ACPToolCall, type ACPToolCallContent, type ACPToolCallLocation, type ACPToolCallStatus, type ACPToolKind, type ACPV1Settings, type ACPValueSource, VERSION, createACP };

package/dist/index.js

@@ -540,9 +540,19 @@ var coldRestoreSchema = z2.object({
   type: z2.literal("cold-restore"),
   acpSessionId: z2.string()
 });
-var outboundMessageSchema = harnessV1BridgeOutboundMessageSchema;
+var acpToolCallCandidateSchema = z2.object({
+  type: z2.literal("acp-tool-call-candidate"),
+  toolCall: z2.custom(
+    (value) => value != null && typeof value === "object" && typeof value.toolCallId === "string" && typeof value.title === "string"
+  )
+});
+var outboundMessageSchema = z2.union([
+  harnessV1BridgeOutboundMessageSchema,
+  acpToolCallCandidateSchema
+]);
 var startMessageSchema = harnessV1BridgeStartBaseSchema.extend({
   prompt: z2.array(acpTextContentBlockSchema),
+  mcpServers: z2.record(z2.string(), z2.unknown()).optional(),
   tools: z2.array(acpSerializableToolSpecSchema).optional(),
   builtinTools: z2.array(builtinToolSchema).readonly().default([]),
   permissionModeMapping: permissionModeMappingSchema.optional(),
@@ -622,6 +632,7 @@ function createACPTurnStartConfig({
   builtinTools,
   permissionMode,
   permissionModeMapping,
+  mcpServers,
   debug,
   authenticationProfile,
   sessionMeta
@@ -633,7 +644,8 @@ function createACPTurnStartConfig({
         authenticationProfile,
         sessionMeta: sessionMeta ?? null,
         builtinTools,
-        permissionModeMapping: permissionModeMapping ?? null
+        permissionModeMapping: permissionModeMapping ?? null,
+        mcpServers: mcpServers ?? null
       })
     ).digest("hex"),
     prompt: [...prompt],
@@ -897,6 +909,14 @@ function createACPV1({
   clientApp,
   lifecycleStateSchema
 }) {
+  if (settings.mcpServers != null && Object.prototype.hasOwnProperty.call(
+    settings.mcpServers,
+    "ai-sdk-harness-tools"
+  )) {
+    throw new Error(
+      'ACP MCP server name "ai-sdk-harness-tools" is reserved for HarnessAgent tools.'
+    );
+  }
   if (!HARNESS_ID_REGEXP.test(settings.harnessId)) {
     throw new Error(
       `ACP harnessId must be a stable kebab-case identifier; received ${JSON.stringify(settings.harnessId)}.`
@@ -1118,6 +1138,8 @@ function createACPV1({
               builtinTools: builtinToolCatalog,
               permissionMode,
               permissionModeMapping,
+              mcpServers: settings.mcpServers,
+              isMcpToolCall: settings.isMcpToolCall,
               initialGuidanceApplied: resolveACPInitialGuidanceApplied({
                 isResume: true,
                 lifecycleState: lifecycleData
@@ -1167,7 +1189,8 @@ function createACPV1({
                   authenticationProfile,
                   sessionMeta: settings.session?.meta,
                   builtinTools: builtinToolCatalog,
-                  permissionModeMapping
+                  permissionModeMapping,
+                  mcpServers: settings.mcpServers
                 });
                 respawnStrategy = {
                   mode: "lossy-rerun",
@@ -1196,6 +1219,7 @@ function createACPV1({
             sessionMeta: settings.session?.meta,
             builtinTools: builtinToolCatalog,
             permissionModeMapping,
+            mcpServers: settings.mcpServers,
             debug: startOptions.observability?.debug
           });
           respawnStrategy = {
@@ -1210,7 +1234,7 @@ function createACPV1({
         override: portOverride,
         harnessId: settings.harnessId
       });
-      const token = randomBytes(32).toString("hex");
+      const token = settings.mintBridgeToken == null ? randomBytes(32).toString("hex") : settings.mintBridgeToken(sandboxSession.id);
       await sandbox.run({
         command: `mkdir -p ${shellQuote(workDir)} ${shellQuote(bridgeStateDir)}`,
         abortSignal: startOptions.abortSignal
@@ -1300,6 +1324,7 @@ function createACPV1({
               builtinTools: builtinToolCatalog,
               permissionMode,
               permissionModeMapping,
+              ...settings.mcpServers == null ? {} : { mcpServers: settings.mcpServers },
               tools: respawnStrategy.turnStartConfig.tools,
               turnStartConfig: respawnStrategy.turnStartConfig,
               recoveryMode: {
@@ -1337,6 +1362,8 @@ function createACPV1({
         builtinTools: builtinToolCatalog,
         permissionMode,
         permissionModeMapping,
+        mcpServers: settings.mcpServers,
+        isMcpToolCall: settings.isMcpToolCall,
         initialGuidanceApplied: resolveACPInitialGuidanceApplied({
           isResume,
           lifecycleState: lifecycleData
@@ -1496,6 +1523,8 @@ function createSession({
   builtinTools,
   permissionMode,
   permissionModeMapping,
+  mcpServers,
+  isMcpToolCall,
   initialGuidanceApplied: initialGuidanceAppliedAtStart,
   skillCatalog,
   skillsFingerprint,
@@ -1542,6 +1571,8 @@ function createSession({
     let abortError;
     let abortListenerAttached = false;
     let openBlock;
+    const dynamicToolCalls = /* @__PURE__ */ new Map();
+    const toolCallClassificationErrors = /* @__PURE__ */ new Map();
     const subscriptions = [];
     const forward = (event) => {
       if (event.type === "text-start" || event.type === "reasoning-start") {
@@ -1574,9 +1605,7 @@ function createSession({
       "reasoning-start",
       "reasoning-delta",
       "reasoning-end",
-      "tool-call",
       "tool-approval-request",
-      "tool-result",
       "file-change",
       "finish-step",
       "raw"
@@ -1604,6 +1633,44 @@ function createSession({
         settle({ error: abortError });
       }
     };
+    subscriptions.push(
+      channel.on("acp-tool-call-candidate", (event) => {
+        try {
+          dynamicToolCalls.set(
+            event.toolCall.toolCallId,
+            isMcpToolCall?.(event.toolCall) === true
+          );
+        } catch (error) {
+          toolCallClassificationErrors.set(event.toolCall.toolCallId, error);
+        }
+      })
+    );
+    subscriptions.push(
+      channel.on("tool-call", (event) => {
+        if (toolCallClassificationErrors.has(event.toolCallId)) {
+          const error = toolCallClassificationErrors.get(event.toolCallId);
+          closeForwardedBlock();
+          forward({ type: "error", error });
+          try {
+            channel.send({ type: "abort" });
+          } catch {
+          }
+          settle({ error });
+          return;
+        }
+        forward(
+          dynamicToolCalls.get(event.toolCallId) === true ? { ...event, dynamic: true } : event
+        );
+      })
+    );
+    subscriptions.push(
+      channel.on("tool-result", (event) => {
+        const dynamic = dynamicToolCalls.get(event.toolCallId) === true;
+        dynamicToolCalls.delete(event.toolCallId);
+        toolCallClassificationErrors.delete(event.toolCallId);
+        forward(dynamic ? { ...event, dynamic: true } : event);
+      })
+    );
     for (const type of eventTypes) {
       subscriptions.push(channel.on(type, (event) => forward(event)));
     }
@@ -1744,6 +1811,7 @@ function createSession({
         builtinTools,
         permissionMode,
         permissionModeMapping,
+        mcpServers,
         debug,
         authenticationProfile,
         sessionMeta
@@ -1769,6 +1837,7 @@ function createSession({
             builtinTools,
             permissionMode,
             permissionModeMapping,
+            ...mcpServers == null ? {} : { mcpServers },
             tools: options.tools == null ? void 0 : turnStartConfig.tools,
             turnStartConfig
           });
@@ -1806,6 +1875,7 @@ function createSession({
             builtinTools: turnStartConfig.builtinTools,
             permissionMode: turnStartConfig.permissionMode,
             permissionModeMapping: turnStartConfig.permissionModeMapping,
+            ...mcpServers == null ? {} : { mcpServers },
             tools: turnStartConfig.tools,
             turnStartConfig,
             recoveryMode: {
@@ -1938,7 +2008,8 @@ function validateACPTurnStartConfig({
   authenticationProfile,
   sessionMeta,
   builtinTools,
-  permissionModeMapping
+  permissionModeMapping,
+  mcpServers
 }) {
   const current = createACPTurnStartConfig({
     prompt: turnStartConfig.prompt,
@@ -1946,6 +2017,7 @@ function validateACPTurnStartConfig({
     builtinTools,
     permissionMode: turnStartConfig.permissionMode,
     permissionModeMapping,
+    mcpServers,
     debug: turnStartConfig.debug,
     authenticationProfile,
     sessionMeta
@@ -1964,6 +2036,7 @@ function validateACPColdSessionConfiguration({
   sessionMeta,
   builtinTools,
   permissionModeMapping,
+  mcpServers,
   debug
 }) {
   const current = createACPTurnStartConfig({
@@ -1972,6 +2045,7 @@ function validateACPColdSessionConfiguration({
     builtinTools,
     permissionMode,
     permissionModeMapping,
+    mcpServers,
     debug,
     authenticationProfile,
     sessionMeta
@@ -2021,7 +2095,7 @@ function deserializeBridgeError({
 }
 
 // src/version.ts
-var VERSION = true ? "1.0.1" : "0.0.0-test";
+var VERSION = true ? "1.0.2" : "0.0.0-test";
 
 // src/acp-harness.ts
 var ACP_CLIENT_APP = {
@@ -2063,6 +2137,14 @@ var acpResumeStateSchema = z4.object({
   skillsFingerprint: z4.string().optional()
 });
 function createACP(settings) {
+  if (settings.mcpServers != null && Object.prototype.hasOwnProperty.call(
+    settings.mcpServers,
+    "ai-sdk-harness-tools"
+  )) {
+    throw new Error(
+      'ACP MCP server name "ai-sdk-harness-tools" is reserved for HarnessAgent tools.'
+    );
+  }
   const version = settings.version ?? "v1";
   switch (version) {
     case "v1": {