ai 3.3.10 → 3.3.12

package/dist/index.mjs

@@ -122,11 +122,16 @@ async function _retryWithExponentialBackoff(f, {
 
 // core/telemetry/assemble-operation-name.ts
 function assembleOperationName({
-  operationName,
+  operationId,
   telemetry
 }) {
   return {
-    "operation.name": `${operationName}${(telemetry == null ? void 0 : telemetry.functionId) != null ? ` ${telemetry.functionId}` : ""}`
+    // standardized operation and resource name:
+    "operation.name": `${operationId}${(telemetry == null ? void 0 : telemetry.functionId) != null ? ` ${telemetry.functionId}` : ""}`,
+    "resource.name": telemetry == null ? void 0 : telemetry.functionId,
+    // detailed, AI SDK specific data:
+    "ai.operationId": operationId,
+    "ai.telemetry.functionId": telemetry == null ? void 0 : telemetry.functionId
   };
 }
 
@@ -146,9 +151,6 @@ function getBaseTelemetryAttributes({
       attributes[`ai.settings.${key}`] = value;
       return attributes;
     }, {}),
-    // special telemetry information
-    "resource.name": telemetry == null ? void 0 : telemetry.functionId,
-    "ai.telemetry.functionId": telemetry == null ? void 0 : telemetry.functionId,
     // add metadata as attributes:
     ...Object.entries((_a12 = telemetry == null ? void 0 : telemetry.metadata) != null ? _a12 : {}).reduce(
       (attributes, [key, value]) => {
@@ -328,7 +330,7 @@ async function embed({
     attributes: selectTelemetryAttributes({
       telemetry,
       attributes: {
-        ...assembleOperationName({ operationName: "ai.embed", telemetry }),
+        ...assembleOperationName({ operationId: "ai.embed", telemetry }),
         ...baseTelemetryAttributes,
         "ai.value": { input: () => JSON.stringify(value) }
       }
@@ -345,7 +347,7 @@ async function embed({
               telemetry,
               attributes: {
                 ...assembleOperationName({
-                  operationName: "ai.embed.doEmbed",
+                  operationId: "ai.embed.doEmbed",
                   telemetry
                 }),
                 ...baseTelemetryAttributes,
@@ -441,7 +443,7 @@ async function embedMany({
     attributes: selectTelemetryAttributes({
       telemetry,
       attributes: {
-        ...assembleOperationName({ operationName: "ai.embedMany", telemetry }),
+        ...assembleOperationName({ operationId: "ai.embedMany", telemetry }),
         ...baseTelemetryAttributes,
         // specific settings that only make sense on the outer level:
         "ai.values": {
@@ -461,7 +463,7 @@ async function embedMany({
               telemetry,
               attributes: {
                 ...assembleOperationName({
-                  operationName: "ai.embedMany.doEmbed",
+                  operationId: "ai.embedMany.doEmbed",
                   telemetry
                 }),
                 ...baseTelemetryAttributes,
@@ -520,7 +522,7 @@ async function embedMany({
               telemetry,
               attributes: {
                 ...assembleOperationName({
-                  operationName: "ai.embedMany.doEmbed",
+                  operationId: "ai.embedMany.doEmbed",
                   telemetry
                 }),
                 ...baseTelemetryAttributes,
@@ -728,15 +730,13 @@ var InvalidDataContentError = class extends AISDKError3 {
 _a3 = symbol3;
 
 // core/prompt/data-content.ts
-function convertDataContentToBase64String(content) {
-  if (typeof content === "string") {
-    return content;
-  }
-  if (content instanceof ArrayBuffer) {
-    return convertUint8ArrayToBase64(new Uint8Array(content));
-  }
-  return convertUint8ArrayToBase64(content);
-}
+import { z } from "zod";
+var dataContentSchema = z.union([
+  z.string(),
+  z.instanceof(Uint8Array),
+  z.instanceof(ArrayBuffer),
+  z.instanceof(Buffer)
+]);
 function convertDataContentToUint8Array(content) {
   if (content instanceof Uint8Array) {
     return content;
@@ -1011,45 +1011,6 @@ async function downloadImages(messages, downloadImplementation) {
   );
 }
 
-// core/prompt/get-validated-prompt.ts
-import { InvalidPromptError } from "@ai-sdk/provider";
-function getValidatedPrompt(prompt) {
-  if (prompt.prompt == null && prompt.messages == null) {
-    throw new InvalidPromptError({
-      prompt,
-      message: "prompt or messages must be defined"
-    });
-  }
-  if (prompt.prompt != null && prompt.messages != null) {
-    throw new InvalidPromptError({
-      prompt,
-      message: "prompt and messages cannot be defined at the same time"
-    });
-  }
-  if (prompt.messages != null) {
-    for (const message of prompt.messages) {
-      if (message.role === "system" && typeof message.content !== "string") {
-        throw new InvalidPromptError({
-          prompt,
-          message: "system message content must be a string"
-        });
-      }
-    }
-  }
-  return prompt.prompt != null ? {
-    type: "prompt",
-    prompt: prompt.prompt,
-    messages: void 0,
-    system: prompt.system
-  } : {
-    type: "messages",
-    prompt: void 0,
-    messages: prompt.messages,
-    // only possible case bc of checks above
-    system: prompt.system
-  };
-}
-
 // errors/invalid-argument-error.ts
 import { AISDKError as AISDKError5 } from "@ai-sdk/provider";
 var name5 = "AI_InvalidArgumentError";
@@ -1191,6 +1152,155 @@ function prepareCallSettings({
   };
 }
 
+// core/prompt/validate-prompt.ts
+import { InvalidPromptError } from "@ai-sdk/provider";
+import { safeValidateTypes } from "@ai-sdk/provider-utils";
+import { z as z6 } from "zod";
+
+// core/prompt/message.ts
+import { z as z5 } from "zod";
+
+// core/types/provider-metadata.ts
+import { z as z3 } from "zod";
+
+// core/types/json-value.ts
+import { z as z2 } from "zod";
+var jsonValueSchema = z2.lazy(
+  () => z2.union([
+    z2.null(),
+    z2.string(),
+    z2.number(),
+    z2.boolean(),
+    z2.record(z2.string(), jsonValueSchema),
+    z2.array(jsonValueSchema)
+  ])
+);
+
+// core/types/provider-metadata.ts
+var providerMetadataSchema = z3.record(
+  z3.string(),
+  z3.record(z3.string(), jsonValueSchema)
+);
+
+// core/prompt/content-part.ts
+import { z as z4 } from "zod";
+var textPartSchema = z4.object({
+  type: z4.literal("text"),
+  text: z4.string(),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var imagePartSchema = z4.object({
+  type: z4.literal("image"),
+  image: z4.union([dataContentSchema, z4.instanceof(URL)]),
+  mimeType: z4.string().optional(),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var toolCallPartSchema = z4.object({
+  type: z4.literal("tool-call"),
+  toolCallId: z4.string(),
+  toolName: z4.string(),
+  args: z4.unknown()
+});
+var toolResultPartSchema = z4.object({
+  type: z4.literal("tool-result"),
+  toolCallId: z4.string(),
+  toolName: z4.string(),
+  result: z4.unknown(),
+  isError: z4.boolean().optional(),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+
+// core/prompt/message.ts
+var coreSystemMessageSchema = z5.object({
+  role: z5.literal("system"),
+  content: z5.string(),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var coreUserMessageSchema = z5.object({
+  role: z5.literal("user"),
+  content: z5.union([
+    z5.string(),
+    z5.array(z5.union([textPartSchema, imagePartSchema]))
+  ]),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var coreAssistantMessageSchema = z5.object({
+  role: z5.literal("assistant"),
+  content: z5.union([
+    z5.string(),
+    z5.array(z5.union([textPartSchema, toolCallPartSchema]))
+  ]),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var coreToolMessageSchema = z5.object({
+  role: z5.literal("tool"),
+  content: z5.array(toolResultPartSchema),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var coreMessageSchema = z5.union([
+  coreSystemMessageSchema,
+  coreUserMessageSchema,
+  coreAssistantMessageSchema,
+  coreToolMessageSchema
+]);
+
+// core/prompt/validate-prompt.ts
+function validatePrompt(prompt) {
+  if (prompt.prompt == null && prompt.messages == null) {
+    throw new InvalidPromptError({
+      prompt,
+      message: "prompt or messages must be defined"
+    });
+  }
+  if (prompt.prompt != null && prompt.messages != null) {
+    throw new InvalidPromptError({
+      prompt,
+      message: "prompt and messages cannot be defined at the same time"
+    });
+  }
+  if (prompt.system != null && typeof prompt.system !== "string") {
+    throw new InvalidPromptError({
+      prompt,
+      message: "system must be a string"
+    });
+  }
+  if (prompt.prompt != null) {
+    if (typeof prompt.prompt !== "string") {
+      throw new InvalidPromptError({
+        prompt,
+        message: "prompt must be a string"
+      });
+    }
+    return {
+      type: "prompt",
+      prompt: prompt.prompt,
+      messages: void 0,
+      system: prompt.system
+    };
+  }
+  if (prompt.messages != null) {
+    const validationResult = safeValidateTypes({
+      value: prompt.messages,
+      schema: z6.array(coreMessageSchema)
+    });
+    if (!validationResult.success) {
+      throw new InvalidPromptError({
+        prompt,
+        message: "messages must be an array of CoreMessage",
+        cause: validationResult.error
+      });
+    }
+    return {
+      type: "messages",
+      prompt: void 0,
+      messages: prompt.messages,
+      // only possible case bc of checks above
+      system: prompt.system
+    };
+  }
+  throw new Error("unreachable");
+}
+
 // core/types/token-usage.ts
 function calculateCompletionTokenUsage(usage) {
   return {
@@ -1301,7 +1411,7 @@ async function generateObject({
       telemetry,
       attributes: {
         ...assembleOperationName({
-          operationName: "ai.generateObject",
+          operationId: "ai.generateObject",
           telemetry
         }),
         ...baseTelemetryAttributes,
@@ -1332,7 +1442,7 @@ async function generateObject({
       let providerMetadata;
       switch (mode) {
         case "json": {
-          const validatedPrompt = getValidatedPrompt({
+          const validatedPrompt = validatePrompt({
             system: model.supportsStructuredOutputs ? system : injectJsonSchemaIntoSystem({
               system,
               schema: schema.jsonSchema
@@ -1352,7 +1462,7 @@ async function generateObject({
                 telemetry,
                 attributes: {
                   ...assembleOperationName({
-                    operationName: "ai.generateObject.doGenerate",
+                    operationId: "ai.generateObject.doGenerate",
                     telemetry
                   }),
                   ...baseTelemetryAttributes,
@@ -1418,7 +1528,7 @@ async function generateObject({
           break;
         }
         case "tool": {
-          const validatedPrompt = getValidatedPrompt({
+          const validatedPrompt = validatePrompt({
             system,
             prompt,
             messages
@@ -1435,7 +1545,7 @@ async function generateObject({
                 telemetry,
                 attributes: {
                   ...assembleOperationName({
-                    operationName: "ai.generateObject.doGenerate",
+                    operationId: "ai.generateObject.doGenerate",
                     telemetry
                   }),
                   ...baseTelemetryAttributes,
@@ -1567,7 +1677,7 @@ var DefaultGenerateObjectResult = class {
 var experimental_generateObject = generateObject;
 
 // core/generate-object/stream-object.ts
-import { safeValidateTypes } from "@ai-sdk/provider-utils";
+import { safeValidateTypes as safeValidateTypes2 } from "@ai-sdk/provider-utils";
 import {
   asSchema as asSchema2,
   isDeepEqualData,
@@ -1677,7 +1787,7 @@ async function streamObject({
       telemetry,
       attributes: {
         ...assembleOperationName({
-          operationName: "ai.streamObject",
+          operationId: "ai.streamObject",
           telemetry
         }),
         ...baseTelemetryAttributes,
@@ -1701,7 +1811,7 @@ async function streamObject({
       let transformer;
       switch (mode) {
         case "json": {
-          const validatedPrompt = getValidatedPrompt({
+          const validatedPrompt = validatePrompt({
             system: model.supportsStructuredOutputs ? system : injectJsonSchemaIntoSystem({
               system,
               schema: schema.jsonSchema
@@ -1741,7 +1851,7 @@ async function streamObject({
           break;
         }
         case "tool": {
-          const validatedPrompt = getValidatedPrompt({
+          const validatedPrompt = validatePrompt({
             system,
             prompt,
             messages
@@ -1792,7 +1902,8 @@ async function streamObject({
       }
       const {
         result: { stream, warnings, rawResponse },
-        doStreamSpan
+        doStreamSpan,
+        startTimestamp
       } = await retry(
         () => recordSpan({
           name: "ai.streamObject.doStream",
@@ -1800,7 +1911,7 @@ async function streamObject({
             telemetry,
             attributes: {
               ...assembleOperationName({
-                operationName: "ai.streamObject.doStream",
+                operationId: "ai.streamObject.doStream",
                 telemetry
               }),
               ...baseTelemetryAttributes,
@@ -1822,8 +1933,10 @@ async function streamObject({
           tracer,
           endWhenDone: false,
           fn: async (doStreamSpan2) => ({
-            result: await model.doStream(callOptions),
-            doStreamSpan: doStreamSpan2
+            startTimestamp: performance.now(),
+            // get before the call
+            doStreamSpan: doStreamSpan2,
+            result: await model.doStream(callOptions)
           })
         })
       );
@@ -1835,7 +1948,8 @@ async function streamObject({
         onFinish,
         rootSpan,
         doStreamSpan,
-        telemetry
+        telemetry,
+        startTimestamp
       });
     }
   });
@@ -1849,7 +1963,8 @@ var DefaultStreamObjectResult = class {
     onFinish,
     rootSpan,
     doStreamSpan,
-    telemetry
+    telemetry,
+    startTimestamp
   }) {
     this.warnings = warnings;
     this.rawResponse = rawResponse;
@@ -1875,8 +1990,14 @@ var DefaultStreamObjectResult = class {
       new TransformStream({
         async transform(chunk, controller) {
           if (firstChunk) {
+            const msToFirstChunk = performance.now() - startTimestamp;
             firstChunk = false;
-            doStreamSpan.addEvent("ai.stream.firstChunk");
+            doStreamSpan.addEvent("ai.stream.firstChunk", {
+              "ai.stream.msToFirstChunk": msToFirstChunk
+            });
+            doStreamSpan.setAttributes({
+              "ai.stream.msToFirstChunk": msToFirstChunk
+            });
           }
           if (typeof chunk === "string") {
             accumulatedText += chunk;
@@ -1912,7 +2033,7 @@ var DefaultStreamObjectResult = class {
               controller.enqueue({ ...chunk, usage });
               resolveUsage(usage);
               resolveProviderMetadata(providerMetadata);
-              const validationResult = safeValidateTypes({
+              const validationResult = safeValidateTypes2({
                 value: latestObject,
                 schema
               });
@@ -2258,7 +2379,7 @@ async function generateText({
       telemetry,
       attributes: {
         ...assembleOperationName({
-          operationName: "ai.generateText",
+          operationId: "ai.generateText",
           telemetry
         }),
         ...baseTelemetryAttributes,
@@ -2273,7 +2394,7 @@ async function generateText({
     fn: async (span) => {
       var _a13, _b, _c, _d;
       const retry = retryWithExponentialBackoff({ maxRetries });
-      const validatedPrompt = getValidatedPrompt({
+      const validatedPrompt = validatePrompt({
         system,
         prompt,
         messages
@@ -2307,7 +2428,7 @@ async function generateText({
               telemetry,
               attributes: {
                 ...assembleOperationName({
-                  operationName: "ai.generateText.doGenerate",
+                  operationId: "ai.generateText.doGenerate",
                   telemetry
                 }),
                 ...baseTelemetryAttributes,
@@ -2451,7 +2572,7 @@ async function executeTools({
           telemetry,
           attributes: {
             ...assembleOperationName({
-              operationName: "ai.toolCall",
+              operationId: "ai.toolCall",
               telemetry
             }),
             "ai.toolCall.name": toolCall.toolName,
@@ -2701,7 +2822,7 @@ function runToolsTransformation({
                   telemetry,
                   attributes: {
                     ...assembleOperationName({
-                      operationName: "ai.toolCall",
+                      operationId: "ai.toolCall",
                       telemetry
                     }),
                     "ai.toolCall.name": toolCall.toolName,
@@ -2838,7 +2959,7 @@ async function streamText({
     attributes: selectTelemetryAttributes({
       telemetry,
       attributes: {
-        ...assembleOperationName({ operationName: "ai.streamText", telemetry }),
+        ...assembleOperationName({ operationId: "ai.streamText", telemetry }),
         ...baseTelemetryAttributes,
         // specific settings that only make sense on the outer level:
         "ai.prompt": {
@@ -2850,14 +2971,15 @@ async function streamText({
     endWhenDone: false,
     fn: async (rootSpan) => {
       const retry = retryWithExponentialBackoff({ maxRetries });
-      const validatedPrompt = getValidatedPrompt({ system, prompt, messages });
+      const validatedPrompt = validatePrompt({ system, prompt, messages });
       const promptMessages = await convertToLanguageModelPrompt({
         prompt: validatedPrompt,
         modelSupportsImageUrls: model.supportsImageUrls
       });
       const {
         result: { stream, warnings, rawResponse },
-        doStreamSpan
+        doStreamSpan,
+        startTimestamp
       } = await retry(
         () => recordSpan({
           name: "ai.streamText.doStream",
@@ -2865,7 +2987,7 @@ async function streamText({
             telemetry,
             attributes: {
               ...assembleOperationName({
-                operationName: "ai.streamText.doStream",
+                operationId: "ai.streamText.doStream",
                 telemetry
               }),
               ...baseTelemetryAttributes,
@@ -2885,22 +3007,22 @@ async function streamText({
           }),
           tracer,
           endWhenDone: false,
-          fn: async (doStreamSpan2) => {
-            return {
-              result: await model.doStream({
-                mode: {
-                  type: "regular",
-                  ...prepareToolsAndToolChoice({ tools, toolChoice })
-                },
-                ...prepareCallSettings(settings),
-                inputFormat: validatedPrompt.type,
-                prompt: promptMessages,
-                abortSignal,
-                headers
-              }),
-              doStreamSpan: doStreamSpan2
-            };
-          }
+          fn: async (doStreamSpan2) => ({
+            startTimestamp: performance.now(),
+            // get before the call
+            doStreamSpan: doStreamSpan2,
+            result: await model.doStream({
+              mode: {
+                type: "regular",
+                ...prepareToolsAndToolChoice({ tools, toolChoice })
+              },
+              ...prepareCallSettings(settings),
+              inputFormat: validatedPrompt.type,
+              prompt: promptMessages,
+              abortSignal,
+              headers
+            })
+          })
         })
       );
       return new DefaultStreamTextResult({
@@ -2917,7 +3039,8 @@ async function streamText({
         onFinish,
         rootSpan,
         doStreamSpan,
-        telemetry
+        telemetry,
+        startTimestamp
       });
     }
   });
@@ -2931,7 +3054,8 @@ var DefaultStreamTextResult = class {
     onFinish,
     rootSpan,
     doStreamSpan,
-    telemetry
+    telemetry,
+    startTimestamp
   }) {
     this.warnings = warnings;
     this.rawResponse = rawResponse;
@@ -2961,8 +3085,14 @@ var DefaultStreamTextResult = class {
       new TransformStream({
         async transform(chunk, controller) {
           if (firstChunk) {
+            const msToFirstChunk = performance.now() - startTimestamp;
             firstChunk = false;
-            doStreamSpan.addEvent("ai.stream.firstChunk");
+            doStreamSpan.addEvent("ai.stream.firstChunk", {
+              "ai.stream.msToFirstChunk": msToFirstChunk
+            });
+            doStreamSpan.setAttributes({
+              "ai.stream.msToFirstChunk": msToFirstChunk
+            });
           }
           if (chunk.type === "text-delta" && chunk.textDelta.length === 0) {
             return;
@@ -4716,10 +4846,7 @@ export {
   StreamingTextResponse,
   TypeValidationError,
   UnsupportedFunctionalityError,
-  convertDataContentToBase64String,
-  convertDataContentToUint8Array,
   convertToCoreMessages,
-  convertUint8ArrayToText,
   cosineSimilarity,
   createCallbacksTransformer,
   createEventStreamTransformer,