ai 3.2.34 → 3.2.35

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 {
@@ -520,157 +722,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
-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/types/token-usage.ts
 function calculateCompletionTokenUsage(usage) {
   return {
@@ -680,12 +731,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 +741,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 +798,7 @@ function injectJsonSchemaIntoSystem({
 // core/generate-object/generate-object.ts
 async function generateObject({
   model,
-  schema,
+  schema: inputSchema,
   mode,
   system,
   prompt,
@@ -737,7 +817,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 +825,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,
@@ -764,7 +844,10 @@ async function generateObject({
       switch (mode) {
         case "json": {
           const validatedPrompt = getValidatedPrompt({
-            system: injectJsonSchemaIntoSystem({ system, schema: jsonSchema }),
+            system: injectJsonSchemaIntoSystem({
+              system,
+              schema: schema.jsonSchema
+            }),
             prompt,
             messages
           });
@@ -836,7 +919,7 @@ async function generateObject({
                       type: "function",
                       name: "json",
                       description: "Respond with a JSON object.",
-                      parameters: jsonSchema
+                      parameters: schema.jsonSchema
                     }
                   },
                   ...prepareCallSettings(settings),
@@ -985,7 +1068,7 @@ var DelayedPromise = class {
 // core/generate-object/stream-object.ts
 async function streamObject({
   model,
-  schema,
+  schema: inputSchema,
   mode,
   system,
   prompt,
@@ -997,7 +1080,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;
   }
@@ -1006,7 +1089,10 @@ async function streamObject({
   switch (mode) {
     case "json": {
       const validatedPrompt = getValidatedPrompt({
-        system: injectJsonSchemaIntoSystem({ system, schema: jsonSchema }),
+        system: injectJsonSchemaIntoSystem({
+          system,
+          schema: schema.jsonSchema
+        }),
         prompt,
         messages
       });
@@ -1046,7 +1132,7 @@ async function streamObject({
             type: "function",
             name: "json",
             description: "Respond with a JSON object.",
-            parameters: jsonSchema
+            parameters: schema.jsonSchema
           }
         },
         ...prepareCallSettings(settings),
@@ -1291,7 +1377,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 +1406,7 @@ function parseToolCall({
   }
   const parseResult = safeParseJSON2({
     text: toolCall.args,
-    schema: tool2.parameters
+    schema: asSchema(tool2.parameters)
   });
   if (parseResult.success === false) {
     throw new InvalidToolArgumentsError({
@@ -3623,6 +3709,7 @@ export {
   generateId2 as generateId,
   generateObject,
   generateText,
+  jsonSchema,
   nanoid,
   parseComplexResponse,
   parseStreamPart,