ai 3.1.8 → 3.1.10

package/dist/index.mjs

@@ -94,6 +94,53 @@ var EmbedResult = class {
   }
 };
 
+// core/util/split-array.ts
+function splitArray(array, chunkSize) {
+  if (chunkSize <= 0) {
+    throw new Error("chunkSize must be greater than 0");
+  }
+  const result = [];
+  for (let i = 0; i < array.length; i += chunkSize) {
+    result.push(array.slice(i, i + chunkSize));
+  }
+  return result;
+}
+
+// core/embed/embed-many.ts
+async function embedMany({
+  model,
+  values,
+  maxRetries,
+  abortSignal
+}) {
+  const retry = retryWithExponentialBackoff({ maxRetries });
+  const maxEmbeddingsPerCall = model.maxEmbeddingsPerCall;
+  if (maxEmbeddingsPerCall == null) {
+    const modelResponse = await retry(
+      () => model.doEmbed({ values, abortSignal })
+    );
+    return new EmbedManyResult({
+      values,
+      embeddings: modelResponse.embeddings
+    });
+  }
+  const valueChunks = splitArray(values, maxEmbeddingsPerCall);
+  const embeddings = [];
+  for (const chunk of valueChunks) {
+    const modelResponse = await retry(
+      () => model.doEmbed({ values: chunk, abortSignal })
+    );
+    embeddings.push(...modelResponse.embeddings);
+  }
+  return new EmbedManyResult({ values, embeddings });
+}
+var EmbedManyResult = class {
+  constructor(options) {
+    this.values = options.values;
+    this.embeddings = options.embeddings;
+  }
+};
+
 // core/generate-object/generate-object.ts
 import { NoObjectGeneratedError } from "@ai-sdk/provider";
 import { safeParseJSON } from "@ai-sdk/provider-utils";
@@ -1593,7 +1640,7 @@ var StreamTextResult = class {
       "Content-Type": "text/plain; charset=utf-8",
       ...init == null ? void 0 : init.headers
     });
-    const reader = this.textStream.pipeThrough(createCallbacksTransformer(void 0)).pipeThrough(createStreamDataTransformer()).getReader();
+    const reader = this.toAIStream().getReader();
     const read = async () => {
       try {
         while (true) {
@@ -1624,15 +1671,14 @@ var StreamTextResult = class {
       "Content-Type": "text/plain; charset=utf-8",
       ...init == null ? void 0 : init.headers
     });
-    const reader = this.textStream.getReader();
+    const reader = this.textStream.pipeThrough(new TextEncoderStream()).getReader();
     const read = async () => {
-      const encoder = new TextEncoder();
       try {
         while (true) {
           const { done, value } = await reader.read();
           if (done)
             break;
-          response.write(encoder.encode(value));
+          response.write(value);
         }
       } catch (error) {
         throw error;
@@ -1662,23 +1708,13 @@ var StreamTextResult = class {
      */
   toTextStreamResponse(init) {
     var _a;
-    const encoder = new TextEncoder();
-    return new Response(
-      this.textStream.pipeThrough(
-        new TransformStream({
-          transform(chunk, controller) {
-            controller.enqueue(encoder.encode(chunk));
-          }
-        })
-      ),
-      {
-        status: (_a = init == null ? void 0 : init.status) != null ? _a : 200,
-        headers: {
-          "Content-Type": "text/plain; charset=utf-8",
-          ...init == null ? void 0 : init.headers
-        }
+    return new Response(this.textStream.pipeThrough(new TextEncoderStream()), {
+      status: (_a = init == null ? void 0 : init.status) != null ? _a : 200,
+      headers: {
+        "Content-Type": "text/plain; charset=utf-8",
+        ...init == null ? void 0 : init.headers
       }
-    );
+    });
   }
 };
 var experimental_streamText = streamText;
@@ -3248,6 +3284,7 @@ export {
   AnthropicStream,
   AssistantResponse,
   CohereStream,
+  EmbedManyResult,
   EmbedResult,
   EmptyResponseBodyError,
   GenerateObjectResult,
@@ -3286,6 +3323,7 @@ export {
   createEventStreamTransformer,
   createStreamDataTransformer,
   embed,
+  embedMany,
   experimental_AssistantResponse,
   experimental_StreamData,
   experimental_StreamingReactResponse,