langsmith 0.4.0 → 0.4.2

package/dist/index.cjs

@@ -13,4 +13,4 @@ var uuid_js_1 = require("./uuid.cjs");
 Object.defineProperty(exports, "uuid7", { enumerable: true, get: function () { return uuid_js_1.uuid7; } });
 Object.defineProperty(exports, "uuid7FromTime", { enumerable: true, get: function () { return uuid_js_1.uuid7FromTime; } });
 // Update using yarn bump-version
-exports.__version__ = "0.4.0";
+exports.__version__ = "0.4.2";

package/dist/index.d.ts

@@ -4,4 +4,4 @@ export { RunTree, type RunTreeConfig } from "./run_trees.js";
 export { overrideFetchImplementation } from "./singletons/fetch.js";
 export { getDefaultProjectName } from "./utils/project.js";
 export { uuid7, uuid7FromTime } from "./uuid.js";
-export declare const __version__ = "0.4.0";
+export declare const __version__ = "0.4.2";

package/dist/index.js

@@ -4,4 +4,4 @@ export { overrideFetchImplementation } from "./singletons/fetch.js";
 export { getDefaultProjectName } from "./utils/project.js";
 export { uuid7, uuid7FromTime } from "./uuid.js";
 // Update using yarn bump-version
-export const __version__ = "0.4.0";
+export const __version__ = "0.4.2";

package/dist/run_trees.cjs

@@ -676,6 +676,7 @@ class RunTree {
                     await childRun.postRun(false);
                 }
             }
+            this.child_runs = [];
         }
         catch (error) {
             console.error(`Error in postRun for run ${this.id}:`, error);
@@ -757,6 +758,7 @@ class RunTree {
                 console.error(`Error in patchRun for run ${this.id}`, error);
             }
         }
+        this.child_runs = [];
     }
     toJSON() {
         return this._convertToCreate(this, undefined, false);

package/dist/run_trees.js

@@ -670,6 +670,7 @@ export class RunTree {
                     await childRun.postRun(false);
                 }
             }
+            this.child_runs = [];
         }
         catch (error) {
             console.error(`Error in postRun for run ${this.id}:`, error);
@@ -751,6 +752,7 @@ export class RunTree {
                 console.error(`Error in patchRun for run ${this.id}`, error);
             }
         }
+        this.child_runs = [];
     }
     toJSON() {
         return this._convertToCreate(this, undefined, false);

package/dist/traceable.cjs

@@ -108,12 +108,14 @@ const _extractUsage = (runData) => {
     return runData.outputs?.usage_metadata ?? usageMetadataFromMetadata;
 };
 async function handleEnd(params) {
-    const { runTree, on_end, postRunPromise, deferredInputs } = params;
+    const { runTree, on_end, postRunPromise, deferredInputs, skipChildPromiseDelay, } = params;
     const onEnd = on_end;
     if (onEnd) {
         onEnd(runTree);
     }
-    await postRunPromise;
+    if (!skipChildPromiseDelay) {
+        await postRunPromise;
+    }
     if (deferredInputs) {
         await runTree?.postRun();
     }
@@ -148,7 +150,7 @@ function isAsyncFn(fn) {
 }
 // Note: This mutates the run tree
 async function handleRunOutputs(params) {
-    const { runTree, rawOutputs, processOutputsFn, on_end, postRunPromise, deferredInputs, } = params;
+    const { runTree, rawOutputs, processOutputsFn, on_end, postRunPromise, deferredInputs, skipChildPromiseDelay, } = params;
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     let outputs;
     if ((0, asserts_js_1.isKVMap)(rawOutputs)) {
@@ -157,7 +159,8 @@ async function handleRunOutputs(params) {
     else {
         outputs = { outputs: rawOutputs };
     }
-    const childRunEndPromises = (0, run_trees_js_1.isRunTree)(runTree) &&
+    const childRunEndPromises = !skipChildPromiseDelay &&
+        (0, run_trees_js_1.isRunTree)(runTree) &&
         constants_js_1._LC_CHILD_RUN_END_PROMISES_KEY in runTree &&
         Array.isArray(runTree[constants_js_1._LC_CHILD_RUN_END_PROMISES_KEY])
         ? Promise.all(runTree[constants_js_1._LC_CHILD_RUN_END_PROMISES_KEY] ?? [])
@@ -609,6 +612,7 @@ function traceable(wrappedFunc, config) {
             }
             async function* wrapAsyncIteratorForTracing(iterator, snapshot) {
                 let finished = false;
+                let hasError = false;
                 const chunks = [];
                 const capturedOtelContext = otel_context.active();
                 try {
@@ -632,12 +636,18 @@ function traceable(wrappedFunc, config) {
                     }
                 }
                 catch (e) {
+                    hasError = true;
                     await currentRunTree?.end(undefined, String(e));
                     throw e;
                 }
                 finally {
-                    if (!finished)
+                    if (!finished) {
+                        // Call return() on the original iterator to trigger cleanup
+                        if (iterator.return) {
+                            await iterator.return(undefined);
+                        }
                         await currentRunTree?.end(undefined, "Cancelled");
+                    }
                     await handleRunOutputs({
                         runTree: currentRunTree,
                         rawOutputs: await handleChunks(chunks),
@@ -645,6 +655,7 @@ function traceable(wrappedFunc, config) {
                         on_end,
                         postRunPromise,
                         deferredInputs,
+                        skipChildPromiseDelay: hasError || !finished,
                     });
                 }
             }
@@ -759,17 +770,14 @@ function traceable(wrappedFunc, config) {
                         return rawOutput;
                     }
                 }, async (error) => {
-                    if ((0, run_trees_js_1.isRunTree)(currentRunTree) &&
-                        constants_js_1._LC_CHILD_RUN_END_PROMISES_KEY in currentRunTree &&
-                        Array.isArray(currentRunTree[constants_js_1._LC_CHILD_RUN_END_PROMISES_KEY])) {
-                        await Promise.all(currentRunTree[constants_js_1._LC_CHILD_RUN_END_PROMISES_KEY] ?? []);
-                    }
+                    // Don't wait for child runs on error - fail fast
                     await currentRunTree?.end(undefined, String(error));
                     await handleEnd({
                         runTree: currentRunTree,
                         postRunPromise,
                         on_end,
                         deferredInputs,
+                        skipChildPromiseDelay: true,
                     });
                     throw error;
                 })

package/dist/traceable.js

@@ -104,12 +104,14 @@ const _extractUsage = (runData) => {
     return runData.outputs?.usage_metadata ?? usageMetadataFromMetadata;
 };
 async function handleEnd(params) {
-    const { runTree, on_end, postRunPromise, deferredInputs } = params;
+    const { runTree, on_end, postRunPromise, deferredInputs, skipChildPromiseDelay, } = params;
     const onEnd = on_end;
     if (onEnd) {
         onEnd(runTree);
     }
-    await postRunPromise;
+    if (!skipChildPromiseDelay) {
+        await postRunPromise;
+    }
     if (deferredInputs) {
         await runTree?.postRun();
     }
@@ -144,7 +146,7 @@ function isAsyncFn(fn) {
 }
 // Note: This mutates the run tree
 async function handleRunOutputs(params) {
-    const { runTree, rawOutputs, processOutputsFn, on_end, postRunPromise, deferredInputs, } = params;
+    const { runTree, rawOutputs, processOutputsFn, on_end, postRunPromise, deferredInputs, skipChildPromiseDelay, } = params;
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     let outputs;
     if (isKVMap(rawOutputs)) {
@@ -153,7 +155,8 @@ async function handleRunOutputs(params) {
     else {
         outputs = { outputs: rawOutputs };
     }
-    const childRunEndPromises = isRunTree(runTree) &&
+    const childRunEndPromises = !skipChildPromiseDelay &&
+        isRunTree(runTree) &&
         _LC_CHILD_RUN_END_PROMISES_KEY in runTree &&
         Array.isArray(runTree[_LC_CHILD_RUN_END_PROMISES_KEY])
         ? Promise.all(runTree[_LC_CHILD_RUN_END_PROMISES_KEY] ?? [])
@@ -605,6 +608,7 @@ export function traceable(wrappedFunc, config) {
             }
             async function* wrapAsyncIteratorForTracing(iterator, snapshot) {
                 let finished = false;
+                let hasError = false;
                 const chunks = [];
                 const capturedOtelContext = otel_context.active();
                 try {
@@ -628,12 +632,18 @@ export function traceable(wrappedFunc, config) {
                     }
                 }
                 catch (e) {
+                    hasError = true;
                     await currentRunTree?.end(undefined, String(e));
                     throw e;
                 }
                 finally {
-                    if (!finished)
+                    if (!finished) {
+                        // Call return() on the original iterator to trigger cleanup
+                        if (iterator.return) {
+                            await iterator.return(undefined);
+                        }
                         await currentRunTree?.end(undefined, "Cancelled");
+                    }
                     await handleRunOutputs({
                         runTree: currentRunTree,
                         rawOutputs: await handleChunks(chunks),
@@ -641,6 +651,7 @@ export function traceable(wrappedFunc, config) {
                         on_end,
                         postRunPromise,
                         deferredInputs,
+                        skipChildPromiseDelay: hasError || !finished,
                     });
                 }
             }
@@ -755,17 +766,14 @@ export function traceable(wrappedFunc, config) {
                         return rawOutput;
                     }
                 }, async (error) => {
-                    if (isRunTree(currentRunTree) &&
-                        _LC_CHILD_RUN_END_PROMISES_KEY in currentRunTree &&
-                        Array.isArray(currentRunTree[_LC_CHILD_RUN_END_PROMISES_KEY])) {
-                        await Promise.all(currentRunTree[_LC_CHILD_RUN_END_PROMISES_KEY] ?? []);
-                    }
+                    // Don't wait for child runs on error - fail fast
                     await currentRunTree?.end(undefined, String(error));
                     await handleEnd({
                         runTree: currentRunTree,
                         postRunPromise,
                         on_end,
                         deferredInputs,
+                        skipChildPromiseDelay: true,
                     });
                     throw error;
                 })

package/dist/wrappers/openai.cjs

@@ -232,6 +232,33 @@ function processChatCompletion(outputs) {
     delete result.usage;
     return result;
 }
+const getChatModelInvocationParamsFn = (provider, useResponsesApi) => {
+    return (payload) => {
+        if (typeof payload !== "object" || payload == null)
+            return undefined;
+        const params = payload;
+        const ls_stop = (typeof params.stop === "string" ? [params.stop] : params.stop) ??
+            undefined;
+        const ls_invocation_params = {};
+        for (const [key, value] of Object.entries(params)) {
+            if (TRACED_INVOCATION_KEYS.includes(key)) {
+                ls_invocation_params[key] = value;
+            }
+        }
+        if (useResponsesApi) {
+            ls_invocation_params.use_responses_api = true;
+        }
+        return {
+            ls_provider: provider,
+            ls_model_type: "chat",
+            ls_model_name: params.model,
+            ls_max_tokens: params.max_completion_tokens ?? params.max_tokens ?? undefined,
+            ls_temperature: params.temperature ?? undefined,
+            ls_stop,
+            ls_invocation_params,
+        };
+    };
+};
 /**
  * Wraps an OpenAI client's completion methods, enabling automatic LangSmith
  * tracing. Method signatures are unchanged, with the exception that you can pass
@@ -279,29 +306,7 @@ const wrapOpenAI = (openai, options) => {
         run_type: "llm",
         aggregator: chatAggregator,
         argsConfigPath: [1, "langsmithExtra"],
-        getInvocationParams: (payload) => {
-            if (typeof payload !== "object" || payload == null)
-                return undefined;
-            // we can safely do so, as the types are not exported in TSC
-            const params = payload;
-            const ls_stop = (typeof params.stop === "string" ? [params.stop] : params.stop) ??
-                undefined;
-            const ls_invocation_params = {};
-            for (const [key, value] of Object.entries(params)) {
-                if (TRACED_INVOCATION_KEYS.includes(key)) {
-                    ls_invocation_params[key] = value;
-                }
-            }
-            return {
-                ls_provider: provider,
-                ls_model_type: "chat",
-                ls_model_name: params.model,
-                ls_max_tokens: params.max_completion_tokens ?? params.max_tokens ?? undefined,
-                ls_temperature: params.temperature ?? undefined,
-                ls_stop,
-                ls_invocation_params,
-            };
-        },
+        getInvocationParams: getChatModelInvocationParamsFn(provider, false),
         processOutputs: processChatCompletion,
         ...options,
     };
@@ -415,17 +420,7 @@ const wrapOpenAI = (openai, options) => {
                 run_type: "llm",
                 aggregator: responsesAggregator,
                 argsConfigPath: [1, "langsmithExtra"],
-                getInvocationParams: (payload) => {
-                    if (typeof payload !== "object" || payload == null)
-                        return undefined;
-                    // Handle responses API parameters
-                    const params = payload;
-                    return {
-                        ls_provider: provider,
-                        ls_model_type: "chat",
-                        ls_model_name: params.model || "unknown",
-                    };
-                },
+                getInvocationParams: getChatModelInvocationParamsFn(provider, true),
                 processOutputs: processChatCompletion,
                 ...options,
             });
@@ -437,17 +432,7 @@ const wrapOpenAI = (openai, options) => {
                 run_type: "llm",
                 aggregator: responsesAggregator,
                 argsConfigPath: [1, "langsmithExtra"],
-                getInvocationParams: (payload) => {
-                    if (typeof payload !== "object" || payload == null)
-                        return undefined;
-                    // Handle responses API parameters
-                    const params = payload;
-                    return {
-                        ls_provider: provider,
-                        ls_model_type: "chat",
-                        ls_model_name: params.model || "unknown",
-                    };
-                },
+                getInvocationParams: getChatModelInvocationParamsFn(provider, true),
                 processOutputs: processChatCompletion,
                 ...options,
             });
@@ -459,17 +444,7 @@ const wrapOpenAI = (openai, options) => {
                 run_type: "llm",
                 aggregator: responsesAggregator,
                 argsConfigPath: [1, "langsmithExtra"],
-                getInvocationParams: (payload) => {
-                    if (typeof payload !== "object" || payload == null)
-                        return undefined;
-                    // Handle responses API parameters
-                    const params = payload;
-                    return {
-                        ls_provider: provider,
-                        ls_model_type: "chat",
-                        ls_model_name: params.model || "unknown",
-                    };
-                },
+                getInvocationParams: getChatModelInvocationParamsFn(provider, true),
                 processOutputs: processChatCompletion,
                 ...options,
             });

package/dist/wrappers/openai.js

@@ -229,6 +229,33 @@ function processChatCompletion(outputs) {
     delete result.usage;
     return result;
 }
+const getChatModelInvocationParamsFn = (provider, useResponsesApi) => {
+    return (payload) => {
+        if (typeof payload !== "object" || payload == null)
+            return undefined;
+        const params = payload;
+        const ls_stop = (typeof params.stop === "string" ? [params.stop] : params.stop) ??
+            undefined;
+        const ls_invocation_params = {};
+        for (const [key, value] of Object.entries(params)) {
+            if (TRACED_INVOCATION_KEYS.includes(key)) {
+                ls_invocation_params[key] = value;
+            }
+        }
+        if (useResponsesApi) {
+            ls_invocation_params.use_responses_api = true;
+        }
+        return {
+            ls_provider: provider,
+            ls_model_type: "chat",
+            ls_model_name: params.model,
+            ls_max_tokens: params.max_completion_tokens ?? params.max_tokens ?? undefined,
+            ls_temperature: params.temperature ?? undefined,
+            ls_stop,
+            ls_invocation_params,
+        };
+    };
+};
 /**
  * Wraps an OpenAI client's completion methods, enabling automatic LangSmith
  * tracing. Method signatures are unchanged, with the exception that you can pass
@@ -276,29 +303,7 @@ export const wrapOpenAI = (openai, options) => {
         run_type: "llm",
         aggregator: chatAggregator,
         argsConfigPath: [1, "langsmithExtra"],
-        getInvocationParams: (payload) => {
-            if (typeof payload !== "object" || payload == null)
-                return undefined;
-            // we can safely do so, as the types are not exported in TSC
-            const params = payload;
-            const ls_stop = (typeof params.stop === "string" ? [params.stop] : params.stop) ??
-                undefined;
-            const ls_invocation_params = {};
-            for (const [key, value] of Object.entries(params)) {
-                if (TRACED_INVOCATION_KEYS.includes(key)) {
-                    ls_invocation_params[key] = value;
-                }
-            }
-            return {
-                ls_provider: provider,
-                ls_model_type: "chat",
-                ls_model_name: params.model,
-                ls_max_tokens: params.max_completion_tokens ?? params.max_tokens ?? undefined,
-                ls_temperature: params.temperature ?? undefined,
-                ls_stop,
-                ls_invocation_params,
-            };
-        },
+        getInvocationParams: getChatModelInvocationParamsFn(provider, false),
         processOutputs: processChatCompletion,
         ...options,
     };
@@ -412,17 +417,7 @@ export const wrapOpenAI = (openai, options) => {
                 run_type: "llm",
                 aggregator: responsesAggregator,
                 argsConfigPath: [1, "langsmithExtra"],
-                getInvocationParams: (payload) => {
-                    if (typeof payload !== "object" || payload == null)
-                        return undefined;
-                    // Handle responses API parameters
-                    const params = payload;
-                    return {
-                        ls_provider: provider,
-                        ls_model_type: "chat",
-                        ls_model_name: params.model || "unknown",
-                    };
-                },
+                getInvocationParams: getChatModelInvocationParamsFn(provider, true),
                 processOutputs: processChatCompletion,
                 ...options,
             });
@@ -434,17 +429,7 @@ export const wrapOpenAI = (openai, options) => {
                 run_type: "llm",
                 aggregator: responsesAggregator,
                 argsConfigPath: [1, "langsmithExtra"],
-                getInvocationParams: (payload) => {
-                    if (typeof payload !== "object" || payload == null)
-                        return undefined;
-                    // Handle responses API parameters
-                    const params = payload;
-                    return {
-                        ls_provider: provider,
-                        ls_model_type: "chat",
-                        ls_model_name: params.model || "unknown",
-                    };
-                },
+                getInvocationParams: getChatModelInvocationParamsFn(provider, true),
                 processOutputs: processChatCompletion,
                 ...options,
             });
@@ -456,17 +441,7 @@ export const wrapOpenAI = (openai, options) => {
                 run_type: "llm",
                 aggregator: responsesAggregator,
                 argsConfigPath: [1, "langsmithExtra"],
-                getInvocationParams: (payload) => {
-                    if (typeof payload !== "object" || payload == null)
-                        return undefined;
-                    // Handle responses API parameters
-                    const params = payload;
-                    return {
-                        ls_provider: provider,
-                        ls_model_type: "chat",
-                        ls_model_name: params.model || "unknown",
-                    };
-                },
+                getInvocationParams: getChatModelInvocationParamsFn(provider, true),
                 processOutputs: processChatCompletion,
                 ...options,
             });

package/package.json

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