ai 6.0.24 → 6.0.26

package/dist/index.js

@@ -89,6 +89,7 @@ __export(src_exports, {
   experimental_generateImage: () => experimental_generateImage,
   experimental_generateSpeech: () => generateSpeech,
   experimental_transcribe: () => transcribe,
+  extractJsonMiddleware: () => extractJsonMiddleware,
   extractReasoningMiddleware: () => extractReasoningMiddleware,
   gateway: () => import_gateway3.gateway,
   generateId: () => import_provider_utils38.generateId,
@@ -1024,7 +1025,7 @@ var import_provider_utils3 = require("@ai-sdk/provider-utils");
 var import_provider_utils4 = require("@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 }) => {
@@ -2995,12 +2996,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 = ({
@@ -3010,6 +3015,7 @@ var object = ({
 }) => {
   const schema = (0, import_provider_utils13.asSchema)(inputSchema);
   return {
+    name: "object",
     responseFormat: (0, import_provider_utils13.resolve)(schema.jsonSchema).then((jsonSchema2) => ({
       type: "json",
       schema: jsonSchema2,
@@ -3059,6 +3065,9 @@ var object = ({
           };
         }
       }
+    },
+    createElementStreamTransform() {
+      return void 0;
     }
   };
 };
@@ -3069,6 +3078,7 @@ var array = ({
 }) => {
   const elementSchema = (0, import_provider_utils13.asSchema)(inputElementSchema);
   return {
+    name: "array",
     // JSON schema that describes an array of elements:
     responseFormat: (0, import_provider_utils13.resolve)(elementSchema.jsonSchema).then((jsonSchema2) => {
       const { $schema, ...itemSchema } = jsonSchema2;
@@ -3158,6 +3168,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]);
+            }
+          }
+        }
+      });
     }
   };
 };
@@ -3167,6 +3189,7 @@ var choice = ({
   description
 }) => {
   return {
+    name: "choice",
     // JSON schema that describes an enumeration:
     responseFormat: Promise.resolve({
       type: "json",
@@ -3233,6 +3256,9 @@ var choice = ({
           }
         }
       }
+    },
+    createElementStreamTransform() {
+      return void 0;
     }
   };
 };
@@ -3241,6 +3267,7 @@ var json = ({
   description
 } = {}) => {
   return {
+    name: "json",
     responseFormat: Promise.resolve({
       type: "json",
       ...name16 != null && { name: name16 },
@@ -3272,6 +3299,9 @@ var json = ({
           return result.value === void 0 ? void 0 : { partial: result.value };
         }
       }
+    },
+    createElementStreamTransform() {
+      return void 0;
     }
   };
 };
@@ -6936,6 +6966,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 import_provider24.UnsupportedFunctionalityError({
+        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;
@@ -10394,6 +10434,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) {
@@ -12181,6 +12345,7 @@ var TextStreamChatTransport = class extends HttpChatTransport {
   experimental_generateImage,
   experimental_generateSpeech,
   experimental_transcribe,
+  extractJsonMiddleware,
   extractReasoningMiddleware,
   gateway,
   generateId,