ai 7.0.22 → 7.0.23

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # ai
 
+## 7.0.23
+
+### Patch Changes
+
+- 930f949: feat(ai): wrap embedMany in tracing channel context
+- Updated dependencies [867f80a]
+  - @ai-sdk/gateway@4.0.17
+
 ## 7.0.22
 
 ### Patch Changes

package/dist/index.d.ts

@@ -4274,7 +4274,7 @@ type RerankingModelCallEndEvent = {
 };
 
 declare const AI_SDK_TELEMETRY_TRACING_CHANNEL = "ai:telemetry";
-type TelemetryTracingEventType = 'generateText' | 'streamText' | 'step' | 'languageModelCall' | 'executeTool' | 'embed' | 'rerank';
+type TelemetryTracingEventType = 'generateText' | 'streamText' | 'step' | 'languageModelCall' | 'executeTool' | 'embed' | 'embedMany' | 'rerank';
 type TelemetryTracingChannelMessage<EVENT = unknown> = {
     readonly type: TelemetryTracingEventType;
     readonly event: EVENT;

package/dist/index.js

@@ -1089,7 +1089,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "7.0.22" : "0.0.0-test";
+var VERSION = true ? "7.0.23" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -11301,7 +11301,7 @@ async function embedMany({
   experimental_onEnd,
   _internal: { generateCallId = originalGenerateCallId5 } = {}
 }) {
-  var _a22, _b;
+  var _a22;
   const model = resolveEmbeddingModel(modelArg);
   const { maxRetries, retry } = prepareRetries({
     maxRetries: maxRetriesArg,
@@ -11317,112 +11317,33 @@ async function embedMany({
   const telemetryDispatcher = createTelemetryDispatcher({
     telemetry
   });
-  await notify({
-    event: {
-      callId,
-      operationId: "ai.embedMany",
-      provider: model.provider,
-      modelId: model.modelId,
-      value: values,
-      maxRetries,
-      headers: headersWithUserAgent,
-      providerOptions
-    },
-    callbacks: [resolvedOnStart, telemetryDispatcher.onStart]
-  });
-  try {
-    const [maxEmbeddingsPerCall, supportsParallelCalls] = await Promise.all([
-      model.maxEmbeddingsPerCall,
-      model.supportsParallelCalls
-    ]);
-    if (maxEmbeddingsPerCall == null || maxEmbeddingsPerCall === Infinity) {
-      const { embeddings: embeddings2, usage, warnings: warnings2, response, providerMetadata: providerMetadata2 } = await retry(async () => {
-        var _a23, _b2;
-        const embedCallId = generateCallId();
-        await notify({
-          event: {
-            callId,
-            embedCallId,
-            operationId: "ai.embedMany.doEmbed",
-            provider: model.provider,
-            modelId: model.modelId,
-            values
-          },
-          callbacks: [telemetryDispatcher.onEmbedStart]
-        });
-        const modelResponse = await model.doEmbed({
-          values,
-          abortSignal,
-          headers: headersWithUserAgent,
-          providerOptions
-        });
-        const embeddings3 = modelResponse.embeddings;
-        const usage2 = (_a23 = modelResponse.usage) != null ? _a23 : { tokens: NaN };
-        await notify({
-          event: {
-            callId,
-            embedCallId,
-            operationId: "ai.embedMany.doEmbed",
-            provider: model.provider,
-            modelId: model.modelId,
-            values,
-            embeddings: embeddings3,
-            usage: usage2
-          },
-          callbacks: [telemetryDispatcher.onEmbedEnd]
-        });
-        return {
-          embeddings: embeddings3,
-          usage: usage2,
-          warnings: (_b2 = modelResponse.warnings) != null ? _b2 : [],
-          providerMetadata: modelResponse.providerMetadata,
-          response: modelResponse.response
-        };
-      });
-      logWarnings({
-        warnings: warnings2,
-        provider: model.provider,
-        model: model.modelId
-      });
+  const runInTracingChannelSpan = (_a22 = telemetryDispatcher.runInTracingChannelSpan) != null ? _a22 : async ({ execute }) => await execute();
+  const startEvent = {
+    callId,
+    operationId: "ai.embedMany",
+    provider: model.provider,
+    modelId: model.modelId,
+    value: values,
+    maxRetries,
+    headers: headersWithUserAgent,
+    providerOptions
+  };
+  return await runInTracingChannelSpan({
+    type: "embedMany",
+    event: startEvent,
+    execute: async () => {
+      var _a23, _b;
       await notify({
-        event: {
-          callId,
-          operationId: "ai.embedMany",
-          provider: model.provider,
-          modelId: model.modelId,
-          value: values,
-          embedding: embeddings2,
-          usage,
-          warnings: warnings2,
-          providerMetadata: providerMetadata2,
-          response: [response]
-        },
-        callbacks: [resolvedOnEnd, telemetryDispatcher.onEnd]
-      });
-      return new DefaultEmbedManyResult({
-        values,
-        embeddings: embeddings2,
-        usage,
-        warnings: warnings2,
-        providerMetadata: providerMetadata2,
-        responses: [response]
+        event: startEvent,
+        callbacks: [resolvedOnStart, telemetryDispatcher.onStart]
       });
-    }
-    const valueChunks = splitArray(values, maxEmbeddingsPerCall);
-    const embeddings = [];
-    const warnings = [];
-    const responses = [];
-    let tokens = 0;
-    let providerMetadata;
-    const parallelChunks = splitArray(
-      valueChunks,
-      supportsParallelCalls ? maxParallelCalls : 1
-    );
-    for (const parallelChunk of parallelChunks) {
-      const results = await Promise.all(
-        parallelChunk.map((chunk) => {
-          return retry(async () => {
-            var _a23, _b2;
+      try {
+        const [maxEmbeddingsPerCall, supportsParallelCalls] = await Promise.all(
+          [model.maxEmbeddingsPerCall, model.supportsParallelCalls]
+        );
+        if (maxEmbeddingsPerCall == null || maxEmbeddingsPerCall === Infinity) {
+          const { embeddings: embeddings2, usage, warnings: warnings2, response, providerMetadata: providerMetadata2 } = await retry(async () => {
+            var _a24, _b2;
             const embedCallId = generateCallId();
             await notify({
               event: {
@@ -11431,18 +11352,18 @@ async function embedMany({
                 operationId: "ai.embedMany.doEmbed",
                 provider: model.provider,
                 modelId: model.modelId,
-                values: chunk
+                values
               },
               callbacks: [telemetryDispatcher.onEmbedStart]
             });
             const modelResponse = await model.doEmbed({
-              values: chunk,
+              values,
               abortSignal,
               headers: headersWithUserAgent,
               providerOptions
             });
-            const chunkEmbeddings = modelResponse.embeddings;
-            const usage = (_a23 = modelResponse.usage) != null ? _a23 : { tokens: NaN };
+            const embeddings3 = modelResponse.embeddings;
+            const usage2 = (_a24 = modelResponse.usage) != null ? _a24 : { tokens: NaN };
             await notify({
               event: {
                 callId,
@@ -11450,75 +11371,162 @@ async function embedMany({
                 operationId: "ai.embedMany.doEmbed",
                 provider: model.provider,
                 modelId: model.modelId,
-                values: chunk,
-                embeddings: chunkEmbeddings,
-                usage
+                values,
+                embeddings: embeddings3,
+                usage: usage2
               },
               callbacks: [telemetryDispatcher.onEmbedEnd]
             });
             return {
-              embeddings: chunkEmbeddings,
-              usage,
+              embeddings: embeddings3,
+              usage: usage2,
               warnings: (_b2 = modelResponse.warnings) != null ? _b2 : [],
               providerMetadata: modelResponse.providerMetadata,
               response: modelResponse.response
             };
           });
-        })
-      );
-      for (const result of results) {
-        embeddings.push(...result.embeddings);
-        warnings.push(...result.warnings);
-        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] = {
-                ...(_a22 = providerMetadata[providerName]) != null ? _a22 : {},
-                ...metadata
-              };
+          logWarnings({
+            warnings: warnings2,
+            provider: model.provider,
+            model: model.modelId
+          });
+          await notify({
+            event: {
+              callId,
+              operationId: "ai.embedMany",
+              provider: model.provider,
+              modelId: model.modelId,
+              value: values,
+              embedding: embeddings2,
+              usage,
+              warnings: warnings2,
+              providerMetadata: providerMetadata2,
+              response: [response]
+            },
+            callbacks: [resolvedOnEnd, telemetryDispatcher.onEnd]
+          });
+          return new DefaultEmbedManyResult({
+            values,
+            embeddings: embeddings2,
+            usage,
+            warnings: warnings2,
+            providerMetadata: providerMetadata2,
+            responses: [response]
+          });
+        }
+        const valueChunks = splitArray(values, maxEmbeddingsPerCall);
+        const embeddings = [];
+        const warnings = [];
+        const responses = [];
+        let tokens = 0;
+        let providerMetadata;
+        const parallelChunks = splitArray(
+          valueChunks,
+          supportsParallelCalls ? maxParallelCalls : 1
+        );
+        for (const parallelChunk of parallelChunks) {
+          const results = await Promise.all(
+            parallelChunk.map((chunk) => {
+              return retry(async () => {
+                var _a24, _b2;
+                const embedCallId = generateCallId();
+                await notify({
+                  event: {
+                    callId,
+                    embedCallId,
+                    operationId: "ai.embedMany.doEmbed",
+                    provider: model.provider,
+                    modelId: model.modelId,
+                    values: chunk
+                  },
+                  callbacks: [telemetryDispatcher.onEmbedStart]
+                });
+                const modelResponse = await model.doEmbed({
+                  values: chunk,
+                  abortSignal,
+                  headers: headersWithUserAgent,
+                  providerOptions
+                });
+                const chunkEmbeddings = modelResponse.embeddings;
+                const usage = (_a24 = modelResponse.usage) != null ? _a24 : { tokens: NaN };
+                await notify({
+                  event: {
+                    callId,
+                    embedCallId,
+                    operationId: "ai.embedMany.doEmbed",
+                    provider: model.provider,
+                    modelId: model.modelId,
+                    values: chunk,
+                    embeddings: chunkEmbeddings,
+                    usage
+                  },
+                  callbacks: [telemetryDispatcher.onEmbedEnd]
+                });
+                return {
+                  embeddings: chunkEmbeddings,
+                  usage,
+                  warnings: (_b2 = modelResponse.warnings) != null ? _b2 : [],
+                  providerMetadata: modelResponse.providerMetadata,
+                  response: modelResponse.response
+                };
+              });
+            })
+          );
+          for (const result of results) {
+            embeddings.push(...result.embeddings);
+            warnings.push(...result.warnings);
+            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] = {
+                    ...(_a23 = providerMetadata[providerName]) != null ? _a23 : {},
+                    ...metadata
+                  };
+                }
+              }
             }
           }
         }
+        logWarnings({
+          warnings,
+          provider: model.provider,
+          model: model.modelId
+        });
+        await notify({
+          event: {
+            callId,
+            operationId: "ai.embedMany",
+            provider: model.provider,
+            modelId: model.modelId,
+            value: values,
+            embedding: embeddings,
+            usage: { tokens },
+            warnings,
+            providerMetadata,
+            response: responses
+          },
+          callbacks: [resolvedOnEnd, telemetryDispatcher.onEnd]
+        });
+        return new DefaultEmbedManyResult({
+          values,
+          embeddings,
+          usage: { tokens },
+          warnings,
+          providerMetadata,
+          responses
+        });
+      } catch (error) {
+        await ((_b = telemetryDispatcher.onError) == null ? void 0 : _b.call(telemetryDispatcher, { callId, error }));
+        throw error;
       }
     }
-    logWarnings({
-      warnings,
-      provider: model.provider,
-      model: model.modelId
-    });
-    await notify({
-      event: {
-        callId,
-        operationId: "ai.embedMany",
-        provider: model.provider,
-        modelId: model.modelId,
-        value: values,
-        embedding: embeddings,
-        usage: { tokens },
-        warnings,
-        providerMetadata,
-        response: responses
-      },
-      callbacks: [resolvedOnEnd, telemetryDispatcher.onEnd]
-    });
-    return new DefaultEmbedManyResult({
-      values,
-      embeddings,
-      usage: { tokens },
-      warnings,
-      providerMetadata,
-      responses
-    });
-  } catch (error) {
-    await ((_b = telemetryDispatcher.onError) == null ? void 0 : _b.call(telemetryDispatcher, { callId, error }));
-    throw error;
-  }
+  });
 }
 var DefaultEmbedManyResult = class {
   constructor(options) {