[JS] Add callable masking (#493)

js/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langsmith",
-  "version": "0.1.9",
+  "version": "0.1.10",
   "description": "Client library to connect to the LangSmith LLM Tracing and Evaluation Platform.",
   "packageManager": "[email protected]",
   "files": [

js/src/client.ts

@@ -363,9 +363,9 @@ export class Client {
 
   private _tenantId: string | null = null;
 
-  private hideInputs?: boolean;
+  private hideInputs?: boolean | ((inputs: KVMap) => KVMap);
 
-  private hideOutputs?: boolean;
+  private hideOutputs?: boolean | ((outputs: KVMap) => KVMap);
 
   private tracingSampleRate?: number;
 
@@ -463,16 +463,28 @@ export class Client {
   }
 
   private processInputs(inputs: KVMap): KVMap {
-    if (this.hideInputs) {
+    if (this.hideInputs === false) {
+      return inputs;
+    }
+    if (this.hideInputs === true) {
       return {};
     }
+    if (typeof this.hideInputs === "function") {
+      return this.hideInputs(inputs);
+    }
     return inputs;
   }
 
   private processOutputs(outputs: KVMap): KVMap {
-    if (this.hideOutputs) {
+    if (this.hideOutputs === false) {
+      return outputs;
+    }
+    if (this.hideOutputs === true) {
       return {};
     }
+    if (typeof this.hideOutputs === "function") {
+      return this.hideOutputs(outputs);
+    }
     return outputs;
   }
 

js/src/index.ts

@@ -11,4 +11,4 @@ export type {
 export { RunTree, type RunTreeConfig } from "./run_trees.js";
 
 // Update using yarn bump-version
-export const __version__ = "0.1.9";
+export const __version__ = "0.1.10";