ai 6.0.24 → 6.0.26

package/dist/index.mjs

@@ -921,7 +921,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.24" : "0.0.0-test";
+var VERSION = true ? "6.0.26" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({ url }) => {
@@ -2907,12 +2907,16 @@ async function parsePartialJson(jsonText) {
 
 // src/generate-text/output.ts
 var text = () => ({
+  name: "text",
   responseFormat: Promise.resolve({ type: "text" }),
   async parseCompleteOutput({ text: text2 }) {
     return text2;
   },
   async parsePartialOutput({ text: text2 }) {
     return { partial: text2 };
+  },
+  createElementStreamTransform() {
+    return void 0;
   }
 });
 var object = ({
@@ -2922,6 +2926,7 @@ var object = ({
 }) => {
   const schema = asSchema2(inputSchema);
   return {
+    name: "object",
     responseFormat: resolve(schema.jsonSchema).then((jsonSchema2) => ({
       type: "json",
       schema: jsonSchema2,
@@ -2971,6 +2976,9 @@ var object = ({
           };
         }
       }
+    },
+    createElementStreamTransform() {
+      return void 0;
     }
   };
 };
@@ -2981,6 +2989,7 @@ var array = ({
 }) => {
   const elementSchema = asSchema2(inputElementSchema);
   return {
+    name: "array",
     // JSON schema that describes an array of elements:
     responseFormat: resolve(elementSchema.jsonSchema).then((jsonSchema2) => {
       const { $schema, ...itemSchema } = jsonSchema2;
@@ -3070,6 +3079,18 @@ var array = ({
           return { partial: parsedElements };
         }
       }
+    },
+    createElementStreamTransform() {
+      let publishedElements = 0;
+      return new TransformStream({
+        transform({ partialOutput }, controller) {
+          if (partialOutput != null) {
+            for (; publishedElements < partialOutput.length; publishedElements++) {
+              controller.enqueue(partialOutput[publishedElements]);
+            }
+          }
+        }
+      });
     }
   };
 };
@@ -3079,6 +3100,7 @@ var choice = ({
   description
 }) => {
   return {
+    name: "choice",
     // JSON schema that describes an enumeration:
     responseFormat: Promise.resolve({
       type: "json",
@@ -3145,6 +3167,9 @@ var choice = ({
           }
         }
       }
+    },
+    createElementStreamTransform() {
+      return void 0;
     }
   };
 };
@@ -3153,6 +3178,7 @@ var json = ({
   description
 } = {}) => {
   return {
+    name: "json",
     responseFormat: Promise.resolve({
       type: "json",
       ...name16 != null && { name: name16 },
@@ -3184,6 +3210,9 @@ var json = ({
           return result.value === void 0 ? void 0 : { partial: result.value };
         }
       }
+    },
+    createElementStreamTransform() {
+      return void 0;
     }
   };
 };
@@ -4306,7 +4335,8 @@ function asContent({
 
 // src/generate-text/stream-text.ts
 import {
-  getErrorMessage as getErrorMessage7
+  getErrorMessage as getErrorMessage7,
+  UnsupportedFunctionalityError as UnsupportedFunctionalityError2
 } from "@ai-sdk/provider";
 import {
   createIdGenerator as createIdGenerator2,
@@ -6860,6 +6890,16 @@ var DefaultStreamTextResult = class {
       )
     );
   }
+  get elementStream() {
+    var _a16, _b, _c;
+    const transform = (_a16 = this.outputSpecification) == null ? void 0 : _a16.createElementStreamTransform();
+    if (transform == null) {
+      throw new UnsupportedFunctionalityError2({
+        functionality: `element streams in ${(_c = (_b = this.outputSpecification) == null ? void 0 : _b.name) != null ? _c : "text"} mode`
+      });
+    }
+    return createAsyncIterableStream(this.teeStream().pipeThrough(transform));
+  }
   get output() {
     return this.finalStep.then((step) => {
       var _a16;
@@ -8685,7 +8725,7 @@ import {
   isJSONArray,
   isJSONObject,
   TypeValidationError as TypeValidationError4,
-  UnsupportedFunctionalityError as UnsupportedFunctionalityError2
+  UnsupportedFunctionalityError as UnsupportedFunctionalityError3
 } from "@ai-sdk/provider";
 import {
   asSchema as asSchema4,
@@ -8710,7 +8750,7 @@ var noSchemaOutputStrategy = {
     } : { success: true, value };
   },
   createElementStream() {
-    throw new UnsupportedFunctionalityError2({
+    throw new UnsupportedFunctionalityError3({
       functionality: "element streams in no-schema mode"
     });
   }
@@ -8732,7 +8772,7 @@ var objectOutputStrategy = (schema) => ({
     return safeValidateTypes4({ value, schema });
   },
   createElementStream() {
-    throw new UnsupportedFunctionalityError2({
+    throw new UnsupportedFunctionalityError3({
       functionality: "element streams in object mode"
     });
   }
@@ -8921,7 +8961,7 @@ var enumOutputStrategy = (enumValues) => {
       };
     },
     createElementStream() {
-      throw new UnsupportedFunctionalityError2({
+      throw new UnsupportedFunctionalityError3({
         functionality: "element streams in enum mode"
       });
     }
@@ -10344,6 +10384,130 @@ function defaultSettingsMiddleware({
   };
 }
 
+// src/middleware/extract-json-middleware.ts
+function defaultTransform(text2) {
+  return text2.replace(/^```(?:json)?\s*\n?/, "").replace(/\n?```\s*$/, "").trim();
+}
+function extractJsonMiddleware(options) {
+  var _a16;
+  const transform = (_a16 = options == null ? void 0 : options.transform) != null ? _a16 : defaultTransform;
+  const hasCustomTransform = (options == null ? void 0 : options.transform) !== void 0;
+  return {
+    specificationVersion: "v3",
+    wrapGenerate: async ({ doGenerate }) => {
+      const { content, ...rest } = await doGenerate();
+      const transformedContent = [];
+      for (const part of content) {
+        if (part.type !== "text") {
+          transformedContent.push(part);
+          continue;
+        }
+        transformedContent.push({
+          ...part,
+          text: transform(part.text)
+        });
+      }
+      return { content: transformedContent, ...rest };
+    },
+    wrapStream: async ({ doStream }) => {
+      const { stream, ...rest } = await doStream();
+      const textBlocks = {};
+      const SUFFIX_BUFFER_SIZE = 12;
+      return {
+        stream: stream.pipeThrough(
+          new TransformStream({
+            transform: (chunk, controller) => {
+              if (chunk.type === "text-start") {
+                textBlocks[chunk.id] = {
+                  startEvent: chunk,
+                  // Custom transforms need to buffer all content
+                  phase: hasCustomTransform ? "buffering" : "prefix",
+                  buffer: "",
+                  prefixStripped: false
+                };
+                return;
+              }
+              if (chunk.type === "text-delta") {
+                const block = textBlocks[chunk.id];
+                if (!block) {
+                  controller.enqueue(chunk);
+                  return;
+                }
+                block.buffer += chunk.delta;
+                if (block.phase === "buffering") {
+                  return;
+                }
+                if (block.phase === "prefix") {
+                  if (block.buffer.length > 0 && !block.buffer.startsWith("`")) {
+                    block.phase = "streaming";
+                    controller.enqueue(block.startEvent);
+                  } else if (block.buffer.startsWith("```")) {
+                    if (block.buffer.includes("\n")) {
+                      const prefixMatch = block.buffer.match(/^```(?:json)?\s*\n/);
+                      if (prefixMatch) {
+                        block.buffer = block.buffer.slice(
+                          prefixMatch[0].length
+                        );
+                        block.prefixStripped = true;
+                        block.phase = "streaming";
+                        controller.enqueue(block.startEvent);
+                      } else {
+                        block.phase = "streaming";
+                        controller.enqueue(block.startEvent);
+                      }
+                    }
+                  } else if (block.buffer.length >= 3 && !block.buffer.startsWith("```")) {
+                    block.phase = "streaming";
+                    controller.enqueue(block.startEvent);
+                  }
+                }
+                if (block.phase === "streaming" && block.buffer.length > SUFFIX_BUFFER_SIZE) {
+                  const toStream = block.buffer.slice(0, -SUFFIX_BUFFER_SIZE);
+                  block.buffer = block.buffer.slice(-SUFFIX_BUFFER_SIZE);
+                  controller.enqueue({
+                    type: "text-delta",
+                    id: chunk.id,
+                    delta: toStream
+                  });
+                }
+                return;
+              }
+              if (chunk.type === "text-end") {
+                const block = textBlocks[chunk.id];
+                if (block) {
+                  if (block.phase === "prefix" || block.phase === "buffering") {
+                    controller.enqueue(block.startEvent);
+                  }
+                  let remaining = block.buffer;
+                  if (block.phase === "buffering") {
+                    remaining = transform(remaining);
+                  } else if (block.prefixStripped) {
+                    remaining = remaining.replace(/\n?```\s*$/, "").trimEnd();
+                  } else {
+                    remaining = transform(remaining);
+                  }
+                  if (remaining.length > 0) {
+                    controller.enqueue({
+                      type: "text-delta",
+                      id: chunk.id,
+                      delta: remaining
+                    });
+                  }
+                  controller.enqueue(chunk);
+                  delete textBlocks[chunk.id];
+                  return;
+                }
+              }
+              controller.enqueue(chunk);
+            }
+          })
+        ),
+        ...rest
+      };
+    }
+  };
+}
+
 // src/util/get-potential-start-index.ts
 function getPotentialStartIndex(text2, searchedText) {
   if (searchedText.length === 0) {
@@ -12145,6 +12309,7 @@ export {
   experimental_generateImage,
   generateSpeech as experimental_generateSpeech,
   transcribe as experimental_transcribe,
+  extractJsonMiddleware,
   extractReasoningMiddleware,
   gateway2 as gateway,
   generateId,