ai 7.0.113 → 7.0.114

package/dist/internal/index.js

@@ -91,7 +91,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "7.0.113" : "0.0.0-test";
+var VERSION = true ? "7.0.114" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -2516,6 +2516,50 @@ function mergeCallbacks(...callbacks) {
 // src/telemetry/create-telemetry-dispatcher.ts
 import { asArray as asArray3 } from "@ai-sdk/provider-utils";
 
+// src/telemetry/filter-included-context.ts
+function filterIncludedContext({
+  context,
+  includeContext
+}) {
+  if (context == null) {
+    return {};
+  }
+  return Object.fromEntries(
+    Object.entries(context).filter(
+      ([key]) => (includeContext == null ? void 0 : includeContext[key]) === true
+    )
+  );
+}
+function filterToolsContext({
+  toolsContext,
+  includeToolsContext
+}) {
+  if (includeToolsContext == null) {
+    return {};
+  }
+  return Object.fromEntries(
+    Object.entries(toolsContext).map(([toolName, toolContext]) => [
+      toolName,
+      filterToolContext({
+        toolName,
+        toolContext,
+        includeToolsContext
+      })
+    ])
+  );
+}
+function filterToolContext({
+  toolName,
+  toolContext,
+  includeToolsContext
+}) {
+  const includeToolContext = includeToolsContext == null ? void 0 : includeToolsContext[toolName];
+  return filterIncludedContext({
+    context: toolContext,
+    includeContext: includeToolContext
+  });
+}
+
 // src/telemetry/tracing-channel.ts
 var AI_SDK_TELEMETRY_TRACING_CHANNEL = "ai:telemetry";
 
@@ -2666,12 +2710,37 @@ function getGlobalTelemetryIntegrations() {
 }
 
 // src/telemetry/create-telemetry-dispatcher.ts
-function augmentEvent(event, telemetry) {
-  return Object.assign(
+function augmentEvent(event, telemetry, filterContext = false) {
+  const augmentedEvent = Object.assign(
     Object.create(Object.getPrototypeOf(event)),
     event,
-    telemetry
+    {
+      recordInputs: telemetry.recordInputs,
+      recordOutputs: telemetry.recordOutputs,
+      functionId: telemetry.functionId
+    }
   );
+  if (filterContext && event != null && typeof event === "object" && "runtimeContext" in event) {
+    augmentedEvent.runtimeContext = filterIncludedContext({
+      context: event.runtimeContext,
+      includeContext: telemetry.includeRuntimeContext
+    });
+  }
+  if (filterContext && event != null && typeof event === "object") {
+    if ("toolsContext" in event) {
+      augmentedEvent.toolsContext = filterToolsContext({
+        toolsContext: event.toolsContext,
+        includeToolsContext: telemetry.includeToolsContext
+      });
+    } else if ("toolContext" in event && event.toolContext != null && "toolCall" in event && event.toolCall != null && typeof event.toolCall === "object" && "toolName" in event.toolCall) {
+      augmentedEvent.toolContext = filterToolContext({
+        toolName: event.toolCall.toolName,
+        toolContext: event.toolContext,
+        includeToolsContext: telemetry.includeToolsContext
+      });
+    }
+  }
+  return augmentedEvent;
 }
 function createTelemetryDispatcher({
   telemetry
@@ -2684,7 +2753,9 @@ function createTelemetryDispatcher({
   const telemetryMetadata = {
     recordInputs: telemetry == null ? void 0 : telemetry.recordInputs,
     recordOutputs: telemetry == null ? void 0 : telemetry.recordOutputs,
-    functionId: telemetry == null ? void 0 : telemetry.functionId
+    functionId: telemetry == null ? void 0 : telemetry.functionId,
+    includeRuntimeContext: telemetry == null ? void 0 : telemetry.includeRuntimeContext,
+    includeToolsContext: telemetry == null ? void 0 : telemetry.includeToolsContext
   };
   const mergeTelemetryCallback = (key) => {
     const integrationCallbacks = integrations.map((integration) => {
@@ -2710,14 +2781,14 @@ function createTelemetryDispatcher({
     runInTracingChannelSpan: async ({ type, event, execute }) => await runWithTracingChannelSpan(
       {
         type,
-        event: augmentEvent(event, telemetryMetadata)
+        event: augmentEvent(event, telemetryMetadata, true)
       },
       execute
     ),
     startTracingChannelContext: ({ type, event, completion }) => openTelemetryChannelSpanContext({
       message: {
         type,
-        event: augmentEvent(event, telemetryMetadata)
+        event: augmentEvent(event, telemetryMetadata, true)
       },
       completion
     }),
@@ -2792,21 +2863,6 @@ function createTelemetryDispatcher({
   };
 }
 
-// src/telemetry/filter-included-context.ts
-function filterIncludedContext({
-  context,
-  includeContext
-}) {
-  if (context == null) {
-    return {};
-  }
-  return Object.fromEntries(
-    Object.entries(context).filter(
-      ([key]) => (includeContext == null ? void 0 : includeContext[key]) === true
-    )
-  );
-}
-
 // src/generate-text/reasoning.ts
 function asReasoningText(reasoningParts) {
   const reasoningText = reasoningParts.map((part) => "text" in part ? part.text : "").join("");
@@ -2943,35 +2999,6 @@ function restrictStepResult({
     providerMetadata: step.providerMetadata
   });
 }
-function filterToolsContext({
-  toolsContext,
-  includeToolsContext
-}) {
-  if (includeToolsContext == null) {
-    return {};
-  }
-  return Object.fromEntries(
-    Object.entries(toolsContext).map(([toolName, toolContext]) => [
-      toolName,
-      filterToolContext({
-        toolName,
-        toolContext,
-        includeToolsContext
-      })
-    ])
-  );
-}
-function filterToolContext({
-  toolName,
-  toolContext,
-  includeToolsContext
-}) {
-  const includeToolContext = includeToolsContext == null ? void 0 : includeToolsContext[toolName];
-  return filterIncludedContext({
-    context: toolContext,
-    includeContext: includeToolContext
-  });
-}
 function createRestrictedTelemetryDispatcher({
   telemetry,
   includeRuntimeContext,