langsmith 0.3.74 → 0.3.76

package/dist/experimental/vercel/index.cjs

@@ -188,9 +188,10 @@ const wrapAISDK = ({ wrapLanguageModel, generateText, streamText, streamObject,
                 const inputFormatter = resolvedLsConfig?.processInputs ?? _formatTracedInputs;
                 return inputFormatter(inputs);
             },
-            processOutputs: (outputs) => {
+            processOutputs: async (outputs) => {
                 if (resolvedLsConfig?.processOutputs) {
-                    return resolvedLsConfig.processOutputs(outputs);
+                    const processedOutputs = await resolvedLsConfig.processOutputs(outputs);
+                    return processedOutputs;
                 }
                 if (outputs.outputs == null || typeof outputs.outputs !== "object") {
                     return outputs;
@@ -262,9 +263,10 @@ const wrapAISDK = ({ wrapLanguageModel, generateText, streamText, streamObject,
                 const inputFormatter = resolvedLsConfig?.processInputs ?? _formatTracedInputs;
                 return inputFormatter(inputs);
             },
-            processOutputs: (outputs) => {
+            processOutputs: async (outputs) => {
                 if (resolvedLsConfig?.processOutputs) {
-                    return resolvedLsConfig.processOutputs(outputs);
+                    const processedOutputs = await resolvedLsConfig.processOutputs(outputs);
+                    return processedOutputs;
                 }
                 if (outputs.outputs == null || typeof outputs.outputs !== "object") {
                     return outputs;
@@ -327,7 +329,8 @@ const wrapAISDK = ({ wrapLanguageModel, generateText, streamText, streamObject,
             processOutputs: async (outputs) => {
                 try {
                     if (resolvedLsConfig?.processOutputs) {
-                        return resolvedLsConfig.processOutputs(outputs);
+                        const processedOutputs = await resolvedLsConfig.processOutputs(outputs);
+                        return processedOutputs;
                     }
                     if (outputs.outputs == null ||
                         typeof outputs.outputs !== "object") {
@@ -409,7 +412,8 @@ const wrapAISDK = ({ wrapLanguageModel, generateText, streamText, streamObject,
             processOutputs: async (outputs) => {
                 try {
                     if (resolvedLsConfig?.processOutputs) {
-                        return resolvedLsConfig.processOutputs(outputs);
+                        const processedOutputs = await resolvedLsConfig.processOutputs(outputs);
+                        return processedOutputs;
                     }
                     if (outputs.outputs == null ||
                         typeof outputs.outputs !== "object") {

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

@@ -97,7 +97,7 @@ export type WrapAISDKConfig<T extends (...args: any[]) => any = (...args: any[])
      * @param outputs.raw - Raw outputs from the AI SDK.
      * @returns A single combined key-value map of processed outputs.
      */
-    processOutputs?: (outputs: Awaited<ReturnType<T>>) => Record<string, unknown>;
+    processOutputs?: (outputs: Awaited<ReturnType<T>>) => Record<string, unknown> | Promise<Record<string, unknown>>;
     /**
      * Apply transformations to AI SDK child LLM run inputs before logging.
      * This function should NOT mutate the inputs.

package/dist/experimental/vercel/index.js

@@ -184,9 +184,10 @@ const wrapAISDK = ({ wrapLanguageModel, generateText, streamText, streamObject,
                 const inputFormatter = resolvedLsConfig?.processInputs ?? _formatTracedInputs;
                 return inputFormatter(inputs);
             },
-            processOutputs: (outputs) => {
+            processOutputs: async (outputs) => {
                 if (resolvedLsConfig?.processOutputs) {
-                    return resolvedLsConfig.processOutputs(outputs);
+                    const processedOutputs = await resolvedLsConfig.processOutputs(outputs);
+                    return processedOutputs;
                 }
                 if (outputs.outputs == null || typeof outputs.outputs !== "object") {
                     return outputs;
@@ -258,9 +259,10 @@ const wrapAISDK = ({ wrapLanguageModel, generateText, streamText, streamObject,
                 const inputFormatter = resolvedLsConfig?.processInputs ?? _formatTracedInputs;
                 return inputFormatter(inputs);
             },
-            processOutputs: (outputs) => {
+            processOutputs: async (outputs) => {
                 if (resolvedLsConfig?.processOutputs) {
-                    return resolvedLsConfig.processOutputs(outputs);
+                    const processedOutputs = await resolvedLsConfig.processOutputs(outputs);
+                    return processedOutputs;
                 }
                 if (outputs.outputs == null || typeof outputs.outputs !== "object") {
                     return outputs;
@@ -323,7 +325,8 @@ const wrapAISDK = ({ wrapLanguageModel, generateText, streamText, streamObject,
             processOutputs: async (outputs) => {
                 try {
                     if (resolvedLsConfig?.processOutputs) {
-                        return resolvedLsConfig.processOutputs(outputs);
+                        const processedOutputs = await resolvedLsConfig.processOutputs(outputs);
+                        return processedOutputs;
                     }
                     if (outputs.outputs == null ||
                         typeof outputs.outputs !== "object") {
@@ -405,7 +408,8 @@ const wrapAISDK = ({ wrapLanguageModel, generateText, streamText, streamObject,
             processOutputs: async (outputs) => {
                 try {
                     if (resolvedLsConfig?.processOutputs) {
-                        return resolvedLsConfig.processOutputs(outputs);
+                        const processedOutputs = await resolvedLsConfig.processOutputs(outputs);
+                        return processedOutputs;
                     }
                     if (outputs.outputs == null ||
                         typeof outputs.outputs !== "object") {

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.74";
+exports.__version__ = "0.3.76";

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.74";
+export declare const __version__ = "0.3.76";

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.74";
+export const __version__ = "0.3.76";

package/dist/traceable.cjs

@@ -91,7 +91,9 @@ const runInputsToMap = (rawInputs) => {
     }
     return inputs;
 };
-const handleRunInputs = (inputs, processInputs) => {
+const handleRunInputs = (
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+inputs, processInputs) => {
     try {
         return processInputs(inputs);
     }
@@ -147,6 +149,7 @@ function isAsyncFn(fn) {
 // Note: This mutates the run tree
 async function handleRunOutputs(params) {
     const { runTree, rawOutputs, processOutputsFn, on_end, postRunPromise, excludeInputs, } = params;
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
     let outputs;
     if ((0, asserts_js_1.isKVMap)(rawOutputs)) {
         outputs = { ...rawOutputs };

package/dist/traceable.d.ts

@@ -2,6 +2,18 @@ import { RunTreeConfig } from "./run_trees.js";
 import { Attachments, InvocationParamsSchema, KVMap } from "./schemas.js";
 import type { TraceableFunction } from "./singletons/types.js";
 import { OTELTracer } from "./experimental/otel/types.js";
+export type ProcessInputs<Args extends unknown[]> = Args extends [] ? Record<string, never> : Args extends [infer Input] ? Input extends KVMap ? Input extends Iterable<infer Item> | AsyncIterable<infer Item> ? {
+    input: Array<Item>;
+} : Input : {
+    input: Input;
+} : {
+    args: Args;
+};
+export type ProcessOutputs<ReturnValue> = ReturnValue extends KVMap ? ReturnValue extends Iterable<infer Item> | AsyncIterable<infer Item> ? {
+    outputs: Array<Item>;
+} : ReturnValue : {
+    outputs: ReturnValue;
+};
 export type TraceableConfig<Func extends (...args: any[]) => any> = Partial<Omit<RunTreeConfig, "inputs" | "outputs">> & {
     aggregator?: (args: any[]) => any;
     argsConfigPath?: [number] | [number, string];
@@ -27,19 +39,31 @@ export type TraceableConfig<Func extends (...args: any[]) => any> = Partial<Omit
      * This function should NOT mutate the inputs.
      * `processInputs` is not inherited by nested traceable functions.
      *
+     * The input to this function is determined as follows based on the
+     * arguments passed to the wrapped function:
+     * - If called with one argument that is an object, it will be the unchanged argument
+     * - If called with one argument that is not an object, it will be `{ input: arg }`
+     * - If called with multiple arguments, it will be `{ args: [...arguments] }`
+     * - If called with no arguments, it will be an empty object `{}`
+     *
      * @param inputs Key-value map of the function inputs.
      * @returns Transformed key-value map
      */
-    processInputs?: (inputs: Readonly<KVMap>) => KVMap;
+    processInputs?: (inputs: Readonly<ProcessInputs<Parameters<Func>>>) => KVMap;
     /**
      * Apply transformations to the outputs before logging.
      * This function should NOT mutate the outputs.
      * `processOutputs` is not inherited by nested traceable functions.
      *
+     * The input to this function is determined as follows based on the
+     * return value of the wrapped function:
+     * - If the return value is an object, it will be the unchanged return value
+     * - If the return value is not an object, it will wrapped as `{ outputs: returnValue }`
+     *
      * @param outputs Key-value map of the function outputs
      * @returns Transformed key-value map
      */
-    processOutputs?: (outputs: Readonly<KVMap>) => KVMap | Promise<KVMap>;
+    processOutputs?: (outputs: Readonly<ProcessOutputs<Awaited<ReturnType<Func>>>>) => KVMap | Promise<KVMap>;
 };
 /**
  * Higher-order function that takes function as input and returns a

package/dist/traceable.js

@@ -87,7 +87,9 @@ const runInputsToMap = (rawInputs) => {
     }
     return inputs;
 };
-const handleRunInputs = (inputs, processInputs) => {
+const handleRunInputs = (
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+inputs, processInputs) => {
     try {
         return processInputs(inputs);
     }
@@ -143,6 +145,7 @@ function isAsyncFn(fn) {
 // Note: This mutates the run tree
 async function handleRunOutputs(params) {
     const { runTree, rawOutputs, processOutputsFn, on_end, postRunPromise, excludeInputs, } = params;
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
     let outputs;
     if (isKVMap(rawOutputs)) {
         outputs = { ...rawOutputs };

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "langsmith",
-  "version": "0.3.74",
-  "description": "Client library to connect to the LangSmith LLM Tracing and Evaluation Platform.",
+  "version": "0.3.76",
+  "description": "Client library to connect to the LangSmith Observability and Evaluation Platform.",
   "packageManager": "yarn@1.22.19",
   "files": [
     "dist/",