@ai-sdk/harness 1.0.71 → 1.0.73

package/dist/agent/index.js

@@ -250,6 +250,7 @@ var HarnessStreamTextResult = class {
     this.toolsContext = options.toolsContext;
     this.providerName = `harness:${options.harnessId}`;
     this.modelId = options.sessionId;
+    this.outputSpecification = options.output;
     let controllerRef;
     const baseStream = new ReadableStream({
       start(c) {
@@ -258,18 +259,13 @@ var HarnessStreamTextResult = class {
       }
     });
     this.fullStreamController = controllerRef;
-    const [forFull, forText] = baseStream.tee();
-    this.stream = forFull;
-    this.fullStream = this.stream;
-    this.textStream = forText.pipeThrough(
+    this.baseStream = options.output == null ? baseStream.pipeThrough(
       new TransformStream({
         transform(part, controller) {
-          if (part.type === "text-delta") {
-            controller.enqueue(part.text);
-          }
+          controller.enqueue({ part, partialOutput: void 0 });
         }
       })
-    );
+    ) : baseStream.pipeThrough(createOutputTransformStream(options.output));
   }
   // ─── Writer-side methods used by the driver ────────────────────────
   /**
@@ -525,6 +521,38 @@ var HarnessStreamTextResult = class {
     }
   }
   // ─── Reader-side public surface (StreamTextResult contract) ────────
+  teeStream() {
+    const [stream, remainingStream] = this.baseStream.tee();
+    this.baseStream = remainingStream;
+    return stream;
+  }
+  get stream() {
+    return createAsyncIterableStream(
+      this.teeStream().pipeThrough(
+        new TransformStream({
+          transform({ part }, controller) {
+            controller.enqueue(part);
+          }
+        })
+      )
+    );
+  }
+  get fullStream() {
+    return this.stream;
+  }
+  get textStream() {
+    return createAsyncIterableStream(
+      this.stream.pipeThrough(
+        new TransformStream({
+          transform(part, controller) {
+            if (part.type === "text-delta") {
+              controller.enqueue(part.text);
+            }
+          }
+        })
+      )
+    );
+  }
   get content() {
     return this._content.promise;
   }
@@ -594,18 +622,48 @@ var HarnessStreamTextResult = class {
   get providerMetadata() {
     return this._providerMetadata.promise;
   }
-  // Output-specification surfaces are not yet supported.
   get experimental_partialOutputStream() {
-    throw notSupportedYet("partial output stream");
+    return this.partialOutputStream;
   }
   get partialOutputStream() {
-    throw notSupportedYet("partial output stream");
+    return createAsyncIterableStream(
+      this.teeStream().pipeThrough(
+        new TransformStream({
+          transform({ partialOutput }, controller) {
+            if (partialOutput != null) {
+              controller.enqueue(partialOutput);
+            }
+          }
+        })
+      )
+    );
   }
   get elementStream() {
-    throw notSupportedYet("element stream");
+    var _a4, _b4, _c;
+    const transform = (_a4 = this.outputSpecification) == null ? void 0 : _a4.createElementStreamTransform();
+    if (transform == null) {
+      throw notSupportedYet(
+        `element streams in ${(_c = (_b4 = this.outputSpecification) == null ? void 0 : _b4.name) != null ? _c : "text"} mode`
+      );
+    }
+    return createAsyncIterableStream(
+      this.teeStream().pipeThrough(transform)
+    );
   }
   get output() {
-    throw notSupportedYet("structured output");
+    return this.finalStep.then((step) => {
+      if (this.outputSpecification == null) {
+        throw notSupportedYet("structured output");
+      }
+      return this.outputSpecification.parseCompleteOutput(
+        { text: step.text },
+        {
+          response: step.response,
+          usage: step.usage,
+          finishReason: step.finishReason
+        }
+      );
+    });
   }
   async consumeStream() {
     const reader = this.fullStream.getReader();
@@ -725,6 +783,73 @@ var HarnessStreamTextResult = class {
     }
   }
 };
+function createOutputTransformStream(output) {
+  let firstTextChunkId;
+  let text = "";
+  let textChunk = "";
+  let textProviderMetadata;
+  let lastPublishedValue = "";
+  const publishTextChunk = (options) => {
+    options.controller.enqueue({
+      part: {
+        type: "text-delta",
+        id: firstTextChunkId,
+        text: textChunk,
+        providerMetadata: textProviderMetadata
+      },
+      partialOutput: options.partialOutput
+    });
+    textChunk = "";
+  };
+  return new TransformStream({
+    async transform(chunk, controller) {
+      var _a4;
+      if (chunk.type === "finish-step" && textChunk.length > 0) {
+        publishTextChunk({ controller });
+      }
+      if (chunk.type !== "text-delta" && chunk.type !== "text-start" && chunk.type !== "text-end") {
+        controller.enqueue({ part: chunk, partialOutput: void 0 });
+        return;
+      }
+      if (firstTextChunkId == null) {
+        firstTextChunkId = chunk.id;
+      } else if (chunk.id !== firstTextChunkId) {
+        controller.enqueue({ part: chunk, partialOutput: void 0 });
+        return;
+      }
+      if (chunk.type === "text-start") {
+        controller.enqueue({ part: chunk, partialOutput: void 0 });
+        return;
+      }
+      if (chunk.type === "text-end") {
+        if (textChunk.length > 0) {
+          publishTextChunk({ controller });
+        }
+        controller.enqueue({ part: chunk, partialOutput: void 0 });
+        return;
+      }
+      text += chunk.text;
+      textChunk += chunk.text;
+      textProviderMetadata = (_a4 = chunk.providerMetadata) != null ? _a4 : textProviderMetadata;
+      if (chunk.text.length === 0 && chunk.providerMetadata != null) {
+        controller.enqueue({ part: chunk, partialOutput: void 0 });
+        return;
+      }
+      const result = await output.parsePartialOutput({ text });
+      if (result === void 0) {
+        return;
+      }
+      const currentValue = typeof result.partial === "string" ? result.partial : JSON.stringify(result.partial);
+      if (currentValue !== lastPublishedValue) {
+        publishTextChunk({
+          controller,
+          partialOutput: result.partial
+        });
+        lastPublishedValue = currentValue;
+      }
+    }
+  });
+}
 function createEmptyPerformance() {
   return {
     effectiveOutputTokensPerSecond: 0,
@@ -1372,7 +1497,8 @@ function runPrompt(input) {
     // toolsContext is not configurable for harnesses; pass undefined cast.
     toolsContext: void 0,
     harnessId: input.harness.harnessId,
-    sessionId: input.session.sessionId
+    sessionId: input.session.sessionId,
+    output: input.output
   });
   const pendingToolApprovals = (_a4 = input.pendingToolApprovals) != null ? _a4 : [];
   const pendingToolResults = (_b4 = input.pendingToolResults) != null ? _b4 : [];
@@ -1413,6 +1539,7 @@ function runPrompt(input) {
     try {
       bridge = await toHarnessStream({
         invoke: input.mode === "continue" ? (emit) => input.session.doContinueTurn({
+          responseFormat: input.responseFormat,
           tools: input.toolSpecs,
           instructions: input.instructions,
           abortSignal: input.abortSignal,
@@ -1425,6 +1552,7 @@ function runPrompt(input) {
           }
           return input.session.doPromptTurn({
             prompt: input.prompt,
+            responseFormat: input.responseFormat,
             tools: input.toolSpecs,
             instructions: input.instructions,
             abortSignal: input.abortSignal,
@@ -2273,6 +2401,8 @@ var HarnessAgentSession = class {
         sessionWorkDir: this.sessionWorkDir,
         runtimeContext: options.runtimeContext,
         abortSignal: options.abortSignal,
+        responseFormat: options.responseFormat,
+        output: options.output,
         telemetry: options.telemetry,
         stopConditions: options.stopConditions,
         toolApproval: this.toolApproval,
@@ -2326,6 +2456,8 @@ var HarnessAgentSession = class {
         sessionWorkDir: this.sessionWorkDir,
         runtimeContext: options.runtimeContext,
         abortSignal: options.abortSignal,
+        responseFormat: options.responseFormat,
+        output: options.output,
         telemetry: options.telemetry,
         stopConditions: options.stopConditions,
         toolApproval: this.toolApproval,
@@ -3342,11 +3474,13 @@ var HarnessAgent = class {
   async generate(options) {
     const turnInput = this._resolveTurnInput(options);
     const runtimeContext = {};
+    const responseFormat = await this._resolveResponseFormat();
     const { result, done } = this._startTurn({
       session: options.session,
       turnInput,
       runtimeContext,
-      abortSignal: options.abortSignal
+      abortSignal: options.abortSignal,
+      responseFormat
     });
     await done;
     return this._toGenerateResult(result);
@@ -3354,11 +3488,13 @@ var HarnessAgent = class {
   async stream(options) {
     const turnInput = this._resolveTurnInput(options);
     const runtimeContext = {};
+    const responseFormat = await this._resolveResponseFormat();
     const { result } = this._startTurn({
       session: options.session,
       turnInput,
       runtimeContext,
-      abortSignal: options.abortSignal
+      abortSignal: options.abortSignal,
+      responseFormat
     });
     return result;
   }
@@ -3370,6 +3506,7 @@ var HarnessAgent = class {
   async continueGenerate(options) {
     var _a4, _b4;
     const runtimeContext = {};
+    const responseFormat = await this._resolveResponseFormat();
     const { result, done } = this._startTurn({
       session: options.session,
       turnInput: {
@@ -3378,7 +3515,8 @@ var HarnessAgent = class {
         toolResultContinuations: (_b4 = options.toolResultContinuations) != null ? _b4 : []
       },
       runtimeContext,
-      abortSignal: options.abortSignal
+      abortSignal: options.abortSignal,
+      responseFormat
     });
     await done;
     return this._toGenerateResult(result);
@@ -3394,6 +3532,7 @@ var HarnessAgent = class {
   async continueStream(options) {
     var _a4, _b4;
     const runtimeContext = {};
+    const responseFormat = await this._resolveResponseFormat();
     const { result } = this._startTurn({
       session: options.session,
       turnInput: {
@@ -3402,7 +3541,8 @@ var HarnessAgent = class {
         toolResultContinuations: (_b4 = options.toolResultContinuations) != null ? _b4 : []
       },
       runtimeContext,
-      abortSignal: options.abortSignal
+      abortSignal: options.abortSignal,
+      responseFormat
     });
     return result;
   }
@@ -3417,6 +3557,8 @@ var HarnessAgent = class {
         builtinToolFiltering: this.builtinToolFiltering,
         runtimeContext: input.runtimeContext,
         abortSignal: input.abortSignal,
+        responseFormat: input.responseFormat,
+        output: this.settings.output,
         telemetry: this.settings.telemetry,
         stopConditions: this.stopConditions,
         toolApprovalContinuations: input.turnInput.toolApprovalContinuations,
@@ -3432,6 +3574,8 @@ var HarnessAgent = class {
       builtinToolFiltering: this.builtinToolFiltering,
       runtimeContext: input.runtimeContext,
       abortSignal: input.abortSignal,
+      responseFormat: input.responseFormat,
+      output: this.settings.output,
       telemetry: this.settings.telemetry,
       stopConditions: this.stopConditions
     });
@@ -3516,20 +3660,34 @@ var HarnessAgent = class {
     return specs;
   }
   async _toGenerateResult(streamResult) {
-    const [steps, usage, responseMessages] = await Promise.all([
+    const [steps, usage, responseMessages, output] = await Promise.all([
       streamResult.steps,
       streamResult.usage,
-      streamResult.responseMessages
+      streamResult.responseMessages,
+      this.settings.output == null ? Promise.resolve(void 0) : streamResult.output
     ]);
-    return new HarnessGenerateTextResult({ steps, usage, responseMessages });
+    return new HarnessGenerateTextResult({ steps, usage, responseMessages, output });
+  }
+  async _resolveResponseFormat() {
+    var _a4;
+    const responseFormat = await ((_a4 = this.settings.output) == null ? void 0 : _a4.responseFormat);
+    if (responseFormat == null || responseFormat.type === "text") {
+      return responseFormat == null ? void 0 : { type: "text" };
+    }
+    return {
+      type: "json",
+      ...responseFormat.schema == null ? {} : { schema: responseFormat.schema },
+      ...responseFormat.name == null ? {} : { name: responseFormat.name },
+      ...responseFormat.description == null ? {} : { description: responseFormat.description }
+    };
   }
 };
 var HarnessGenerateTextResult = class {
   constructor(options) {
-    this.output = void 0;
     this.steps = options.steps;
     this.usage = options.usage;
     this.responseMessages = options.responseMessages;
+    this.output = options.output;
   }
   get finalStep() {
     return this.steps.at(-1);
@@ -3691,10 +3849,11 @@ async function prepareSandboxForHarness(options) {
       "prepareSandboxForHarness: at least one harness must be provided."
     );
   }
-  const harnesses = [...options.harnesses].sort(
-    (a, b) => a.harnessId.localeCompare(b.harnessId)
-  );
-  assertUniqueHarnessIds(harnesses);
+  const harnesses = [
+    ...new Map(
+      options.harnesses.map((harness) => [harness.harnessId, harness])
+    ).values()
+  ].sort((a, b) => a.harnessId.localeCompare(b.harnessId));
   const workDir = sandboxConfig.workDir == null ? void 0 : normalizeSandboxWorkDir(sandboxConfig.workDir);
   const recipeIdentities = {};
   const skippedHarnessIds = [];
@@ -3741,17 +3900,6 @@ async function prepareSandboxForHarness(options) {
     skippedHarnessIds
   };
 }
-function assertUniqueHarnessIds(harnesses) {
-  const seen = /* @__PURE__ */ new Set();
-  for (const harness of harnesses) {
-    if (seen.has(harness.harnessId)) {
-      throw new Error(
-        `prepareSandboxForHarness: duplicate harness id "${harness.harnessId}".`
-      );
-    }
-    seen.add(harness.harnessId);
-  }
-}
 async function resolvePreparedSandboxIdentity({
   recipeIdentities,
   bootstrapHash,