ai 3.2.34 → 3.2.36

package/dist/index.mjs

@@ -13,6 +13,157 @@ import {
 } from "@ai-sdk/ui-utils";
 import { generateId as generateIdImpl } from "@ai-sdk/provider-utils";
 
+// core/telemetry/get-base-telemetry-attributes.ts
+function getBaseTelemetryAttributes({
+  operationName,
+  model,
+  settings,
+  telemetry,
+  headers
+}) {
+  var _a;
+  return {
+    "ai.model.provider": model.provider,
+    "ai.model.id": model.modelId,
+    // settings:
+    ...Object.entries(settings).reduce((attributes, [key, value]) => {
+      attributes[`ai.settings.${key}`] = value;
+      return attributes;
+    }, {}),
+    // special telemetry information
+    "operation.name": operationName,
+    "resource.name": telemetry == null ? void 0 : telemetry.functionId,
+    "ai.telemetry.functionId": telemetry == null ? void 0 : telemetry.functionId,
+    // add metadata as attributes:
+    ...Object.entries((_a = telemetry == null ? void 0 : telemetry.metadata) != null ? _a : {}).reduce(
+      (attributes, [key, value]) => {
+        attributes[`ai.telemetry.metadata.${key}`] = value;
+        return attributes;
+      },
+      {}
+    ),
+    // request headers
+    ...Object.entries(headers != null ? headers : {}).reduce((attributes, [key, value]) => {
+      if (value !== void 0) {
+        attributes[`ai.request.headers.${key}`] = value;
+      }
+      return attributes;
+    }, {})
+  };
+}
+
+// core/telemetry/get-tracer.ts
+import { trace } from "@opentelemetry/api";
+
+// core/telemetry/noop-tracer.ts
+var noopTracer = {
+  startSpan() {
+    return noopSpan;
+  },
+  startActiveSpan(name, arg1, arg2, arg3) {
+    if (typeof arg1 === "function") {
+      return arg1(noopSpan);
+    }
+    if (typeof arg2 === "function") {
+      return arg2(noopSpan);
+    }
+    if (typeof arg3 === "function") {
+      return arg3(noopSpan);
+    }
+  }
+};
+var noopSpan = {
+  spanContext() {
+    return noopSpanContext;
+  },
+  setAttribute() {
+    return this;
+  },
+  setAttributes() {
+    return this;
+  },
+  addEvent() {
+    return this;
+  },
+  addLink() {
+    return this;
+  },
+  addLinks() {
+    return this;
+  },
+  setStatus() {
+    return this;
+  },
+  updateName() {
+    return this;
+  },
+  end() {
+    return this;
+  },
+  isRecording() {
+    return false;
+  },
+  recordException() {
+    return this;
+  }
+};
+var noopSpanContext = {
+  traceId: "",
+  spanId: "",
+  traceFlags: 0
+};
+
+// core/telemetry/get-tracer.ts
+var testTracer = void 0;
+function getTracer({ isEnabled }) {
+  if (!isEnabled) {
+    return noopTracer;
+  }
+  if (testTracer) {
+    return testTracer;
+  }
+  return trace.getTracer("ai");
+}
+
+// core/telemetry/record-span.ts
+import { SpanStatusCode } from "@opentelemetry/api";
+function recordSpan({
+  name,
+  tracer,
+  attributes,
+  fn,
+  endWhenDone = true
+}) {
+  return tracer.startActiveSpan(name, { attributes }, async (span) => {
+    try {
+      const result = await fn(span);
+      if (endWhenDone) {
+        span.end();
+      }
+      return result;
+    } catch (error) {
+      try {
+        if (error instanceof Error) {
+          span.recordException({
+            name: error.name,
+            message: error.message,
+            stack: error.stack
+          });
+          span.setStatus({
+            code: SpanStatusCode.ERROR,
+            message: error.message
+          });
+        } else {
+          span.setStatus({ code: SpanStatusCode.ERROR });
+        }
+      } finally {
+        span.end();
+      }
+      throw error;
+    }
+  });
+}
+
 // core/util/retry-with-exponential-backoff.ts
 import { APICallError, RetryError } from "@ai-sdk/provider";
 import { getErrorMessage, isAbortError } from "@ai-sdk/provider-utils";
@@ -81,18 +232,69 @@ async function embed({
   value,
   maxRetries,
   abortSignal,
-  headers
+  headers,
+  experimental_telemetry: telemetry
 }) {
   var _a;
-  const retry = retryWithExponentialBackoff({ maxRetries });
-  const modelResponse = await retry(
-    () => model.doEmbed({ values: [value], abortSignal, headers })
-  );
-  return new DefaultEmbedResult({
-    value,
-    embedding: modelResponse.embeddings[0],
-    usage: (_a = modelResponse.usage) != null ? _a : { tokens: NaN },
-    rawResponse: modelResponse.rawResponse
+  const baseTelemetryAttributes = getBaseTelemetryAttributes({
+    operationName: "ai.embed",
+    model,
+    telemetry,
+    headers,
+    settings: { maxRetries }
+  });
+  const tracer = getTracer({ isEnabled: (_a = telemetry == null ? void 0 : telemetry.isEnabled) != null ? _a : false });
+  return recordSpan({
+    name: "ai.embed",
+    attributes: {
+      ...baseTelemetryAttributes,
+      // specific settings that only make sense on the outer level:
+      "ai.value": JSON.stringify(value)
+    },
+    tracer,
+    fn: async (span) => {
+      const retry = retryWithExponentialBackoff({ maxRetries });
+      const { embedding, usage, rawResponse } = await retry(
+        () => (
+          // nested spans to align with the embedMany telemetry data:
+          recordSpan({
+            name: "ai.embed.doEmbed",
+            attributes: {
+              ...baseTelemetryAttributes,
+              // specific settings that only make sense on the outer level:
+              "ai.values": [JSON.stringify(value)]
+            },
+            tracer,
+            fn: async (doEmbedSpan) => {
+              var _a2;
+              const modelResponse = await model.doEmbed({
+                values: [value],
+                abortSignal,
+                headers
+              });
+              const embedding2 = modelResponse.embeddings[0];
+              const usage2 = (_a2 = modelResponse.usage) != null ? _a2 : { tokens: NaN };
+              doEmbedSpan.setAttributes({
+                "ai.embeddings": modelResponse.embeddings.map(
+                  (embedding3) => JSON.stringify(embedding3)
+                ),
+                "ai.usage.tokens": usage2.tokens
+              });
+              return {
+                embedding: embedding2,
+                usage: usage2,
+                rawResponse: modelResponse.rawResponse
+              };
+            }
+          })
+        )
+      );
+      span.setAttributes({
+        "ai.embedding": JSON.stringify(embedding),
+        "ai.usage.tokens": usage.tokens
+      });
+      return new DefaultEmbedResult({ value, embedding, usage, rawResponse });
+    }
   });
 }
 var DefaultEmbedResult = class {
@@ -122,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) {
@@ -161,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] },
@@ -177,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 {
@@ -244,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": {
@@ -261,7 +579,9 @@ function convertToLanguageModelPrompt(prompt) {
     }
     case "messages": {
       languageModelMessages.push(
-        ...prompt.messages.map(convertToLanguageModelMessage)
+        ...prompt.messages.map(
+          (message) => convertToLanguageModelMessage(message, downloadedImages)
+        )
       );
       break;
     }
@@ -272,7 +592,7 @@ function convertToLanguageModelPrompt(prompt) {
   }
   return languageModelMessages;
 }
-function convertToLanguageModelMessage(message) {
+function convertToLanguageModelMessage(message, downloadedImages) {
   const role = message.role;
   switch (role) {
     case "system": {
@@ -289,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 {
@@ -308,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 {
@@ -347,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)
                 };
               }
             }
@@ -379,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";
@@ -467,208 +824,57 @@ function prepareCallSettings({
   }
   if (presencePenalty != null) {
     if (typeof presencePenalty !== "number") {
-      throw new InvalidArgumentError({
-        parameter: "presencePenalty",
-        value: presencePenalty,
-        message: "presencePenalty must be a number"
-      });
-    }
-  }
-  if (frequencyPenalty != null) {
-    if (typeof frequencyPenalty !== "number") {
-      throw new InvalidArgumentError({
-        parameter: "frequencyPenalty",
-        value: frequencyPenalty,
-        message: "frequencyPenalty must be a number"
-      });
-    }
-  }
-  if (seed != null) {
-    if (!Number.isInteger(seed)) {
-      throw new InvalidArgumentError({
-        parameter: "seed",
-        value: seed,
-        message: "seed must be an integer"
-      });
-    }
-  }
-  if (maxRetries != null) {
-    if (!Number.isInteger(maxRetries)) {
-      throw new InvalidArgumentError({
-        parameter: "maxRetries",
-        value: maxRetries,
-        message: "maxRetries must be an integer"
-      });
-    }
-    if (maxRetries < 0) {
-      throw new InvalidArgumentError({
-        parameter: "maxRetries",
-        value: maxRetries,
-        message: "maxRetries must be >= 0"
-      });
-    }
-  }
-  return {
-    maxTokens,
-    temperature: temperature != null ? temperature : 0,
-    topP,
-    presencePenalty,
-    frequencyPenalty,
-    stopSequences: stopSequences != null && stopSequences.length > 0 ? stopSequences : void 0,
-    seed,
-    maxRetries: maxRetries != null ? maxRetries : 2
-  };
-}
-
-// core/telemetry/get-base-telemetry-attributes.ts
-function getBaseTelemetryAttributes({
-  operationName,
-  model,
-  settings,
-  telemetry,
-  headers
-}) {
-  var _a;
-  return {
-    "ai.model.provider": model.provider,
-    "ai.model.id": model.modelId,
-    // settings:
-    ...Object.entries(settings).reduce((attributes, [key, value]) => {
-      attributes[`ai.settings.${key}`] = value;
-      return attributes;
-    }, {}),
-    // special telemetry information
-    "operation.name": operationName,
-    "resource.name": telemetry == null ? void 0 : telemetry.functionId,
-    "ai.telemetry.functionId": telemetry == null ? void 0 : telemetry.functionId,
-    // add metadata as attributes:
-    ...Object.entries((_a = telemetry == null ? void 0 : telemetry.metadata) != null ? _a : {}).reduce(
-      (attributes, [key, value]) => {
-        attributes[`ai.telemetry.metadata.${key}`] = value;
-        return attributes;
-      },
-      {}
-    ),
-    // request headers
-    ...Object.entries(headers != null ? headers : {}).reduce((attributes, [key, value]) => {
-      if (value !== void 0) {
-        attributes[`ai.request.headers.${key}`] = value;
-      }
-      return attributes;
-    }, {})
-  };
-}
-
-// core/telemetry/get-tracer.ts
-import { trace } from "@opentelemetry/api";
-
-// core/telemetry/noop-tracer.ts
-var noopTracer = {
-  startSpan() {
-    return noopSpan;
-  },
-  startActiveSpan(name, arg1, arg2, arg3) {
-    if (typeof arg1 === "function") {
-      return arg1(noopSpan);
-    }
-    if (typeof arg2 === "function") {
-      return arg2(noopSpan);
-    }
-    if (typeof arg3 === "function") {
-      return arg3(noopSpan);
-    }
-  }
-};
-var noopSpan = {
-  spanContext() {
-    return noopSpanContext;
-  },
-  setAttribute() {
-    return this;
-  },
-  setAttributes() {
-    return this;
-  },
-  addEvent() {
-    return this;
-  },
-  addLink() {
-    return this;
-  },
-  addLinks() {
-    return this;
-  },
-  setStatus() {
-    return this;
-  },
-  updateName() {
-    return this;
-  },
-  end() {
-    return this;
-  },
-  isRecording() {
-    return false;
-  },
-  recordException() {
-    return this;
-  }
-};
-var noopSpanContext = {
-  traceId: "",
-  spanId: "",
-  traceFlags: 0
-};
-
-// core/telemetry/get-tracer.ts
-var testTracer = void 0;
-function getTracer({ isEnabled }) {
-  if (!isEnabled) {
-    return noopTracer;
+      throw new InvalidArgumentError({
+        parameter: "presencePenalty",
+        value: presencePenalty,
+        message: "presencePenalty must be a number"
+      });
+    }
   }
-  if (testTracer) {
-    return testTracer;
+  if (frequencyPenalty != null) {
+    if (typeof frequencyPenalty !== "number") {
+      throw new InvalidArgumentError({
+        parameter: "frequencyPenalty",
+        value: frequencyPenalty,
+        message: "frequencyPenalty must be a number"
+      });
+    }
   }
-  return trace.getTracer("ai");
-}
-
-// core/telemetry/record-span.ts
-import { SpanStatusCode } from "@opentelemetry/api";
-function recordSpan({
-  name,
-  tracer,
-  attributes,
-  fn,
-  endWhenDone = true
-}) {
-  return tracer.startActiveSpan(name, { attributes }, async (span) => {
-    try {
-      const result = await fn(span);
-      if (endWhenDone) {
-        span.end();
-      }
-      return result;
-    } catch (error) {
-      try {
-        if (error instanceof Error) {
-          span.recordException({
-            name: error.name,
-            message: error.message,
-            stack: error.stack
-          });
-          span.setStatus({
-            code: SpanStatusCode.ERROR,
-            message: error.message
-          });
-        } else {
-          span.setStatus({ code: SpanStatusCode.ERROR });
-        }
-      } finally {
-        span.end();
-      }
-      throw error;
+  if (seed != null) {
+    if (!Number.isInteger(seed)) {
+      throw new InvalidArgumentError({
+        parameter: "seed",
+        value: seed,
+        message: "seed must be an integer"
+      });
     }
-  });
+  }
+  if (maxRetries != null) {
+    if (!Number.isInteger(maxRetries)) {
+      throw new InvalidArgumentError({
+        parameter: "maxRetries",
+        value: maxRetries,
+        message: "maxRetries must be an integer"
+      });
+    }
+    if (maxRetries < 0) {
+      throw new InvalidArgumentError({
+        parameter: "maxRetries",
+        value: maxRetries,
+        message: "maxRetries must be >= 0"
+      });
+    }
+  }
+  return {
+    maxTokens,
+    temperature: temperature != null ? temperature : 0,
+    topP,
+    presencePenalty,
+    frequencyPenalty,
+    stopSequences: stopSequences != null && stopSequences.length > 0 ? stopSequences : void 0,
+    seed,
+    maxRetries: maxRetries != null ? maxRetries : 2
+  };
 }
 
 // core/types/token-usage.ts
@@ -680,12 +886,6 @@ function calculateCompletionTokenUsage(usage) {
   };
 }
 
-// core/util/convert-zod-to-json-schema.ts
-import zodToJsonSchema from "zod-to-json-schema";
-function convertZodToJSONSchema(zodSchema) {
-  return zodToJsonSchema(zodSchema);
-}
-
 // core/util/prepare-response-headers.ts
 function prepareResponseHeaders(init, { contentType }) {
   var _a;
@@ -696,6 +896,41 @@ function prepareResponseHeaders(init, { contentType }) {
   return headers;
 }
 
+// core/util/schema.ts
+import { validatorSymbol } from "@ai-sdk/provider-utils";
+import zodToJsonSchema from "zod-to-json-schema";
+var schemaSymbol = Symbol("vercel.ai.schema");
+function jsonSchema(jsonSchema2, {
+  validate
+} = {}) {
+  return {
+    [schemaSymbol]: true,
+    _type: void 0,
+    // should never be used directly
+    [validatorSymbol]: true,
+    jsonSchema: jsonSchema2,
+    validate
+  };
+}
+function isSchema(value) {
+  return typeof value === "object" && value !== null && schemaSymbol in value && value[schemaSymbol] === true && "jsonSchema" in value && "validate" in value;
+}
+function asSchema(schema) {
+  return isSchema(schema) ? schema : zodSchema(schema);
+}
+function zodSchema(zodSchema2) {
+  return jsonSchema(
+    // we assume that zodToJsonSchema will return a valid JSONSchema7:
+    zodToJsonSchema(zodSchema2),
+    {
+      validate: (value) => {
+        const result = zodSchema2.safeParse(value);
+        return result.success ? { success: true, value: result.data } : { success: false, error: result.error };
+      }
+    }
+  );
+}
+
 // core/generate-object/inject-json-schema-into-system.ts
 var DEFAULT_SCHEMA_PREFIX = "JSON schema:";
 var DEFAULT_SCHEMA_SUFFIX = "You MUST answer with a JSON object that matches the JSON schema above.";
@@ -718,7 +953,7 @@ function injectJsonSchemaIntoSystem({
 // core/generate-object/generate-object.ts
 async function generateObject({
   model,
-  schema,
+  schema: inputSchema,
   mode,
   system,
   prompt,
@@ -737,7 +972,7 @@ async function generateObject({
     headers,
     settings: { ...settings, maxRetries }
   });
-  const jsonSchema = convertZodToJSONSchema(schema);
+  const schema = asSchema(inputSchema);
   const tracer = getTracer({ isEnabled: (_a = telemetry == null ? void 0 : telemetry.isEnabled) != null ? _a : false });
   return recordSpan({
     name: "ai.generateObject",
@@ -745,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(jsonSchema),
+      "ai.schema": JSON.stringify(schema.jsonSchema),
       "ai.settings.mode": mode
     },
     tracer,
@@ -764,11 +999,17 @@ async function generateObject({
       switch (mode) {
         case "json": {
           const validatedPrompt = getValidatedPrompt({
-            system: injectJsonSchemaIntoSystem({ system, schema: jsonSchema }),
+            system: injectJsonSchemaIntoSystem({
+              system,
+              schema: schema.jsonSchema
+            }),
             prompt,
             messages
           });
-          const promptMessages = convertToLanguageModelPrompt(validatedPrompt);
+          const promptMessages = await convertToLanguageModelPrompt({
+            prompt: validatedPrompt,
+            modelSupportsImageUrls: model.supportsImageUrls
+          });
           const inputFormat = validatedPrompt.type;
           const generateResult = await retry(
             () => recordSpan({
@@ -816,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({
@@ -836,7 +1080,7 @@ async function generateObject({
                       type: "function",
                       name: "json",
                       description: "Respond with a JSON object.",
-                      parameters: jsonSchema
+                      parameters: schema.jsonSchema
                     }
                   },
                   ...prepareCallSettings(settings),
@@ -985,7 +1229,7 @@ var DelayedPromise = class {
 // core/generate-object/stream-object.ts
 async function streamObject({
   model,
-  schema,
+  schema: inputSchema,
   mode,
   system,
   prompt,
@@ -993,98 +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 jsonSchema = convertZodToJSONSchema(schema);
-  if (mode === "auto" || mode == null) {
-    mode = model.defaultObjectGenerationMode;
-  }
-  let callOptions;
-  let transformer;
-  switch (mode) {
-    case "json": {
-      const validatedPrompt = getValidatedPrompt({
-        system: injectJsonSchemaIntoSystem({ system, 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;
-          }
+  const schema = asSchema(inputSchema);
+  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: 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 {
@@ -1093,7 +1395,9 @@ var DefaultStreamObjectResult = class {
     warnings,
     rawResponse,
     schema,
-    onFinish
+    onFinish,
+    rootSpan,
+    doStreamSpan
   }) {
     this.warnings = warnings;
     this.rawResponse = rawResponse;
@@ -1108,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;
@@ -1165,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,
@@ -1178,6 +1499,8 @@ var DefaultStreamObjectResult = class {
             }));
           } catch (error2) {
             controller.error(error2);
+          } finally {
+            rootSpan.end();
           }
         }
       })
@@ -1291,7 +1614,7 @@ function prepareToolsAndToolChoice({
       type: "function",
       name,
       description: tool2.description,
-      parameters: convertZodToJSONSchema(tool2.parameters)
+      parameters: asSchema(tool2.parameters).jsonSchema
     })),
     toolChoice: toolChoice == null ? { type: "auto" } : typeof toolChoice === "string" ? { type: toolChoice } : { type: "tool", toolName: toolChoice.toolName }
   };
@@ -1320,7 +1643,7 @@ function parseToolCall({
   }
   const parseResult = safeParseJSON2({
     text: toolCall.args,
-    schema: tool2.parameters
+    schema: asSchema(tool2.parameters)
   });
   if (parseResult.success === false) {
     throw new InvalidToolArgumentsError({
@@ -1384,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 = [];
@@ -1457,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:
@@ -1867,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
@@ -3623,6 +3954,7 @@ export {
   generateId2 as generateId,
   generateObject,
   generateText,
+  jsonSchema,
   nanoid,
   parseComplexResponse,
   parseStreamPart,