ai 5.0.0-beta.32 → 5.0.0-beta.33

package/dist/index.mjs

@@ -1823,12 +1823,15 @@ function createToolModelOutput({
   if (errorMode === "text") {
     return { type: "error-text", value: getErrorMessage4(output) };
   } else if (errorMode === "json") {
-    return { type: "error-json", value: output };
+    return { type: "error-json", value: toJSONValue(output) };
   }
   if (tool3 == null ? void 0 : tool3.toModelOutput) {
     return tool3.toModelOutput(output);
   }
-  return typeof output === "string" ? { type: "text", value: output } : { type: "json", value: output };
+  return typeof output === "string" ? { type: "text", value: output } : { type: "json", value: toJSONValue(output) };
+}
+function toJSONValue(value) {
+  return value === void 0 ? null : value;
 }
 
 // src/generate-text/to-response-messages.ts
@@ -5341,7 +5344,7 @@ async function embed({
     }),
     tracer,
     fn: async (span) => {
-      const { embedding, usage, response } = await retry(
+      const { embedding, usage, response, providerMetadata } = await retry(
         () => (
           // nested spans to align with the embedMany telemetry data:
           recordSpan({
@@ -5385,6 +5388,7 @@ async function embed({
               return {
                 embedding: embedding2,
                 usage: usage2,
+                providerMetadata: modelResponse.providerMetadata,
                 response: modelResponse.response
               };
             }
@@ -5404,6 +5408,7 @@ async function embed({
         value,
         embedding,
         usage,
+        providerMetadata,
         response
       });
     }
@@ -5414,6 +5419,7 @@ var DefaultEmbedResult = class {
     this.value = options.value;
     this.embedding = options.embedding;
     this.usage = options.usage;
+    this.providerMetadata = options.providerMetadata;
     this.response = options.response;
   }
 };
@@ -5471,58 +5477,64 @@ async function embedMany({
     }),
     tracer,
     fn: async (span) => {
+      var _a16;
       const [maxEmbeddingsPerCall, supportsParallelCalls] = await Promise.all([
         model.maxEmbeddingsPerCall,
         model.supportsParallelCalls
       ]);
       if (maxEmbeddingsPerCall == null || maxEmbeddingsPerCall === Infinity) {
-        const { embeddings: embeddings2, usage, response } = await retry(() => {
-          return recordSpan({
-            name: "ai.embedMany.doEmbed",
-            attributes: selectTelemetryAttributes({
-              telemetry,
-              attributes: {
-                ...assembleOperationName({
-                  operationId: "ai.embedMany.doEmbed",
-                  telemetry
-                }),
-                ...baseTelemetryAttributes,
-                // specific settings that only make sense on the outer level:
-                "ai.values": {
-                  input: () => values.map((value) => JSON.stringify(value))
+        const { embeddings: embeddings2, usage, response, providerMetadata: providerMetadata2 } = await retry(
+          () => {
+            return recordSpan({
+              name: "ai.embedMany.doEmbed",
+              attributes: selectTelemetryAttributes({
+                telemetry,
+                attributes: {
+                  ...assembleOperationName({
+                    operationId: "ai.embedMany.doEmbed",
+                    telemetry
+                  }),
+                  ...baseTelemetryAttributes,
+                  // specific settings that only make sense on the outer level:
+                  "ai.values": {
+                    input: () => values.map((value) => JSON.stringify(value))
+                  }
                 }
+              }),
+              tracer,
+              fn: async (doEmbedSpan) => {
+                var _a17;
+                const modelResponse = await model.doEmbed({
+                  values,
+                  abortSignal,
+                  headers,
+                  providerOptions
+                });
+                const embeddings3 = modelResponse.embeddings;
+                const usage2 = (_a17 = modelResponse.usage) != null ? _a17 : { tokens: NaN };
+                doEmbedSpan.setAttributes(
+                  selectTelemetryAttributes({
+                    telemetry,
+                    attributes: {
+                      "ai.embeddings": {
+                        output: () => embeddings3.map(
+                          (embedding) => JSON.stringify(embedding)
+                        )
+                      },
+                      "ai.usage.tokens": usage2.tokens
+                    }
+                  })
+                );
+                return {
+                  embeddings: embeddings3,
+                  usage: usage2,
+                  providerMetadata: modelResponse.providerMetadata,
+                  response: modelResponse.response
+                };
               }
-            }),
-            tracer,
-            fn: async (doEmbedSpan) => {
-              var _a16;
-              const modelResponse = await model.doEmbed({
-                values,
-                abortSignal,
-                headers,
-                providerOptions
-              });
-              const embeddings3 = modelResponse.embeddings;
-              const usage2 = (_a16 = modelResponse.usage) != null ? _a16 : { tokens: NaN };
-              doEmbedSpan.setAttributes(
-                selectTelemetryAttributes({
-                  telemetry,
-                  attributes: {
-                    "ai.embeddings": {
-                      output: () => embeddings3.map((embedding) => JSON.stringify(embedding))
-                    },
-                    "ai.usage.tokens": usage2.tokens
-                  }
-                })
-              );
-              return {
-                embeddings: embeddings3,
-                usage: usage2,
-                response: modelResponse.response
-              };
-            }
-          });
-        });
+            });
+          }
+        );
         span.setAttributes(
           selectTelemetryAttributes({
             telemetry,
@@ -5538,6 +5550,7 @@ async function embedMany({
           values,
           embeddings: embeddings2,
           usage,
+          providerMetadata: providerMetadata2,
           responses: [response]
         });
       }
@@ -5545,6 +5558,7 @@ async function embedMany({
       const embeddings = [];
       const responses = [];
       let tokens = 0;
+      let providerMetadata;
       const parallelChunks = splitArray(
         valueChunks,
         supportsParallelCalls ? maxParallelCalls : 1
@@ -5571,7 +5585,7 @@ async function embedMany({
                 }),
                 tracer,
                 fn: async (doEmbedSpan) => {
-                  var _a16;
+                  var _a17;
                   const modelResponse = await model.doEmbed({
                     values: chunk,
                     abortSignal,
@@ -5579,7 +5593,7 @@ async function embedMany({
                     providerOptions
                   });
                   const embeddings2 = modelResponse.embeddings;
-                  const usage = (_a16 = modelResponse.usage) != null ? _a16 : { tokens: NaN };
+                  const usage = (_a17 = modelResponse.usage) != null ? _a17 : { tokens: NaN };
                   doEmbedSpan.setAttributes(
                     selectTelemetryAttributes({
                       telemetry,
@@ -5596,6 +5610,7 @@ async function embedMany({
                   return {
                     embeddings: embeddings2,
                     usage,
+                    providerMetadata: modelResponse.providerMetadata,
                     response: modelResponse.response
                   };
                 }
@@ -5607,6 +5622,20 @@ async function embedMany({
           embeddings.push(...result.embeddings);
           responses.push(result.response);
           tokens += result.usage.tokens;
+          if (result.providerMetadata) {
+            if (!providerMetadata) {
+              providerMetadata = { ...result.providerMetadata };
+            } else {
+              for (const [providerName, metadata] of Object.entries(
+                result.providerMetadata
+              )) {
+                providerMetadata[providerName] = {
+                  ...(_a16 = providerMetadata[providerName]) != null ? _a16 : {},
+                  ...metadata
+                };
+              }
+            }
+          }
         }
       }
       span.setAttributes(
@@ -5624,6 +5653,7 @@ async function embedMany({
         values,
         embeddings,
         usage: { tokens },
+        providerMetadata,
         responses
       });
     }
@@ -5634,6 +5664,7 @@ var DefaultEmbedManyResult = class {
     this.values = options.values;
     this.embeddings = options.embeddings;
     this.usage = options.usage;
+    this.providerMetadata = options.providerMetadata;
     this.responses = options.responses;
   }
 };