ai 3.2.35 → 3.2.36

package/dist/index.mjs

@@ -324,32 +324,112 @@ async function embedMany({
   values,
   maxRetries,
   abortSignal,
-  headers
+  headers,
+  experimental_telemetry: telemetry
 }) {
-  var _a, _b, _c;
-  const retry = retryWithExponentialBackoff({ maxRetries });
-  const maxEmbeddingsPerCall = model.maxEmbeddingsPerCall;
-  if (maxEmbeddingsPerCall == null) {
-    const modelResponse = await retry(
-      () => model.doEmbed({ values, abortSignal, headers })
-    );
-    return new DefaultEmbedManyResult({
-      values,
-      embeddings: modelResponse.embeddings,
-      usage: (_a = modelResponse.usage) != null ? _a : { tokens: NaN }
-    });
-  }
-  const valueChunks = splitArray(values, maxEmbeddingsPerCall);
-  const embeddings = [];
-  let tokens = 0;
-  for (const chunk of valueChunks) {
-    const modelResponse = await retry(
-      () => model.doEmbed({ values: chunk, abortSignal, headers })
-    );
-    embeddings.push(...modelResponse.embeddings);
-    tokens += (_c = (_b = modelResponse.usage) == null ? void 0 : _b.tokens) != null ? _c : NaN;
-  }
-  return new DefaultEmbedManyResult({ values, embeddings, usage: { tokens } });
+  var _a;
+  const baseTelemetryAttributes = getBaseTelemetryAttributes({
+    operationName: "ai.embedMany",
+    model,
+    telemetry,
+    headers,
+    settings: { maxRetries }
+  });
+  const tracer = getTracer({ isEnabled: (_a = telemetry == null ? void 0 : telemetry.isEnabled) != null ? _a : false });
+  return recordSpan({
+    name: "ai.embedMany",
+    attributes: {
+      ...baseTelemetryAttributes,
+      // specific settings that only make sense on the outer level:
+      "ai.values": values.map((value) => JSON.stringify(value))
+    },
+    tracer,
+    fn: async (span) => {
+      const retry = retryWithExponentialBackoff({ maxRetries });
+      const maxEmbeddingsPerCall = model.maxEmbeddingsPerCall;
+      if (maxEmbeddingsPerCall == null) {
+        const { embeddings: embeddings2, usage } = await retry(() => {
+          return recordSpan({
+            name: "ai.embedMany.doEmbed",
+            attributes: {
+              ...baseTelemetryAttributes,
+              // specific settings that only make sense on the outer level:
+              "ai.values": values.map((value) => JSON.stringify(value))
+            },
+            tracer,
+            fn: async (doEmbedSpan) => {
+              var _a2;
+              const modelResponse = await model.doEmbed({
+                values,
+                abortSignal,
+                headers
+              });
+              const embeddings3 = modelResponse.embeddings;
+              const usage2 = (_a2 = modelResponse.usage) != null ? _a2 : { tokens: NaN };
+              doEmbedSpan.setAttributes({
+                "ai.embeddings": embeddings3.map(
+                  (embedding) => JSON.stringify(embedding)
+                ),
+                "ai.usage.tokens": usage2.tokens
+              });
+              return { embeddings: embeddings3, usage: usage2 };
+            }
+          });
+        });
+        span.setAttributes({
+          "ai.embeddings": embeddings2.map(
+            (embedding) => JSON.stringify(embedding)
+          ),
+          "ai.usage.tokens": usage.tokens
+        });
+        return new DefaultEmbedManyResult({ values, embeddings: embeddings2, usage });
+      }
+      const valueChunks = splitArray(values, maxEmbeddingsPerCall);
+      const embeddings = [];
+      let tokens = 0;
+      for (const chunk of valueChunks) {
+        const { embeddings: responseEmbeddings, usage } = await retry(() => {
+          return recordSpan({
+            name: "ai.embedMany.doEmbed",
+            attributes: {
+              ...baseTelemetryAttributes,
+              // specific settings that only make sense on the outer level:
+              "ai.values": chunk.map((value) => JSON.stringify(value))
+            },
+            tracer,
+            fn: async (doEmbedSpan) => {
+              var _a2;
+              const modelResponse = await model.doEmbed({
+                values: chunk,
+                abortSignal,
+                headers
+              });
+              const embeddings2 = modelResponse.embeddings;
+              const usage2 = (_a2 = modelResponse.usage) != null ? _a2 : { tokens: NaN };
+              doEmbedSpan.setAttributes({
+                "ai.embeddings": embeddings2.map(
+                  (embedding) => JSON.stringify(embedding)
+                ),
+                "ai.usage.tokens": usage2.tokens
+              });
+              return { embeddings: embeddings2, usage: usage2 };
+            }
+          });
+        });
+        embeddings.push(...responseEmbeddings);
+        tokens += usage.tokens;
+      }
+      span.setAttributes({
+        "ai.embeddings": embeddings.map((embedding) => JSON.stringify(embedding)),
+        "ai.usage.tokens": tokens
+      });
+      return new DefaultEmbedManyResult({
+        values,
+        embeddings,
+        usage: { tokens }
+      });
+    }
+  });
 }
 var DefaultEmbedManyResult = class {
   constructor(options) {
@@ -363,6 +443,9 @@ var DefaultEmbedManyResult = class {
 import { NoObjectGeneratedError } from "@ai-sdk/provider";
 import { safeParseJSON } from "@ai-sdk/provider-utils";
 
+// core/prompt/convert-to-language-model-prompt.ts
+import { getErrorMessage as getErrorMessage2 } from "@ai-sdk/provider-utils";
+
 // core/util/detect-image-mimetype.ts
 var mimeTypeSignatures = [
   { mimeType: "image/gif", bytes: [71, 73, 70] },
@@ -379,6 +462,35 @@ function detectImageMimeType(image) {
   return void 0;
 }
 
+// core/util/download.ts
+import { DownloadError } from "@ai-sdk/provider";
+async function download({
+  url,
+  fetchImplementation = fetch
+}) {
+  var _a;
+  const urlText = url.toString();
+  try {
+    const response = await fetchImplementation(urlText);
+    if (!response.ok) {
+      throw new DownloadError({
+        url: urlText,
+        statusCode: response.status,
+        statusText: response.statusText
+      });
+    }
+    return {
+      data: new Uint8Array(await response.arrayBuffer()),
+      mimeType: (_a = response.headers.get("content-type")) != null ? _a : void 0
+    };
+  } catch (error) {
+    if (DownloadError.isDownloadError(error)) {
+      throw error;
+    }
+    throw new DownloadError({ url: urlText, cause: error });
+  }
+}
+
 // core/prompt/data-content.ts
 import { InvalidDataContentError } from "@ai-sdk/provider";
 import {
@@ -446,12 +558,16 @@ var InvalidMessageRoleError = class extends Error {
 };
 
 // core/prompt/convert-to-language-model-prompt.ts
-import { getErrorMessage as getErrorMessage2 } from "@ai-sdk/provider-utils";
-function convertToLanguageModelPrompt(prompt) {
+async function convertToLanguageModelPrompt({
+  prompt,
+  modelSupportsImageUrls = true,
+  downloadImplementation = download
+}) {
   const languageModelMessages = [];
   if (prompt.system != null) {
     languageModelMessages.push({ role: "system", content: prompt.system });
   }
+  const downloadedImages = modelSupportsImageUrls || prompt.messages == null ? null : await downloadImages(prompt.messages, downloadImplementation);
   const promptType = prompt.type;
   switch (promptType) {
     case "prompt": {
@@ -463,7 +579,9 @@ function convertToLanguageModelPrompt(prompt) {
     }
     case "messages": {
       languageModelMessages.push(
-        ...prompt.messages.map(convertToLanguageModelMessage)
+        ...prompt.messages.map(
+          (message) => convertToLanguageModelMessage(message, downloadedImages)
+        )
       );
       break;
     }
@@ -474,7 +592,7 @@ function convertToLanguageModelPrompt(prompt) {
   }
   return languageModelMessages;
 }
-function convertToLanguageModelMessage(message) {
+function convertToLanguageModelMessage(message, downloadedImages) {
   const role = message.role;
   switch (role) {
     case "system": {
@@ -491,18 +609,27 @@ function convertToLanguageModelMessage(message) {
         role: "user",
         content: message.content.map(
           (part) => {
-            var _a;
+            var _a, _b, _c;
             switch (part.type) {
               case "text": {
                 return part;
               }
               case "image": {
                 if (part.image instanceof URL) {
-                  return {
-                    type: "image",
-                    image: part.image,
-                    mimeType: part.mimeType
-                  };
+                  if (downloadedImages == null) {
+                    return {
+                      type: "image",
+                      image: part.image,
+                      mimeType: part.mimeType
+                    };
+                  } else {
+                    const downloadedImage = downloadedImages[part.image.toString()];
+                    return {
+                      type: "image",
+                      image: downloadedImage.data,
+                      mimeType: (_a = part.mimeType) != null ? _a : downloadedImage.mimeType
+                    };
+                  }
                 }
                 if (typeof part.image === "string") {
                   try {
@@ -510,11 +637,20 @@ function convertToLanguageModelMessage(message) {
                     switch (url.protocol) {
                       case "http:":
                       case "https:": {
-                        return {
-                          type: "image",
-                          image: url,
-                          mimeType: part.mimeType
-                        };
+                        if (downloadedImages == null) {
+                          return {
+                            type: "image",
+                            image: url,
+                            mimeType: part.mimeType
+                          };
+                        } else {
+                          const downloadedImage = downloadedImages[part.image];
+                          return {
+                            type: "image",
+                            image: downloadedImage.data,
+                            mimeType: (_b = part.mimeType) != null ? _b : downloadedImage.mimeType
+                          };
+                        }
                       }
                       case "data:": {
                         try {
@@ -549,7 +685,7 @@ function convertToLanguageModelMessage(message) {
                 return {
                   type: "image",
                   image: imageUint8,
-                  mimeType: (_a = part.mimeType) != null ? _a : detectImageMimeType(imageUint8)
+                  mimeType: (_c = part.mimeType) != null ? _c : detectImageMimeType(imageUint8)
                 };
               }
             }
@@ -581,6 +717,25 @@ function convertToLanguageModelMessage(message) {
     }
   }
 }
+async function downloadImages(messages, downloadImplementation) {
+  const urls = messages.filter((message) => message.role === "user").map((message) => message.content).filter(
+    (content) => Array.isArray(content)
+  ).flat().filter((part) => part.type === "image").map((part) => part.image).map(
+    (part) => (
+      // support string urls in image parts:
+      typeof part === "string" && (part.startsWith("http:") || part.startsWith("https:")) ? new URL(part) : part
+    )
+  ).filter((image) => image instanceof URL);
+  const downloadedImages = await Promise.all(
+    urls.map(async (url) => ({
+      url,
+      data: await downloadImplementation({ url })
+    }))
+  );
+  return Object.fromEntries(
+    downloadedImages.map(({ url, data }) => [url.toString(), data])
+  );
+}
 
 // core/prompt/get-validated-prompt.ts
 import { InvalidPromptError } from "@ai-sdk/provider";
@@ -825,7 +980,7 @@ async function generateObject({
       ...baseTelemetryAttributes,
       // specific settings that only make sense on the outer level:
       "ai.prompt": JSON.stringify({ system, prompt, messages }),
-      "ai.settings.jsonSchema": JSON.stringify(schema.jsonSchema),
+      "ai.schema": JSON.stringify(schema.jsonSchema),
       "ai.settings.mode": mode
     },
     tracer,
@@ -851,7 +1006,10 @@ async function generateObject({
             prompt,
             messages
           });
-          const promptMessages = convertToLanguageModelPrompt(validatedPrompt);
+          const promptMessages = await convertToLanguageModelPrompt({
+            prompt: validatedPrompt,
+            modelSupportsImageUrls: model.supportsImageUrls
+          });
           const inputFormat = validatedPrompt.type;
           const generateResult = await retry(
             () => recordSpan({
@@ -899,7 +1057,10 @@ async function generateObject({
             prompt,
             messages
           });
-          const promptMessages = convertToLanguageModelPrompt(validatedPrompt);
+          const promptMessages = await convertToLanguageModelPrompt({
+            prompt: validatedPrompt,
+            modelSupportsImageUrls: model.supportsImageUrls
+          });
           const inputFormat = validatedPrompt.type;
           const generateResult = await retry(
             () => recordSpan({
@@ -1076,101 +1237,156 @@ async function streamObject({
   maxRetries,
   abortSignal,
   headers,
+  experimental_telemetry: telemetry,
   onFinish,
   ...settings
 }) {
+  var _a;
+  const baseTelemetryAttributes = getBaseTelemetryAttributes({
+    operationName: "ai.streamObject",
+    model,
+    telemetry,
+    headers,
+    settings: { ...settings, maxRetries }
+  });
+  const tracer = getTracer({ isEnabled: (_a = telemetry == null ? void 0 : telemetry.isEnabled) != null ? _a : false });
   const retry = retryWithExponentialBackoff({ maxRetries });
   const schema = asSchema(inputSchema);
-  if (mode === "auto" || mode == null) {
-    mode = model.defaultObjectGenerationMode;
-  }
-  let callOptions;
-  let transformer;
-  switch (mode) {
-    case "json": {
-      const validatedPrompt = getValidatedPrompt({
-        system: injectJsonSchemaIntoSystem({
-          system,
-          schema: schema.jsonSchema
-        }),
-        prompt,
-        messages
-      });
-      callOptions = {
-        mode: { type: "object-json" },
-        ...prepareCallSettings(settings),
-        inputFormat: validatedPrompt.type,
-        prompt: convertToLanguageModelPrompt(validatedPrompt),
-        abortSignal,
-        headers
-      };
-      transformer = {
-        transform: (chunk, controller) => {
-          switch (chunk.type) {
-            case "text-delta":
-              controller.enqueue(chunk.textDelta);
-              break;
-            case "finish":
-            case "error":
-              controller.enqueue(chunk);
-              break;
-          }
+  return recordSpan({
+    name: "ai.streamObject",
+    attributes: {
+      ...baseTelemetryAttributes,
+      // specific settings that only make sense on the outer level:
+      "ai.prompt": JSON.stringify({ system, prompt, messages }),
+      "ai.schema": JSON.stringify(schema.jsonSchema),
+      "ai.settings.mode": mode
+    },
+    tracer,
+    endWhenDone: false,
+    fn: async (rootSpan) => {
+      if (mode === "auto" || mode == null) {
+        mode = model.defaultObjectGenerationMode;
+      }
+      let callOptions;
+      let transformer;
+      switch (mode) {
+        case "json": {
+          const validatedPrompt = getValidatedPrompt({
+            system: injectJsonSchemaIntoSystem({
+              system,
+              schema: schema.jsonSchema
+            }),
+            prompt,
+            messages
+          });
+          callOptions = {
+            mode: { type: "object-json" },
+            ...prepareCallSettings(settings),
+            inputFormat: validatedPrompt.type,
+            prompt: await convertToLanguageModelPrompt({
+              prompt: validatedPrompt,
+              modelSupportsImageUrls: model.supportsImageUrls
+            }),
+            abortSignal,
+            headers
+          };
+          transformer = {
+            transform: (chunk, controller) => {
+              switch (chunk.type) {
+                case "text-delta":
+                  controller.enqueue(chunk.textDelta);
+                  break;
+                case "finish":
+                case "error":
+                  controller.enqueue(chunk);
+                  break;
+              }
+            }
+          };
+          break;
         }
-      };
-      break;
-    }
-    case "tool": {
-      const validatedPrompt = getValidatedPrompt({
-        system,
-        prompt,
-        messages
-      });
-      callOptions = {
-        mode: {
-          type: "object-tool",
-          tool: {
-            type: "function",
-            name: "json",
-            description: "Respond with a JSON object.",
-            parameters: schema.jsonSchema
-          }
-        },
-        ...prepareCallSettings(settings),
-        inputFormat: validatedPrompt.type,
-        prompt: convertToLanguageModelPrompt(validatedPrompt),
-        abortSignal,
-        headers
-      };
-      transformer = {
-        transform(chunk, controller) {
-          switch (chunk.type) {
-            case "tool-call-delta":
-              controller.enqueue(chunk.argsTextDelta);
-              break;
-            case "finish":
-            case "error":
-              controller.enqueue(chunk);
-              break;
-          }
+        case "tool": {
+          const validatedPrompt = getValidatedPrompt({
+            system,
+            prompt,
+            messages
+          });
+          callOptions = {
+            mode: {
+              type: "object-tool",
+              tool: {
+                type: "function",
+                name: "json",
+                description: "Respond with a JSON object.",
+                parameters: schema.jsonSchema
+              }
+            },
+            ...prepareCallSettings(settings),
+            inputFormat: validatedPrompt.type,
+            prompt: await convertToLanguageModelPrompt({
+              prompt: validatedPrompt,
+              modelSupportsImageUrls: model.supportsImageUrls
+            }),
+            abortSignal,
+            headers
+          };
+          transformer = {
+            transform(chunk, controller) {
+              switch (chunk.type) {
+                case "tool-call-delta":
+                  controller.enqueue(chunk.argsTextDelta);
+                  break;
+                case "finish":
+                case "error":
+                  controller.enqueue(chunk);
+                  break;
+              }
+            }
+          };
+          break;
         }
-      };
-      break;
-    }
-    case void 0: {
-      throw new Error("Model does not have a default object generation mode.");
-    }
-    default: {
-      const _exhaustiveCheck = mode;
-      throw new Error(`Unsupported mode: ${_exhaustiveCheck}`);
+        case void 0: {
+          throw new Error(
+            "Model does not have a default object generation mode."
+          );
+        }
+        default: {
+          const _exhaustiveCheck = mode;
+          throw new Error(`Unsupported mode: ${_exhaustiveCheck}`);
+        }
+      }
+      const {
+        result: { stream, warnings, rawResponse },
+        doStreamSpan
+      } = await retry(
+        () => recordSpan({
+          name: "ai.streamObject.doStream",
+          attributes: {
+            ...baseTelemetryAttributes,
+            "ai.prompt.format": callOptions.inputFormat,
+            "ai.prompt.messages": JSON.stringify(callOptions.prompt),
+            "ai.settings.mode": mode
+          },
+          tracer,
+          endWhenDone: false,
+          fn: async (doStreamSpan2) => {
+            return {
+              result: await model.doStream(callOptions),
+              doStreamSpan: doStreamSpan2
+            };
+          }
+        })
+      );
+      return new DefaultStreamObjectResult({
+        stream: stream.pipeThrough(new TransformStream(transformer)),
+        warnings,
+        rawResponse,
+        schema,
+        onFinish,
+        rootSpan,
+        doStreamSpan
+      });
     }
-  }
-  const result = await retry(() => model.doStream(callOptions));
-  return new DefaultStreamObjectResult({
-    stream: result.stream.pipeThrough(new TransformStream(transformer)),
-    warnings: result.warnings,
-    rawResponse: result.rawResponse,
-    schema,
-    onFinish
   });
 }
 var DefaultStreamObjectResult = class {
@@ -1179,7 +1395,9 @@ var DefaultStreamObjectResult = class {
     warnings,
     rawResponse,
     schema,
-    onFinish
+    onFinish,
+    rootSpan,
+    doStreamSpan
   }) {
     this.warnings = warnings;
     this.rawResponse = rawResponse;
@@ -1194,10 +1412,15 @@ var DefaultStreamObjectResult = class {
     let accumulatedText = "";
     let delta = "";
     let latestObject = void 0;
+    let firstChunk = true;
     const self = this;
     this.originalStream = stream.pipeThrough(
       new TransformStream({
         async transform(chunk, controller) {
+          if (firstChunk) {
+            firstChunk = false;
+            doStreamSpan.addEvent("ai.stream.firstChunk");
+          }
           if (typeof chunk === "string") {
             accumulatedText += chunk;
             delta += chunk;
@@ -1251,12 +1474,24 @@ var DefaultStreamObjectResult = class {
         // invoke onFinish callback and resolve toolResults promise when the stream is about to close:
         async flush(controller) {
           try {
+            const finalUsage = usage != null ? usage : {
+              promptTokens: NaN,
+              completionTokens: NaN,
+              totalTokens: NaN
+            };
+            doStreamSpan.setAttributes({
+              "ai.usage.promptTokens": finalUsage.promptTokens,
+              "ai.usage.completionTokens": finalUsage.completionTokens,
+              "ai.result.object": JSON.stringify(object)
+            });
+            doStreamSpan.end();
+            rootSpan.setAttributes({
+              "ai.usage.promptTokens": finalUsage.promptTokens,
+              "ai.usage.completionTokens": finalUsage.completionTokens,
+              "ai.result.object": JSON.stringify(object)
+            });
             await (onFinish == null ? void 0 : onFinish({
-              usage: usage != null ? usage : {
-                promptTokens: NaN,
-                completionTokens: NaN,
-                totalTokens: NaN
-              },
+              usage: finalUsage,
               object,
               error,
               rawResponse,
@@ -1264,6 +1499,8 @@ var DefaultStreamObjectResult = class {
             }));
           } catch (error2) {
             controller.error(error2);
+          } finally {
+            rootSpan.end();
           }
         }
       })
@@ -1470,7 +1707,10 @@ async function generateText({
         ...prepareToolsAndToolChoice({ tools, toolChoice })
       };
       const callSettings = prepareCallSettings(settings);
-      const promptMessages = convertToLanguageModelPrompt(validatedPrompt);
+      const promptMessages = await convertToLanguageModelPrompt({
+        prompt: validatedPrompt,
+        modelSupportsImageUrls: model.supportsImageUrls
+      });
       let currentModelResponse;
       let currentToolCalls = [];
       let currentToolResults = [];
@@ -1543,7 +1783,9 @@ async function generateText({
         });
         responseMessages.push(...newResponseMessages);
         promptMessages.push(
-          ...newResponseMessages.map(convertToLanguageModelMessage)
+          ...newResponseMessages.map(
+            (message) => convertToLanguageModelMessage(message, null)
+          )
         );
       } while (
         // there are tool calls:
@@ -1953,7 +2195,10 @@ async function streamText({
     fn: async (rootSpan) => {
       const retry = retryWithExponentialBackoff({ maxRetries });
       const validatedPrompt = getValidatedPrompt({ system, prompt, messages });
-      const promptMessages = convertToLanguageModelPrompt(validatedPrompt);
+      const promptMessages = await convertToLanguageModelPrompt({
+        prompt: validatedPrompt,
+        modelSupportsImageUrls: model.supportsImageUrls
+      });
       const {
         result: { stream, warnings, rawResponse },
         doStreamSpan