langsmith 0.4.0 → 0.4.1

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.1";

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.1";

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.1";

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

@@ -636,8 +636,13 @@ function traceable(wrappedFunc, config) {
                     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),

package/dist/traceable.js

@@ -632,8 +632,13 @@ export function traceable(wrappedFunc, config) {
                     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),

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.1",
   "description": "Client library to connect to the LangSmith Observability and Evaluation Platform.",
   "packageManager": "yarn@1.22.19",
   "files": [