ai 4.0.0-canary.6 → 4.0.0-canary.8

package/dist/index.mjs

@@ -1490,11 +1490,14 @@ 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
   };
 }
 
@@ -3115,8 +3118,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,
@@ -3388,7 +3390,6 @@ async function generateText({
           messages: responseMessages
         },
         logprobs: currentModelResponse.logprobs,
-        responseMessages,
         steps,
         providerMetadata: currentModelResponse.providerMetadata
       });
@@ -3465,7 +3466,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;
@@ -4030,7 +4030,7 @@ var DefaultStreamTextResult = class {
     generateId: generateId3,
     tools
   }) {
-    this.warnings = warnings;
+    this.rawWarnings = warnings;
     this.rawResponse = rawResponse;
     const { resolve: resolveUsage, promise: usagePromise } = createResolvablePromise();
     this.usage = usagePromise;
@@ -4053,11 +4053,8 @@ 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 { resolve: resolveWarnings, promise: warningsPromise } = createResolvablePromise();
+    this.warnings = warningsPromise;
     const {
       stream: stitchableStream,
       addStream,
@@ -4211,7 +4208,7 @@ var DefaultStreamTextResult = class {
             },
             // invoke onFinish callback and resolve toolResults promise when the stream is about to close:
             async flush(controller) {
-              var _a11;
+              var _a11, _b;
               const stepToolCallsJson = stepToolCalls.length > 0 ? JSON.stringify(stepToolCalls) : void 0;
               let nextStepType = "done";
               if (currentStep + 1 < maxSteps) {
@@ -4303,7 +4300,7 @@ var DefaultStreamTextResult = class {
                 toolResults: stepToolResults,
                 finishReason: stepFinishReason,
                 usage: stepUsage,
-                warnings: self.warnings,
+                warnings: self.rawWarnings,
                 logprobs: stepLogProbs,
                 request: stepRequest,
                 response: {
@@ -4328,7 +4325,7 @@ var DefaultStreamTextResult = class {
                   doStreamSpan: doStreamSpan3,
                   startTimestampMs: startTimestamp2
                 } = await startStep({ responseMessages });
-                self.warnings = result.warnings;
+                self.rawWarnings = result.warnings;
                 self.rawResponse = result.rawResponse;
                 addStepStream({
                   stream: result.stream,
@@ -4383,7 +4380,7 @@ var DefaultStreamTextResult = class {
                   messages: responseMessages
                 });
                 resolveSteps(stepResults);
-                resolveResponseMessages(responseMessages);
+                resolveWarnings((_b = self.rawWarnings) != null ? _b : []);
                 await (onFinish == null ? void 0 : onFinish({
                   finishReason: stepFinishReason,
                   logprobs: stepLogProbs,
@@ -4723,15 +4720,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) {
@@ -4767,10 +4760,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 NoSuchModelError4({
         modelId: id,
@@ -4882,8 +4875,6 @@ function AssistantResponse({ threadId, messageId }, process2) {
       );
       try {
         await process2({
-          threadId,
-          messageId,
           sendMessage,
           sendDataMessage,
           forwardStream
@@ -4906,12 +4897,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
 });
@@ -5020,13 +5009,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({
@@ -5151,9 +5135,6 @@ export {
   createStreamDataTransformer,
   embed,
   embedMany,
-  experimental_AssistantResponse,
-  experimental_StreamData,
-  experimental_createModelRegistry,
   experimental_createProviderRegistry,
   experimental_customProvider,
   experimental_wrapLanguageModel,