ai 4.0.0-canary.2 → 4.0.0-canary.4

package/dist/index.js

@@ -21,7 +21,6 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 var streams_exports = {};
 __export(streams_exports, {
   AISDKError: () => import_provider13.AISDKError,
-  AIStream: () => AIStream,
   APICallError: () => import_provider13.APICallError,
   AssistantResponse: () => AssistantResponse,
   DownloadError: () => DownloadError,
@@ -43,14 +42,12 @@ __export(streams_exports, {
   NoSuchProviderError: () => NoSuchProviderError,
   NoSuchToolError: () => NoSuchToolError,
   RetryError: () => RetryError,
-  StreamData: () => StreamData2,
+  StreamData: () => StreamData,
   StreamingTextResponse: () => StreamingTextResponse,
   TypeValidationError: () => import_provider13.TypeValidationError,
   UnsupportedFunctionalityError: () => import_provider13.UnsupportedFunctionalityError,
   convertToCoreMessages: () => convertToCoreMessages,
   cosineSimilarity: () => cosineSimilarity,
-  createCallbacksTransformer: () => createCallbacksTransformer,
-  createEventStreamTransformer: () => createEventStreamTransformer,
   createStreamDataTransformer: () => createStreamDataTransformer,
   embed: () => embed,
   embedMany: () => embedMany,
@@ -59,32 +56,26 @@ __export(streams_exports, {
   experimental_createModelRegistry: () => experimental_createModelRegistry,
   experimental_createProviderRegistry: () => experimental_createProviderRegistry,
   experimental_customProvider: () => experimental_customProvider,
-  experimental_generateObject: () => experimental_generateObject,
-  experimental_generateText: () => experimental_generateText,
-  experimental_streamObject: () => experimental_streamObject,
-  experimental_streamText: () => experimental_streamText,
   experimental_wrapLanguageModel: () => experimental_wrapLanguageModel,
-  formatStreamPart: () => import_ui_utils9.formatStreamPart,
-  generateId: () => generateId2,
+  formatStreamPart: () => import_ui_utils10.formatStreamPart,
+  generateId: () => import_provider_utils11.generateId,
   generateObject: () => generateObject,
   generateText: () => generateText,
-  jsonSchema: () => import_ui_utils6.jsonSchema,
-  parseStreamPart: () => import_ui_utils9.parseStreamPart,
-  processDataProtocolResponse: () => import_ui_utils9.processDataProtocolResponse,
-  readDataStream: () => import_ui_utils9.readDataStream,
-  readableFromAsyncIterable: () => readableFromAsyncIterable,
+  jsonSchema: () => import_ui_utils7.jsonSchema,
+  parseStreamPart: () => import_ui_utils10.parseStreamPart,
+  processDataProtocolResponse: () => import_ui_utils10.processDataProtocolResponse,
+  readDataStream: () => import_ui_utils10.readDataStream,
   streamObject: () => streamObject,
   streamText: () => streamText,
   streamToResponse: () => streamToResponse,
-  tool: () => tool,
-  trimStartOfStreamHelper: () => trimStartOfStreamHelper
+  tool: () => tool
 });
 module.exports = __toCommonJS(streams_exports);
-var import_ui_utils9 = require("@ai-sdk/ui-utils");
-var import_provider_utils10 = require("@ai-sdk/provider-utils");
+var import_ui_utils10 = require("@ai-sdk/ui-utils");
+var import_provider_utils11 = require("@ai-sdk/provider-utils");
 
 // core/index.ts
-var import_ui_utils6 = require("@ai-sdk/ui-utils");
+var import_ui_utils7 = require("@ai-sdk/ui-utils");
 
 // util/retry-with-exponential-backoff.ts
 var import_provider2 = require("@ai-sdk/provider");
@@ -1469,9 +1460,7 @@ function convertToCoreMessages(messages, options) {
         });
         break;
       }
-      case "function":
-      case "data":
-      case "tool": {
+      case "data": {
         break;
       }
       default: {
@@ -2361,7 +2350,6 @@ var DefaultGenerateObjectResult = class {
     });
   }
 };
-var experimental_generateObject = generateObject;
 
 // core/generate-object/stream-object.ts
 var import_provider_utils6 = require("@ai-sdk/provider-utils");
@@ -2981,7 +2969,6 @@ var DefaultStreamObjectResult = class {
     });
   }
 };
-var experimental_streamObject = streamObject;
 
 // core/generate-text/generate-text.ts
 var import_provider_utils8 = require("@ai-sdk/provider-utils");
@@ -3012,12 +2999,6 @@ var InvalidToolArgumentsError = class extends import_provider11.AISDKError {
   static isInstance(error) {
     return import_provider11.AISDKError.hasMarker(error, marker8);
   }
-  /**
-   * @deprecated use `isInstance` instead
-   */
-  static isInvalidToolArgumentsError(error) {
-    return error instanceof Error && error.name === name8 && typeof error.toolName === "string" && typeof error.toolArgs === "string";
-  }
 };
 _a8 = symbol8;
 
@@ -3193,9 +3174,7 @@ async function generateText({
   maxRetries,
   abortSignal,
   headers,
-  maxAutomaticRoundtrips = 0,
-  maxToolRoundtrips = maxAutomaticRoundtrips,
-  maxSteps = maxToolRoundtrips != null ? maxToolRoundtrips + 1 : 1,
+  maxSteps = 1,
   experimental_continuationSteps,
   experimental_continueSteps: continueSteps = experimental_continuationSteps != null ? experimental_continuationSteps : false,
   experimental_telemetry: telemetry,
@@ -3563,7 +3542,6 @@ var DefaultGenerateTextResult = class {
     this.request = options.request;
     this.response = options.response;
     this.responseMessages = options.responseMessages;
-    this.roundtrips = options.steps;
     this.steps = options.steps;
     this.experimental_providerMetadata = options.providerMetadata;
     this.rawResponse = {
@@ -3572,10 +3550,10 @@ var DefaultGenerateTextResult = class {
     this.logprobs = options.logprobs;
   }
 };
-var experimental_generateText = generateText;
 
 // core/generate-text/stream-text.ts
 var import_provider_utils9 = require("@ai-sdk/provider-utils");
+var import_ui_utils6 = require("@ai-sdk/ui-utils");
 
 // core/util/create-stitchable-stream.ts
 function createStitchableStream() {
@@ -3932,8 +3910,7 @@ async function streamText({
   maxRetries,
   abortSignal,
   headers,
-  maxToolRoundtrips = 0,
-  maxSteps = maxToolRoundtrips != null ? maxToolRoundtrips + 1 : 1,
+  maxSteps = 1,
   experimental_continueSteps: continueSteps = false,
   experimental_telemetry: telemetry,
   experimental_providerMetadata: providerMetadata,
@@ -4579,37 +4556,18 @@ var DefaultStreamTextResult = class {
       }
     });
   }
-  toAIStream(callbacks = {}) {
-    return this.toDataStreamInternal({ callbacks });
-  }
   toDataStreamInternal({
-    callbacks = {},
     getErrorMessage: getErrorMessage3 = () => "",
     // mask error messages for safety by default
     sendUsage = true
   } = {}) {
     let aggregatedResponse = "";
     const callbackTransformer = new TransformStream({
-      async start() {
-        if (callbacks.onStart)
-          await callbacks.onStart();
-      },
       async transform(chunk, controller) {
         controller.enqueue(chunk);
         if (chunk.type === "text-delta") {
-          const textDelta = chunk.textDelta;
-          aggregatedResponse += textDelta;
-          if (callbacks.onToken)
-            await callbacks.onToken(textDelta);
-          if (callbacks.onText)
-            await callbacks.onText(textDelta);
+          aggregatedResponse += chunk.textDelta;
         }
-      },
-      async flush() {
-        if (callbacks.onCompletion)
-          await callbacks.onCompletion(aggregatedResponse);
-        if (callbacks.onFinal)
-          await callbacks.onFinal(aggregatedResponse);
       }
     });
     const streamPartsTransformer = new TransformStream({
@@ -4617,12 +4575,12 @@ var DefaultStreamTextResult = class {
         const chunkType = chunk.type;
         switch (chunkType) {
           case "text-delta": {
-            controller.enqueue((0, import_ui_utils9.formatStreamPart)("text", chunk.textDelta));
+            controller.enqueue((0, import_ui_utils6.formatStreamPart)("text", chunk.textDelta));
             break;
           }
           case "tool-call-streaming-start": {
             controller.enqueue(
-              (0, import_ui_utils9.formatStreamPart)("tool_call_streaming_start", {
+              (0, import_ui_utils6.formatStreamPart)("tool_call_streaming_start", {
                 toolCallId: chunk.toolCallId,
                 toolName: chunk.toolName
               })
@@ -4631,7 +4589,7 @@ var DefaultStreamTextResult = class {
           }
           case "tool-call-delta": {
             controller.enqueue(
-              (0, import_ui_utils9.formatStreamPart)("tool_call_delta", {
+              (0, import_ui_utils6.formatStreamPart)("tool_call_delta", {
                 toolCallId: chunk.toolCallId,
                 argsTextDelta: chunk.argsTextDelta
               })
@@ -4640,7 +4598,7 @@ var DefaultStreamTextResult = class {
           }
           case "tool-call": {
             controller.enqueue(
-              (0, import_ui_utils9.formatStreamPart)("tool_call", {
+              (0, import_ui_utils6.formatStreamPart)("tool_call", {
                 toolCallId: chunk.toolCallId,
                 toolName: chunk.toolName,
                 args: chunk.args
@@ -4650,7 +4608,7 @@ var DefaultStreamTextResult = class {
           }
           case "tool-result": {
             controller.enqueue(
-              (0, import_ui_utils9.formatStreamPart)("tool_result", {
+              (0, import_ui_utils6.formatStreamPart)("tool_result", {
                 toolCallId: chunk.toolCallId,
                 result: chunk.result
               })
@@ -4659,13 +4617,13 @@ var DefaultStreamTextResult = class {
           }
           case "error": {
             controller.enqueue(
-              (0, import_ui_utils9.formatStreamPart)("error", getErrorMessage3(chunk.error))
+              (0, import_ui_utils6.formatStreamPart)("error", getErrorMessage3(chunk.error))
             );
             break;
           }
           case "step-finish": {
             controller.enqueue(
-              (0, import_ui_utils9.formatStreamPart)("finish_step", {
+              (0, import_ui_utils6.formatStreamPart)("finish_step", {
                 finishReason: chunk.finishReason,
                 usage: sendUsage ? {
                   promptTokens: chunk.usage.promptTokens,
@@ -4678,7 +4636,7 @@ var DefaultStreamTextResult = class {
           }
           case "finish": {
             controller.enqueue(
-              (0, import_ui_utils9.formatStreamPart)("finish_message", {
+              (0, import_ui_utils6.formatStreamPart)("finish_message", {
                 finishReason: chunk.finishReason,
                 usage: sendUsage ? {
                   promptTokens: chunk.usage.promptTokens,
@@ -4697,9 +4655,6 @@ var DefaultStreamTextResult = class {
     });
     return this.fullStream.pipeThrough(callbackTransformer).pipeThrough(streamPartsTransformer).pipeThrough(new TextEncoderStream());
   }
-  pipeAIStreamToResponse(response, init) {
-    return this.pipeDataStreamToResponse(response, init);
-  }
   pipeDataStreamToResponse(response, options) {
     const init = options == null ? void 0 : "init" in options ? options.init : {
       headers: "headers" in options ? options.headers : void 0,
@@ -4731,9 +4686,6 @@ var DefaultStreamTextResult = class {
       stream: this.textStream.pipeThrough(new TextEncoderStream())
     });
   }
-  toAIStreamResponse(options) {
-    return this.toDataStreamResponse(options);
-  }
   toDataStream(options) {
     const stream = this.toDataStreamInternal({
       getErrorMessage: options == null ? void 0 : options.getErrorMessage,
@@ -4773,7 +4725,6 @@ var DefaultStreamTextResult = class {
     });
   }
 };
-var experimental_streamText = streamText;
 
 // core/middleware/wrap-language-model.ts
 var experimental_wrapLanguageModel = ({
@@ -4960,123 +4911,8 @@ function magnitude(vector) {
   return Math.sqrt(dotProduct(vector, vector));
 }
 
-// streams/ai-stream.ts
-var import_eventsource_parser = require("eventsource-parser");
-function createEventStreamTransformer(customParser) {
-  const textDecoder = new TextDecoder();
-  let eventSourceParser;
-  return new TransformStream({
-    async start(controller) {
-      eventSourceParser = (0, import_eventsource_parser.createParser)(
-        (event) => {
-          if ("data" in event && event.type === "event" && event.data === "[DONE]" || // Replicate doesn't send [DONE] but does send a 'done' event
-          // @see https://replicate.com/docs/streaming
-          event.event === "done") {
-            controller.terminate();
-            return;
-          }
-          if ("data" in event) {
-            const parsedMessage = customParser ? customParser(event.data, {
-              event: event.event
-            }) : event.data;
-            if (parsedMessage)
-              controller.enqueue(parsedMessage);
-          }
-        }
-      );
-    },
-    transform(chunk) {
-      eventSourceParser.feed(textDecoder.decode(chunk));
-    }
-  });
-}
-function createCallbacksTransformer(cb) {
-  const textEncoder = new TextEncoder();
-  let aggregatedResponse = "";
-  const callbacks = cb || {};
-  return new TransformStream({
-    async start() {
-      if (callbacks.onStart)
-        await callbacks.onStart();
-    },
-    async transform(message, controller) {
-      const content = typeof message === "string" ? message : message.content;
-      controller.enqueue(textEncoder.encode(content));
-      aggregatedResponse += content;
-      if (callbacks.onToken)
-        await callbacks.onToken(content);
-      if (callbacks.onText && typeof message === "string") {
-        await callbacks.onText(message);
-      }
-    },
-    async flush() {
-      if (callbacks.onCompletion) {
-        await callbacks.onCompletion(aggregatedResponse);
-      }
-    }
-  });
-}
-function trimStartOfStreamHelper() {
-  let isStreamStart = true;
-  return (text) => {
-    if (isStreamStart) {
-      text = text.trimStart();
-      if (text)
-        isStreamStart = false;
-    }
-    return text;
-  };
-}
-function AIStream(response, customParser, callbacks) {
-  if (!response.ok) {
-    if (response.body) {
-      const reader = response.body.getReader();
-      return new ReadableStream({
-        async start(controller) {
-          const { done, value } = await reader.read();
-          if (!done) {
-            const errorText = new TextDecoder().decode(value);
-            controller.error(new Error(`Response error: ${errorText}`));
-          }
-        }
-      });
-    } else {
-      return new ReadableStream({
-        start(controller) {
-          controller.error(new Error("Response error: No response body"));
-        }
-      });
-    }
-  }
-  const responseBodyStream = response.body || createEmptyReadableStream();
-  return responseBodyStream.pipeThrough(createEventStreamTransformer(customParser)).pipeThrough(createCallbacksTransformer(callbacks));
-}
-function createEmptyReadableStream() {
-  return new ReadableStream({
-    start(controller) {
-      controller.close();
-    }
-  });
-}
-function readableFromAsyncIterable(iterable) {
-  let it = iterable[Symbol.asyncIterator]();
-  return new ReadableStream({
-    async pull(controller) {
-      const { done, value } = await it.next();
-      if (done)
-        controller.close();
-      else
-        controller.enqueue(value);
-    },
-    async cancel(reason) {
-      var _a11;
-      await ((_a11 = it.return) == null ? void 0 : _a11.call(it, reason));
-    }
-  });
-}
-
 // streams/assistant-response.ts
-var import_ui_utils7 = require("@ai-sdk/ui-utils");
+var import_ui_utils8 = require("@ai-sdk/ui-utils");
 function AssistantResponse({ threadId, messageId }, process2) {
   const stream = new ReadableStream({
     async start(controller) {
@@ -5084,17 +4920,17 @@ function AssistantResponse({ threadId, messageId }, process2) {
       const textEncoder = new TextEncoder();
       const sendMessage = (message) => {
         controller.enqueue(
-          textEncoder.encode((0, import_ui_utils7.formatStreamPart)("assistant_message", message))
+          textEncoder.encode((0, import_ui_utils8.formatStreamPart)("assistant_message", message))
         );
       };
       const sendDataMessage = (message) => {
         controller.enqueue(
-          textEncoder.encode((0, import_ui_utils7.formatStreamPart)("data_message", message))
+          textEncoder.encode((0, import_ui_utils8.formatStreamPart)("data_message", message))
         );
       };
       const sendError = (errorMessage) => {
         controller.enqueue(
-          textEncoder.encode((0, import_ui_utils7.formatStreamPart)("error", errorMessage))
+          textEncoder.encode((0, import_ui_utils8.formatStreamPart)("error", errorMessage))
         );
       };
       const forwardStream = async (stream2) => {
@@ -5105,7 +4941,7 @@ function AssistantResponse({ threadId, messageId }, process2) {
             case "thread.message.created": {
               controller.enqueue(
                 textEncoder.encode(
-                  (0, import_ui_utils7.formatStreamPart)("assistant_message", {
+                  (0, import_ui_utils8.formatStreamPart)("assistant_message", {
                     id: value.data.id,
                     role: "assistant",
                     content: [{ type: "text", text: { value: "" } }]
@@ -5119,7 +4955,7 @@ function AssistantResponse({ threadId, messageId }, process2) {
               if ((content == null ? void 0 : content.type) === "text" && ((_b = content.text) == null ? void 0 : _b.value) != null) {
                 controller.enqueue(
                   textEncoder.encode(
-                    (0, import_ui_utils7.formatStreamPart)("text", content.text.value)
+                    (0, import_ui_utils8.formatStreamPart)("text", content.text.value)
                   )
                 );
               }
@@ -5136,7 +4972,7 @@ function AssistantResponse({ threadId, messageId }, process2) {
       };
       controller.enqueue(
         textEncoder.encode(
-          (0, import_ui_utils7.formatStreamPart)("assistant_control_data", {
+          (0, import_ui_utils8.formatStreamPart)("assistant_control_data", {
             threadId,
             messageId
           })
@@ -5178,14 +5014,40 @@ __export(langchain_adapter_exports, {
   toDataStreamResponse: () => toDataStreamResponse
 });
 
+// streams/stream-callbacks.ts
+function createCallbacksTransformer(callbacks = {}) {
+  const textEncoder = new TextEncoder();
+  let aggregatedResponse = "";
+  return new TransformStream({
+    async start() {
+      if (callbacks.onStart)
+        await callbacks.onStart();
+    },
+    async transform(message, controller) {
+      controller.enqueue(textEncoder.encode(message));
+      aggregatedResponse += message;
+      if (callbacks.onToken)
+        await callbacks.onToken(message);
+      if (callbacks.onText && typeof message === "string") {
+        await callbacks.onText(message);
+      }
+    },
+    async flush() {
+      if (callbacks.onCompletion) {
+        await callbacks.onCompletion(aggregatedResponse);
+      }
+    }
+  });
+}
+
 // streams/stream-data.ts
-var import_ui_utils8 = require("@ai-sdk/ui-utils");
+var import_ui_utils9 = require("@ai-sdk/ui-utils");
 
 // util/constants.ts
 var HANGING_STREAM_WARNING_TIME_MS = 15 * 1e3;
 
 // streams/stream-data.ts
-var StreamData2 = class {
+var StreamData = class {
   constructor() {
     this.encoder = new TextEncoder();
     this.controller = null;
@@ -5231,7 +5093,7 @@ var StreamData2 = class {
       throw new Error("Stream controller is not initialized.");
     }
     this.controller.enqueue(
-      this.encoder.encode((0, import_ui_utils8.formatStreamPart)("data", [value]))
+      this.encoder.encode((0, import_ui_utils9.formatStreamPart)("data", [value]))
     );
   }
   appendMessageAnnotation(value) {
@@ -5242,7 +5104,7 @@ var StreamData2 = class {
       throw new Error("Stream controller is not initialized.");
     }
     this.controller.enqueue(
-      this.encoder.encode((0, import_ui_utils8.formatStreamPart)("message_annotations", [value]))
+      this.encoder.encode((0, import_ui_utils9.formatStreamPart)("message_annotations", [value]))
     );
   }
 };
@@ -5252,11 +5114,11 @@ function createStreamDataTransformer() {
   return new TransformStream({
     transform: async (chunk, controller) => {
       const message = decoder.decode(chunk);
-      controller.enqueue(encoder.encode((0, import_ui_utils8.formatStreamPart)("text", message)));
+      controller.enqueue(encoder.encode((0, import_ui_utils9.formatStreamPart)("text", message)));
     }
   });
 }
-var experimental_StreamData = class extends StreamData2 {
+var experimental_StreamData = class extends StreamData {
 };
 
 // streams/langchain-adapter.ts
@@ -5320,8 +5182,16 @@ __export(llamaindex_adapter_exports, {
   toDataStream: () => toDataStream2,
   toDataStreamResponse: () => toDataStreamResponse2
 });
+var import_provider_utils10 = require("@ai-sdk/provider-utils");
 function toDataStream2(stream, callbacks) {
-  return toReadableStream(stream).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
+  const trimStart = trimStartOfStream();
+  return (0, import_provider_utils10.convertAsyncIteratorToReadableStream)(stream[Symbol.asyncIterator]()).pipeThrough(
+    new TransformStream({
+      async transform(message, controller) {
+        controller.enqueue(trimStart(message.delta));
+      }
+    })
+  ).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
 }
 function toDataStreamResponse2(stream, options = {}) {
   var _a11;
@@ -5337,23 +5207,16 @@ function toDataStreamResponse2(stream, options = {}) {
     })
   });
 }
-function toReadableStream(res) {
-  const it = res[Symbol.asyncIterator]();
-  const trimStartOfStream = trimStartOfStreamHelper();
-  return new ReadableStream({
-    async pull(controller) {
-      var _a11;
-      const { value, done } = await it.next();
-      if (done) {
-        controller.close();
-        return;
-      }
-      const text = trimStartOfStream((_a11 = value.delta) != null ? _a11 : "");
-      if (text) {
-        controller.enqueue(text);
-      }
+function trimStartOfStream() {
+  let isStreamStart = true;
+  return (text) => {
+    if (isStreamStart) {
+      text = text.trimStart();
+      if (text)
+        isStreamStart = false;
     }
-  });
+    return text;
+  };
 }
 
 // streams/stream-to-response.ts
@@ -5397,13 +5260,9 @@ var StreamingTextResponse = class extends Response {
     });
   }
 };
-
-// streams/index.ts
-var generateId2 = import_provider_utils10.generateId;
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AISDKError,
-  AIStream,
   APICallError,
   AssistantResponse,
   DownloadError,
@@ -5431,8 +5290,6 @@ var generateId2 = import_provider_utils10.generateId;
   UnsupportedFunctionalityError,
   convertToCoreMessages,
   cosineSimilarity,
-  createCallbacksTransformer,
-  createEventStreamTransformer,
   createStreamDataTransformer,
   embed,
   embedMany,
@@ -5441,10 +5298,6 @@ var generateId2 = import_provider_utils10.generateId;
   experimental_createModelRegistry,
   experimental_createProviderRegistry,
   experimental_customProvider,
-  experimental_generateObject,
-  experimental_generateText,
-  experimental_streamObject,
-  experimental_streamText,
   experimental_wrapLanguageModel,
   formatStreamPart,
   generateId,
@@ -5454,11 +5307,9 @@ var generateId2 = import_provider_utils10.generateId;
   parseStreamPart,
   processDataProtocolResponse,
   readDataStream,
-  readableFromAsyncIterable,
   streamObject,
   streamText,
   streamToResponse,
-  tool,
-  trimStartOfStreamHelper
+  tool
 });
 //# sourceMappingURL=index.js.map