ai 3.2.34 → 3.2.35

package/dist/index.js

@@ -91,6 +91,7 @@ __export(streams_exports, {
   generateId: () => generateId2,
   generateObject: () => generateObject,
   generateText: () => generateText,
+  jsonSchema: () => jsonSchema,
   nanoid: () => nanoid,
   parseComplexResponse: () => import_ui_utils6.parseComplexResponse,
   parseStreamPart: () => import_ui_utils6.parseStreamPart,
@@ -104,7 +105,158 @@ __export(streams_exports, {
 });
 module.exports = __toCommonJS(streams_exports);
 var import_ui_utils6 = require("@ai-sdk/ui-utils");
-var import_provider_utils7 = require("@ai-sdk/provider-utils");
+var import_provider_utils8 = require("@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
+var import_api = require("@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 import_api.trace.getTracer("ai");
+}
+
+// core/telemetry/record-span.ts
+var import_api2 = require("@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: import_api2.SpanStatusCode.ERROR,
+            message: error.message
+          });
+        } else {
+          span.setStatus({ code: import_api2.SpanStatusCode.ERROR });
+        }
+      } finally {
+        span.end();
+      }
+      throw error;
+    }
+  });
+}
 
 // core/util/retry-with-exponential-backoff.ts
 var import_provider = require("@ai-sdk/provider");
@@ -174,18 +326,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 {
@@ -252,7 +455,7 @@ var DefaultEmbedManyResult = class {
 
 // core/generate-object/generate-object.ts
 var import_provider5 = require("@ai-sdk/provider");
-var import_provider_utils4 = require("@ai-sdk/provider-utils");
+var import_provider_utils5 = require("@ai-sdk/provider-utils");
 
 // core/util/detect-image-mimetype.ts
 var mimeTypeSignatures = [
@@ -610,157 +813,6 @@ function prepareCallSettings({
   };
 }
 
-// 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
-var import_api = require("@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 import_api.trace.getTracer("ai");
-}
-
-// core/telemetry/record-span.ts
-var import_api2 = require("@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: import_api2.SpanStatusCode.ERROR,
-            message: error.message
-          });
-        } else {
-          span.setStatus({ code: import_api2.SpanStatusCode.ERROR });
-        }
-      } finally {
-        span.end();
-      }
-      throw error;
-    }
-  });
-}
-
 // core/types/token-usage.ts
 function calculateCompletionTokenUsage(usage) {
   return {
@@ -770,12 +822,6 @@ function calculateCompletionTokenUsage(usage) {
   };
 }
 
-// core/util/convert-zod-to-json-schema.ts
-var import_zod_to_json_schema = __toESM(require("zod-to-json-schema"));
-function convertZodToJSONSchema(zodSchema) {
-  return (0, import_zod_to_json_schema.default)(zodSchema);
-}
-
 // core/util/prepare-response-headers.ts
 function prepareResponseHeaders(init, { contentType }) {
   var _a;
@@ -786,6 +832,41 @@ function prepareResponseHeaders(init, { contentType }) {
   return headers;
 }
 
+// core/util/schema.ts
+var import_provider_utils4 = require("@ai-sdk/provider-utils");
+var import_zod_to_json_schema = __toESM(require("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
+    [import_provider_utils4.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:
+    (0, import_zod_to_json_schema.default)(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.";
@@ -808,7 +889,7 @@ function injectJsonSchemaIntoSystem({
 // core/generate-object/generate-object.ts
 async function generateObject({
   model,
-  schema,
+  schema: inputSchema,
   mode,
   system,
   prompt,
@@ -827,7 +908,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",
@@ -835,7 +916,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.settings.jsonSchema": JSON.stringify(schema.jsonSchema),
       "ai.settings.mode": mode
     },
     tracer,
@@ -854,7 +935,10 @@ async function generateObject({
       switch (mode) {
         case "json": {
           const validatedPrompt = getValidatedPrompt({
-            system: injectJsonSchemaIntoSystem({ system, schema: jsonSchema }),
+            system: injectJsonSchemaIntoSystem({
+              system,
+              schema: schema.jsonSchema
+            }),
             prompt,
             messages
           });
@@ -926,7 +1010,7 @@ async function generateObject({
                       type: "function",
                       name: "json",
                       description: "Respond with a JSON object.",
-                      parameters: jsonSchema
+                      parameters: schema.jsonSchema
                     }
                   },
                   ...prepareCallSettings(settings),
@@ -968,7 +1052,7 @@ async function generateObject({
           throw new Error(`Unsupported mode: ${_exhaustiveCheck}`);
         }
       }
-      const parseResult = (0, import_provider_utils4.safeParseJSON)({ text: result, schema });
+      const parseResult = (0, import_provider_utils5.safeParseJSON)({ text: result, schema });
       if (!parseResult.success) {
         throw parseResult.error;
       }
@@ -1011,7 +1095,7 @@ var DefaultGenerateObjectResult = class {
 var experimental_generateObject = generateObject;
 
 // core/generate-object/stream-object.ts
-var import_provider_utils5 = require("@ai-sdk/provider-utils");
+var import_provider_utils6 = require("@ai-sdk/provider-utils");
 var import_ui_utils = require("@ai-sdk/ui-utils");
 
 // core/util/async-iterable-stream.ts
@@ -1072,7 +1156,7 @@ var DelayedPromise = class {
 // core/generate-object/stream-object.ts
 async function streamObject({
   model,
-  schema,
+  schema: inputSchema,
   mode,
   system,
   prompt,
@@ -1084,7 +1168,7 @@ async function streamObject({
   ...settings
 }) {
   const retry = retryWithExponentialBackoff({ maxRetries });
-  const jsonSchema = convertZodToJSONSchema(schema);
+  const schema = asSchema(inputSchema);
   if (mode === "auto" || mode == null) {
     mode = model.defaultObjectGenerationMode;
   }
@@ -1093,7 +1177,10 @@ async function streamObject({
   switch (mode) {
     case "json": {
       const validatedPrompt = getValidatedPrompt({
-        system: injectJsonSchemaIntoSystem({ system, schema: jsonSchema }),
+        system: injectJsonSchemaIntoSystem({
+          system,
+          schema: schema.jsonSchema
+        }),
         prompt,
         messages
       });
@@ -1133,7 +1220,7 @@ async function streamObject({
             type: "function",
             name: "json",
             description: "Respond with a JSON object.",
-            parameters: jsonSchema
+            parameters: schema.jsonSchema
           }
         },
         ...prepareCallSettings(settings),
@@ -1230,7 +1317,7 @@ var DefaultStreamObjectResult = class {
               usage = calculateCompletionTokenUsage(chunk.usage);
               controller.enqueue({ ...chunk, usage });
               resolveUsage(usage);
-              const validationResult = (0, import_provider_utils5.safeValidateTypes)({
+              const validationResult = (0, import_provider_utils6.safeValidateTypes)({
                 value: latestObject,
                 schema
               });
@@ -1378,7 +1465,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 }
   };
@@ -1386,7 +1473,7 @@ function prepareToolsAndToolChoice({
 
 // core/generate-text/tool-call.ts
 var import_provider6 = require("@ai-sdk/provider");
-var import_provider_utils6 = require("@ai-sdk/provider-utils");
+var import_provider_utils7 = require("@ai-sdk/provider-utils");
 function parseToolCall({
   toolCall,
   tools
@@ -1402,9 +1489,9 @@ function parseToolCall({
       availableTools: Object.keys(tools)
     });
   }
-  const parseResult = (0, import_provider_utils6.safeParseJSON)({
+  const parseResult = (0, import_provider_utils7.safeParseJSON)({
     text: toolCall.args,
-    schema: tool2.parameters
+    schema: asSchema(tool2.parameters)
   });
   if (parseResult.success === false) {
     throw new import_provider6.InvalidToolArgumentsError({
@@ -3619,8 +3706,8 @@ var StreamingTextResponse = class extends Response {
 };
 
 // streams/index.ts
-var generateId2 = import_provider_utils7.generateId;
-var nanoid = import_provider_utils7.generateId;
+var generateId2 = import_provider_utils8.generateId;
+var nanoid = import_provider_utils8.generateId;
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AIStream,
@@ -3684,6 +3771,7 @@ var nanoid = import_provider_utils7.generateId;
   generateId,
   generateObject,
   generateText,
+  jsonSchema,
   nanoid,
   parseComplexResponse,
   parseStreamPart,