langsmith 0.10.8 → 0.10.9

package/dist/index.d.ts

@@ -1,12 +1,12 @@
 export { Client, type ClientConfig, type LangSmithTracingClientInterface, } from "./client.js";
 export type { Dataset, Example, TracerSession, Run, Feedback, FeedbackConfigSchema, RetrieverOutput, } from "./schemas.js";
 export { RunTree, type RunTreeConfig, type WriteReplica } from "./run_trees.js";
-export { address, type Address, type Environment, EnvAddressError, } from "./address.js";
+export { AgentAddress, ExperimentAddress, EvaluatorAddress, type Address, type ApiAddress, type Environment, EnvAddressError, } from "./address.js";
 export { overrideFetchImplementation } from "./singletons/fetch.js";
 export { getDefaultProjectName } from "./utils/project.js";
 export { computeRunIdForSecondaryReplica, uuid7, uuid7FromTime, } from "./uuid.js";
 export { isTracingEnabled } from "./utils/guard.js";
 export { Cache, PromptCache, type CacheConfig, type CacheMetrics, configureGlobalPromptCache, promptCacheSingleton, } from "./utils/prompt_cache/index.js";
 export { LangsmithError, APIError, APIUserAbortError, APIConnectionError, APIConnectionTimeoutError, BadRequestError, AuthenticationError, PermissionDeniedError, NotFoundError, ConflictError, UnprocessableEntityError, RateLimitError, InternalServerError, } from "./_openapi_client/core/error.js";
-export declare const __version__ = "0.10.8";
+export declare const __version__ = "0.10.9";
 export declare const LS_MESSAGE_VIEW_EXCLUDE: "ls_message_view_exclude";

package/dist/index.js

@@ -1,6 +1,6 @@
 export { Client, } from "./client.js";
 export { RunTree } from "./run_trees.js";
-export { address, EnvAddressError, } from "./address.js";
+export { AgentAddress, ExperimentAddress, EvaluatorAddress, EnvAddressError, } from "./address.js";
 export { overrideFetchImplementation } from "./singletons/fetch.js";
 export { getDefaultProjectName } from "./utils/project.js";
 export { computeRunIdForSecondaryReplica, uuid7, uuid7FromTime, } from "./uuid.js";
@@ -8,6 +8,6 @@ export { isTracingEnabled } from "./utils/guard.js";
 export { Cache, PromptCache, configureGlobalPromptCache, promptCacheSingleton, } from "./utils/prompt_cache/index.js";
 export { LangsmithError, APIError, APIUserAbortError, APIConnectionError, APIConnectionTimeoutError, BadRequestError, AuthenticationError, PermissionDeniedError, NotFoundError, ConflictError, UnprocessableEntityError, RateLimitError, InternalServerError, } from "./_openapi_client/core/error.js";
 // Update using pnpm bump-version
-export const __version__ = "0.10.8";
+export const __version__ = "0.10.9";
 // Metadata key to hide a traced run from LangSmith's Messages View.
 export const LS_MESSAGE_VIEW_EXCLUDE = "ls_message_view_exclude";

package/dist/run_trees.cjs

@@ -14,7 +14,7 @@ const env_js_3 = require("./utils/env.cjs");
 const warn_js_1 = require("./utils/warn.cjs");
 const _uuid_js_1 = require("./utils/_uuid.cjs");
 const index_js_1 = require("./utils/uuid/src/index.cjs");
-const agent_addressing_js_1 = require("./utils/agent_addressing.cjs");
+const addressing_js_1 = require("./utils/addressing.cjs");
 const address_js_1 = require("./address.cjs");
 const TIMESTAMP_LENGTH = 36;
 // DNS namespace for UUID v5 (same as Python's uuid.NAMESPACE_DNS)
@@ -52,15 +52,10 @@ const HEADER_SAFE_REPLICA_FIELDS = new Set([
     "reroot",
     "address",
 ]);
-// The address string, URL-encoded.
-const LANGSMITH_ADDRESS = "langsmith-address";
-/** Validates an untrusted header address; `undefined` if unusable. */
+/** Rebuilds an untrusted header address; `undefined` if unusable. */
 function addressFromHeader(value) {
-    if (value == null) {
-        return undefined;
-    }
     try {
-        return (0, address_js_1.parse)(value);
+        return address_js_1.AgentAddress._fromLrn(value);
     }
     catch {
         // Values are untrusted, so not logged.
@@ -140,7 +135,7 @@ class Baggage {
             else if (key === "langsmith-project") {
                 project_name = value;
             }
-            else if (key === LANGSMITH_ADDRESS) {
+            else if (key === "langsmith-address") {
                 address = addressFromHeader(value);
             }
             else if (key === "langsmith-replicas") {
@@ -152,13 +147,15 @@ class Baggage {
                     const filtered = filterReplicaForHeaders(replica);
                     const raw = filtered.address;
                     delete filtered.address;
-                    const address = addressFromHeader(raw);
                     // A project wins over an address; a replica with an invalid
                     // address and no project is dropped.
                     if (filtered.projectName || raw == null) {
                         return [filtered];
                     }
-                    return address ? [{ ...filtered, address }] : [];
+                    const replicaAddress = addressFromHeader(raw);
+                    return replicaAddress
+                        ? [{ ...filtered, address: replicaAddress }]
+                        : [];
                 });
             }
         }
@@ -176,7 +173,7 @@ class Baggage {
             items.push(`langsmith-project=${encodeURIComponent(this.project_name)}`);
         }
         if (this.address) {
-            items.push(`${LANGSMITH_ADDRESS}=${encodeURIComponent(this.address)}`);
+            items.push(`langsmith-address=${encodeURIComponent(this.address._toLrn())}`);
         }
         return items.join(",");
     }
@@ -426,23 +423,23 @@ class RunTree {
     }
     /** Falls back to the env; a bad env leaves the run untraced. */
     _resolveAddressing(config) {
-        const address = (0, agent_addressing_js_1.checkAddress)(config.address);
+        const address = (0, address_js_1.ensureAgent)(config.address);
         const project = config.project_name || undefined;
-        (0, agent_addressing_js_1.rejectConflicting)(project, address);
+        (0, addressing_js_1.rejectConflicting)(project, address);
         if (project || address) {
             this.project_name = project;
             this.address = address;
             return;
         }
         try {
-            [this.project_name, this.address] = (0, agent_addressing_js_1.resolveFromEnv)();
+            [this.project_name, this.address] = (0, addressing_js_1.resolveFromEnv)();
         }
         catch (e) {
             if (!(e instanceof address_js_1.EnvAddressError)) {
                 throw e;
             }
             if ((0, env_js_1.isEnvTracingEnabled)(this.tracingEnabled)) {
-                (0, agent_addressing_js_1.logUntraced)(e);
+                (0, addressing_js_1.logUntraced)(e);
             }
             this.project_name = undefined;
             this.address = undefined;
@@ -652,9 +649,9 @@ class RunTree {
     }
     /** A replica naming neither inherits the run tree's destination. */
     _replicaAddressing(replica) {
-        const address = (0, agent_addressing_js_1.checkAddress)(replica.address);
+        const address = (0, address_js_1.ensureAgent)(replica.address);
         if (replica.projectName != null) {
-            (0, agent_addressing_js_1.rejectConflicting)(replica.projectName, address);
+            (0, addressing_js_1.rejectConflicting)(replica.projectName, address);
             return [replica.projectName, undefined];
         }
         if (address) {
@@ -669,14 +666,14 @@ class RunTree {
         if (primary === undefined &&
             !this.replicas?.some((r) => r.primary === true) &&
             projectName === this.project_name &&
-            address === this.address) {
+            address?._toLrn() === this.address?._toLrn()) {
             return {
                 ...baseRun,
                 session_name: projectName,
                 address,
             };
         }
-        const seed = projectName ?? address ?? "agent//";
+        const seed = projectName ?? address?._toLrn() ?? "agent//";
         // Apply reroot logic before ID remapping
         if (reroot) {
             if (distributedParentId) {
@@ -1197,7 +1194,7 @@ function _ensureWriteReplicas(replicas) {
                     updates: replica[1],
                 };
             }
-            const address = (0, agent_addressing_js_1.checkAddress)(replica.address);
+            const address = (0, address_js_1.ensureAgent)(replica.address);
             return address ? { ...replica, address } : replica;
         })
         : _getWriteReplicasFromEnv();

package/dist/run_trees.d.ts

@@ -1,6 +1,6 @@
 import { Client } from "./client.js";
 import { Attachments, BaseRun, KVMap, RunCreate } from "./schemas.js";
-import { type Address } from "./address.js";
+import { AgentAddress } from "./address.js";
 export declare function convertToDottedOrderFormat(epoch: number, runId: string, executionOrder?: number): {
     dottedOrder: string;
     microsecondPrecisionDatestring: string;
@@ -11,7 +11,7 @@ export interface RunTreeConfig {
     id?: string;
     project_name?: string;
     /** (beta) Send the run to this address instead of a project. */
-    address?: Address;
+    address?: AgentAddress;
     parent_run?: RunTree;
     parent_run_id?: string;
     child_runs?: RunTree[];
@@ -64,7 +64,7 @@ export type WriteReplica = {
     workspaceId?: string;
     projectName?: string;
     /** (beta) Send the replica to this address instead of a project. */
-    address?: Address;
+    address?: AgentAddress;
     /** Whether this replica keeps the original run IDs. */
     primary?: boolean;
     updates?: KVMap | undefined;
@@ -87,7 +87,7 @@ export declare class RunTree implements BaseRun {
     /** Unset for a run sent to an `address`. */
     project_name?: string;
     /** (beta) Set instead of `project_name` for an addressed run. */
-    address?: Address;
+    address?: AgentAddress;
     parent_run?: RunTree;
     parent_run_id?: string;
     child_runs: RunTree[];

package/dist/run_trees.js

@@ -8,8 +8,8 @@ import { getLangSmithEnvironmentVariable } from "./utils/env.js";
 import { warnOnce } from "./utils/warn.js";
 import { uuid7FromTime, nonCryptographicUuid7Deterministic, } from "./utils/_uuid.js";
 import { v5 as uuidv5 } from "./utils/uuid/src/index.js";
-import { checkAddress, logUntraced, rejectConflicting, resolveFromEnv, } from "./utils/agent_addressing.js";
-import { EnvAddressError, parse as parseAddress, } from "./address.js";
+import { logUntraced, rejectConflicting, resolveFromEnv, } from "./utils/addressing.js";
+import { AgentAddress, EnvAddressError, ensureAgent } from "./address.js";
 const TIMESTAMP_LENGTH = 36;
 // DNS namespace for UUID v5 (same as Python's uuid.NAMESPACE_DNS)
 const UUID_NAMESPACE_DNS = "6ba7b810-9dad-11d1-80b4-00c04fd430c8";
@@ -46,15 +46,10 @@ const HEADER_SAFE_REPLICA_FIELDS = new Set([
     "reroot",
     "address",
 ]);
-// The address string, URL-encoded.
-const LANGSMITH_ADDRESS = "langsmith-address";
-/** Validates an untrusted header address; `undefined` if unusable. */
+/** Rebuilds an untrusted header address; `undefined` if unusable. */
 function addressFromHeader(value) {
-    if (value == null) {
-        return undefined;
-    }
     try {
-        return parseAddress(value);
+        return AgentAddress._fromLrn(value);
     }
     catch {
         // Values are untrusted, so not logged.
@@ -134,7 +129,7 @@ class Baggage {
             else if (key === "langsmith-project") {
                 project_name = value;
             }
-            else if (key === LANGSMITH_ADDRESS) {
+            else if (key === "langsmith-address") {
                 address = addressFromHeader(value);
             }
             else if (key === "langsmith-replicas") {
@@ -146,13 +141,15 @@ class Baggage {
                     const filtered = filterReplicaForHeaders(replica);
                     const raw = filtered.address;
                     delete filtered.address;
-                    const address = addressFromHeader(raw);
                     // A project wins over an address; a replica with an invalid
                     // address and no project is dropped.
                     if (filtered.projectName || raw == null) {
                         return [filtered];
                     }
-                    return address ? [{ ...filtered, address }] : [];
+                    const replicaAddress = addressFromHeader(raw);
+                    return replicaAddress
+                        ? [{ ...filtered, address: replicaAddress }]
+                        : [];
                 });
             }
         }
@@ -170,7 +167,7 @@ class Baggage {
             items.push(`langsmith-project=${encodeURIComponent(this.project_name)}`);
         }
         if (this.address) {
-            items.push(`${LANGSMITH_ADDRESS}=${encodeURIComponent(this.address)}`);
+            items.push(`langsmith-address=${encodeURIComponent(this.address._toLrn())}`);
         }
         return items.join(",");
     }
@@ -420,7 +417,7 @@ export class RunTree {
     }
     /** Falls back to the env; a bad env leaves the run untraced. */
     _resolveAddressing(config) {
-        const address = checkAddress(config.address);
+        const address = ensureAgent(config.address);
         const project = config.project_name || undefined;
         rejectConflicting(project, address);
         if (project || address) {
@@ -646,7 +643,7 @@ export class RunTree {
     }
     /** A replica naming neither inherits the run tree's destination. */
     _replicaAddressing(replica) {
-        const address = checkAddress(replica.address);
+        const address = ensureAgent(replica.address);
         if (replica.projectName != null) {
             rejectConflicting(replica.projectName, address);
             return [replica.projectName, undefined];
@@ -663,14 +660,14 @@ export class RunTree {
         if (primary === undefined &&
             !this.replicas?.some((r) => r.primary === true) &&
             projectName === this.project_name &&
-            address === this.address) {
+            address?._toLrn() === this.address?._toLrn()) {
             return {
                 ...baseRun,
                 session_name: projectName,
                 address,
             };
         }
-        const seed = projectName ?? address ?? "agent//";
+        const seed = projectName ?? address?._toLrn() ?? "agent//";
         // Apply reroot logic before ID remapping
         if (reroot) {
             if (distributedParentId) {
@@ -1190,7 +1187,7 @@ function _ensureWriteReplicas(replicas) {
                     updates: replica[1],
                 };
             }
-            const address = checkAddress(replica.address);
+            const address = ensureAgent(replica.address);
             return address ? { ...replica, address } : replica;
         })
         : _getWriteReplicasFromEnv();

package/dist/sandbox/client.cjs

@@ -15,6 +15,7 @@ const errors_js_1 = require("./errors.cjs");
 const helpers_js_1 = require("./helpers.cjs");
 const mounts_js_1 = require("./mounts.cjs");
 const index_js_2 = require("../utils/uuid/src/index.cjs");
+const tracing_js_1 = require("./tracing.cjs");
 /**
  * Sleep that can be interrupted by an AbortSignal.
  * Resolves after `ms` milliseconds or rejects immediately if the signal fires.
@@ -305,6 +306,12 @@ class SandboxClient {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "_sandboxIds", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: new Map()
+        });
         this._baseUrl = (config.apiEndpoint ?? getDefaultApiEndpoint()).replace(/\/$/, "");
         this._apiKey = config.apiKey ?? getDefaultApiKey();
         this._defaultHeaders = { ...(config.headers ?? {}) };
@@ -398,6 +405,25 @@ class SandboxClient {
     getDefaultHeaders() {
         return { ...this._defaultHeaders };
     }
+    _traceSandbox(name) {
+        const id = this._sandboxIds.get(name) ?? ((0, index_js_2.validate)(name) ? name : undefined);
+        (0, tracing_js_1.addSandboxMetadata)(id);
+        return id;
+    }
+    _sandboxFromData(data) {
+        if (data.id)
+            this._sandboxIds.set(data.name, data.id);
+        (0, tracing_js_1.addSandboxMetadata)(data.id);
+        return new sandbox_js_1.Sandbox(data, this);
+    }
+    _forgetSandbox(name) {
+        const id = this._sandboxIds.get(name) ?? name;
+        for (const [cachedName, cachedId] of this._sandboxIds) {
+            if (cachedName === name || cachedId === id) {
+                this._sandboxIds.delete(cachedName);
+            }
+        }
+    }
     /**
      * JSON POST helper. Sends JSON body, checks response status,
      * and returns the Response for further processing.
@@ -406,6 +432,7 @@ class SandboxClient {
      * @internal
      */
     _boxUrl(name, ...segments) {
+        this._traceSandbox(name);
         const suffix = segments.length ? `/${segments.join("/")}` : "";
         return `${this._baseUrl}/boxes/${encodeURIComponent(name)}${suffix}`;
     }
@@ -454,7 +481,7 @@ class SandboxClient {
         return new service_url_js_1.ServiceUrl(data, () => this.serviceUrl(name, {
             port,
             expiresInSeconds,
-        }));
+        }), this._traceSandbox(name));
     }
     async _postJson(url, body, options) {
         const response = await this._fetch(url, {
@@ -534,7 +561,7 @@ class SandboxClient {
             await (0, helpers_js_1.handleSandboxCreationError)(response);
         }
         const data = (await response.json());
-        return new sandbox_js_1.Sandbox(data, this);
+        return this._sandboxFromData(data);
     }
     /**
      * Get a Sandbox by name.
@@ -555,7 +582,7 @@ class SandboxClient {
             await (0, helpers_js_1.handleClientHttpError)(response);
         }
         const data = (await response.json());
-        return new sandbox_js_1.Sandbox(data, this);
+        return this._sandboxFromData(data);
     }
     /**
      * List all Sandboxes.
@@ -625,7 +652,8 @@ class SandboxClient {
             await (0, helpers_js_1.handleClientHttpError)(response);
         }
         const data = (await response.json());
-        return new sandbox_js_1.Sandbox(data, this);
+        this._forgetSandbox(name);
+        return this._sandboxFromData(data);
     }
     /**
      * Delete a Sandbox.
@@ -642,6 +670,7 @@ class SandboxClient {
             }
             await (0, helpers_js_1.handleClientHttpError)(response);
         }
+        this._forgetSandbox(name);
     }
     /**
      * Get the provisioning status of a sandbox.

package/dist/sandbox/client.d.ts

@@ -40,6 +40,7 @@ export declare class SandboxClient {
     private _fetchImpl;
     private _caller;
     private _registriesClient?;
+    private _sandboxIds;
     constructor(config?: SandboxClientConfig);
     private _apiRoot;
     /**
@@ -65,6 +66,9 @@ export declare class SandboxClient {
      * ```
      */
     get registries(): Registries;
+    private _traceSandbox;
+    private _sandboxFromData;
+    private _forgetSandbox;
     /**
      * Generate an authenticated URL for an HTTP service running in a sandbox.
      *

package/dist/sandbox/client.js

@@ -11,7 +11,8 @@ import { Sandbox } from "./sandbox.js";
 import { LangSmithResourceCreationError, LangSmithResourceNameConflictError, LangSmithResourceNotFoundError, LangSmithResourceTimeoutError, LangSmithSandboxAPIError, LangSmithValidationError, } from "./errors.js";
 import { handleClientHttpError, handleSandboxCreationError, throwIfNotReady, validateTtl, } from "./helpers.js";
 import { validateMountConfigProxyConfig } from "./mounts.js";
-import { v4 as uuidv4 } from "../utils/uuid/src/index.js";
+import { v4 as uuidv4, validate as isUuid } from "../utils/uuid/src/index.js";
+import { addSandboxMetadata } from "./tracing.js";
 /**
  * Sleep that can be interrupted by an AbortSignal.
  * Resolves after `ms` milliseconds or rejects immediately if the signal fires.
@@ -302,6 +303,12 @@ export class SandboxClient {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "_sandboxIds", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: new Map()
+        });
         this._baseUrl = (config.apiEndpoint ?? getDefaultApiEndpoint()).replace(/\/$/, "");
         this._apiKey = config.apiKey ?? getDefaultApiKey();
         this._defaultHeaders = { ...(config.headers ?? {}) };
@@ -395,6 +402,25 @@ export class SandboxClient {
     getDefaultHeaders() {
         return { ...this._defaultHeaders };
     }
+    _traceSandbox(name) {
+        const id = this._sandboxIds.get(name) ?? (isUuid(name) ? name : undefined);
+        addSandboxMetadata(id);
+        return id;
+    }
+    _sandboxFromData(data) {
+        if (data.id)
+            this._sandboxIds.set(data.name, data.id);
+        addSandboxMetadata(data.id);
+        return new Sandbox(data, this);
+    }
+    _forgetSandbox(name) {
+        const id = this._sandboxIds.get(name) ?? name;
+        for (const [cachedName, cachedId] of this._sandboxIds) {
+            if (cachedName === name || cachedId === id) {
+                this._sandboxIds.delete(cachedName);
+            }
+        }
+    }
     /**
      * JSON POST helper. Sends JSON body, checks response status,
      * and returns the Response for further processing.
@@ -403,6 +429,7 @@ export class SandboxClient {
      * @internal
      */
     _boxUrl(name, ...segments) {
+        this._traceSandbox(name);
         const suffix = segments.length ? `/${segments.join("/")}` : "";
         return `${this._baseUrl}/boxes/${encodeURIComponent(name)}${suffix}`;
     }
@@ -451,7 +478,7 @@ export class SandboxClient {
         return new ServiceUrl(data, () => this.serviceUrl(name, {
             port,
             expiresInSeconds,
-        }));
+        }), this._traceSandbox(name));
     }
     async _postJson(url, body, options) {
         const response = await this._fetch(url, {
@@ -531,7 +558,7 @@ export class SandboxClient {
             await handleSandboxCreationError(response);
         }
         const data = (await response.json());
-        return new Sandbox(data, this);
+        return this._sandboxFromData(data);
     }
     /**
      * Get a Sandbox by name.
@@ -552,7 +579,7 @@ export class SandboxClient {
             await handleClientHttpError(response);
         }
         const data = (await response.json());
-        return new Sandbox(data, this);
+        return this._sandboxFromData(data);
     }
     /**
      * List all Sandboxes.
@@ -622,7 +649,8 @@ export class SandboxClient {
             await handleClientHttpError(response);
         }
         const data = (await response.json());
-        return new Sandbox(data, this);
+        this._forgetSandbox(name);
+        return this._sandboxFromData(data);
     }
     /**
      * Delete a Sandbox.
@@ -639,6 +667,7 @@ export class SandboxClient {
             }
             await handleClientHttpError(response);
         }
+        this._forgetSandbox(name);
     }
     /**
      * Get the provisioning status of a sandbox.

package/dist/sandbox/command_handle.cjs

@@ -7,6 +7,7 @@
  */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.CommandHandle = void 0;
+const tracing_js_1 = require("./tracing.cjs");
 const errors_js_1 = require("./errors.cjs");
 /**
  * Whether a 'started' frame proves *this* command was reattached to.
@@ -205,6 +206,7 @@ class CommandHandle {
      * The final execution result. Drains the stream if not already exhausted.
      */
     get result() {
+        (0, tracing_js_1.addSandboxMetadata)(this._sandbox.id);
         return this._getResult();
     }
     async _getResult() {
@@ -268,6 +270,7 @@ class CommandHandle {
      * - After kill():                  no reconnect, error propagates
      */
     async *[Symbol.asyncIterator]() {
+        (0, tracing_js_1.addSandboxMetadata)(this._sandbox.id);
         this._reconnectAttempts = 0;
         while (true) {
             try {
@@ -283,6 +286,8 @@ class CommandHandle {
                             chunk.offset + new TextEncoder().encode(chunk.data).length;
                         this._onStderr?.(chunk.data);
                     }
+                    // Each resumption runs in the consumer's current context.
+                    (0, tracing_js_1.addSandboxMetadata)(this._sandbox.id);
                     yield chunk;
                 }
                 return; // Stream ended normally (exit message received)
@@ -325,6 +330,7 @@ class CommandHandle {
      * subsequently yield an exit message with a non-zero exit code.
      */
     kill() {
+        (0, tracing_js_1.addSandboxMetadata)(this._sandbox.id);
         if (this._control) {
             this._control.sendKill();
         }
@@ -333,6 +339,7 @@ class CommandHandle {
      * Write data to the command's stdin.
      */
     sendInput(data) {
+        (0, tracing_js_1.addSandboxMetadata)(this._sandbox.id);
         if (this._stdinClosed) {
             throw new errors_js_1.LangSmithSandboxOperationError("stdin is closed for this command. Non-PTY commands close stdin by " +
                 "default so a command that reads it sees EOF instead of hanging; " +
@@ -350,6 +357,7 @@ class CommandHandle {
      * EOT byte (0x04) with {@link sendInput} instead.
      */
     closeInput() {
+        (0, tracing_js_1.addSandboxMetadata)(this._sandbox.id);
         if (this._pty || this._stdinClosed)
             return;
         this._stdinClosed = true;

package/dist/sandbox/command_handle.js

@@ -4,6 +4,7 @@
  *
  * Port of Python's AsyncCommandHandle to TypeScript.
  */
+import { addSandboxMetadata } from "./tracing.js";
 import { LangSmithSandboxConnectionError, LangSmithSandboxServerReloadError, LangSmithSandboxOperationError, LangSmithStreamEndedBeforeStartedError, } from "./errors.js";
 /**
  * Whether a 'started' frame proves *this* command was reattached to.
@@ -202,6 +203,7 @@ export class CommandHandle {
      * The final execution result. Drains the stream if not already exhausted.
      */
     get result() {
+        addSandboxMetadata(this._sandbox.id);
         return this._getResult();
     }
     async _getResult() {
@@ -265,6 +267,7 @@ export class CommandHandle {
      * - After kill():                  no reconnect, error propagates
      */
     async *[Symbol.asyncIterator]() {
+        addSandboxMetadata(this._sandbox.id);
         this._reconnectAttempts = 0;
         while (true) {
             try {
@@ -280,6 +283,8 @@ export class CommandHandle {
                             chunk.offset + new TextEncoder().encode(chunk.data).length;
                         this._onStderr?.(chunk.data);
                     }
+                    // Each resumption runs in the consumer's current context.
+                    addSandboxMetadata(this._sandbox.id);
                     yield chunk;
                 }
                 return; // Stream ended normally (exit message received)
@@ -322,6 +327,7 @@ export class CommandHandle {
      * subsequently yield an exit message with a non-zero exit code.
      */
     kill() {
+        addSandboxMetadata(this._sandbox.id);
         if (this._control) {
             this._control.sendKill();
         }
@@ -330,6 +336,7 @@ export class CommandHandle {
      * Write data to the command's stdin.
      */
     sendInput(data) {
+        addSandboxMetadata(this._sandbox.id);
         if (this._stdinClosed) {
             throw new LangSmithSandboxOperationError("stdin is closed for this command. Non-PTY commands close stdin by " +
                 "default so a command that reads it sees EOF instead of hanging; " +
@@ -347,6 +354,7 @@ export class CommandHandle {
      * EOT byte (0x04) with {@link sendInput} instead.
      */
     closeInput() {
+        addSandboxMetadata(this._sandbox.id);
         if (this._pty || this._stdinClosed)
             return;
         this._stdinClosed = true;