langsmith 0.3.62-rc.2 → 0.3.63

package/dist/experimental/vercel/index.js

@@ -1,4 +1,4 @@
-import { LangSmithMiddleware } from "./middleware.js";
+import { LangSmithMiddleware, } from "./middleware.js";
 import { convertMessageToTracedFormat } from "./utils.js";
 import { isTraceableFunction, traceable } from "../../traceable.js";
 const _wrapTools = (tools, lsConfig) => {
@@ -55,6 +55,69 @@ const _formatTracedInputs = (params) => {
         return { ...rest, prompt, messages };
     }
 };
+const _mergeConfig = (baseConfig, runtimeConfig) => {
+    return {
+        ...baseConfig,
+        ...runtimeConfig,
+        metadata: {
+            ...baseConfig?.metadata,
+            ...runtimeConfig?.metadata,
+        },
+    };
+};
+const _extractChildRunConfig = (lsConfig) => {
+    const { id, name, parent_run_id, start_time, end_time, attachments, dotted_order, processInputs, processOutputs, processChildLLMRunInputs, processChildLLMRunOutputs, ...inheritedConfig } = lsConfig ?? {};
+    const childConfig = inheritedConfig;
+    if (processChildLLMRunInputs) {
+        childConfig.processInputs = processChildLLMRunInputs;
+    }
+    if (processChildLLMRunOutputs) {
+        childConfig.processOutputs = processChildLLMRunOutputs;
+    }
+    return childConfig;
+};
+const _resolveConfigs = (baseLsConfig, runtimeLsConfig) => {
+    const baseChildRunConfig = _extractChildRunConfig(baseLsConfig);
+    const runtimeChildLLMRunConfig = _extractChildRunConfig(runtimeLsConfig);
+    const resolvedLsConfig = _mergeConfig(baseLsConfig, runtimeLsConfig);
+    const resolvedChildLLMRunConfig = _mergeConfig(baseChildRunConfig, runtimeChildLLMRunConfig);
+    const { processInputs: _processInputs, processOutputs: _processOutputs, ...resolvedToolConfig } = resolvedChildLLMRunConfig;
+    return {
+        resolvedLsConfig,
+        resolvedChildLLMRunConfig,
+        resolvedToolConfig,
+    };
+};
+/**
+ * Wraps LangSmith config in a way that matches AI SDK provider types.
+ *
+ * ```ts
+ * import { createLangSmithProviderOptions } from "langsmith/experimental/vercel";
+ * import * as ai from "ai";
+ *
+ * const lsConfig = createLangSmithProviderOptions<typeof ai.generateText>({
+ *   // Will have appropriate typing
+ *   processInputs: (inputs) => {
+ *     const { messages } = inputs;
+ *     return {
+ *       messages: messages?.map((message) => ({
+ *         ...message,
+ *         content: "REDACTED",
+ *       })),
+ *       prompt: "REDACTED",
+ *     };
+ *   },
+ * });
+ * ```
+ *
+ * Note: AI SDK expects only JSON values in an object for
+ * provider options, but LangSmith's config may contain non-JSON values.
+ * These are not passed to the underlying AI SDK model, so it is safe to
+ * cast the typing here.
+ */
+export const createLangSmithProviderOptions = (lsConfig) => {
+    return (lsConfig ?? {});
+};
 /**
  * Wraps Vercel AI SDK 5 functions with LangSmith tracing capabilities.
  *
@@ -71,8 +134,7 @@ const _formatTracedInputs = (params) => {
  * @returns returns.streamText - Wrapped streamText function that traces calls to LangSmith
  * @returns returns.streamObject - Wrapped streamObject function that traces calls to LangSmith
  */
-const wrapAISDK = ({ wrapLanguageModel, generateText, streamText, streamObject, generateObject, }, lsConfig) => {
-    const { id, name, parent_run_id, start_time, end_time, attachments, dotted_order, ...inheritedConfig } = lsConfig ?? {};
+const wrapAISDK = ({ wrapLanguageModel, generateText, streamText, streamObject, generateObject, }, baseLsConfig) => {
     /**
      * Wrapped version of AI SDK 5's generateText with LangSmith tracing.
      *
@@ -93,6 +155,8 @@ const wrapAISDK = ({ wrapLanguageModel, generateText, streamText, streamObject,
      */
     const wrappedGenerateText = async (...args) => {
         const params = args[0];
+        const { langsmith: runtimeLsConfig, ...providerOptions } = params.providerOptions ?? {};
+        const { resolvedLsConfig, resolvedChildLLMRunConfig, resolvedToolConfig } = _resolveConfigs(baseLsConfig, runtimeLsConfig);
         const traceableFunc = traceable(async (...args) => {
             const [params, ...rest] = args;
             const wrappedModel = wrapLanguageModel({
@@ -100,23 +164,30 @@ const wrapAISDK = ({ wrapLanguageModel, generateText, streamText, streamObject,
                 middleware: LangSmithMiddleware({
                     name: _getModelDisplayName(params.model),
                     modelId: _getModelId(params.model),
-                    lsConfig: inheritedConfig,
+                    lsConfig: resolvedChildLLMRunConfig,
                 }),
             });
             return generateText({
                 ...params,
-                tools: _wrapTools(params.tools, inheritedConfig),
+                providerOptions,
+                tools: _wrapTools(params.tools, resolvedToolConfig),
                 model: wrappedModel,
             }, ...rest);
         }, {
             name: _getModelDisplayName(params.model),
-            ...lsConfig,
+            ...resolvedLsConfig,
             metadata: {
                 ai_sdk_method: "ai.generateText",
-                ...lsConfig?.metadata,
+                ...resolvedLsConfig?.metadata,
+            },
+            processInputs: (inputs) => {
+                const inputFormatter = resolvedLsConfig?.processInputs ?? _formatTracedInputs;
+                return inputFormatter(inputs);
             },
-            processInputs: (inputs) => _formatTracedInputs(inputs),
             processOutputs: (outputs) => {
+                if (resolvedLsConfig?.processOutputs) {
+                    return resolvedLsConfig.processOutputs(outputs);
+                }
                 if (outputs.outputs == null || typeof outputs.outputs !== "object") {
                     return outputs;
                 }
@@ -159,6 +230,8 @@ const wrapAISDK = ({ wrapLanguageModel, generateText, streamText, streamObject,
      */
     const wrappedGenerateObject = async (...args) => {
         const params = args[0];
+        const { langsmith: runtimeLsConfig, ...providerOptions } = params.providerOptions ?? {};
+        const { resolvedLsConfig, resolvedChildLLMRunConfig } = _resolveConfigs(baseLsConfig, runtimeLsConfig);
         const traceableFunc = traceable(async (...args) => {
             const [params, ...rest] = args;
             const wrappedModel = wrapLanguageModel({
@@ -166,22 +239,29 @@ const wrapAISDK = ({ wrapLanguageModel, generateText, streamText, streamObject,
                 middleware: LangSmithMiddleware({
                     name: _getModelDisplayName(params.model),
                     modelId: _getModelId(params.model),
-                    lsConfig: inheritedConfig,
+                    lsConfig: resolvedChildLLMRunConfig,
                 }),
             });
             return generateObject({
                 ...params,
+                providerOptions,
                 model: wrappedModel,
             }, ...rest);
         }, {
             name: _getModelDisplayName(params.model),
-            ...lsConfig,
+            ...resolvedLsConfig,
             metadata: {
                 ai_sdk_method: "ai.generateObject",
-                ...lsConfig?.metadata,
+                ...resolvedLsConfig?.metadata,
+            },
+            processInputs: (inputs) => {
+                const inputFormatter = resolvedLsConfig?.processInputs ?? _formatTracedInputs;
+                return inputFormatter(inputs);
             },
-            processInputs: (inputs) => _formatTracedInputs(inputs),
             processOutputs: (outputs) => {
+                if (resolvedLsConfig?.processOutputs) {
+                    return resolvedLsConfig.processOutputs(outputs);
+                }
                 if (outputs.outputs == null || typeof outputs.outputs !== "object") {
                     return outputs;
                 }
@@ -211,6 +291,8 @@ const wrapAISDK = ({ wrapLanguageModel, generateText, streamText, streamObject,
      */
     const wrappedStreamText = (...args) => {
         const params = args[0];
+        const { langsmith: runtimeLsConfig, ...providerOptions } = params.providerOptions ?? {};
+        const { resolvedLsConfig, resolvedChildLLMRunConfig, resolvedToolConfig } = _resolveConfigs(baseLsConfig, runtimeLsConfig);
         const traceableFunc = traceable((...args) => {
             const [params, ...rest] = args;
             const wrappedModel = wrapLanguageModel({
@@ -218,23 +300,30 @@ const wrapAISDK = ({ wrapLanguageModel, generateText, streamText, streamObject,
                 middleware: LangSmithMiddleware({
                     name: _getModelDisplayName(params.model),
                     modelId: _getModelId(params.model),
-                    lsConfig: inheritedConfig,
+                    lsConfig: resolvedChildLLMRunConfig,
                 }),
             });
             return streamText({
                 ...params,
-                tools: _wrapTools(params.tools, inheritedConfig),
+                providerOptions,
+                tools: _wrapTools(params.tools, resolvedToolConfig),
                 model: wrappedModel,
             }, ...rest);
         }, {
             name: _getModelDisplayName(params.model),
-            ...lsConfig,
+            ...resolvedLsConfig,
             metadata: {
                 ai_sdk_method: "ai.streamText",
-                ...lsConfig?.metadata,
+                ...resolvedLsConfig?.metadata,
+            },
+            processInputs: (inputs) => {
+                const inputFormatter = resolvedLsConfig?.processInputs ?? _formatTracedInputs;
+                return inputFormatter(inputs);
             },
-            processInputs: (inputs) => _formatTracedInputs(inputs),
             processOutputs: async (outputs) => {
+                if (resolvedLsConfig?.processOutputs) {
+                    return resolvedLsConfig.processOutputs(outputs);
+                }
                 if (outputs.outputs == null || typeof outputs.outputs !== "object") {
                     return outputs;
                 }
@@ -271,6 +360,8 @@ const wrapAISDK = ({ wrapLanguageModel, generateText, streamText, streamObject,
      */
     const wrappedStreamObject = (...args) => {
         const params = args[0];
+        const { langsmith: runtimeLsConfig, ...providerOptions } = params.providerOptions ?? {};
+        const { resolvedLsConfig, resolvedChildLLMRunConfig } = _resolveConfigs(baseLsConfig, runtimeLsConfig);
         const traceableFunc = traceable((...args) => {
             const [params, ...rest] = args;
             const wrappedModel = wrapLanguageModel({
@@ -278,22 +369,29 @@ const wrapAISDK = ({ wrapLanguageModel, generateText, streamText, streamObject,
                 middleware: LangSmithMiddleware({
                     name: _getModelDisplayName(params.model),
                     modelId: _getModelId(params.model),
-                    lsConfig: inheritedConfig,
+                    lsConfig: resolvedChildLLMRunConfig,
                 }),
             });
             return streamObject({
                 ...params,
+                providerOptions,
                 model: wrappedModel,
             }, ...rest);
         }, {
             name: _getModelDisplayName(params.model),
-            ...lsConfig,
+            ...resolvedLsConfig,
             metadata: {
                 ai_sdk_method: "ai.streamObject",
-                ...lsConfig?.metadata,
+                ...resolvedLsConfig?.metadata,
+            },
+            processInputs: (inputs) => {
+                const inputFormatter = resolvedLsConfig?.processInputs ?? _formatTracedInputs;
+                return inputFormatter(inputs);
             },
-            processInputs: (inputs) => _formatTracedInputs(inputs),
             processOutputs: async (outputs) => {
+                if (resolvedLsConfig?.processOutputs) {
+                    return resolvedLsConfig.processOutputs(outputs);
+                }
                 if (outputs.outputs == null || typeof outputs.outputs !== "object") {
                     return outputs;
                 }

package/dist/experimental/vercel/middleware.cjs

@@ -15,7 +15,6 @@ const _formatTracedInputs = (params) => {
     }
     return rest;
 };
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
 const _formatTracedOutputs = (outputs) => {
     const formattedOutputs = { ...outputs };
     if (formattedOutputs.role == null) {
@@ -75,11 +74,13 @@ function LangSmithMiddleware(config) {
                 },
                 processInputs: (inputs) => {
                     const typedInputs = inputs;
-                    return _formatTracedInputs(typedInputs);
+                    const inputFormatter = lsConfig?.processInputs ?? _formatTracedInputs;
+                    return inputFormatter(typedInputs);
                 },
                 processOutputs: (outputs) => {
                     const typedOutputs = outputs;
-                    return _formatTracedOutputs(typedOutputs);
+                    const outputFormatter = lsConfig?.processOutputs ?? _formatTracedOutputs;
+                    return outputFormatter(typedOutputs);
                 },
             });
             const res = await traceableFunc(params);
@@ -91,6 +92,8 @@ function LangSmithMiddleware(config) {
             if (parentRunTree != null &&
                 typeof parentRunTree === "object" &&
                 typeof parentRunTree.createChild === "function") {
+                const inputFormatter = lsConfig?.processInputs ?? _formatTracedInputs;
+                const formattedInputs = inputFormatter(params);
                 runTree = parentRunTree?.createChild({
                     ...lsConfig,
                     name: name ?? "ai.doStream",
@@ -100,7 +103,7 @@ function LangSmithMiddleware(config) {
                         ai_sdk_method: "ai.doStream",
                         ...lsConfig?.metadata,
                     },
-                    inputs: _formatTracedInputs(params),
+                    inputs: formattedInputs,
                 });
             }
             await runTree?.postRun();
@@ -170,17 +173,21 @@ function LangSmithMiddleware(config) {
                                 }
                             }, {
                                 content: "",
-                                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                                role: "assistant",
                                 tool_calls: [],
                             });
-                            await runTree?.end(_formatTracedOutputs(output));
+                            const outputFormatter = lsConfig?.processOutputs ?? utils_js_1.convertMessageToTracedFormat;
+                            const formattedOutputs = outputFormatter(output);
+                            await runTree?.end(formattedOutputs);
                         }
                         catch (error) {
                             await runTree?.end(undefined, error.message ?? String(error));
                             throw error;
                         }
                         finally {
-                            await runTree?.patchRun();
+                            await runTree?.patchRun({
+                                excludeInputs: true,
+                            });
                         }
                     },
                 });
@@ -191,7 +198,9 @@ function LangSmithMiddleware(config) {
             }
             catch (error) {
                 await runTree?.end(undefined, error.message ?? String(error));
-                await runTree?.patchRun();
+                await runTree?.patchRun({
+                    excludeInputs: true,
+                });
                 throw error;
             }
         },

package/dist/experimental/vercel/middleware.d.ts

@@ -1,10 +1,27 @@
-import type { LanguageModelV2Middleware } from "@ai-sdk/provider";
+import type { LanguageModelV2Middleware, SharedV2ProviderMetadata, LanguageModelV2FinishReason } from "@ai-sdk/provider";
 import type { RunTreeConfig } from "../../run_trees.js";
+export type AggregatedDoStreamOutput = {
+    content: string;
+    role: "assistant";
+    tool_calls: {
+        id: string;
+        type: "function";
+        function: {
+            name: string;
+            arguments: string;
+        };
+    }[];
+    providerMetadata?: SharedV2ProviderMetadata;
+    finishReason?: LanguageModelV2FinishReason;
+};
 /**
  * AI SDK middleware that wraps an AI SDK 5 model and adds LangSmith tracing.
  */
 export declare function LangSmithMiddleware(config?: {
     name: string;
     modelId?: string;
-    lsConfig?: Partial<Omit<RunTreeConfig, "inputs" | "outputs" | "run_type">>;
+    lsConfig?: Partial<Omit<RunTreeConfig, "inputs" | "outputs" | "run_type">> & {
+        processInputs?: (inputs: Record<string, unknown>) => Record<string, unknown>;
+        processOutputs?: (outputs: Record<string, unknown>) => Record<string, unknown>;
+    };
 }): LanguageModelV2Middleware;

package/dist/experimental/vercel/middleware.js

@@ -12,7 +12,6 @@ const _formatTracedInputs = (params) => {
     }
     return rest;
 };
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
 const _formatTracedOutputs = (outputs) => {
     const formattedOutputs = { ...outputs };
     if (formattedOutputs.role == null) {
@@ -72,11 +71,13 @@ export function LangSmithMiddleware(config) {
                 },
                 processInputs: (inputs) => {
                     const typedInputs = inputs;
-                    return _formatTracedInputs(typedInputs);
+                    const inputFormatter = lsConfig?.processInputs ?? _formatTracedInputs;
+                    return inputFormatter(typedInputs);
                 },
                 processOutputs: (outputs) => {
                     const typedOutputs = outputs;
-                    return _formatTracedOutputs(typedOutputs);
+                    const outputFormatter = lsConfig?.processOutputs ?? _formatTracedOutputs;
+                    return outputFormatter(typedOutputs);
                 },
             });
             const res = await traceableFunc(params);
@@ -88,6 +89,8 @@ export function LangSmithMiddleware(config) {
             if (parentRunTree != null &&
                 typeof parentRunTree === "object" &&
                 typeof parentRunTree.createChild === "function") {
+                const inputFormatter = lsConfig?.processInputs ?? _formatTracedInputs;
+                const formattedInputs = inputFormatter(params);
                 runTree = parentRunTree?.createChild({
                     ...lsConfig,
                     name: name ?? "ai.doStream",
@@ -97,7 +100,7 @@ export function LangSmithMiddleware(config) {
                         ai_sdk_method: "ai.doStream",
                         ...lsConfig?.metadata,
                     },
-                    inputs: _formatTracedInputs(params),
+                    inputs: formattedInputs,
                 });
             }
             await runTree?.postRun();
@@ -167,17 +170,21 @@ export function LangSmithMiddleware(config) {
                                 }
                             }, {
                                 content: "",
-                                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                                role: "assistant",
                                 tool_calls: [],
                             });
-                            await runTree?.end(_formatTracedOutputs(output));
+                            const outputFormatter = lsConfig?.processOutputs ?? convertMessageToTracedFormat;
+                            const formattedOutputs = outputFormatter(output);
+                            await runTree?.end(formattedOutputs);
                         }
                         catch (error) {
                             await runTree?.end(undefined, error.message ?? String(error));
                             throw error;
                         }
                         finally {
-                            await runTree?.patchRun();
+                            await runTree?.patchRun({
+                                excludeInputs: true,
+                            });
                         }
                     },
                 });
@@ -188,7 +195,9 @@ export function LangSmithMiddleware(config) {
             }
             catch (error) {
                 await runTree?.end(undefined, error.message ?? String(error));
-                await runTree?.patchRun();
+                await runTree?.patchRun({
+                    excludeInputs: true,
+                });
                 throw error;
             }
         },

package/dist/index.cjs

@@ -10,4 +10,4 @@ Object.defineProperty(exports, "overrideFetchImplementation", { enumerable: true
 var project_js_1 = require("./utils/project.cjs");
 Object.defineProperty(exports, "getDefaultProjectName", { enumerable: true, get: function () { return project_js_1.getDefaultProjectName; } });
 // Update using yarn bump-version
-exports.__version__ = "0.3.62-rc.2";
+exports.__version__ = "0.3.63";

package/dist/index.d.ts

@@ -3,4 +3,4 @@ export type { Dataset, Example, TracerSession, Run, Feedback, RetrieverOutput, }
 export { RunTree, type RunTreeConfig } from "./run_trees.js";
 export { overrideFetchImplementation } from "./singletons/fetch.js";
 export { getDefaultProjectName } from "./utils/project.js";
-export declare const __version__ = "0.3.62-rc.2";
+export declare const __version__ = "0.3.63";

package/dist/index.js

@@ -3,4 +3,4 @@ export { RunTree } from "./run_trees.js";
 export { overrideFetchImplementation } from "./singletons/fetch.js";
 export { getDefaultProjectName } from "./utils/project.js";
 // Update using yarn bump-version
-export const __version__ = "0.3.62-rc.2";
+export const __version__ = "0.3.63";

package/dist/langchain.cjs

@@ -81,6 +81,7 @@ async function getLangchainCallbacks(currentRunTree) {
 /**
  * RunnableTraceable is a Runnable that wraps a traceable function.
  * This allows adding Langsmith traced functions into LangChain sequences.
+ * @deprecated Wrap or pass directly instead.
  */
 class RunnableTraceable extends runnables_1.Runnable {
     constructor(fields) {
@@ -111,6 +112,9 @@ class RunnableTraceable extends runnables_1.Runnable {
     async invoke(input, options) {
         const [config] = this._getOptionsList(options ?? {}, 1);
         const callbacks = await (0, runnables_1.getCallbackManagerForConfig)(config);
+        // Avoid start time ties - this is old, deprecated code used only in tests
+        // and recent perf improvements have made this necessary.
+        await new Promise((resolve) => setImmediate(resolve));
         return (await this.func((0, runnables_1.patchConfig)(config, { callbacks }), input));
     }
     async *_streamIterator(input, options) {

package/dist/langchain.d.ts

@@ -15,6 +15,7 @@ type AnyTraceableFunction = TraceableFunction<(...any: any[]) => any>;
 /**
  * RunnableTraceable is a Runnable that wraps a traceable function.
  * This allows adding Langsmith traced functions into LangChain sequences.
+ * @deprecated Wrap or pass directly instead.
  */
 export declare class RunnableTraceable<RunInput, RunOutput> extends Runnable<RunInput, RunOutput> {
     lc_serializable: boolean;

package/dist/langchain.js

@@ -77,6 +77,7 @@ export async function getLangchainCallbacks(currentRunTree) {
 /**
  * RunnableTraceable is a Runnable that wraps a traceable function.
  * This allows adding Langsmith traced functions into LangChain sequences.
+ * @deprecated Wrap or pass directly instead.
  */
 export class RunnableTraceable extends Runnable {
     constructor(fields) {
@@ -107,6 +108,9 @@ export class RunnableTraceable extends Runnable {
     async invoke(input, options) {
         const [config] = this._getOptionsList(options ?? {}, 1);
         const callbacks = await getCallbackManagerForConfig(config);
+        // Avoid start time ties - this is old, deprecated code used only in tests
+        // and recent perf improvements have made this necessary.
+        await new Promise((resolve) => setImmediate(resolve));
         return (await this.func(patchConfig(config, { callbacks }), input));
     }
     async *_streamIterator(input, options) {

package/dist/run_trees.cjs

@@ -547,12 +547,12 @@ class RunTree {
             console.error(`Error in postRun for run ${this.id}:`, error);
         }
     }
-    async patchRun() {
+    async patchRun(options) {
         if (this.replicas && this.replicas.length > 0) {
             for (const { projectName, apiKey, apiUrl, updates } of this.replicas) {
                 const runData = this._remapForProject(projectName ?? this.project_name);
-                await this.client.updateRun(runData.id, {
-                    inputs: runData.inputs,
+                const updatePayload = {
+                    id: runData.id,
                     outputs: runData.outputs,
                     error: runData.error,
                     parent_run_id: runData.parent_run_id,
@@ -566,7 +566,14 @@ class RunTree {
                     extra: runData.extra,
                     attachments: this.attachments,
                     ...updates,
-                }, {
+                };
+                // Important that inputs is not a key in the run update
+                // if excluded because it will overwrite the run create if the
+                // two operations are merged during batching
+                if (!options?.excludeInputs) {
+                    updatePayload.inputs = runData.inputs;
+                }
+                await this.client.updateRun(runData.id, updatePayload, {
                     apiKey,
                     apiUrl,
                 });
@@ -577,7 +584,6 @@ class RunTree {
                 const runUpdate = {
                     end_time: this.end_time,
                     error: this.error,
-                    inputs: this.inputs,
                     outputs: this.outputs,
                     parent_run_id: this.parent_run?.id ?? this.parent_run_id,
                     reference_example_id: this.reference_example_id,
@@ -589,6 +595,12 @@ class RunTree {
                     attachments: this.attachments,
                     session_name: this.project_name,
                 };
+                // Important that inputs is not a key in the run update
+                // if excluded because it will overwrite the run create if the
+                // two operations are merged during batching
+                if (!options?.excludeInputs) {
+                    runUpdate.inputs = this.inputs;
+                }
                 await this.client.updateRun(this.id, runUpdate);
             }
             catch (error) {

package/dist/run_trees.d.ts

@@ -107,7 +107,9 @@ export declare class RunTree implements BaseRun {
     private _convertToCreate;
     private _remapForProject;
     postRun(excludeChildRuns?: boolean): Promise<void>;
-    patchRun(): Promise<void>;
+    patchRun(options?: {
+        excludeInputs?: boolean;
+    }): Promise<void>;
     toJSON(): RunCreate & {
         id: string;
     };

package/dist/run_trees.js

@@ -508,12 +508,12 @@ export class RunTree {
             console.error(`Error in postRun for run ${this.id}:`, error);
         }
     }
-    async patchRun() {
+    async patchRun(options) {
         if (this.replicas && this.replicas.length > 0) {
             for (const { projectName, apiKey, apiUrl, updates } of this.replicas) {
                 const runData = this._remapForProject(projectName ?? this.project_name);
-                await this.client.updateRun(runData.id, {
-                    inputs: runData.inputs,
+                const updatePayload = {
+                    id: runData.id,
                     outputs: runData.outputs,
                     error: runData.error,
                     parent_run_id: runData.parent_run_id,
@@ -527,7 +527,14 @@ export class RunTree {
                     extra: runData.extra,
                     attachments: this.attachments,
                     ...updates,
-                }, {
+                };
+                // Important that inputs is not a key in the run update
+                // if excluded because it will overwrite the run create if the
+                // two operations are merged during batching
+                if (!options?.excludeInputs) {
+                    updatePayload.inputs = runData.inputs;
+                }
+                await this.client.updateRun(runData.id, updatePayload, {
                     apiKey,
                     apiUrl,
                 });
@@ -538,7 +545,6 @@ export class RunTree {
                 const runUpdate = {
                     end_time: this.end_time,
                     error: this.error,
-                    inputs: this.inputs,
                     outputs: this.outputs,
                     parent_run_id: this.parent_run?.id ?? this.parent_run_id,
                     reference_example_id: this.reference_example_id,
@@ -550,6 +556,12 @@ export class RunTree {
                     attachments: this.attachments,
                     session_name: this.project_name,
                 };
+                // Important that inputs is not a key in the run update
+                // if excluded because it will overwrite the run create if the
+                // two operations are merged during batching
+                if (!options?.excludeInputs) {
+                    runUpdate.inputs = this.inputs;
+                }
                 await this.client.updateRun(this.id, runUpdate);
             }
             catch (error) {

package/dist/singletons/fetch.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports._getFetchImplementation = exports._globalFetchImplementationIsNodeFetch = exports.overrideFetchImplementation = void 0;
+exports._getFetchImplementation = exports._globalFetchImplementationIsNodeFetch = exports.clearFetchImplementation = exports.overrideFetchImplementation = void 0;
 const env_js_1 = require("../utils/env.cjs");
 // Wrap the default fetch call due to issues with illegal invocations
 // in some environments:
@@ -18,6 +18,10 @@ const overrideFetchImplementation = (fetch) => {
     globalThis[LANGSMITH_FETCH_IMPLEMENTATION_KEY] = fetch;
 };
 exports.overrideFetchImplementation = overrideFetchImplementation;
+const clearFetchImplementation = () => {
+    delete globalThis[LANGSMITH_FETCH_IMPLEMENTATION_KEY];
+};
+exports.clearFetchImplementation = clearFetchImplementation;
 const _globalFetchImplementationIsNodeFetch = () => {
     const fetchImpl = globalThis[LANGSMITH_FETCH_IMPLEMENTATION_KEY];
     if (!fetchImpl)

package/dist/singletons/fetch.d.ts

@@ -5,4 +5,5 @@
  * @param fetch The new fetch functino to use.
  */
 export declare const overrideFetchImplementation: (fetch: (...args: any[]) => any) => void;
+export declare const clearFetchImplementation: () => void;
 export declare const _globalFetchImplementationIsNodeFetch: () => boolean;