ai 3.0.19 → 3.0.20

package/dist/index.mjs

@@ -1739,9 +1739,7 @@ var StreamTextResult = class {
   @returns an `AIStream` object.
      */
   toAIStream(callbacks) {
-    return readableFromAsyncIterable(this.textStream).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(
-      createStreamDataTransformer(callbacks == null ? void 0 : callbacks.experimental_streamData)
-    );
+    return readableFromAsyncIterable(this.textStream).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
   }
 };
 
@@ -1946,7 +1944,6 @@ function createChunkDecoder(complex) {
   };
 }
 var isStreamStringEqualToType = (type, value) => value.startsWith(`${StreamStringPrefixes[type]}:`) && value.endsWith("\n");
-var COMPLEX_HEADER = "X-Experimental-Stream-Data";
 
 // streams/ai-stream.ts
 import {
@@ -2073,7 +2070,7 @@ function readableFromAsyncIterable(iterable) {
 }
 
 // streams/stream-data.ts
-var experimental_StreamData = class {
+var StreamData = class {
   constructor() {
     this.encoder = new TextEncoder();
     this.controller = null;
@@ -2159,14 +2156,7 @@ var experimental_StreamData = class {
     this.messageAnnotations.push(value);
   }
 };
-function createStreamDataTransformer(experimental_streamData) {
-  if (!experimental_streamData) {
-    return new TransformStream({
-      transform: async (chunk, controller) => {
-        controller.enqueue(chunk);
-      }
-    });
-  }
+function createStreamDataTransformer() {
   const encoder = new TextEncoder();
   const decoder = new TextDecoder();
   return new TransformStream({
@@ -2176,6 +2166,8 @@ function createStreamDataTransformer(experimental_streamData) {
     }
   });
 }
+var experimental_StreamData = class extends StreamData {
+};
 
 // streams/anthropic-stream.ts
 function parseAnthropicStream() {
@@ -2215,16 +2207,16 @@ async function* streamable(stream) {
 }
 function AnthropicStream(res, cb) {
   if (Symbol.asyncIterator in res) {
-    return readableFromAsyncIterable(streamable(res)).pipeThrough(createCallbacksTransformer(cb)).pipeThrough(createStreamDataTransformer(cb == null ? void 0 : cb.experimental_streamData));
+    return readableFromAsyncIterable(streamable(res)).pipeThrough(createCallbacksTransformer(cb)).pipeThrough(createStreamDataTransformer());
   } else {
     return AIStream(res, parseAnthropicStream(), cb).pipeThrough(
-      createStreamDataTransformer(cb == null ? void 0 : cb.experimental_streamData)
+      createStreamDataTransformer()
     );
   }
 }
 
 // streams/assistant-response.ts
-function experimental_AssistantResponse({ threadId, messageId }, process2) {
+function AssistantResponse({ threadId, messageId }, process2) {
   const stream = new ReadableStream({
     async start(controller) {
       var _a;
@@ -2315,6 +2307,7 @@ function experimental_AssistantResponse({ threadId, messageId }, process2) {
     }
   });
 }
+var experimental_AssistantResponse = AssistantResponse;
 
 // streams/aws-bedrock-stream.ts
 async function* asDeltaIterable(response, extractTextDeltaFromChunk) {
@@ -2342,16 +2335,7 @@ function AWSBedrockAnthropicStream(response, callbacks) {
   return AWSBedrockStream(response, callbacks, (chunk) => chunk.completion);
 }
 function AWSBedrockCohereStream(response, callbacks) {
-  return AWSBedrockStream(
-    response,
-    callbacks,
-    // As of 2023-11-17, Bedrock does not support streaming for Cohere,
-    // so we take the full generation:
-    (chunk) => {
-      var _a, _b;
-      return (_b = (_a = chunk.generations) == null ? void 0 : _a[0]) == null ? void 0 : _b.text;
-    }
-  );
+  return AWSBedrockStream(response, callbacks, (chunk) => chunk == null ? void 0 : chunk.text);
 }
 function AWSBedrockLlama2Stream(response, callbacks) {
   return AWSBedrockStream(response, callbacks, (chunk) => chunk.generation);
@@ -2359,9 +2343,7 @@ function AWSBedrockLlama2Stream(response, callbacks) {
 function AWSBedrockStream(response, callbacks, extractTextDeltaFromChunk) {
   return readableFromAsyncIterable(
     asDeltaIterable(response, extractTextDeltaFromChunk)
-  ).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(
-    createStreamDataTransformer(callbacks == null ? void 0 : callbacks.experimental_streamData)
-  );
+  ).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
 }
 
 // streams/cohere-stream.ts
@@ -2416,13 +2398,9 @@ async function* streamable2(stream) {
 }
 function CohereStream(reader, callbacks) {
   if (Symbol.asyncIterator in reader) {
-    return readableFromAsyncIterable(streamable2(reader)).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(
-      createStreamDataTransformer(callbacks == null ? void 0 : callbacks.experimental_streamData)
-    );
+    return readableFromAsyncIterable(streamable2(reader)).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
   } else {
-    return createParser2(reader).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(
-      createStreamDataTransformer(callbacks == null ? void 0 : callbacks.experimental_streamData)
-    );
+    return createParser2(reader).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
   }
 }
 
@@ -2441,7 +2419,7 @@ async function* streamable3(response) {
   }
 }
 function GoogleGenerativeAIStream(response, cb) {
-  return readableFromAsyncIterable(streamable3(response)).pipeThrough(createCallbacksTransformer(cb)).pipeThrough(createStreamDataTransformer(cb == null ? void 0 : cb.experimental_streamData));
+  return readableFromAsyncIterable(streamable3(response)).pipeThrough(createCallbacksTransformer(cb)).pipeThrough(createStreamDataTransformer());
 }
 
 // streams/huggingface-stream.ts
@@ -2469,9 +2447,7 @@ function createParser3(res) {
   });
 }
 function HuggingFaceStream(res, callbacks) {
-  return createParser3(res).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(
-    createStreamDataTransformer(callbacks == null ? void 0 : callbacks.experimental_streamData)
-  );
+  return createParser3(res).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
 }
 
 // streams/inkeep-stream.ts
@@ -2508,7 +2484,7 @@ function InkeepStream(res, callbacks) {
     }
   };
   return AIStream(res, inkeepEventParser, passThroughCallbacks).pipeThrough(
-    createStreamDataTransformer(passThroughCallbacks == null ? void 0 : passThroughCallbacks.experimental_streamData)
+    createStreamDataTransformer()
   );
 }
 
@@ -2533,9 +2509,7 @@ function LangChainStream(callbacks) {
     }
   };
   return {
-    stream: stream.readable.pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(
-      createStreamDataTransformer(callbacks == null ? void 0 : callbacks.experimental_streamData)
-    ),
+    stream: stream.readable.pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer()),
     writer,
     handlers: {
       handleLLMNewToken: async (token) => {
@@ -2586,9 +2560,7 @@ async function* streamable4(stream) {
 }
 function MistralStream(response, callbacks) {
   const stream = readableFromAsyncIterable(streamable4(response));
-  return stream.pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(
-    createStreamDataTransformer(callbacks == null ? void 0 : callbacks.experimental_streamData)
-  );
+  return stream.pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
 }
 
 // streams/openai-stream.ts
@@ -2732,9 +2704,7 @@ function OpenAIStream(res, callbacks) {
     const functionCallTransformer = createFunctionCallTransformer(cb);
     return stream.pipeThrough(functionCallTransformer);
   } else {
-    return stream.pipeThrough(
-      createStreamDataTransformer(cb == null ? void 0 : cb.experimental_streamData)
-    );
+    return stream.pipeThrough(createStreamDataTransformer());
   }
 }
 function createFunctionCallTransformer(callbacks) {
@@ -2744,7 +2714,6 @@ function createFunctionCallTransformer(callbacks) {
   let aggregatedFinalCompletionResponse = "";
   let isFunctionStreamingIn = false;
   let functionCallMessages = callbacks[__internal__OpenAIFnMessagesSymbol] || [];
-  const isComplexMode = callbacks == null ? void 0 : callbacks.experimental_streamData;
   const decode = createChunkDecoder();
   return new TransformStream({
     async transform(chunk, controller) {
@@ -2759,7 +2728,7 @@ function createFunctionCallTransformer(callbacks) {
       }
       if (!isFunctionStreamingIn) {
         controller.enqueue(
-          isComplexMode ? textEncoder.encode(formatStreamPart("text", message)) : chunk
+          textEncoder.encode(formatStreamPart("text", message))
         );
         return;
       } else {
@@ -2870,17 +2839,17 @@ function createFunctionCallTransformer(callbacks) {
           if (!functionResponse) {
             controller.enqueue(
               textEncoder.encode(
-                isComplexMode ? formatStreamPart(
+                formatStreamPart(
                   payload.function_call ? "function_call" : "tool_calls",
                   // parse to prevent double-encoding:
                   JSON.parse(aggregatedResponse)
-                ) : aggregatedResponse
+                )
               )
             );
             return;
           } else if (typeof functionResponse === "string") {
             controller.enqueue(
-              isComplexMode ? textEncoder.encode(formatStreamPart("text", functionResponse)) : textEncoder.encode(functionResponse)
+              textEncoder.encode(formatStreamPart("text", functionResponse))
             );
             aggregatedFinalCompletionResponse = functionResponse;
             return;
@@ -2930,7 +2899,7 @@ async function ReplicateStream(res, cb, options) {
     }
   });
   return AIStream(eventStream, void 0, cb).pipeThrough(
-    createStreamDataTransformer(cb == null ? void 0 : cb.experimental_streamData)
+    createStreamDataTransformer()
   );
 }
 
@@ -3091,74 +3060,41 @@ async function parseComplexResponse({
 // streams/streaming-react-response.ts
 var experimental_StreamingReactResponse = class {
   constructor(res, options) {
-    var _a;
+    var _a, _b;
     let resolveFunc = () => {
     };
     let next = new Promise((resolve) => {
       resolveFunc = resolve;
     });
-    if (options == null ? void 0 : options.data) {
-      const processedStream = res.pipeThrough(
-        options.data.stream
-      );
-      let lastPayload = void 0;
-      parseComplexResponse({
-        reader: processedStream.getReader(),
-        update: (merged, data) => {
-          var _a2, _b, _c;
-          const content2 = (_b = (_a2 = merged[0]) == null ? void 0 : _a2.content) != null ? _b : "";
-          const ui = ((_c = options == null ? void 0 : options.ui) == null ? void 0 : _c.call(options, { content: content2, data })) || content2;
-          const payload = { ui, content: content2 };
-          const resolvePrevious = resolveFunc;
-          const nextRow = new Promise((resolve) => {
-            resolveFunc = resolve;
-          });
-          resolvePrevious({
-            next: nextRow,
-            ...payload
+    const processedStream = (options == null ? void 0 : options.data) != null ? res.pipeThrough((_a = options == null ? void 0 : options.data) == null ? void 0 : _a.stream) : res;
+    let lastPayload = void 0;
+    parseComplexResponse({
+      reader: processedStream.getReader(),
+      update: (merged, data) => {
+        var _a2, _b2, _c;
+        const content = (_b2 = (_a2 = merged[0]) == null ? void 0 : _a2.content) != null ? _b2 : "";
+        const ui = ((_c = options == null ? void 0 : options.ui) == null ? void 0 : _c.call(options, { content, data })) || content;
+        const payload = { ui, content };
+        const resolvePrevious = resolveFunc;
+        const nextRow = new Promise((resolve) => {
+          resolveFunc = resolve;
+        });
+        resolvePrevious({
+          next: nextRow,
+          ...payload
+        });
+        lastPayload = payload;
+      },
+      generateId: (_b = options == null ? void 0 : options.generateId) != null ? _b : generateId,
+      onFinish: () => {
+        if (lastPayload !== void 0) {
+          resolveFunc({
+            next: null,
+            ...lastPayload
           });
-          lastPayload = payload;
-        },
-        generateId: (_a = options.generateId) != null ? _a : generateId,
-        onFinish: () => {
-          if (lastPayload !== void 0) {
-            resolveFunc({
-              next: null,
-              ...lastPayload
-            });
-          }
         }
-      });
-      return next;
-    }
-    let content = "";
-    const decode = createChunkDecoder();
-    const reader = res.getReader();
-    async function readChunk() {
-      var _a2;
-      const { done, value } = await reader.read();
-      if (!done) {
-        content += decode(value);
       }
-      const ui = ((_a2 = options == null ? void 0 : options.ui) == null ? void 0 : _a2.call(options, { content })) || content;
-      const payload = {
-        ui,
-        content
-      };
-      const resolvePrevious = resolveFunc;
-      const nextRow = done ? null : new Promise((resolve) => {
-        resolveFunc = resolve;
-      });
-      resolvePrevious({
-        next: nextRow,
-        ...payload
-      });
-      if (done) {
-        return;
-      }
-      await readChunk();
-    }
-    readChunk();
+    });
     return next;
   }
 };
@@ -3175,7 +3111,6 @@ var StreamingTextResponse = class extends Response {
       status: 200,
       headers: {
         "Content-Type": "text/plain; charset=utf-8",
-        [COMPLEX_HEADER]: data ? "true" : "false",
         ...init == null ? void 0 : init.headers
       }
     });
@@ -3207,7 +3142,7 @@ export {
   AWSBedrockLlama2Stream,
   AWSBedrockStream,
   AnthropicStream,
-  COMPLEX_HEADER,
+  AssistantResponse,
   CohereStream,
   GenerateObjectResult,
   GenerateTextResult,
@@ -3218,6 +3153,7 @@ export {
   MistralStream,
   OpenAIStream,
   ReplicateStream,
+  StreamData,
   StreamObjectResult,
   StreamTextResult,
   StreamingTextResponse,