langsmith 0.1.60 → 0.1.62

package/dist/client.cjs

@@ -116,11 +116,15 @@ class Queue {
         return this.items.length;
     }
     push(item) {
-        // this.items.push is synchronous with promise creation:
-        // https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/Promise/Promise
-        return new Promise((resolve) => {
-            this.items.push([item, resolve]);
-        });
+        let itemPromiseResolve;
+        const itemPromise = new Promise((resolve) => {
+            // Setting itemPromiseResolve is synchronous with promise creation:
+            // https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/Promise/Promise
+            itemPromiseResolve = resolve;
+        });
+        // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+        this.items.push([item, itemPromiseResolve, itemPromise]);
+        return itemPromise;
     }
     pop(upToN) {
         if (upToN < 1) {
@@ -270,6 +274,12 @@ class Client {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "blockOnRootRunFinalization", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
         const defaultConfig = Client.getDefaultClientConfig();
         this.tracingSampleRate = getTracingSamplingRate();
         this.apiUrl = trimQuotes(config.apiUrl ?? defaultConfig.apiUrl) ?? "";
@@ -292,6 +302,8 @@ class Client {
         this.hideOutputs =
             config.hideOutputs ?? config.anonymizer ?? defaultConfig.hideOutputs;
         this.autoBatchTracing = config.autoBatchTracing ?? this.autoBatchTracing;
+        this.blockOnRootRunFinalization =
+            config.blockOnRootRunFinalization ?? this.blockOnRootRunFinalization;
         this.pendingAutoBatchedRunLimit =
             config.pendingAutoBatchedRunLimit ?? this.pendingAutoBatchedRunLimit;
         this.fetchOptions = config.fetchOptions || {};
@@ -698,7 +710,9 @@ class Client {
         if (this.autoBatchTracing &&
             data.trace_id !== undefined &&
             data.dotted_order !== undefined) {
-            if (run.end_time !== undefined && data.parent_run_id === undefined) {
+            if (run.end_time !== undefined &&
+                data.parent_run_id === undefined &&
+                this.blockOnRootRunFinalization) {
                 // Trigger a batch as soon as a root trace ends and block to ensure trace finishes
                 // in serverless environments.
                 await this.processRunOperation({ action: "update", item: data }, true);
@@ -2656,5 +2670,29 @@ class Client {
             throw new Error(`Invalid public ${kind} URL or token: ${urlOrToken}`);
         }
     }
+    /**
+     * Awaits all pending trace batches. Useful for environments where
+     * you need to be sure that all tracing requests finish before execution ends,
+     * such as serverless environments.
+     *
+     * @example
+     * ```
+     * import { Client } from "langsmith";
+     *
+     * const client = new Client();
+     *
+     * try {
+     *   // Tracing happens here
+     *   ...
+     * } finally {
+     *   await client.awaitPendingTraceBatches();
+     * }
+     * ```
+     *
+     * @returns A promise that resolves once all currently pending traces have sent.
+     */
+    awaitPendingTraceBatches() {
+        return Promise.all(this.autoBatchQueue.items.map(([, , promise]) => promise));
+    }
 }
 exports.Client = Client;

package/dist/client.d.ts

@@ -12,6 +12,7 @@ export interface ClientConfig {
     hideOutputs?: boolean | ((outputs: KVMap) => KVMap);
     autoBatchTracing?: boolean;
     pendingAutoBatchedRunLimit?: number;
+    blockOnRootRunFinalization?: boolean;
     fetchOptions?: RequestInit;
 }
 /**
@@ -157,7 +158,7 @@ export type CreateExampleOptions = {
     sourceRunId?: string;
 };
 export declare class Queue<T> {
-    items: [T, () => void][];
+    items: [T, () => void, Promise<void>][];
     get size(): number;
     push(item: T): Promise<void>;
     pop(upToN: number): [T[], () => void];
@@ -185,6 +186,7 @@ export declare class Client {
     private serverInfo;
     private fetchOptions;
     private settings;
+    private blockOnRootRunFinalization;
     constructor(config?: ClientConfig);
     static getDefaultClientConfig(): {
         apiUrl: string;
@@ -736,5 +738,27 @@ export declare class Client {
         datasetName?: string;
     }): Promise<void>;
     private parseTokenOrUrl;
+    /**
+     * Awaits all pending trace batches. Useful for environments where
+     * you need to be sure that all tracing requests finish before execution ends,
+     * such as serverless environments.
+     *
+     * @example
+     * ```
+     * import { Client } from "langsmith";
+     *
+     * const client = new Client();
+     *
+     * try {
+     *   // Tracing happens here
+     *   ...
+     * } finally {
+     *   await client.awaitPendingTraceBatches();
+     * }
+     * ```
+     *
+     * @returns A promise that resolves once all currently pending traces have sent.
+     */
+    awaitPendingTraceBatches(): Promise<void[]>;
 }
 export {};

package/dist/client.js

@@ -90,11 +90,15 @@ export class Queue {
         return this.items.length;
     }
     push(item) {
-        // this.items.push is synchronous with promise creation:
-        // https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/Promise/Promise
-        return new Promise((resolve) => {
-            this.items.push([item, resolve]);
-        });
+        let itemPromiseResolve;
+        const itemPromise = new Promise((resolve) => {
+            // Setting itemPromiseResolve is synchronous with promise creation:
+            // https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/Promise/Promise
+            itemPromiseResolve = resolve;
+        });
+        // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+        this.items.push([item, itemPromiseResolve, itemPromise]);
+        return itemPromise;
     }
     pop(upToN) {
         if (upToN < 1) {
@@ -243,6 +247,12 @@ export class Client {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "blockOnRootRunFinalization", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
         const defaultConfig = Client.getDefaultClientConfig();
         this.tracingSampleRate = getTracingSamplingRate();
         this.apiUrl = trimQuotes(config.apiUrl ?? defaultConfig.apiUrl) ?? "";
@@ -265,6 +275,8 @@ export class Client {
         this.hideOutputs =
             config.hideOutputs ?? config.anonymizer ?? defaultConfig.hideOutputs;
         this.autoBatchTracing = config.autoBatchTracing ?? this.autoBatchTracing;
+        this.blockOnRootRunFinalization =
+            config.blockOnRootRunFinalization ?? this.blockOnRootRunFinalization;
         this.pendingAutoBatchedRunLimit =
             config.pendingAutoBatchedRunLimit ?? this.pendingAutoBatchedRunLimit;
         this.fetchOptions = config.fetchOptions || {};
@@ -671,7 +683,9 @@ export class Client {
         if (this.autoBatchTracing &&
             data.trace_id !== undefined &&
             data.dotted_order !== undefined) {
-            if (run.end_time !== undefined && data.parent_run_id === undefined) {
+            if (run.end_time !== undefined &&
+                data.parent_run_id === undefined &&
+                this.blockOnRootRunFinalization) {
                 // Trigger a batch as soon as a root trace ends and block to ensure trace finishes
                 // in serverless environments.
                 await this.processRunOperation({ action: "update", item: data }, true);
@@ -2629,4 +2643,28 @@ export class Client {
             throw new Error(`Invalid public ${kind} URL or token: ${urlOrToken}`);
         }
     }
+    /**
+     * Awaits all pending trace batches. Useful for environments where
+     * you need to be sure that all tracing requests finish before execution ends,
+     * such as serverless environments.
+     *
+     * @example
+     * ```
+     * import { Client } from "langsmith";
+     *
+     * const client = new Client();
+     *
+     * try {
+     *   // Tracing happens here
+     *   ...
+     * } finally {
+     *   await client.awaitPendingTraceBatches();
+     * }
+     * ```
+     *
+     * @returns A promise that resolves once all currently pending traces have sent.
+     */
+    awaitPendingTraceBatches() {
+        return Promise.all(this.autoBatchQueue.items.map(([, , promise]) => promise));
+    }
 }

package/dist/index.cjs

@@ -8,4 +8,4 @@ Object.defineProperty(exports, "RunTree", { enumerable: true, get: function () {
 var fetch_js_1 = require("./singletons/fetch.cjs");
 Object.defineProperty(exports, "overrideFetchImplementation", { enumerable: true, get: function () { return fetch_js_1.overrideFetchImplementation; } });
 // Update using yarn bump-version
-exports.__version__ = "0.1.60";
+exports.__version__ = "0.1.62";

package/dist/index.d.ts

@@ -2,4 +2,4 @@ export { Client, type ClientConfig } from "./client.js";
 export type { Dataset, Example, TracerSession, Run, Feedback, RetrieverOutput, } from "./schemas.js";
 export { RunTree, type RunTreeConfig } from "./run_trees.js";
 export { overrideFetchImplementation } from "./singletons/fetch.js";
-export declare const __version__ = "0.1.60";
+export declare const __version__ = "0.1.62";

package/dist/index.js

@@ -2,4 +2,4 @@ export { Client } from "./client.js";
 export { RunTree } from "./run_trees.js";
 export { overrideFetchImplementation } from "./singletons/fetch.js";
 // Update using yarn bump-version
-export const __version__ = "0.1.60";
+export const __version__ = "0.1.62";

package/dist/run_trees.cjs

@@ -364,31 +364,41 @@ class RunTree {
         return persistedRun;
     }
     async postRun(excludeChildRuns = true) {
-        const runtimeEnv = await (0, env_js_1.getRuntimeEnvironment)();
-        const runCreate = await this._convertToCreate(this, runtimeEnv, true);
-        await this.client.createRun(runCreate);
-        if (!excludeChildRuns) {
-            (0, warn_js_1.warnOnce)("Posting with excludeChildRuns=false is deprecated and will be removed in a future version.");
-            for (const childRun of this.child_runs) {
-                await childRun.postRun(false);
+        try {
+            const runtimeEnv = await (0, env_js_1.getRuntimeEnvironment)();
+            const runCreate = await this._convertToCreate(this, runtimeEnv, true);
+            await this.client.createRun(runCreate);
+            if (!excludeChildRuns) {
+                (0, warn_js_1.warnOnce)("Posting with excludeChildRuns=false is deprecated and will be removed in a future version.");
+                for (const childRun of this.child_runs) {
+                    await childRun.postRun(false);
+                }
             }
         }
+        catch (error) {
+            console.error(`Error in postRun for run ${this.id}:`, error);
+        }
     }
     async patchRun() {
-        const runUpdate = {
-            end_time: this.end_time,
-            error: this.error,
-            inputs: this.inputs,
-            outputs: this.outputs,
-            parent_run_id: this.parent_run?.id,
-            reference_example_id: this.reference_example_id,
-            extra: this.extra,
-            events: this.events,
-            dotted_order: this.dotted_order,
-            trace_id: this.trace_id,
-            tags: this.tags,
-        };
-        await this.client.updateRun(this.id, runUpdate);
+        try {
+            const runUpdate = {
+                end_time: this.end_time,
+                error: this.error,
+                inputs: this.inputs,
+                outputs: this.outputs,
+                parent_run_id: this.parent_run?.id,
+                reference_example_id: this.reference_example_id,
+                extra: this.extra,
+                events: this.events,
+                dotted_order: this.dotted_order,
+                trace_id: this.trace_id,
+                tags: this.tags,
+            };
+            await this.client.updateRun(this.id, runUpdate);
+        }
+        catch (error) {
+            console.error(`Error in patchRun for run ${this.id}`, error);
+        }
     }
     toJSON() {
         return this._convertToCreate(this, undefined, false);

package/dist/run_trees.js

@@ -337,31 +337,41 @@ export class RunTree {
         return persistedRun;
     }
     async postRun(excludeChildRuns = true) {
-        const runtimeEnv = await getRuntimeEnvironment();
-        const runCreate = await this._convertToCreate(this, runtimeEnv, true);
-        await this.client.createRun(runCreate);
-        if (!excludeChildRuns) {
-            warnOnce("Posting with excludeChildRuns=false is deprecated and will be removed in a future version.");
-            for (const childRun of this.child_runs) {
-                await childRun.postRun(false);
+        try {
+            const runtimeEnv = await getRuntimeEnvironment();
+            const runCreate = await this._convertToCreate(this, runtimeEnv, true);
+            await this.client.createRun(runCreate);
+            if (!excludeChildRuns) {
+                warnOnce("Posting with excludeChildRuns=false is deprecated and will be removed in a future version.");
+                for (const childRun of this.child_runs) {
+                    await childRun.postRun(false);
+                }
             }
         }
+        catch (error) {
+            console.error(`Error in postRun for run ${this.id}:`, error);
+        }
     }
     async patchRun() {
-        const runUpdate = {
-            end_time: this.end_time,
-            error: this.error,
-            inputs: this.inputs,
-            outputs: this.outputs,
-            parent_run_id: this.parent_run?.id,
-            reference_example_id: this.reference_example_id,
-            extra: this.extra,
-            events: this.events,
-            dotted_order: this.dotted_order,
-            trace_id: this.trace_id,
-            tags: this.tags,
-        };
-        await this.client.updateRun(this.id, runUpdate);
+        try {
+            const runUpdate = {
+                end_time: this.end_time,
+                error: this.error,
+                inputs: this.inputs,
+                outputs: this.outputs,
+                parent_run_id: this.parent_run?.id,
+                reference_example_id: this.reference_example_id,
+                extra: this.extra,
+                events: this.events,
+                dotted_order: this.dotted_order,
+                trace_id: this.trace_id,
+                tags: this.tags,
+            };
+            await this.client.updateRun(this.id, runUpdate);
+        }
+        catch (error) {
+            console.error(`Error in patchRun for run ${this.id}`, error);
+        }
     }
     toJSON() {
         return this._convertToCreate(this, undefined, false);

package/dist/traceable.cjs

@@ -428,13 +428,18 @@ function traceable(wrappedFunc, config) {
                     }
                     if ((0, asserts_js_1.isGenerator)(wrappedFunc) && (0, asserts_js_1.isIteratorLike)(rawOutput)) {
                         const chunks = gatherAll(rawOutput);
-                        await currentRunTree?.end(handleRunOutputs(await handleChunks(chunks.reduce((memo, { value, done }) => {
-                            if (!done || typeof value !== "undefined") {
-                                memo.push(value);
-                            }
-                            return memo;
-                        }, []))));
-                        await handleEnd();
+                        try {
+                            await currentRunTree?.end(handleRunOutputs(await handleChunks(chunks.reduce((memo, { value, done }) => {
+                                if (!done || typeof value !== "undefined") {
+                                    memo.push(value);
+                                }
+                                return memo;
+                            }, []))));
+                            await handleEnd();
+                        }
+                        catch (e) {
+                            console.error("Error occurred during handleEnd:", e);
+                        }
                         return (function* () {
                             for (const ret of chunks) {
                                 if (ret.done)

package/dist/traceable.js

@@ -425,13 +425,18 @@ export function traceable(wrappedFunc, config) {
                     }
                     if (isGenerator(wrappedFunc) && isIteratorLike(rawOutput)) {
                         const chunks = gatherAll(rawOutput);
-                        await currentRunTree?.end(handleRunOutputs(await handleChunks(chunks.reduce((memo, { value, done }) => {
-                            if (!done || typeof value !== "undefined") {
-                                memo.push(value);
-                            }
-                            return memo;
-                        }, []))));
-                        await handleEnd();
+                        try {
+                            await currentRunTree?.end(handleRunOutputs(await handleChunks(chunks.reduce((memo, { value, done }) => {
+                                if (!done || typeof value !== "undefined") {
+                                    memo.push(value);
+                                }
+                                return memo;
+                            }, []))));
+                            await handleEnd();
+                        }
+                        catch (e) {
+                            console.error("Error occurred during handleEnd:", e);
+                        }
                         return (function* () {
                             for (const ret of chunks) {
                                 if (ret.done)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langsmith",
-  "version": "0.1.60",
+  "version": "0.1.62",
   "description": "Client library to connect to the LangSmith LLM Tracing and Evaluation Platform.",
   "packageManager": "yarn@1.22.19",
   "files": [