ai 6.0.110 → 6.0.111

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # ai
 
+## 6.0.111
+
+### Patch Changes
+
+- 2129c82: feat(ai): register global telemetry integrations
+
 ## 6.0.110
 
 ### Patch Changes

package/dist/index.d.mts

@@ -1079,17 +1079,22 @@ type OnFinishEvent<TOOLS extends ToolSet = ToolSet> = StepResult<TOOLS> & {
     readonly metadata: Record<string, unknown> | undefined;
 };
 
+/**
+ * A callback function that can be used to notify listeners.
+ */
+type Listener<EVENT> = (event: EVENT) => PromiseLike<void> | void;
+
 /**
  * Implement this interface to create custom telemetry integrations.
  * Methods can be sync or return a PromiseLike.
  */
-interface TelemetryIntegration<TOOLS extends ToolSet = ToolSet, OUTPUT extends Output = Output> {
-    onStart?(event: OnStartEvent<TOOLS, OUTPUT>): PromiseLike<void> | void;
-    onStepStart?(event: OnStepStartEvent<TOOLS, OUTPUT>): PromiseLike<void> | void;
-    onToolCallStart?(event: OnToolCallStartEvent<TOOLS>): PromiseLike<void> | void;
-    onToolCallFinish?(event: OnToolCallFinishEvent<TOOLS>): PromiseLike<void> | void;
-    onStepFinish?(event: OnStepFinishEvent<TOOLS>): PromiseLike<void> | void;
-    onFinish?(event: OnFinishEvent<TOOLS>): PromiseLike<void> | void;
+interface TelemetryIntegration {
+    onStart?: Listener<OnStartEvent<ToolSet, Output>>;
+    onStepStart?: Listener<OnStepStartEvent<ToolSet, Output>>;
+    onToolCallStart?: Listener<OnToolCallStartEvent<ToolSet>>;
+    onToolCallFinish?: Listener<OnToolCallFinishEvent<ToolSet>>;
+    onStepFinish?: Listener<OnStepFinishEvent<ToolSet>>;
+    onFinish?: Listener<OnFinishEvent<ToolSet>>;
 }
 
 /**

package/dist/index.d.ts

@@ -1079,17 +1079,22 @@ type OnFinishEvent<TOOLS extends ToolSet = ToolSet> = StepResult<TOOLS> & {
     readonly metadata: Record<string, unknown> | undefined;
 };
 
+/**
+ * A callback function that can be used to notify listeners.
+ */
+type Listener<EVENT> = (event: EVENT) => PromiseLike<void> | void;
+
 /**
  * Implement this interface to create custom telemetry integrations.
  * Methods can be sync or return a PromiseLike.
  */
-interface TelemetryIntegration<TOOLS extends ToolSet = ToolSet, OUTPUT extends Output = Output> {
-    onStart?(event: OnStartEvent<TOOLS, OUTPUT>): PromiseLike<void> | void;
-    onStepStart?(event: OnStepStartEvent<TOOLS, OUTPUT>): PromiseLike<void> | void;
-    onToolCallStart?(event: OnToolCallStartEvent<TOOLS>): PromiseLike<void> | void;
-    onToolCallFinish?(event: OnToolCallFinishEvent<TOOLS>): PromiseLike<void> | void;
-    onStepFinish?(event: OnStepFinishEvent<TOOLS>): PromiseLike<void> | void;
-    onFinish?(event: OnFinishEvent<TOOLS>): PromiseLike<void> | void;
+interface TelemetryIntegration {
+    onStart?: Listener<OnStartEvent<ToolSet, Output>>;
+    onStepStart?: Listener<OnStepStartEvent<ToolSet, Output>>;
+    onToolCallStart?: Listener<OnToolCallStartEvent<ToolSet>>;
+    onToolCallFinish?: Listener<OnToolCallFinishEvent<ToolSet>>;
+    onStepFinish?: Listener<OnStepFinishEvent<ToolSet>>;
+    onFinish?: Listener<OnFinishEvent<ToolSet>>;
 }
 
 /**

package/dist/index.js

@@ -633,6 +633,8 @@ function asArray(value) {
 // src/util/notify.ts
 async function notify(options) {
   for (const callback of asArray(options.callbacks)) {
+    if (callback == null)
+      continue;
     try {
       await callback(options.event);
     } catch (_ignored) {
@@ -1228,7 +1230,7 @@ var import_provider_utils3 = require("@ai-sdk/provider-utils");
 var import_provider_utils4 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "6.0.110" : "0.0.0-test";
+var VERSION = true ? "6.0.111" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -2434,6 +2436,65 @@ function stringifyForTelemetry(prompt) {
   );
 }
 
+// src/telemetry/telemetry-integration-registry.ts
+function registerTelemetryIntegration(integration) {
+  if (!globalThis.AI_SDK_TELEMETRY_INTEGRATIONS) {
+    globalThis.AI_SDK_TELEMETRY_INTEGRATIONS = [];
+  }
+  globalThis.AI_SDK_TELEMETRY_INTEGRATIONS.push(integration);
+}
+function getGlobalTelemetryIntegrations() {
+  var _a21;
+  return (_a21 = globalThis.AI_SDK_TELEMETRY_INTEGRATIONS) != null ? _a21 : [];
+}
+
+// src/telemetry/get-global-telemetry-integration.ts
+function bindTelemetryIntegration(integration) {
+  var _a21, _b, _c, _d, _e, _f;
+  return {
+    onStart: (_a21 = integration.onStart) == null ? void 0 : _a21.bind(integration),
+    onStepStart: (_b = integration.onStepStart) == null ? void 0 : _b.bind(integration),
+    onToolCallStart: (_c = integration.onToolCallStart) == null ? void 0 : _c.bind(integration),
+    onToolCallFinish: (_d = integration.onToolCallFinish) == null ? void 0 : _d.bind(integration),
+    onStepFinish: (_e = integration.onStepFinish) == null ? void 0 : _e.bind(integration),
+    onFinish: (_f = integration.onFinish) == null ? void 0 : _f.bind(integration)
+  };
+}
+function getGlobalTelemetryIntegration() {
+  const globalIntegrations = getGlobalTelemetryIntegrations();
+  return (integrations) => {
+    const localIntegrations = asArray(integrations);
+    const allIntegrations = [...globalIntegrations, ...localIntegrations];
+    function createTelemetryComposite(getListenerFromIntegration) {
+      const listeners = allIntegrations.map(getListenerFromIntegration).filter(Boolean);
+      return async (event) => {
+        for (const listener of listeners) {
+          try {
+            await listener(event);
+          } catch (_ignored) {
+          }
+        }
+      };
+    }
+    return {
+      onStart: createTelemetryComposite((integration) => integration.onStart),
+      onStepStart: createTelemetryComposite(
+        (integration) => integration.onStepStart
+      ),
+      onToolCallStart: createTelemetryComposite(
+        (integration) => integration.onToolCallStart
+      ),
+      onToolCallFinish: createTelemetryComposite(
+        (integration) => integration.onToolCallFinish
+      ),
+      onStepFinish: createTelemetryComposite(
+        (integration) => integration.onStepFinish
+      ),
+      onFinish: createTelemetryComposite((integration) => integration.onFinish)
+    };
+  };
+}
+
 // src/types/usage.ts
 function asLanguageModelUsage(usage) {
   return {
@@ -4052,6 +4113,7 @@ async function generateText({
   ...settings
 }) {
   const model = resolveLanguageModel(modelArg);
+  const createGlobalTelemetry = getGlobalTelemetryIntegration();
   const stopConditions = asArray(stopWhen);
   const totalTimeoutMs = getTotalTimeoutMs(timeout);
   const stepTimeoutMs = getStepTimeoutMs(timeout);
@@ -4082,6 +4144,7 @@ async function generateText({
     prompt,
     messages
   });
+  const globalTelemetry = createGlobalTelemetry(telemetry == null ? void 0 : telemetry.integrations);
   await notify({
     event: {
       model: modelInfo,
@@ -4111,7 +4174,10 @@ async function generateText({
       metadata: telemetry == null ? void 0 : telemetry.metadata,
       experimental_context
     },
-    callbacks: onStart
+    callbacks: [
+      onStart,
+      globalTelemetry.onStart
+    ]
   });
   const tracer = getTracer(telemetry);
   try {
@@ -4156,8 +4222,14 @@ async function generateText({
             experimental_context,
             stepNumber: 0,
             model: modelInfo,
-            onToolCallStart,
-            onToolCallFinish
+            onToolCallStart: [
+              onToolCallStart,
+              globalTelemetry.onToolCallStart
+            ],
+            onToolCallFinish: [
+              onToolCallFinish,
+              globalTelemetry.onToolCallFinish
+            ]
           });
           const toolContent = [];
           for (const output2 of toolOutputs) {
@@ -4283,7 +4355,10 @@ async function generateText({
                 metadata: telemetry == null ? void 0 : telemetry.metadata,
                 experimental_context
               },
-              callbacks: onStepStart
+              callbacks: [
+                onStepStart,
+                globalTelemetry.onStepStart
+              ]
             });
             currentModelResponse = await retry(
               () => {
@@ -4461,8 +4536,14 @@ async function generateText({
                   experimental_context,
                   stepNumber: steps.length,
                   model: stepModelInfo,
-                  onToolCallStart,
-                  onToolCallFinish
+                  onToolCallStart: [
+                    onToolCallStart,
+                    globalTelemetry.onToolCallStart
+                  ],
+                  onToolCallFinish: [
+                    onToolCallFinish,
+                    globalTelemetry.onToolCallFinish
+                  ]
                 })
               );
             }
@@ -4529,7 +4610,10 @@ async function generateText({
               model: stepModelInfo.modelId
             });
             steps.push(currentStepResult);
-            await notify({ event: currentStepResult, callbacks: onStepFinish });
+            await notify({
+              event: currentStepResult,
+              callbacks: [onStepFinish, globalTelemetry.onStepFinish]
+            });
           } finally {
             if (stepTimeoutId != null) {
               clearTimeout(stepTimeoutId);
@@ -4610,7 +4694,10 @@ async function generateText({
             steps,
             totalUsage
           },
-          callbacks: onFinish
+          callbacks: [
+            onFinish,
+            globalTelemetry.onFinish
+          ]
         });
         let resolvedOutput;
         if (lastStep.finishReason === "stop") {
@@ -6501,6 +6588,8 @@ var DefaultStreamTextResult = class {
     this.outputSpecification = output;
     this.includeRawChunks = includeRawChunks;
     this.tools = tools;
+    const createGlobalTelemetry = getGlobalTelemetryIntegration();
+    const globalTelemetry = createGlobalTelemetry(telemetry == null ? void 0 : telemetry.integrations);
     let stepFinish;
     let recordedContent = [];
     const recordedResponseMessages = [];
@@ -6648,7 +6737,10 @@ var DefaultStreamTextResult = class {
             },
             providerMetadata: part.providerMetadata
           });
-          await notify({ event: currentStepResult, callbacks: onStepFinish });
+          await notify({
+            event: currentStepResult,
+            callbacks: [onStepFinish, globalTelemetry.onStepFinish]
+          });
           logWarnings({
             warnings: recordedWarnings,
             provider: modelInfo.provider,
@@ -6712,7 +6804,10 @@ var DefaultStreamTextResult = class {
               providerMetadata: finalStep.providerMetadata,
               steps: recordedSteps
             },
-            callbacks: onFinish
+            callbacks: [
+              onFinish,
+              globalTelemetry.onFinish
+            ]
           });
           rootSpan.setAttributes(
             await selectTelemetryAttributes({
@@ -6869,7 +6964,10 @@ var DefaultStreamTextResult = class {
             ...callbackTelemetryProps,
             experimental_context
           },
-          callbacks: onStart
+          callbacks: [
+            onStart,
+            globalTelemetry.onStart
+          ]
         });
         const initialMessages = initialPrompt.messages;
         const initialResponseMessages = [];
@@ -6919,8 +7017,14 @@ var DefaultStreamTextResult = class {
                   experimental_context,
                   stepNumber: recordedSteps.length,
                   model: modelInfo,
-                  onToolCallStart,
-                  onToolCallFinish,
+                  onToolCallStart: [
+                    onToolCallStart,
+                    globalTelemetry.onToolCallStart
+                  ],
+                  onToolCallFinish: [
+                    onToolCallFinish,
+                    globalTelemetry.onToolCallFinish
+                  ],
                   onPreliminaryToolResult: (result2) => {
                     toolExecutionStepStreamController == null ? void 0 : toolExecutionStepStreamController.enqueue(result2);
                   }
@@ -7071,7 +7175,10 @@ var DefaultStreamTextResult = class {
                 ...callbackTelemetryProps,
                 experimental_context
               },
-              callbacks: onStepStart
+              callbacks: [
+                onStepStart,
+                globalTelemetry.onStepStart
+              ]
             });
             const {
               result: { stream: stream2, response, request },
@@ -7147,8 +7254,14 @@ var DefaultStreamTextResult = class {
               generateId: generateId2,
               stepNumber: recordedSteps.length,
               model: stepModelInfo,
-              onToolCallStart,
-              onToolCallFinish
+              onToolCallStart: [
+                onToolCallStart,
+                globalTelemetry.onToolCallStart
+              ],
+              onToolCallFinish: [
+                onToolCallFinish,
+                globalTelemetry.onToolCallFinish
+              ]
             });
             const stepRequest = ((_i = include == null ? void 0 : include.requestBody) != null ? _i : true) ? request != null ? request : {} : { ...request, body: void 0 };
             const stepToolCalls = [];
@@ -13292,27 +13405,6 @@ var TextStreamChatTransport = class extends HttpChatTransport {
     });
   }
 };
-
-// src/telemetry/telemetry-integration-registry.ts
-function registerTelemetryIntegration(integration) {
-  if (!globalThis.AI_SDK_TELEMETRY_INTEGRATIONS) {
-    globalThis.AI_SDK_TELEMETRY_INTEGRATIONS = [];
-  }
-  globalThis.AI_SDK_TELEMETRY_INTEGRATIONS.push(integration);
-}
-
-// src/telemetry/get-global-telemetry-integration.ts
-function bindTelemetryIntegration(integration) {
-  var _a21, _b, _c, _d, _e, _f;
-  return {
-    onStart: (_a21 = integration.onStart) == null ? void 0 : _a21.bind(integration),
-    onStepStart: (_b = integration.onStepStart) == null ? void 0 : _b.bind(integration),
-    onToolCallStart: (_c = integration.onToolCallStart) == null ? void 0 : _c.bind(integration),
-    onToolCallFinish: (_d = integration.onToolCallFinish) == null ? void 0 : _d.bind(integration),
-    onStepFinish: (_e = integration.onStepFinish) == null ? void 0 : _e.bind(integration),
-    onFinish: (_f = integration.onFinish) == null ? void 0 : _f.bind(integration)
-  };
-}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AISDKError,