ai 4.0.0-canary.5 → 4.0.0-canary.7

package/dist/index.js

@@ -50,9 +50,6 @@ __export(streams_exports, {
   createStreamDataTransformer: () => createStreamDataTransformer,
   embed: () => embed,
   embedMany: () => embedMany,
-  experimental_AssistantResponse: () => experimental_AssistantResponse,
-  experimental_StreamData: () => experimental_StreamData,
-  experimental_createModelRegistry: () => experimental_createModelRegistry,
   experimental_createProviderRegistry: () => experimental_createProviderRegistry,
   experimental_customProvider: () => experimental_customProvider,
   experimental_wrapLanguageModel: () => experimental_wrapLanguageModel,
@@ -1546,28 +1543,30 @@ function standardizePrompt({
 }
 
 // core/types/usage.ts
-function calculateLanguageModelUsage(usage) {
+function calculateLanguageModelUsage({
+  promptTokens,
+  completionTokens
+}) {
   return {
-    promptTokens: usage.promptTokens,
-    completionTokens: usage.completionTokens,
-    totalTokens: usage.promptTokens + usage.completionTokens
+    promptTokens,
+    completionTokens,
+    totalTokens: promptTokens + completionTokens
   };
 }
 
 // core/util/prepare-response-headers.ts
-function prepareResponseHeaders(init, {
+function prepareResponseHeaders(headers, {
   contentType,
   dataStreamVersion
 }) {
-  var _a11;
-  const headers = new Headers((_a11 = init == null ? void 0 : init.headers) != null ? _a11 : {});
-  if (!headers.has("Content-Type")) {
-    headers.set("Content-Type", contentType);
+  const responseHeaders = new Headers(headers != null ? headers : {});
+  if (!responseHeaders.has("Content-Type")) {
+    responseHeaders.set("Content-Type", contentType);
   }
   if (dataStreamVersion !== void 0) {
-    headers.set("X-Vercel-AI-Data-Stream", dataStreamVersion);
+    responseHeaders.set("X-Vercel-AI-Data-Stream", dataStreamVersion);
   }
-  return headers;
+  return responseHeaders;
 }
 
 // core/generate-object/inject-json-instruction.ts
@@ -2328,7 +2327,7 @@ var DefaultGenerateObjectResult = class {
     var _a11;
     return new Response(JSON.stringify(this.object), {
       status: (_a11 = init == null ? void 0 : init.status) != null ? _a11 : 200,
-      headers: prepareResponseHeaders(init, {
+      headers: prepareResponseHeaders(init == null ? void 0 : init.headers, {
         contentType: "application/json; charset=utf-8"
       })
     });
@@ -2399,23 +2398,23 @@ function now() {
 }
 
 // core/util/prepare-outgoing-http-headers.ts
-function prepareOutgoingHttpHeaders(init, {
+function prepareOutgoingHttpHeaders(headers, {
   contentType,
   dataStreamVersion
 }) {
-  const headers = {};
-  if ((init == null ? void 0 : init.headers) != null) {
-    for (const [key, value] of Object.entries(init.headers)) {
-      headers[key] = value;
+  const outgoingHeaders = {};
+  if (headers != null) {
+    for (const [key, value] of Object.entries(headers)) {
+      outgoingHeaders[key] = value;
     }
   }
-  if (headers["Content-Type"] == null) {
-    headers["Content-Type"] = contentType;
+  if (outgoingHeaders["Content-Type"] == null) {
+    outgoingHeaders["Content-Type"] = contentType;
   }
   if (dataStreamVersion !== void 0) {
-    headers["X-Vercel-AI-Data-Stream"] = dataStreamVersion;
+    outgoingHeaders["X-Vercel-AI-Data-Stream"] = dataStreamVersion;
   }
-  return headers;
+  return outgoingHeaders;
 }
 
 // core/util/write-to-server-response.ts
@@ -2930,7 +2929,7 @@ var DefaultStreamObjectResult = class {
       response,
       status: init == null ? void 0 : init.status,
       statusText: init == null ? void 0 : init.statusText,
-      headers: prepareOutgoingHttpHeaders(init, {
+      headers: prepareOutgoingHttpHeaders(init == null ? void 0 : init.headers, {
         contentType: "text/plain; charset=utf-8"
       }),
       stream: this.textStream.pipeThrough(new TextEncoderStream())
@@ -2940,7 +2939,7 @@ var DefaultStreamObjectResult = class {
     var _a11;
     return new Response(this.textStream.pipeThrough(new TextEncoderStream()), {
       status: (_a11 = init == null ? void 0 : init.status) != null ? _a11 : 200,
-      headers: prepareResponseHeaders(init, {
+      headers: prepareResponseHeaders(init == null ? void 0 : init.headers, {
         contentType: "text/plain; charset=utf-8"
       })
     });
@@ -3152,8 +3151,7 @@ async function generateText({
   abortSignal,
   headers,
   maxSteps = 1,
-  experimental_continuationSteps,
-  experimental_continueSteps: continueSteps = experimental_continuationSteps != null ? experimental_continuationSteps : false,
+  experimental_continueSteps: continueSteps = false,
   experimental_telemetry: telemetry,
   experimental_providerMetadata: providerMetadata,
   experimental_activeTools: activeTools,
@@ -3425,7 +3423,6 @@ async function generateText({
           messages: responseMessages
         },
         logprobs: currentModelResponse.logprobs,
-        responseMessages,
         steps,
         providerMetadata: currentModelResponse.providerMetadata
       });
@@ -3502,7 +3499,6 @@ var DefaultGenerateTextResult = class {
     this.warnings = options.warnings;
     this.request = options.request;
     this.response = options.response;
-    this.responseMessages = options.responseMessages;
     this.steps = options.steps;
     this.experimental_providerMetadata = options.providerMetadata;
     this.logprobs = options.logprobs;
@@ -4090,11 +4086,6 @@ var DefaultStreamTextResult = class {
     this.request = requestPromise;
     const { resolve: resolveResponse, promise: responsePromise } = createResolvablePromise();
     this.response = responsePromise;
-    const {
-      resolve: resolveResponseMessages,
-      promise: responseMessagesPromise
-    } = createResolvablePromise();
-    this.responseMessages = responseMessagesPromise;
     const {
       stream: stitchableStream,
       addStream,
@@ -4420,7 +4411,6 @@ var DefaultStreamTextResult = class {
                   messages: responseMessages
                 });
                 resolveSteps(stepResults);
-                resolveResponseMessages(responseMessages);
                 await (onFinish == null ? void 0 : onFinish({
                   finishReason: stepFinishReason,
                   logprobs: stepLogProbs,
@@ -4595,20 +4585,19 @@ var DefaultStreamTextResult = class {
     });
     return this.fullStream.pipeThrough(callbackTransformer).pipeThrough(streamPartsTransformer).pipeThrough(new TextEncoderStream());
   }
-  pipeDataStreamToResponse(response, options) {
-    const init = options == null ? void 0 : "init" in options ? options.init : {
-      headers: "headers" in options ? options.headers : void 0,
-      status: "status" in options ? options.status : void 0,
-      statusText: "statusText" in options ? options.statusText : void 0
-    };
-    const data = options == null ? void 0 : "data" in options ? options.data : void 0;
-    const getErrorMessage3 = options == null ? void 0 : "getErrorMessage" in options ? options.getErrorMessage : void 0;
-    const sendUsage = options == null ? void 0 : "sendUsage" in options ? options.sendUsage : void 0;
+  pipeDataStreamToResponse(response, {
+    status,
+    statusText,
+    headers,
+    data,
+    getErrorMessage: getErrorMessage3,
+    sendUsage
+  } = {}) {
     writeToServerResponse({
       response,
-      status: init == null ? void 0 : init.status,
-      statusText: init == null ? void 0 : init.statusText,
-      headers: prepareOutgoingHttpHeaders(init, {
+      status,
+      statusText,
+      headers: prepareOutgoingHttpHeaders(headers, {
         contentType: "text/plain; charset=utf-8",
         dataStreamVersion: "v1"
       }),
@@ -4620,7 +4609,7 @@ var DefaultStreamTextResult = class {
       response,
       status: init == null ? void 0 : init.status,
       statusText: init == null ? void 0 : init.statusText,
-      headers: prepareOutgoingHttpHeaders(init, {
+      headers: prepareOutgoingHttpHeaders(init == null ? void 0 : init.headers, {
         contentType: "text/plain; charset=utf-8"
       }),
       stream: this.textStream.pipeThrough(new TextEncoderStream())
@@ -4633,22 +4622,20 @@ var DefaultStreamTextResult = class {
     });
     return (options == null ? void 0 : options.data) ? mergeStreams(options == null ? void 0 : options.data.stream, stream) : stream;
   }
-  toDataStreamResponse(options) {
-    var _a11;
-    const init = options == null ? void 0 : "init" in options ? options.init : {
-      headers: "headers" in options ? options.headers : void 0,
-      status: "status" in options ? options.status : void 0,
-      statusText: "statusText" in options ? options.statusText : void 0
-    };
-    const data = options == null ? void 0 : "data" in options ? options.data : void 0;
-    const getErrorMessage3 = options == null ? void 0 : "getErrorMessage" in options ? options.getErrorMessage : void 0;
-    const sendUsage = options == null ? void 0 : "sendUsage" in options ? options.sendUsage : void 0;
+  toDataStreamResponse({
+    headers,
+    status,
+    statusText,
+    data,
+    getErrorMessage: getErrorMessage3,
+    sendUsage
+  } = {}) {
     return new Response(
       this.toDataStream({ data, getErrorMessage: getErrorMessage3, sendUsage }),
       {
-        status: (_a11 = init == null ? void 0 : init.status) != null ? _a11 : 200,
-        statusText: init == null ? void 0 : init.statusText,
-        headers: prepareResponseHeaders(init, {
+        status,
+        statusText,
+        headers: prepareResponseHeaders(headers, {
           contentType: "text/plain; charset=utf-8",
           dataStreamVersion: "v1"
         })
@@ -4659,7 +4646,7 @@ var DefaultStreamTextResult = class {
     var _a11;
     return new Response(this.textStream.pipeThrough(new TextEncoderStream()), {
       status: (_a11 = init == null ? void 0 : init.status) != null ? _a11 : 200,
-      headers: prepareResponseHeaders(init, {
+      headers: prepareResponseHeaders(init == null ? void 0 : init.headers, {
         contentType: "text/plain; charset=utf-8"
       })
     });
@@ -4763,15 +4750,11 @@ function experimental_createProviderRegistry(providers) {
   }
   return registry;
 }
-var experimental_createModelRegistry = experimental_createProviderRegistry;
 var DefaultProviderRegistry = class {
   constructor() {
     this.providers = {};
   }
-  registerProvider({
-    id,
-    provider
-  }) {
+  registerProvider({ id, provider }) {
     this.providers[id] = provider;
   }
   getProvider(id) {
@@ -4807,10 +4790,10 @@ var DefaultProviderRegistry = class {
     return model;
   }
   textEmbeddingModel(id) {
-    var _a11, _b, _c;
+    var _a11;
     const [providerId, modelId] = this.splitId(id, "textEmbeddingModel");
     const provider = this.getProvider(providerId);
-    const model = (_c = (_a11 = provider.textEmbeddingModel) == null ? void 0 : _a11.call(provider, modelId)) != null ? _c : "textEmbedding" in provider ? (_b = provider.textEmbedding) == null ? void 0 : _b.call(provider, modelId) : void 0;
+    const model = (_a11 = provider.textEmbeddingModel) == null ? void 0 : _a11.call(provider, modelId);
     if (model == null) {
       throw new import_provider16.NoSuchModelError({
         modelId: id,
@@ -4920,8 +4903,6 @@ function AssistantResponse({ threadId, messageId }, process2) {
       );
       try {
         await process2({
-          threadId,
-          messageId,
           sendMessage,
           sendDataMessage,
           forwardStream
@@ -4944,12 +4925,10 @@ function AssistantResponse({ threadId, messageId }, process2) {
     }
   });
 }
-var experimental_AssistantResponse = AssistantResponse;
 
 // streams/langchain-adapter.ts
 var langchain_adapter_exports = {};
 __export(langchain_adapter_exports, {
-  toAIStream: () => toAIStream,
   toDataStream: () => toDataStream,
   toDataStreamResponse: () => toDataStreamResponse
 });
@@ -5058,13 +5037,8 @@ function createStreamDataTransformer() {
     }
   });
 }
-var experimental_StreamData = class extends StreamData {
-};
 
 // streams/langchain-adapter.ts
-function toAIStream(stream, callbacks) {
-  return toDataStream(stream, callbacks);
-}
 function toDataStream(stream, callbacks) {
   return stream.pipeThrough(
     new TransformStream({
@@ -5097,7 +5071,7 @@ function toDataStreamResponse(stream, options) {
   return new Response(responseStream, {
     status: (_a11 = init == null ? void 0 : init.status) != null ? _a11 : 200,
     statusText: init == null ? void 0 : init.statusText,
-    headers: prepareResponseHeaders(init, {
+    headers: prepareResponseHeaders(init == null ? void 0 : init.headers, {
       contentType: "text/plain; charset=utf-8",
       dataStreamVersion: "v1"
     })
@@ -5141,7 +5115,7 @@ function toDataStreamResponse2(stream, options = {}) {
   return new Response(responseStream, {
     status: (_a11 = init == null ? void 0 : init.status) != null ? _a11 : 200,
     statusText: init == null ? void 0 : init.statusText,
-    headers: prepareResponseHeaders(init, {
+    headers: prepareResponseHeaders(init == null ? void 0 : init.headers, {
       contentType: "text/plain; charset=utf-8",
       dataStreamVersion: "v1"
     })
@@ -5190,9 +5164,6 @@ function trimStartOfStream() {
   createStreamDataTransformer,
   embed,
   embedMany,
-  experimental_AssistantResponse,
-  experimental_StreamData,
-  experimental_createModelRegistry,
   experimental_createProviderRegistry,
   experimental_customProvider,
   experimental_wrapLanguageModel,