langsmith 0.7.0 → 0.7.2

package/dist/sandbox/client.cjs

@@ -89,6 +89,12 @@ class SandboxClient {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "_defaultHeaders", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "_fetchImpl", {
             enumerable: true,
             configurable: true,
@@ -103,6 +109,7 @@ class SandboxClient {
         });
         this._baseUrl = (config.apiEndpoint ?? getDefaultApiEndpoint()).replace(/\/$/, "");
         this._apiKey = config.apiKey ?? getDefaultApiKey();
+        this._defaultHeaders = { ...(config.headers ?? {}) };
         this._fetchImpl = (0, fetch_js_1._getFetchImplementation)();
         this._caller = new async_caller_js_1.AsyncCaller({
             maxRetries: config.maxRetries ?? 3,
@@ -122,6 +129,11 @@ class SandboxClient {
         if (this._apiKey) {
             headers.set("X-Api-Key", this._apiKey);
         }
+        for (const [name, value] of Object.entries(this._defaultHeaders)) {
+            if (!headers.has(name)) {
+                headers.set(name, value);
+            }
+        }
         return this._caller.call(() => this._fetchImpl(url, {
             ...init,
             headers,
@@ -134,6 +146,15 @@ class SandboxClient {
     getApiKey() {
         return this._apiKey;
     }
+    /**
+     * Get the constructor-supplied default headers. Used by the WebSocket exec
+     * path so headers like `X-Service-Key` set on the client are attached to
+     * the WS upgrade request.
+     * @internal
+     */
+    getDefaultHeaders() {
+        return { ...this._defaultHeaders };
+    }
     /**
      * JSON POST helper. Sends JSON body, checks response status,
      * and returns the Response for further processing.

package/dist/sandbox/client.d.ts

@@ -34,6 +34,7 @@ import { Sandbox } from "./sandbox.js";
 export declare class SandboxClient {
     private _baseUrl;
     private _apiKey?;
+    private _defaultHeaders;
     private _fetchImpl;
     private _caller;
     constructor(config?: SandboxClientConfig);

package/dist/sandbox/client.js

@@ -86,6 +86,12 @@ export class SandboxClient {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "_defaultHeaders", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "_fetchImpl", {
             enumerable: true,
             configurable: true,
@@ -100,6 +106,7 @@ export class SandboxClient {
         });
         this._baseUrl = (config.apiEndpoint ?? getDefaultApiEndpoint()).replace(/\/$/, "");
         this._apiKey = config.apiKey ?? getDefaultApiKey();
+        this._defaultHeaders = { ...(config.headers ?? {}) };
         this._fetchImpl = _getFetchImplementation();
         this._caller = new AsyncCaller({
             maxRetries: config.maxRetries ?? 3,
@@ -119,6 +126,11 @@ export class SandboxClient {
         if (this._apiKey) {
             headers.set("X-Api-Key", this._apiKey);
         }
+        for (const [name, value] of Object.entries(this._defaultHeaders)) {
+            if (!headers.has(name)) {
+                headers.set(name, value);
+            }
+        }
         return this._caller.call(() => this._fetchImpl(url, {
             ...init,
             headers,
@@ -131,6 +143,15 @@ export class SandboxClient {
     getApiKey() {
         return this._apiKey;
     }
+    /**
+     * Get the constructor-supplied default headers. Used by the WebSocket exec
+     * path so headers like `X-Service-Key` set on the client are attached to
+     * the WS upgrade request.
+     * @internal
+     */
+    getDefaultHeaders() {
+        return { ...this._defaultHeaders };
+    }
     /**
      * JSON POST helper. Sends JSON body, checks response status,
      * and returns the Response for further processing.

package/dist/sandbox/sandbox.cjs

@@ -227,6 +227,7 @@ class Sandbox {
     async _runWs(command, options = {}) {
         const { timeout = 60, env, cwd, shell = "/bin/bash", onStdout, onStderr, idleTimeout, killOnDisconnect, ttlSeconds, pty, } = options;
         const dataplaneUrl = this.requireDataplaneUrl();
+        const clientHeaders = this._client.getDefaultHeaders();
         const [stream, control] = await (0, ws_execute_js_1.runWsStream)(dataplaneUrl, this._client.getApiKey(), command, {
             timeout,
             env,
@@ -238,6 +239,9 @@ class Sandbox {
             killOnDisconnect,
             ttlSeconds,
             pty,
+            ...(Object.keys(clientHeaders).length > 0
+                ? { headers: clientHeaders }
+                : {}),
         });
         const handle = new command_handle_js_1.CommandHandle(stream, control, this);
         await handle._ensureStarted();
@@ -292,7 +296,14 @@ class Sandbox {
     async reconnect(commandId, options = {}) {
         const { stdoutOffset = 0, stderrOffset = 0 } = options;
         const dataplaneUrl = this.requireDataplaneUrl();
-        const [stream, control] = await (0, ws_execute_js_1.reconnectWsStream)(dataplaneUrl, this._client.getApiKey(), commandId, { stdoutOffset, stderrOffset });
+        const clientHeaders = this._client.getDefaultHeaders();
+        const [stream, control] = await (0, ws_execute_js_1.reconnectWsStream)(dataplaneUrl, this._client.getApiKey(), commandId, {
+            stdoutOffset,
+            stderrOffset,
+            ...(Object.keys(clientHeaders).length > 0
+                ? { headers: clientHeaders }
+                : {}),
+        });
         return new command_handle_js_1.CommandHandle(stream, control, this, {
             commandId,
             stdoutOffset,

package/dist/sandbox/sandbox.js

@@ -224,6 +224,7 @@ export class Sandbox {
     async _runWs(command, options = {}) {
         const { timeout = 60, env, cwd, shell = "/bin/bash", onStdout, onStderr, idleTimeout, killOnDisconnect, ttlSeconds, pty, } = options;
         const dataplaneUrl = this.requireDataplaneUrl();
+        const clientHeaders = this._client.getDefaultHeaders();
         const [stream, control] = await runWsStream(dataplaneUrl, this._client.getApiKey(), command, {
             timeout,
             env,
@@ -235,6 +236,9 @@ export class Sandbox {
             killOnDisconnect,
             ttlSeconds,
             pty,
+            ...(Object.keys(clientHeaders).length > 0
+                ? { headers: clientHeaders }
+                : {}),
         });
         const handle = new CommandHandle(stream, control, this);
         await handle._ensureStarted();
@@ -289,7 +293,14 @@ export class Sandbox {
     async reconnect(commandId, options = {}) {
         const { stdoutOffset = 0, stderrOffset = 0 } = options;
         const dataplaneUrl = this.requireDataplaneUrl();
-        const [stream, control] = await reconnectWsStream(dataplaneUrl, this._client.getApiKey(), commandId, { stdoutOffset, stderrOffset });
+        const clientHeaders = this._client.getDefaultHeaders();
+        const [stream, control] = await reconnectWsStream(dataplaneUrl, this._client.getApiKey(), commandId, {
+            stdoutOffset,
+            stderrOffset,
+            ...(Object.keys(clientHeaders).length > 0
+                ? { headers: clientHeaders }
+                : {}),
+        });
         return new CommandHandle(stream, control, this, {
             commandId,
             stdoutOffset,

package/dist/sandbox/types.d.ts

@@ -108,6 +108,13 @@ export interface SandboxClientConfig {
      * Defaults to Infinity (no limit).
      */
     maxConcurrency?: number;
+    /**
+     * Optional default headers attached to every request on this client,
+     * including the data-plane `/execute` HTTP endpoint and the `/execute/ws`
+     * WebSocket upgrade. Use this to pass additional auth headers
+     * (e.g. `X-Service-Key`).
+     */
+    headers?: Record<string, string>;
 }
 /**
  * A single chunk of streaming output from command execution.
@@ -163,6 +170,11 @@ export interface WsRunOptions {
     ttlSeconds?: number;
     /** Whether to allocate a PTY. */
     pty?: boolean;
+    /**
+     * Additional headers attached to the WebSocket upgrade request. Merged on
+     * top of any default headers the SandboxClient was constructed with.
+     */
+    headers?: Record<string, string>;
 }
 /**
  * Options for running a command in a sandbox.

package/dist/sandbox/ws_execute.cjs

@@ -38,11 +38,15 @@ function buildWsUrl(dataplaneUrl) {
 /**
  * Build auth headers for the WebSocket upgrade request.
  */
-function buildAuthHeaders(apiKey) {
+function buildAuthHeaders(apiKey, extraHeaders) {
+    const headers = {};
     if (apiKey) {
-        return { "X-Api-Key": apiKey };
+        headers["X-Api-Key"] = apiKey;
     }
-    return {};
+    if (extraHeaders) {
+        Object.assign(headers, extraHeaders);
+    }
+    return headers;
 }
 // =============================================================================
 // Stream Control
@@ -238,9 +242,9 @@ async function* readWsMessages(ws) {
  * data arrives in addition to yielding the messages.
  */
 async function runWsStream(dataplaneUrl, apiKey, command, options = {}) {
-    const { timeout = 60, env, cwd, shell = "/bin/bash", onStdout, onStderr, commandId, idleTimeout = 300, killOnDisconnect = false, ttlSeconds = 600, pty, } = options;
+    const { timeout = 60, env, cwd, shell = "/bin/bash", onStdout, onStderr, commandId, idleTimeout = 300, killOnDisconnect = false, ttlSeconds = 600, pty, headers: extraHeaders, } = options;
     const wsUrl = buildWsUrl(dataplaneUrl);
-    const headers = buildAuthHeaders(apiKey);
+    const headers = buildAuthHeaders(apiKey, extraHeaders);
     const control = new WSStreamControl();
     async function* stream() {
         let ws;
@@ -308,9 +312,9 @@ async function runWsStream(dataplaneUrl, apiKey, command, options = {}) {
  * No 'started' message is sent on reconnection.
  */
 async function reconnectWsStream(dataplaneUrl, apiKey, commandId, options = {}) {
-    const { stdoutOffset = 0, stderrOffset = 0 } = options;
+    const { stdoutOffset = 0, stderrOffset = 0, headers: extraHeaders } = options;
     const wsUrl = buildWsUrl(dataplaneUrl);
-    const headers = buildAuthHeaders(apiKey);
+    const headers = buildAuthHeaders(apiKey, extraHeaders);
     const control = new WSStreamControl();
     async function* stream() {
         let ws;

package/dist/sandbox/ws_execute.d.ts

@@ -13,7 +13,7 @@ export declare function buildWsUrl(dataplaneUrl: string): string;
 /**
  * Build auth headers for the WebSocket upgrade request.
  */
-export declare function buildAuthHeaders(apiKey: string | undefined): Record<string, string>;
+export declare function buildAuthHeaders(apiKey: string | undefined, extraHeaders?: Record<string, string>): Record<string, string>;
 /**
  * Control interface for an active WebSocket stream.
  *
@@ -66,5 +66,6 @@ export declare function runWsStream(dataplaneUrl: string, apiKey: string | undef
 export declare function reconnectWsStream(dataplaneUrl: string, apiKey: string | undefined, commandId: string, options?: {
     stdoutOffset?: number;
     stderrOffset?: number;
+    headers?: Record<string, string>;
 }): Promise<[AsyncIterableIterator<WsMessage>, WSStreamControl]>;
 export {};

package/dist/sandbox/ws_execute.js

@@ -30,11 +30,15 @@ export function buildWsUrl(dataplaneUrl) {
 /**
  * Build auth headers for the WebSocket upgrade request.
  */
-export function buildAuthHeaders(apiKey) {
+export function buildAuthHeaders(apiKey, extraHeaders) {
+    const headers = {};
     if (apiKey) {
-        return { "X-Api-Key": apiKey };
+        headers["X-Api-Key"] = apiKey;
     }
-    return {};
+    if (extraHeaders) {
+        Object.assign(headers, extraHeaders);
+    }
+    return headers;
 }
 // =============================================================================
 // Stream Control
@@ -229,9 +233,9 @@ async function* readWsMessages(ws) {
  * data arrives in addition to yielding the messages.
  */
 export async function runWsStream(dataplaneUrl, apiKey, command, options = {}) {
-    const { timeout = 60, env, cwd, shell = "/bin/bash", onStdout, onStderr, commandId, idleTimeout = 300, killOnDisconnect = false, ttlSeconds = 600, pty, } = options;
+    const { timeout = 60, env, cwd, shell = "/bin/bash", onStdout, onStderr, commandId, idleTimeout = 300, killOnDisconnect = false, ttlSeconds = 600, pty, headers: extraHeaders, } = options;
     const wsUrl = buildWsUrl(dataplaneUrl);
-    const headers = buildAuthHeaders(apiKey);
+    const headers = buildAuthHeaders(apiKey, extraHeaders);
     const control = new WSStreamControl();
     async function* stream() {
         let ws;
@@ -299,9 +303,9 @@ export async function runWsStream(dataplaneUrl, apiKey, command, options = {}) {
  * No 'started' message is sent on reconnection.
  */
 export async function reconnectWsStream(dataplaneUrl, apiKey, commandId, options = {}) {
-    const { stdoutOffset = 0, stderrOffset = 0 } = options;
+    const { stdoutOffset = 0, stderrOffset = 0, headers: extraHeaders } = options;
     const wsUrl = buildWsUrl(dataplaneUrl);
-    const headers = buildAuthHeaders(apiKey);
+    const headers = buildAuthHeaders(apiKey, extraHeaders);
     const control = new WSStreamControl();
     async function* stream() {
         let ws;

package/dist/utils/types.cjs

@@ -0,0 +1,13 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.isPrimitive = exports.isRecord = void 0;
+const isRecord = (value) => {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+};
+exports.isRecord = isRecord;
+const isPrimitive = (value) => {
+    return (typeof value === "string" ||
+        typeof value === "number" ||
+        typeof value === "boolean");
+};
+exports.isPrimitive = isPrimitive;

package/dist/utils/types.d.ts

@@ -0,0 +1,2 @@
+export declare const isRecord: (value: unknown) => value is Record<string, unknown>;
+export declare const isPrimitive: (value: unknown) => value is string | number | boolean;

package/dist/utils/types.js

@@ -0,0 +1,8 @@
+export const isRecord = (value) => {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+};
+export const isPrimitive = (value) => {
+    return (typeof value === "string" ||
+        typeof value === "number" ||
+        typeof value === "boolean");
+};

package/dist/utils/vercel.cjs

@@ -4,6 +4,7 @@ exports.extractOutputTokenDetails = extractOutputTokenDetails;
 exports.extractInputTokenDetails = extractInputTokenDetails;
 exports.extractUsageMetadata = extractUsageMetadata;
 const usage_js_1 = require("./usage.cjs");
+const types_js_1 = require("./types.cjs");
 function extractTraceableServiceTier(providerMetadata) {
     if (providerMetadata?.openai != null &&
         typeof providerMetadata.openai === "object") {
@@ -19,6 +20,14 @@ function extractTraceableServiceTier(providerMetadata) {
 function isLanguageModelV3Usage(usage) {
     return usage.inputTokens != null && typeof usage.inputTokens === "object";
 }
+function isTelemetryLanguageModelUsage(usage) {
+    if (!(0, types_js_1.isRecord)(usage))
+        return false;
+    return ((usage.inputTokens == null || typeof usage.inputTokens === "number") &&
+        (0, types_js_1.isRecord)(usage.inputTokenDetails) &&
+        (usage.outputTokens == null || typeof usage.outputTokens === "number") &&
+        (0, types_js_1.isRecord)(usage.outputTokenDetails));
+}
 function extractAISDK6OutputTokenDetails(usage, providerMetadata) {
     const openAIServiceTier = extractTraceableServiceTier(providerMetadata ?? {});
     const outputTokenDetailsKeyPrefix = openAIServiceTier
@@ -42,32 +51,51 @@ function extractAISDK6OutputTokenDetails(usage, providerMetadata) {
     }
     return outputTokenDetails;
 }
+// TODO: verify service tier token counting
+function extractTelemetryOutputTokenDetails(usage, providerMetadata) {
+    const result = {};
+    const serviceTier = extractTraceableServiceTier(providerMetadata ?? {});
+    const prefix = serviceTier ? `${serviceTier}_` : "";
+    const reasoning = usage.outputTokenDetails?.reasoningTokens ?? undefined;
+    if (typeof reasoning === "number" && reasoning > 0) {
+        const key = `${prefix}reasoning`;
+        result[key] = reasoning;
+    }
+    // Apply service tier logic
+    if (serviceTier != null) {
+        // Avoid counting reasoning tokens towards the output token count
+        // since service tier tokens are already priced differently
+        result[serviceTier] = (usage.outputTokens ?? 0) - (reasoning ?? 0);
+    }
+    return result;
+}
 function extractOutputTokenDetails(usage, providerMetadata) {
     if (usage == null) {
         return {};
     }
+    // AI SDK 7: Use Telemetry Language Model Usage
+    if (isTelemetryLanguageModelUsage(usage)) {
+        return extractTelemetryOutputTokenDetails(usage, providerMetadata);
+    }
     // AI SDK 6: Check for built-in outputTokens breakdown first
     if (isLanguageModelV3Usage(usage)) {
         // Return AI SDK 6 results (even if empty, to prevent falling through to SDK 5 logic)
         return extractAISDK6OutputTokenDetails(usage, providerMetadata);
     }
+    // AI SDK <= 5
     const openAIServiceTier = extractTraceableServiceTier(providerMetadata ?? {});
-    const outputTokenDetailsKeyPrefix = openAIServiceTier
-        ? `${openAIServiceTier}_`
-        : "";
-    const outputTokenDetails = {};
+    const prefix = openAIServiceTier ? `${openAIServiceTier}_` : "";
+    const result = {};
     if (typeof usage?.reasoningTokens === "number") {
-        outputTokenDetails[`${outputTokenDetailsKeyPrefix}reasoning`] =
-            usage.reasoningTokens;
+        result[`${prefix}reasoning`] = usage.reasoningTokens;
     }
     if (openAIServiceTier && typeof usage?.outputTokens === "number") {
         // Avoid counting reasoning tokens towards the output token count
         // since service tier tokens are already priced differently
-        outputTokenDetails[openAIServiceTier] =
-            usage.outputTokens -
-                (outputTokenDetails[`${outputTokenDetailsKeyPrefix}reasoning`] ?? 0);
+        result[openAIServiceTier] =
+            usage.outputTokens - (result[`${prefix}reasoning`] ?? 0);
     }
-    return outputTokenDetails;
+    return result;
 }
 function extractAISDK6InputTokenDetails(usage, providerMetadata) {
     let inputTokenDetails = {};
@@ -116,10 +144,40 @@ function extractAISDK6InputTokenDetails(usage, providerMetadata) {
     }
     return inputTokenDetails;
 }
+function extractTelemetryInputTokenDetails(usage, providerMetadata) {
+    const result = {};
+    const cacheRead = usage.inputTokenDetails?.cacheReadTokens ?? undefined;
+    const cacheWrite = usage.inputTokenDetails?.cacheWriteTokens ?? undefined;
+    if (typeof cacheRead === "number" && cacheRead > 0) {
+        result.cache_read = cacheRead;
+    }
+    if (typeof cacheWrite === "number" && cacheWrite > 0) {
+        result.cache_creation = cacheWrite;
+    }
+    const openAIServiceTier = extractTraceableServiceTier(providerMetadata ?? {});
+    const prefix = openAIServiceTier
+        ? `${openAIServiceTier}_`
+        : "";
+    if (openAIServiceTier && typeof usage.inputTokens === "number") {
+        if (result.cache_read != null) {
+            result[`${prefix}cache_read`] = result.cache_read;
+            delete result.cache_read;
+        }
+        if (typeof usage?.inputTokens === "number") {
+            result[openAIServiceTier] =
+                usage.inputTokens - (result[`${prefix}cache_read`] ?? 0);
+        }
+    }
+    return result;
+}
 function extractInputTokenDetails(usage, providerMetadata) {
     if (usage == null) {
         return {};
     }
+    // AI SDK 7: Use Telemetry Language Model Usage
+    if (isTelemetryLanguageModelUsage(usage)) {
+        return extractTelemetryInputTokenDetails(usage, providerMetadata);
+    }
     // AI SDK 6: Check for built-in inputTokens breakdown first
     if (isLanguageModelV3Usage(usage)) {
         // Return AI SDK 6 results (even if empty, to prevent falling through to SDK 5 logic)

package/dist/utils/vercel.d.ts

@@ -1,7 +1,22 @@
 import type { LanguageModelV2Usage, LanguageModelV3Usage } from "@ai-sdk/provider";
 import { KVMap } from "../schemas.js";
-export declare function extractOutputTokenDetails(usage?: Partial<LanguageModelV2Usage> | Partial<LanguageModelV3Usage>, providerMetadata?: Record<string, unknown>): Record<string, number>;
-export declare function extractInputTokenDetails(usage?: Partial<LanguageModelV2Usage> | Partial<LanguageModelV3Usage>, providerMetadata?: Record<string, unknown>): Record<string, number>;
+export declare function extractOutputTokenDetails(usage?: Partial<LanguageModelV2Usage> | Partial<LanguageModelV3Usage>, providerMetadata?: Record<string, unknown>): Record<string, number> | {
+    reasoning?: number;
+    flex?: number;
+    flex_reasoning?: number;
+    priority?: number;
+    priority_reasoning?: number;
+};
+export declare function extractInputTokenDetails(usage?: Partial<LanguageModelV2Usage> | Partial<LanguageModelV3Usage>, providerMetadata?: Record<string, unknown>): Record<string, number> | {
+    cache_read?: number;
+    cache_creation?: number;
+    flex?: number;
+    flex_cache_read?: number;
+    flex_cache_creation?: number;
+    priority?: number;
+    priority_cache_read?: number;
+    priority_cache_creation?: number;
+};
 export declare function extractUsageMetadata(span?: {
     status?: {
         code: number;

package/dist/utils/vercel.js

@@ -1,4 +1,5 @@
 import { convertAnthropicUsageToInputTokenDetails } from "./usage.js";
+import { isRecord } from "./types.js";
 function extractTraceableServiceTier(providerMetadata) {
     if (providerMetadata?.openai != null &&
         typeof providerMetadata.openai === "object") {
@@ -14,6 +15,14 @@ function extractTraceableServiceTier(providerMetadata) {
 function isLanguageModelV3Usage(usage) {
     return usage.inputTokens != null && typeof usage.inputTokens === "object";
 }
+function isTelemetryLanguageModelUsage(usage) {
+    if (!isRecord(usage))
+        return false;
+    return ((usage.inputTokens == null || typeof usage.inputTokens === "number") &&
+        isRecord(usage.inputTokenDetails) &&
+        (usage.outputTokens == null || typeof usage.outputTokens === "number") &&
+        isRecord(usage.outputTokenDetails));
+}
 function extractAISDK6OutputTokenDetails(usage, providerMetadata) {
     const openAIServiceTier = extractTraceableServiceTier(providerMetadata ?? {});
     const outputTokenDetailsKeyPrefix = openAIServiceTier
@@ -37,32 +46,51 @@ function extractAISDK6OutputTokenDetails(usage, providerMetadata) {
     }
     return outputTokenDetails;
 }
+// TODO: verify service tier token counting
+function extractTelemetryOutputTokenDetails(usage, providerMetadata) {
+    const result = {};
+    const serviceTier = extractTraceableServiceTier(providerMetadata ?? {});
+    const prefix = serviceTier ? `${serviceTier}_` : "";
+    const reasoning = usage.outputTokenDetails?.reasoningTokens ?? undefined;
+    if (typeof reasoning === "number" && reasoning > 0) {
+        const key = `${prefix}reasoning`;
+        result[key] = reasoning;
+    }
+    // Apply service tier logic
+    if (serviceTier != null) {
+        // Avoid counting reasoning tokens towards the output token count
+        // since service tier tokens are already priced differently
+        result[serviceTier] = (usage.outputTokens ?? 0) - (reasoning ?? 0);
+    }
+    return result;
+}
 export function extractOutputTokenDetails(usage, providerMetadata) {
     if (usage == null) {
         return {};
     }
+    // AI SDK 7: Use Telemetry Language Model Usage
+    if (isTelemetryLanguageModelUsage(usage)) {
+        return extractTelemetryOutputTokenDetails(usage, providerMetadata);
+    }
     // AI SDK 6: Check for built-in outputTokens breakdown first
     if (isLanguageModelV3Usage(usage)) {
         // Return AI SDK 6 results (even if empty, to prevent falling through to SDK 5 logic)
         return extractAISDK6OutputTokenDetails(usage, providerMetadata);
     }
+    // AI SDK <= 5
     const openAIServiceTier = extractTraceableServiceTier(providerMetadata ?? {});
-    const outputTokenDetailsKeyPrefix = openAIServiceTier
-        ? `${openAIServiceTier}_`
-        : "";
-    const outputTokenDetails = {};
+    const prefix = openAIServiceTier ? `${openAIServiceTier}_` : "";
+    const result = {};
     if (typeof usage?.reasoningTokens === "number") {
-        outputTokenDetails[`${outputTokenDetailsKeyPrefix}reasoning`] =
-            usage.reasoningTokens;
+        result[`${prefix}reasoning`] = usage.reasoningTokens;
     }
     if (openAIServiceTier && typeof usage?.outputTokens === "number") {
         // Avoid counting reasoning tokens towards the output token count
         // since service tier tokens are already priced differently
-        outputTokenDetails[openAIServiceTier] =
-            usage.outputTokens -
-                (outputTokenDetails[`${outputTokenDetailsKeyPrefix}reasoning`] ?? 0);
+        result[openAIServiceTier] =
+            usage.outputTokens - (result[`${prefix}reasoning`] ?? 0);
     }
-    return outputTokenDetails;
+    return result;
 }
 function extractAISDK6InputTokenDetails(usage, providerMetadata) {
     let inputTokenDetails = {};
@@ -111,10 +139,40 @@ function extractAISDK6InputTokenDetails(usage, providerMetadata) {
     }
     return inputTokenDetails;
 }
+function extractTelemetryInputTokenDetails(usage, providerMetadata) {
+    const result = {};
+    const cacheRead = usage.inputTokenDetails?.cacheReadTokens ?? undefined;
+    const cacheWrite = usage.inputTokenDetails?.cacheWriteTokens ?? undefined;
+    if (typeof cacheRead === "number" && cacheRead > 0) {
+        result.cache_read = cacheRead;
+    }
+    if (typeof cacheWrite === "number" && cacheWrite > 0) {
+        result.cache_creation = cacheWrite;
+    }
+    const openAIServiceTier = extractTraceableServiceTier(providerMetadata ?? {});
+    const prefix = openAIServiceTier
+        ? `${openAIServiceTier}_`
+        : "";
+    if (openAIServiceTier && typeof usage.inputTokens === "number") {
+        if (result.cache_read != null) {
+            result[`${prefix}cache_read`] = result.cache_read;
+            delete result.cache_read;
+        }
+        if (typeof usage?.inputTokens === "number") {
+            result[openAIServiceTier] =
+                usage.inputTokens - (result[`${prefix}cache_read`] ?? 0);
+        }
+    }
+    return result;
+}
 export function extractInputTokenDetails(usage, providerMetadata) {
     if (usage == null) {
         return {};
     }
+    // AI SDK 7: Use Telemetry Language Model Usage
+    if (isTelemetryLanguageModelUsage(usage)) {
+        return extractTelemetryInputTokenDetails(usage, providerMetadata);
+    }
     // AI SDK 6: Check for built-in inputTokens breakdown first
     if (isLanguageModelV3Usage(usage)) {
         // Return AI SDK 6 results (even if empty, to prevent falling through to SDK 5 logic)

package/experimental/sandbox.cjs

@@ -0,0 +1 @@
+module.exports = require('../dist/sandbox/index.cjs');

package/experimental/sandbox.d.cts

@@ -0,0 +1 @@
+export * from '../dist/sandbox/index.js'

package/experimental/sandbox.d.ts

@@ -0,0 +1 @@
+export * from '../dist/sandbox/index.js'

package/experimental/sandbox.js

@@ -0,0 +1 @@
+export * from '../dist/sandbox/index.js'