langsmith 0.5.8 → 0.5.10

package/dist/client.cjs

@@ -50,6 +50,26 @@ const index_js_2 = require("./utils/prompt_cache/index.cjs");
 const fsUtils = __importStar(require("./utils/fs.cjs"));
 const fetch_js_1 = require("./singletons/fetch.cjs");
 const index_js_3 = require("./utils/fast-safe-stringify/index.cjs");
+/**
+ * Catches timestamps without a timezone suffix.
+ */
+function _ensureUTCTimestamp(ts) {
+    if (typeof ts === "string" &&
+        ts.length > 0 &&
+        !ts.includes("Z") &&
+        !ts.includes("+") &&
+        !ts.includes("-", 10)) {
+        return ts + "Z";
+    }
+    return ts;
+}
+function _normalizeRunTimestamps(run) {
+    return {
+        ...run,
+        start_time: _ensureUTCTimestamp(run.start_time),
+        end_time: _ensureUTCTimestamp(run.end_time),
+    };
+}
 function mergeRuntimeEnvIntoRun(run, cachedEnvVars, omitTracedRuntimeInfo) {
     if (omitTracedRuntimeInfo) {
         return run;
@@ -1542,7 +1562,7 @@ class Client {
     }
     async readRun(runId, { loadChildRuns } = { loadChildRuns: false }) {
         (0, _uuid_js_1.assertUuid)(runId);
-        let run = await this._get(`/runs/${runId}`);
+        let run = _normalizeRunTimestamps(await this._get(`/runs/${runId}`));
         if (loadChildRuns) {
             run = await this._loadChildRuns(run);
         }
@@ -1761,20 +1781,21 @@ class Client {
         }
         let runsYielded = 0;
         for await (const runs of this._getCursorPaginatedList("/runs/query", body)) {
+            const normalized = runs.map(_normalizeRunTimestamps);
             if (limit) {
                 if (runsYielded >= limit) {
                     break;
                 }
-                if (runs.length + runsYielded > limit) {
-                    const newRuns = runs.slice(0, limit - runsYielded);
+                if (normalized.length + runsYielded > limit) {
+                    const newRuns = normalized.slice(0, limit - runsYielded);
                     yield* newRuns;
                     break;
                 }
-                runsYielded += runs.length;
-                yield* runs;
+                runsYielded += normalized.length;
+                yield* normalized;
             }
             else {
-                yield* runs;
+                yield* normalized;
             }
         }
     }
@@ -1891,7 +1912,8 @@ class Client {
         }
         const threadsMap = new Map();
         for await (const runs of this._getCursorPaginatedList("/runs/query", bodyQuery)) {
-            for (const run of runs) {
+            for (const raw of runs) {
+                const run = _normalizeRunTimestamps(raw);
                 const tid = run.thread_id;
                 if (tid) {
                     const list = threadsMap.get(tid) ?? [];
@@ -2064,8 +2086,8 @@ class Client {
             await (0, error_js_1.raiseForStatus)(res, "list shared runs");
             return res;
         });
-        const runs = await response.json();
-        return runs;
+        const runs = (await response.json());
+        return runs.map(_normalizeRunTimestamps);
     }
     async readDatasetSharedSchema(datasetId, datasetName) {
         if (!datasetId && !datasetName) {
@@ -3714,7 +3736,8 @@ class Client {
             await (0, error_js_1.raiseForStatus)(res, "get run from annotation queue");
             return res;
         });
-        return response.json();
+        const run = await response.json();
+        return _normalizeRunTimestamps(run);
     }
     /**
      * Delete a run from an an annotation queue.

package/dist/client.js

@@ -13,6 +13,26 @@ import { promptCacheSingleton, } from "./utils/prompt_cache/index.js";
 import * as fsUtils from "./utils/fs.js";
 import { _shouldStreamForGlobalFetchImplementation, _getFetchImplementation, } from "./singletons/fetch.js";
 import { serialize as serializePayloadForTracing } from "./utils/fast-safe-stringify/index.js";
+/**
+ * Catches timestamps without a timezone suffix.
+ */
+function _ensureUTCTimestamp(ts) {
+    if (typeof ts === "string" &&
+        ts.length > 0 &&
+        !ts.includes("Z") &&
+        !ts.includes("+") &&
+        !ts.includes("-", 10)) {
+        return ts + "Z";
+    }
+    return ts;
+}
+function _normalizeRunTimestamps(run) {
+    return {
+        ...run,
+        start_time: _ensureUTCTimestamp(run.start_time),
+        end_time: _ensureUTCTimestamp(run.end_time),
+    };
+}
 export function mergeRuntimeEnvIntoRun(run, cachedEnvVars, omitTracedRuntimeInfo) {
     if (omitTracedRuntimeInfo) {
         return run;
@@ -1504,7 +1524,7 @@ export class Client {
     }
     async readRun(runId, { loadChildRuns } = { loadChildRuns: false }) {
         assertUuid(runId);
-        let run = await this._get(`/runs/${runId}`);
+        let run = _normalizeRunTimestamps(await this._get(`/runs/${runId}`));
         if (loadChildRuns) {
             run = await this._loadChildRuns(run);
         }
@@ -1723,20 +1743,21 @@ export class Client {
         }
         let runsYielded = 0;
         for await (const runs of this._getCursorPaginatedList("/runs/query", body)) {
+            const normalized = runs.map(_normalizeRunTimestamps);
             if (limit) {
                 if (runsYielded >= limit) {
                     break;
                 }
-                if (runs.length + runsYielded > limit) {
-                    const newRuns = runs.slice(0, limit - runsYielded);
+                if (normalized.length + runsYielded > limit) {
+                    const newRuns = normalized.slice(0, limit - runsYielded);
                     yield* newRuns;
                     break;
                 }
-                runsYielded += runs.length;
-                yield* runs;
+                runsYielded += normalized.length;
+                yield* normalized;
             }
             else {
-                yield* runs;
+                yield* normalized;
             }
         }
     }
@@ -1853,7 +1874,8 @@ export class Client {
         }
         const threadsMap = new Map();
         for await (const runs of this._getCursorPaginatedList("/runs/query", bodyQuery)) {
-            for (const run of runs) {
+            for (const raw of runs) {
+                const run = _normalizeRunTimestamps(raw);
                 const tid = run.thread_id;
                 if (tid) {
                     const list = threadsMap.get(tid) ?? [];
@@ -2026,8 +2048,8 @@ export class Client {
             await raiseForStatus(res, "list shared runs");
             return res;
         });
-        const runs = await response.json();
-        return runs;
+        const runs = (await response.json());
+        return runs.map(_normalizeRunTimestamps);
     }
     async readDatasetSharedSchema(datasetId, datasetName) {
         if (!datasetId && !datasetName) {
@@ -3676,7 +3698,8 @@ export class Client {
             await raiseForStatus(res, "get run from annotation queue");
             return res;
         });
-        return response.json();
+        const run = await response.json();
+        return _normalizeRunTimestamps(run);
     }
     /**
      * Delete a run from an an annotation queue.

package/dist/experimental/sandbox/client.cjs

@@ -106,6 +106,13 @@ class SandboxClient {
             headers,
         }));
     }
+    /**
+     * Get the API key for WebSocket authentication.
+     * @internal
+     */
+    getApiKey() {
+        return this._apiKey;
+    }
     // =========================================================================
     // Volume Operations
     // =========================================================================
@@ -530,21 +537,24 @@ class SandboxClient {
      * ```
      */
     async createSandbox(templateName, options = {}) {
-        const { name, timeout = 30 } = options;
+        const { name, timeout = 30, waitForReady = true } = options;
         const url = `${this._baseUrl}/boxes`;
         const payload = {
             template_name: templateName,
-            wait_for_ready: true,
-            timeout,
+            wait_for_ready: waitForReady,
         };
+        if (waitForReady) {
+            payload.timeout = timeout;
+        }
         if (name) {
             payload.name = name;
         }
+        const httpTimeout = waitForReady ? (timeout + 30) * 1000 : 30 * 1000;
         const response = await this._fetch(url, {
             method: "POST",
             headers: { "Content-Type": "application/json" },
             body: JSON.stringify(payload),
-            signal: AbortSignal.timeout((timeout + 30) * 1000),
+            signal: AbortSignal.timeout(httpTimeout),
         });
         if (!response.ok) {
             await (0, helpers_js_1.handleSandboxCreationError)(response);
@@ -635,5 +645,64 @@ class SandboxClient {
             await (0, helpers_js_1.handleClientHttpError)(response);
         }
     }
+    /**
+     * Get the provisioning status of a sandbox.
+     *
+     * This is a lightweight endpoint designed for polling during async creation.
+     * Use this instead of getSandbox() when you only need the status.
+     *
+     * @param name - Sandbox name.
+     * @returns ResourceStatus with status and optional status_message.
+     * @throws LangSmithResourceNotFoundError if sandbox not found.
+     */
+    async getSandboxStatus(name) {
+        const url = `${this._baseUrl}/boxes/${encodeURIComponent(name)}/status`;
+        const response = await this._fetch(url);
+        if (!response.ok) {
+            if (response.status === 404) {
+                throw new errors_js_1.LangSmithResourceNotFoundError(`Sandbox '${name}' not found`, "sandbox");
+            }
+            await (0, helpers_js_1.handleClientHttpError)(response);
+        }
+        return (await response.json());
+    }
+    /**
+     * Wait for a sandbox to become ready.
+     *
+     * Polls getSandboxStatus() until the sandbox reaches "ready" or "failed" status,
+     * then returns the full Sandbox object.
+     *
+     * @param name - Sandbox name.
+     * @param options - Polling options (timeout, pollInterval).
+     * @returns Ready Sandbox.
+     * @throws LangSmithResourceCreationError if sandbox status becomes "failed".
+     * @throws LangSmithResourceTimeoutError if timeout expires while still provisioning.
+     * @throws LangSmithResourceNotFoundError if sandbox not found.
+     *
+     * @example
+     * ```typescript
+     * const sandbox = await client.createSandbox("python-sandbox", { waitForReady: false });
+     * // ... do other work ...
+     * const readySandbox = await client.waitForSandbox(sandbox.name);
+     * ```
+     */
+    async waitForSandbox(name, options = {}) {
+        const { timeout = 120, pollInterval = 1.0 } = options;
+        const deadline = Date.now() + timeout * 1000;
+        let lastStatus = "provisioning";
+        while (Date.now() < deadline) {
+            const statusResult = await this.getSandboxStatus(name);
+            lastStatus = statusResult.status;
+            if (statusResult.status === "ready") {
+                return this.getSandbox(name);
+            }
+            if (statusResult.status === "failed") {
+                throw new errors_js_1.LangSmithResourceCreationError(statusResult.status_message ?? `Sandbox '${name}' creation failed`, "sandbox");
+            }
+            // Wait before polling again
+            await new Promise((resolve) => setTimeout(resolve, pollInterval * 1000));
+        }
+        throw new errors_js_1.LangSmithResourceTimeoutError(`Sandbox '${name}' did not become ready within ${timeout}s`, "sandbox", lastStatus);
+    }
 }
 exports.SandboxClient = SandboxClient;

package/dist/experimental/sandbox/client.d.ts

@@ -1,7 +1,7 @@
 /**
  * Main SandboxClient class for interacting with the sandbox server API.
  */
-import type { CreatePoolOptions, CreateSandboxOptions, CreateTemplateOptions, CreateVolumeOptions, Pool, SandboxClientConfig, SandboxTemplate, UpdatePoolOptions, UpdateTemplateOptions, UpdateVolumeOptions, Volume } from "./types.js";
+import type { CreatePoolOptions, CreateSandboxOptions, CreateTemplateOptions, CreateVolumeOptions, Pool, ResourceStatus, SandboxClientConfig, SandboxTemplate, UpdatePoolOptions, UpdateTemplateOptions, UpdateVolumeOptions, Volume, WaitForSandboxOptions } from "./types.js";
 import { Sandbox } from "./sandbox.js";
 /**
  * Client for interacting with the Sandbox Server API.
@@ -237,4 +237,36 @@ export declare class SandboxClient {
      * @throws LangSmithResourceNotFoundError if sandbox not found.
      */
     deleteSandbox(name: string): Promise<void>;
+    /**
+     * Get the provisioning status of a sandbox.
+     *
+     * This is a lightweight endpoint designed for polling during async creation.
+     * Use this instead of getSandbox() when you only need the status.
+     *
+     * @param name - Sandbox name.
+     * @returns ResourceStatus with status and optional status_message.
+     * @throws LangSmithResourceNotFoundError if sandbox not found.
+     */
+    getSandboxStatus(name: string): Promise<ResourceStatus>;
+    /**
+     * Wait for a sandbox to become ready.
+     *
+     * Polls getSandboxStatus() until the sandbox reaches "ready" or "failed" status,
+     * then returns the full Sandbox object.
+     *
+     * @param name - Sandbox name.
+     * @param options - Polling options (timeout, pollInterval).
+     * @returns Ready Sandbox.
+     * @throws LangSmithResourceCreationError if sandbox status becomes "failed".
+     * @throws LangSmithResourceTimeoutError if timeout expires while still provisioning.
+     * @throws LangSmithResourceNotFoundError if sandbox not found.
+     *
+     * @example
+     * ```typescript
+     * const sandbox = await client.createSandbox("python-sandbox", { waitForReady: false });
+     * // ... do other work ...
+     * const readySandbox = await client.waitForSandbox(sandbox.name);
+     * ```
+     */
+    waitForSandbox(name: string, options?: WaitForSandboxOptions): Promise<Sandbox>;
 }

package/dist/experimental/sandbox/client.js

@@ -5,7 +5,7 @@ import { getLangSmithEnvironmentVariable } from "../../utils/env.js";
 import { _getFetchImplementation } from "../../singletons/fetch.js";
 import { AsyncCaller } from "../../utils/async_caller.js";
 import { Sandbox } from "./sandbox.js";
-import { LangSmithResourceNameConflictError, LangSmithResourceNotFoundError, LangSmithSandboxAPIError, } from "./errors.js";
+import { LangSmithResourceCreationError, LangSmithResourceNameConflictError, LangSmithResourceNotFoundError, LangSmithResourceTimeoutError, LangSmithSandboxAPIError, } from "./errors.js";
 import { handleClientHttpError, handleConflictError, handlePoolError, handleResourceInUseError, handleSandboxCreationError, handleVolumeCreationError, } from "./helpers.js";
 /**
  * Get the default sandbox API endpoint from environment.
@@ -103,6 +103,13 @@ export class SandboxClient {
             headers,
         }));
     }
+    /**
+     * Get the API key for WebSocket authentication.
+     * @internal
+     */
+    getApiKey() {
+        return this._apiKey;
+    }
     // =========================================================================
     // Volume Operations
     // =========================================================================
@@ -527,21 +534,24 @@ export class SandboxClient {
      * ```
      */
     async createSandbox(templateName, options = {}) {
-        const { name, timeout = 30 } = options;
+        const { name, timeout = 30, waitForReady = true } = options;
         const url = `${this._baseUrl}/boxes`;
         const payload = {
             template_name: templateName,
-            wait_for_ready: true,
-            timeout,
+            wait_for_ready: waitForReady,
         };
+        if (waitForReady) {
+            payload.timeout = timeout;
+        }
         if (name) {
             payload.name = name;
         }
+        const httpTimeout = waitForReady ? (timeout + 30) * 1000 : 30 * 1000;
         const response = await this._fetch(url, {
             method: "POST",
             headers: { "Content-Type": "application/json" },
             body: JSON.stringify(payload),
-            signal: AbortSignal.timeout((timeout + 30) * 1000),
+            signal: AbortSignal.timeout(httpTimeout),
         });
         if (!response.ok) {
             await handleSandboxCreationError(response);
@@ -632,4 +642,63 @@ export class SandboxClient {
             await handleClientHttpError(response);
         }
     }
+    /**
+     * Get the provisioning status of a sandbox.
+     *
+     * This is a lightweight endpoint designed for polling during async creation.
+     * Use this instead of getSandbox() when you only need the status.
+     *
+     * @param name - Sandbox name.
+     * @returns ResourceStatus with status and optional status_message.
+     * @throws LangSmithResourceNotFoundError if sandbox not found.
+     */
+    async getSandboxStatus(name) {
+        const url = `${this._baseUrl}/boxes/${encodeURIComponent(name)}/status`;
+        const response = await this._fetch(url);
+        if (!response.ok) {
+            if (response.status === 404) {
+                throw new LangSmithResourceNotFoundError(`Sandbox '${name}' not found`, "sandbox");
+            }
+            await handleClientHttpError(response);
+        }
+        return (await response.json());
+    }
+    /**
+     * Wait for a sandbox to become ready.
+     *
+     * Polls getSandboxStatus() until the sandbox reaches "ready" or "failed" status,
+     * then returns the full Sandbox object.
+     *
+     * @param name - Sandbox name.
+     * @param options - Polling options (timeout, pollInterval).
+     * @returns Ready Sandbox.
+     * @throws LangSmithResourceCreationError if sandbox status becomes "failed".
+     * @throws LangSmithResourceTimeoutError if timeout expires while still provisioning.
+     * @throws LangSmithResourceNotFoundError if sandbox not found.
+     *
+     * @example
+     * ```typescript
+     * const sandbox = await client.createSandbox("python-sandbox", { waitForReady: false });
+     * // ... do other work ...
+     * const readySandbox = await client.waitForSandbox(sandbox.name);
+     * ```
+     */
+    async waitForSandbox(name, options = {}) {
+        const { timeout = 120, pollInterval = 1.0 } = options;
+        const deadline = Date.now() + timeout * 1000;
+        let lastStatus = "provisioning";
+        while (Date.now() < deadline) {
+            const statusResult = await this.getSandboxStatus(name);
+            lastStatus = statusResult.status;
+            if (statusResult.status === "ready") {
+                return this.getSandbox(name);
+            }
+            if (statusResult.status === "failed") {
+                throw new LangSmithResourceCreationError(statusResult.status_message ?? `Sandbox '${name}' creation failed`, "sandbox");
+            }
+            // Wait before polling again
+            await new Promise((resolve) => setTimeout(resolve, pollInterval * 1000));
+        }
+        throw new LangSmithResourceTimeoutError(`Sandbox '${name}' did not become ready within ${timeout}s`, "sandbox", lastStatus);
+    }
 }