@temporalio/langsmith 1.20.0

package/lib/workflow-interceptors.js

@@ -0,0 +1,350 @@
+"use strict";
+/**
+ * Workflow-isolate interceptors and the deterministic LangSmith context
+ * provider. Loaded into the workflow bundle (via the plugin's `workflowModules`)
+ * and runs entirely inside the V8 isolate, where `node:async_hooks` is
+ * unavailable so a synchronous stack-based context provider stands in for
+ * LangSmith's async-context store.
+ *
+ * Internal: the worker loads this module by specifier and the bundler aliases
+ * `node:async_hooks` to it — it is not a hand-import API.
+ *
+ * @module
+ * @internal
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AsyncLocalStorage = void 0;
+exports.interceptors = interceptors;
+const traceable_1 = require("langsmith/singletons/traceable");
+const workflow_1 = require("@temporalio/workflow");
+const propagation_1 = require("./propagation");
+const run_tree_1 = require("./run-tree");
+function readConfig() {
+    try {
+        if (typeof __TEMPORAL_LANGSMITH_CONFIG__ === 'string') {
+            return JSON.parse(__TEMPORAL_LANGSMITH_CONFIG__);
+        }
+    }
+    catch {
+        /* fall through to default */
+    }
+    return { addTemporalRuns: false };
+}
+/**
+ * Synchronous, isolate-safe replacement for LangSmith's async-context store,
+ * shared by the installed provider, the {@link AsyncLocalStorage} shim, and
+ * every interceptor so they all read and write one store. `workflowAmbient`
+ * persists across `await` boundaries; `stack` tracks synchronous `traceable`
+ * nesting. Under `Promise.all` fan-out parenting falls back to the ambient —
+ * trace-shape-only non-determinism, never workflow history.
+ */
+class WorkflowContextManager {
+    workflowAmbient;
+    stack = [];
+    getStore() {
+        return this.stack.length > 0 ? this.stack[this.stack.length - 1] : this.workflowAmbient;
+    }
+    /**
+     * Push `context`, run `fn`, pop in `finally`, and return `fn`'s result — the
+     * `AsyncLocalStorage.run` contract langsmith relies on.
+     */
+    run(context, fn) {
+        this.stack.push(context);
+        try {
+            return fn();
+        }
+        finally {
+            this.stack.pop();
+        }
+    }
+    /** The ambient run an outbound interceptor should propagate / parent under. */
+    ambient() {
+        return this.getStore();
+    }
+    /** Run an async scope with a dedicated ambient (workflow execute / handler). */
+    async withAmbient(run, fn) {
+        const prev = this.workflowAmbient;
+        this.workflowAmbient = run;
+        try {
+            return await fn();
+        }
+        finally {
+            this.workflowAmbient = prev;
+        }
+    }
+}
+let providerInstalled = false;
+function ensureProviderInstalled(manager) {
+    if (providerInstalled) {
+        return;
+    }
+    providerInstalled = true;
+    // LangSmith's `AsyncLocalStorageInterface` requires exactly two members,
+    // `getStore` and `run` (it never calls `enterWith`), so we provide only those.
+    // The cast bridges the generic `run<T>` signature to the interface's
+    // `run: (ctx, () => void) => void`.
+    traceable_1.AsyncLocalStorageProviderSingleton.initializeGlobalInstance({
+        getStore: () => manager.getStore(),
+        run: (context, fn) => manager.run(context, fn),
+    });
+}
+const sharedManager = new WorkflowContextManager();
+ensureProviderInstalled(sharedManager);
+/**
+ * Isolate-safe stand-in for Node's `AsyncLocalStorage`. The plugin's bundler
+ * aliases `node:async_hooks` (absent in the isolate) to this module so
+ * LangSmith's `import { AsyncLocalStorage } from "node:async_hooks"` resolves
+ * here. Every instance delegates to {@link sharedManager}.
+ *
+ * @internal
+ */
+class AsyncLocalStorage {
+    getStore() {
+        return sharedManager.getStore();
+    }
+    run(store, fn) {
+        return sharedManager.run(store, fn);
+    }
+    enterWith(_store) {
+        /* no-op: parenting inside the isolate is stack-scoped via run() */
+    }
+    /**
+     * Best-effort equivalent of `AsyncLocalStorage.snapshot()` (used only by
+     * LangSmith's async-generator streaming paths). Runs the callback in the
+     * current synchronous scope; the isolate has no async context to capture.
+     */
+    static snapshot() {
+        return (fn, ...args) => fn(...args);
+    }
+}
+exports.AsyncLocalStorage = AsyncLocalStorage;
+/** Reconstruct the propagated parent run from a Payload-keyed header map. */
+function reconstructParent(headers) {
+    return (0, run_tree_1.runTreeFromContext)((0, propagation_1.readContextHeader)(headers));
+}
+/**
+ * Re-wrap a reconstructed (plain) parent so its `createChild` produces
+ * replay-safe children. Carries the same id/trace/dotted so descendants parent
+ * under the real propagated run; never emitted itself.
+ */
+function asReplaySafeParent(parent, generateNewUuid4) {
+    if (!parent) {
+        return undefined;
+    }
+    return new run_tree_1.ReplaySafeRunTree({
+        name: parent.name,
+        run_type: parent.run_type,
+        id: parent.id,
+        trace_id: parent.trace_id,
+        dotted_order: parent.dotted_order,
+        parent_run_id: parent.parent_run_id,
+        project_name: parent.project_name,
+    }, generateNewUuid4);
+}
+/**
+ * A placeholder parent for the no-propagated-parent case. Installed as the ambient
+ * so a workflow-body `traceable` always nests via `createChild`; never emitted,
+ * and its children are independent roots (see {@link _RootReplaySafeRunTreeFactory}).
+ */
+function placeholderRoot(generateNewUuid4) {
+    return new run_tree_1._RootReplaySafeRunTreeFactory({ name: (0, workflow_1.workflowInfo)().workflowType, run_type: run_tree_1.RUN_TYPE.CHAIN }, generateNewUuid4);
+}
+/** Emit a short-lived marker run as a child of `parent`. */
+async function emitMarker(parent, name, inputs) {
+    const marker = parent.createChild({ name, run_type: run_tree_1.RUN_TYPE.CHAIN, inputs });
+    await (0, run_tree_1.emitMarkerRun)(marker);
+    return marker;
+}
+function buildReplaySafeRunTree(config, params) {
+    return new run_tree_1.ReplaySafeRunTree({
+        name: params.name,
+        run_type: params.runType,
+        parent_run: params.parent,
+        project_name: config.projectName,
+        tags: config.tags,
+        extra: { metadata: (0, propagation_1.scrubSensitive)(config.metadata) ?? {} },
+        inputs: params.inputs,
+    }, params.generateNewUuid4);
+}
+class LangSmithWorkflowInbound {
+    ctx;
+    config;
+    constructor(ctx, config) {
+        this.ctx = ctx;
+        this.config = config;
+    }
+    async execute(input, next) {
+        const parent = reconstructParent(input.headers);
+        return this.runInbound((0, run_tree_1.runWorkflowRunName)((0, workflow_1.workflowInfo)().workflowType), run_tree_1.RUN_TYPE.CHAIN, parent, { args: input.args }, () => next(input));
+    }
+    async handleSignal(input, next) {
+        const parent = reconstructParent(input.headers);
+        await this.runInbound((0, run_tree_1.handleSignalRunName)(input.signalName), run_tree_1.RUN_TYPE.CHAIN, parent, { args: input.args }, () => next(input), true);
+    }
+    async handleQuery(input, next) {
+        if ((0, propagation_1.isInternalQuery)(input.queryName)) {
+            return next(input);
+        }
+        const parent = reconstructParent(input.headers);
+        return this.runInbound((0, run_tree_1.handleQueryRunName)(input.queryName), run_tree_1.RUN_TYPE.CHAIN, parent, { args: input.args }, () => next(input), true, 
+        // Read-only: mint run ids from the nondeterministic unsafe random source so the
+        // handler never draws from the Workflow main PRNG (which a clean replay would not advance).
+        (0, workflow_1.workflowInfo)().unsafe.random.uuid4);
+    }
+    async handleUpdate(input, next) {
+        const parent = reconstructParent(input.headers);
+        return this.runInbound((0, run_tree_1.handleUpdateRunName)(input.name), run_tree_1.RUN_TYPE.CHAIN, parent, { args: input.args }, () => next(input), true);
+    }
+    validateUpdate(input, next) {
+        // Read-only: mint run ids from the nondeterministic unsafe random source so the validator
+        // never draws from the Workflow main PRNG (which a clean replay would not advance).
+        const generateNewUuid4 = (0, workflow_1.workflowInfo)().unsafe.random.uuid4;
+        if (!this.config.addTemporalRuns) {
+            // Propagation only: install the reconstructed parent (or a placeholder parent
+            // when none was propagated, to keep a validator-body `traceable` off the
+            // no-parent `crypto` path) so the validator body nests under the update's
+            // trace. Validators are synchronous, so install via the stack-based `run`,
+            // not the async `withAmbient`.
+            const ambient = asReplaySafeParent(reconstructParent(input.headers), generateNewUuid4) ?? placeholderRoot(generateNewUuid4);
+            this.ctx.run(ambient, () => next(input));
+            return;
+        }
+        const run = buildReplaySafeRunTree(this.config, {
+            name: (0, run_tree_1.validateUpdateRunName)(input.name),
+            runType: run_tree_1.RUN_TYPE.CHAIN,
+            parent: asReplaySafeParent(reconstructParent(input.headers), generateNewUuid4),
+            inputs: { args: input.args },
+            generateNewUuid4,
+        });
+        // Validators are synchronous and must not be made async; emit start/end
+        // around the synchronous call, with the run installed on the stack so a
+        // `traceable` in the validator body nests under it.
+        void run.postRun();
+        try {
+            this.ctx.run(run, () => next(input));
+            void run.end({});
+        }
+        catch (err) {
+            void run.end(undefined, (0, run_tree_1.describeError)(err));
+            void run.patchRun();
+            throw err;
+        }
+        void run.patchRun();
+    }
+    /**
+     * Open the inbound run, install it as the active run for `next`, and close it.
+     *
+     * `scoped` handlers install the run on the synchronous stack so a handler running
+     * concurrently with the workflow body cannot leak its run into the body's ambient;
+     * `execute` (scoped: false) installs a persistent ambient that survives the body's
+     * awaits.
+     */
+    async runInbound(name, runType, parent, inputs, next, scoped = false, generateNewUuid4) {
+        if (!this.config.addTemporalRuns) {
+            // Propagation only: install the reconstructed parent as the active run so
+            // user `traceable` runs nest under it; never emit a Temporal-operation run.
+            // With no propagated parent, install a placeholder parent instead of
+            // `undefined` so a workflow-body `traceable` takes LangSmith's
+            // `createChild` branch (deterministic id) rather than the no-parent branch
+            // that mints a uuid via `crypto`, which the isolate lacks.
+            const ambient = asReplaySafeParent(parent, generateNewUuid4) ?? placeholderRoot(generateNewUuid4);
+            return scoped ? this.ctx.run(ambient, next) : this.ctx.withAmbient(ambient, next);
+        }
+        const run = buildReplaySafeRunTree(this.config, {
+            name,
+            runType,
+            parent: asReplaySafeParent(parent, generateNewUuid4),
+            inputs,
+            generateNewUuid4,
+        });
+        await run.postRun();
+        const body = async () => {
+            try {
+                const result = await next();
+                await run.end((0, run_tree_1.asOutputs)(result));
+                await run.patchRun();
+                return result;
+            }
+            catch (err) {
+                await run.end(undefined, (0, run_tree_1.describeError)(err));
+                await run.patchRun();
+                throw err;
+            }
+        };
+        return scoped ? this.ctx.run(run, body) : this.ctx.withAmbient(run, body);
+    }
+}
+class LangSmithWorkflowOutbound {
+    ctx;
+    config;
+    constructor(ctx, config) {
+        this.ctx = ctx;
+        this.config = config;
+    }
+    async scheduleActivity(input, next) {
+        return this.peerMarker((0, run_tree_1.startActivityRunName)(input.activityType), input, (headers) => next({ ...input, headers }));
+    }
+    async scheduleLocalActivity(input, next) {
+        return this.peerMarker((0, run_tree_1.startActivityRunName)(input.activityType), input, (headers) => next({ ...input, headers }));
+    }
+    async startChildWorkflowExecution(input, next) {
+        return this.peerMarker((0, run_tree_1.startChildWorkflowRunName)(input.workflowType), input, (headers) => next({ ...input, headers }));
+    }
+    continueAsNew(input, next) {
+        // Don't propagate a placeholder root: it is never emitted, so the successor's
+        // runs would dangle under a nonexistent parent. Let it install its own.
+        const ambient = this.ctx.ambient();
+        const context = ambient instanceof run_tree_1._RootReplaySafeRunTreeFactory ? undefined : (0, run_tree_1.runHeaders)(ambient);
+        const headers = (0, propagation_1.withContextHeader)(input.headers, context);
+        return next({ ...input, headers });
+    }
+    async signalWorkflow(input, next) {
+        const isChild = input.target.type === 'child';
+        const name = isChild
+            ? (0, run_tree_1.signalChildWorkflowRunName)(input.signalName)
+            : (0, run_tree_1.signalExternalWorkflowRunName)(input.signalName);
+        return this.parentMarker(name, input, (headers) => next({ ...input, headers }));
+    }
+    async startNexusOperation(input, next) {
+        const ambient = this.ctx.ambient();
+        if (this.config.addTemporalRuns && ambient instanceof run_tree_1.ReplaySafeRunTree) {
+            await emitMarker(ambient, (0, run_tree_1.startNexusOperationRunName)(input.service, input.operation), {});
+        }
+        const ctx = (0, run_tree_1.runHeaders)(ambient);
+        const headers = ctx ? { ...input.headers, [propagation_1.HEADER_KEY]: (0, propagation_1.encodeContextString)(ctx) } : input.headers;
+        return next({ ...input, headers });
+    }
+    /** Sibling-marker operations: marker is a peer of the remote run; propagate ambient. */
+    async peerMarker(name, input, next) {
+        const ambient = this.ctx.ambient();
+        if (this.config.addTemporalRuns && ambient instanceof run_tree_1.ReplaySafeRunTree) {
+            await emitMarker(ambient, name, { args: input.args ?? [] });
+        }
+        const headers = (0, propagation_1.withContextHeader)(input.headers, (0, run_tree_1.runHeaders)(ambient));
+        return next(headers);
+    }
+    /** Parent-marker operations: remote handler nests under the marker; propagate marker. */
+    async parentMarker(name, input, next) {
+        const ambient = this.ctx.ambient();
+        let propagate = ambient;
+        if (this.config.addTemporalRuns && ambient instanceof run_tree_1.ReplaySafeRunTree) {
+            propagate = await emitMarker(ambient, name, { args: input.args ?? [] });
+        }
+        const headers = (0, propagation_1.withContextHeader)(input.headers, (0, run_tree_1.runHeaders)(propagate));
+        return next(headers);
+    }
+}
+/**
+ * Workflow interceptors factory. Loaded by the Temporal worker for every
+ * workflow in the bundle (registered via the plugin's `workflowModules`).
+ *
+ * @internal
+ */
+function interceptors() {
+    const config = readConfig();
+    return {
+        inbound: [new LangSmithWorkflowInbound(sharedManager, config)],
+        outbound: [new LangSmithWorkflowOutbound(sharedManager, config)],
+    };
+}
+//# sourceMappingURL=workflow-interceptors.js.map

package/lib/workflow-interceptors.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow-interceptors.js","sourceRoot":"","sources":["../src/workflow-interceptors.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;GAYG;;;AAigBH,oCAMC;AApgBD,8DAAoF;AASpF,mDAY8B;AAC9B,+CAOuB;AACvB,yCAmBoB;AAqBpB,SAAS,UAAU;IACjB,IAAI,CAAC;QACH,IAAI,OAAO,6BAA6B,KAAK,QAAQ,EAAE,CAAC;YACtD,OAAO,IAAI,CAAC,KAAK,CAAC,6BAA6B,CAA4B,CAAC;QAC9E,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,6BAA6B;IAC/B,CAAC;IACD,OAAO,EAAE,eAAe,EAAE,KAAK,EAAE,CAAC;AACpC,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,sBAAsB;IAClB,eAAe,CAAsB;IAC5B,KAAK,GAA4B,EAAE,CAAC;IAErD,QAAQ;QACN,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC;IAC1F,CAAC;IAED;;;OAGG;IACH,GAAG,CAAI,OAA4B,EAAE,EAAW;QAC9C,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACzB,IAAI,CAAC;YACH,OAAO,EAAE,EAAE,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;IAED,+EAA+E;IAC/E,OAAO;QACL,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC;IACzB,CAAC;IAED,gFAAgF;IAChF,KAAK,CAAC,WAAW,CAAI,GAAwB,EAAE,EAAoB;QACjE,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC;QAClC,IAAI,CAAC,eAAe,GAAG,GAAG,CAAC;QAC3B,IAAI,CAAC;YACH,OAAO,MAAM,EAAE,EAAE,CAAC;QACpB,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;QAC9B,CAAC;IACH,CAAC;CACF;AAED,IAAI,iBAAiB,GAAG,KAAK,CAAC;AAC9B,SAAS,uBAAuB,CAAC,OAA+B;IAC9D,IAAI,iBAAiB,EAAE,CAAC;QACtB,OAAO;IACT,CAAC;IACD,iBAAiB,GAAG,IAAI,CAAC;IACzB,yEAAyE;IACzE,+EAA+E;IAC/E,qEAAqE;IACrE,oCAAoC;IACpC,8CAAkC,CAAC,wBAAwB,CAAC;QAC1D,QAAQ,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,QAAQ,EAAE;QAClC,GAAG,EAAE,CAAI,OAA4B,EAAE,EAAW,EAAK,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,OAA8B,EAAE,EAAE,CAAC;KACrB,CAAC,CAAC;AAC1F,CAAC;AAED,MAAM,aAAa,GAAG,IAAI,sBAAsB,EAAE,CAAC;AACnD,uBAAuB,CAAC,aAAa,CAAC,CAAC;AAEvC;;;;;;;GAOG;AACH,MAAa,iBAAiB;IAC5B,QAAQ;QACN,OAAO,aAAa,CAAC,QAAQ,EAA8B,CAAC;IAC9D,CAAC;IAED,GAAG,CAAI,KAAQ,EAAE,EAA2B;QAC1C,OAAO,aAAa,CAAC,GAAG,CAAC,KAAuC,EAAE,EAAa,CAAC,CAAC;IACnF,CAAC;IAED,SAAS,CAAC,MAAS;QACjB,mEAAmE;IACrE,CAAC;IAED;;;;OAIG;IACH,MAAM,CAAC,QAAQ;QACb,OAAO,CAAC,EAAE,EAAE,GAAG,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;IACtC,CAAC;CACF;AArBD,8CAqBC;AAED,6EAA6E;AAC7E,SAAS,iBAAiB,CAAC,OAA4C;IACrE,OAAO,IAAA,6BAAkB,EAAC,IAAA,+BAAiB,EAAC,OAA4C,CAAC,CAAC,CAAC;AAC7F,CAAC;AAED;;;;GAIG;AACH,SAAS,kBAAkB,CACzB,MAA2B,EAC3B,gBAA+B;IAE/B,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,IAAI,4BAAiB,CAC1B;QACE,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,EAAE,EAAE,MAAM,CAAC,EAAE;QACb,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,YAAY,EAAE,MAAM,CAAC,YAAY;QACjC,aAAa,EAAE,MAAM,CAAC,aAAa;QACnC,YAAY,EAAE,MAAM,CAAC,YAAY;KAClC,EACD,gBAAgB,CACjB,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,SAAS,eAAe,CAAC,gBAA+B;IACtD,OAAO,IAAI,wCAA6B,CACtC,EAAE,IAAI,EAAE,IAAA,uBAAY,GAAE,CAAC,YAAY,EAAE,QAAQ,EAAE,mBAAQ,CAAC,KAAK,EAAE,EAC/D,gBAAgB,CACjB,CAAC;AACJ,CAAC;AAED,4DAA4D;AAC5D,KAAK,UAAU,UAAU,CACvB,MAAyB,EACzB,IAAY,EACZ,MAA+B;IAE/B,MAAM,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,mBAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;IAC9E,MAAM,IAAA,wBAAa,EAAC,MAAM,CAAC,CAAC;IAC5B,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,sBAAsB,CAC7B,MAA+B,EAC/B,MAMC;IAED,OAAO,IAAI,4BAAiB,CAC1B;QACE,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,QAAQ,EAAE,MAAM,CAAC,OAAO;QACxB,UAAU,EAAE,MAAM,CAAC,MAAM;QACzB,YAAY,EAAE,MAAM,CAAC,WAAW;QAChC,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,KAAK,EAAE,EAAE,QAAQ,EAAE,IAAA,4BAAc,EAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE;QAC1D,MAAM,EAAE,MAAM,CAAC,MAAM;KACtB,EACD,MAAM,CAAC,gBAAgB,CACxB,CAAC;AACJ,CAAC;AAED,MAAM,wBAAwB;IAET;IACA;IAFnB,YACmB,GAA2B,EAC3B,MAA+B;QAD/B,QAAG,GAAH,GAAG,CAAwB;QAC3B,WAAM,GAAN,MAAM,CAAyB;IAC/C,CAAC;IAEJ,KAAK,CAAC,OAAO,CAAC,KAA2B,EAAE,IAAsD;QAC/F,MAAM,MAAM,GAAG,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAChD,OAAO,IAAI,CAAC,UAAU,CACpB,IAAA,6BAAkB,EAAC,IAAA,uBAAY,GAAE,CAAC,YAAY,CAAC,EAC/C,mBAAQ,CAAC,KAAK,EACd,MAAM,EACN,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,EACpB,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAClB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,KAAkB,EAAE,IAA2D;QAChG,MAAM,MAAM,GAAG,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAChD,MAAM,IAAI,CAAC,UAAU,CACnB,IAAA,8BAAmB,EAAC,KAAK,CAAC,UAAU,CAAC,EACrC,mBAAQ,CAAC,KAAK,EACd,MAAM,EACN,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,EACpB,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,EACjB,IAAI,CACL,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAAiB,EAAE,IAA0D;QAC7F,IAAI,IAAA,6BAAe,EAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;YACrC,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC;QACrB,CAAC;QACD,MAAM,MAAM,GAAG,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAChD,OAAO,IAAI,CAAC,UAAU,CACpB,IAAA,6BAAkB,EAAC,KAAK,CAAC,SAAS,CAAC,EACnC,mBAAQ,CAAC,KAAK,EACd,MAAM,EACN,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,EACpB,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,EACjB,IAAI;QACJ,gFAAgF;QAChF,4FAA4F;QAC5F,IAAA,uBAAY,GAAE,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CACnC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,KAAkB,EAClB,IAA2D;QAE3D,MAAM,MAAM,GAAG,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAChD,OAAO,IAAI,CAAC,UAAU,CACpB,IAAA,8BAAmB,EAAC,KAAK,CAAC,IAAI,CAAC,EAC/B,mBAAQ,CAAC,KAAK,EACd,MAAM,EACN,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,EACpB,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,EACjB,IAAI,CACL,CAAC;IACJ,CAAC;IAED,cAAc,CAAC,KAAkB,EAAE,IAA6D;QAC9F,0FAA0F;QAC1F,oFAAoF;QACpF,MAAM,gBAAgB,GAAG,IAAA,uBAAY,GAAE,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC;QAC5D,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;YACjC,8EAA8E;YAC9E,yEAAyE;YACzE,0EAA0E;YAC1E,2EAA2E;YAC3E,+BAA+B;YAC/B,MAAM,OAAO,GACX,kBAAkB,CAAC,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,gBAAgB,CAAC,IAAI,eAAe,CAAC,gBAAgB,CAAC,CAAC;YAC9G,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;YACzC,OAAO;QACT,CAAC;QACD,MAAM,GAAG,GAAG,sBAAsB,CAAC,IAAI,CAAC,MAAM,EAAE;YAC9C,IAAI,EAAE,IAAA,gCAAqB,EAAC,KAAK,CAAC,IAAI,CAAC;YACvC,OAAO,EAAE,mBAAQ,CAAC,KAAK;YACvB,MAAM,EAAE,kBAAkB,CAAC,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,gBAAgB,CAAC;YAC9E,MAAM,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE;YAC5B,gBAAgB;SACjB,CAAC,CAAC;QACH,wEAAwE;QACxE,wEAAwE;QACxE,oDAAoD;QACpD,KAAK,GAAG,CAAC,OAAO,EAAE,CAAC;QACnB,IAAI,CAAC;YACH,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;YACrC,KAAK,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACnB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,KAAK,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,IAAA,wBAAa,EAAC,GAAG,CAAC,CAAC,CAAC;YAC5C,KAAK,GAAG,CAAC,QAAQ,EAAE,CAAC;YACpB,MAAM,GAAG,CAAC;QACZ,CAAC;QACD,KAAK,GAAG,CAAC,QAAQ,EAAE,CAAC;IACtB,CAAC;IAED;;;;;;;OAOG;IACK,KAAK,CAAC,UAAU,CACtB,IAAY,EACZ,OAAe,EACf,MAA2B,EAC3B,MAA+B,EAC/B,IAA4B,EAC5B,MAAM,GAAG,KAAK,EACd,gBAA+B;QAE/B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;YACjC,0EAA0E;YAC1E,4EAA4E;YAC5E,qEAAqE;YACrE,+DAA+D;YAC/D,2EAA2E;YAC3E,2DAA2D;YAC3D,MAAM,OAAO,GAAG,kBAAkB,CAAC,MAAM,EAAE,gBAAgB,CAAC,IAAI,eAAe,CAAC,gBAAgB,CAAC,CAAC;YAClG,OAAO,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACpF,CAAC;QACD,MAAM,GAAG,GAAG,sBAAsB,CAAC,IAAI,CAAC,MAAM,EAAE;YAC9C,IAAI;YACJ,OAAO;YACP,MAAM,EAAE,kBAAkB,CAAC,MAAM,EAAE,gBAAgB,CAAC;YACpD,MAAM;YACN,gBAAgB;SACjB,CAAC,CAAC;QACH,MAAM,GAAG,CAAC,OAAO,EAAE,CAAC;QACpB,MAAM,IAAI,GAAG,KAAK,IAAsB,EAAE;YACxC,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,IAAI,EAAE,CAAC;gBAC5B,MAAM,GAAG,CAAC,GAAG,CAAC,IAAA,oBAAS,EAAC,MAAM,CAAC,CAAC,CAAC;gBACjC,MAAM,GAAG,CAAC,QAAQ,EAAE,CAAC;gBACrB,OAAO,MAAM,CAAC;YAChB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,IAAA,wBAAa,EAAC,GAAG,CAAC,CAAC,CAAC;gBAC7C,MAAM,GAAG,CAAC,QAAQ,EAAE,CAAC;gBACrB,MAAM,GAAG,CAAC;YACZ,CAAC;QACH,CAAC,CAAC;QACF,OAAO,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;IAC5E,CAAC;CACF;AAED,MAAM,yBAAyB;IAEV;IACA;IAFnB,YACmB,GAA2B,EAC3B,MAA+B;QAD/B,QAAG,GAAH,GAAG,CAAwB;QAC3B,WAAM,GAAN,MAAM,CAAyB;IAC/C,CAAC;IAEJ,KAAK,CAAC,gBAAgB,CACpB,KAAoB,EACpB,IAAgE;QAEhE,OAAO,IAAI,CAAC,UAAU,CAAC,IAAA,+BAAoB,EAAC,KAAK,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,CAAC,OAAgB,EAAE,EAAE,CAC3F,IAAI,CAAC,EAAE,GAAG,KAAK,EAAE,OAAO,EAAE,CAAC,CAC5B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,qBAAqB,CACzB,KAAyB,EACzB,IAAqE;QAErE,OAAO,IAAI,CAAC,UAAU,CAAC,IAAA,+BAAoB,EAAC,KAAK,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,CAAC,OAAgB,EAAE,EAAE,CAC3F,IAAI,CAAC,EAAE,GAAG,KAAK,EAAE,OAAO,EAAE,CAAC,CAC5B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,2BAA2B,CAC/B,KAAuC,EACvC,IAA2E;QAE3E,OAAO,IAAI,CAAC,UAAU,CAAC,IAAA,oCAAyB,EAAC,KAAK,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,CAAC,OAAgB,EAAE,EAAE,CAChG,IAAI,CAAC,EAAE,GAAG,KAAK,EAAE,OAAO,EAAE,CAAC,CAC5B,CAAC;IACJ,CAAC;IAED,aAAa,CACX,KAAyB,EACzB,IAA6D;QAE7D,8EAA8E;QAC9E,wEAAwE;QACxE,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACnC,MAAM,OAAO,GAAG,OAAO,YAAY,wCAA6B,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAA,qBAAU,EAAC,OAAO,CAAC,CAAC;QACnG,MAAM,OAAO,GAAG,IAAA,+BAAiB,EAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC1D,OAAO,IAAI,CAAC,EAAE,GAAG,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,cAAc,CAClB,KAA0B,EAC1B,IAA8D;QAE9D,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,OAAO,CAAC;QAC9C,MAAM,IAAI,GAAG,OAAO;YAClB,CAAC,CAAC,IAAA,qCAA0B,EAAC,KAAK,CAAC,UAAU,CAAC;YAC9C,CAAC,CAAC,IAAA,wCAA6B,EAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QACpD,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,OAAgB,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,GAAG,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;IAC3F,CAAC;IAED,KAAK,CAAC,mBAAmB,CACvB,KAA+B,EAC/B,IAAmE;QAEnE,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACnC,IAAI,IAAI,CAAC,MAAM,CAAC,eAAe,IAAI,OAAO,YAAY,4BAAiB,EAAE,CAAC;YACxE,MAAM,UAAU,CAAC,OAAO,EAAE,IAAA,qCAA0B,EAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,SAAS,CAAC,EAAE,EAAE,CAAC,CAAC;QAC5F,CAAC;QACD,MAAM,GAAG,GAAG,IAAA,qBAAU,EAAC,OAAO,CAAC,CAAC;QAChC,MAAM,OAAO,GAAG,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,wBAAU,CAAC,EAAE,IAAA,iCAAmB,EAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC;QACnG,OAAO,IAAI,CAAC,EAAE,GAAG,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;IACrC,CAAC;IAED,wFAAwF;IAChF,KAAK,CAAC,UAAU,CACtB,IAAY,EACZ,KAA6C,EAC7C,IAAsC;QAEtC,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACnC,IAAI,IAAI,CAAC,MAAM,CAAC,eAAe,IAAI,OAAO,YAAY,4BAAiB,EAAE,CAAC;YACxE,MAAM,UAAU,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;QAC9D,CAAC;QACD,MAAM,OAAO,GAAG,IAAA,+BAAiB,EAAC,KAAK,CAAC,OAAO,EAAE,IAAA,qBAAU,EAAC,OAAO,CAAC,CAAC,CAAC;QACtE,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;IACvB,CAAC;IAED,yFAAyF;IACjF,KAAK,CAAC,YAAY,CACxB,IAAY,EACZ,KAA6C,EAC7C,IAAsC;QAEtC,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACnC,IAAI,SAAS,GAAwB,OAAO,CAAC;QAC7C,IAAI,IAAI,CAAC,MAAM,CAAC,eAAe,IAAI,OAAO,YAAY,4BAAiB,EAAE,CAAC;YACxE,SAAS,GAAG,MAAM,UAAU,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;QAC1E,CAAC;QACD,MAAM,OAAO,GAAG,IAAA,+BAAiB,EAAC,KAAK,CAAC,OAAO,EAAE,IAAA,qBAAU,EAAC,SAAS,CAAC,CAAC,CAAC;QACxE,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;IACvB,CAAC;CACF;AAED;;;;;GAKG;AACH,SAAgB,YAAY;IAC1B,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,OAAO;QACL,OAAO,EAAE,CAAC,IAAI,wBAAwB,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;QAC9D,QAAQ,EAAE,CAAC,IAAI,yBAAyB,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;KACjE,CAAC;AACJ,CAAC"}

package/package.json

@@ -0,0 +1,88 @@
+{
+  "name": "@temporalio/langsmith",
+  "version": "1.20.0",
+  "description": "Temporal LangSmith observability integration package",
+  "main": "lib/index.js",
+  "types": "./lib/index.d.ts",
+  "typesVersions": {
+    "*": {
+      "workflow-interceptors": [
+        "./lib/workflow-interceptors.d.ts"
+      ]
+    }
+  },
+  "ava": {
+    "files": [
+      "./lib/__tests__/test-*.js"
+    ],
+    "timeout": "120s",
+    "concurrency": 1,
+    "workerThreads": false
+  },
+  "exports": {
+    ".": {
+      "types": "./lib/index.d.ts",
+      "import": "./lib/index.js",
+      "require": "./lib/index.js",
+      "default": "./lib/index.js"
+    },
+    "./workflow-interceptors": {
+      "types": "./lib/workflow-interceptors.d.ts",
+      "import": "./lib/workflow-interceptors.js",
+      "require": "./lib/workflow-interceptors.js",
+      "default": "./lib/workflow-interceptors.js"
+    }
+  },
+  "keywords": [
+    "temporal",
+    "workflow",
+    "ai",
+    "langsmith",
+    "observability",
+    "tracing"
+  ],
+  "author": "Temporal Technologies Inc. <sdk@temporal.io>",
+  "license": "MIT",
+  "dependencies": {
+    "@temporalio/activity": "1.20.0",
+    "@temporalio/client": "1.20.0",
+    "@temporalio/plugin": "1.20.0",
+    "@temporalio/workflow": "1.20.0",
+    "@temporalio/common": "1.20.0",
+    "@temporalio/worker": "1.20.0"
+  },
+  "peerDependencies": {
+    "langsmith": "^0.7.9"
+  },
+  "devDependencies": {
+    "ava": "^5.3.1",
+    "langsmith": "^0.7.9",
+    "nexus-rpc": "^0.0.2",
+    "@temporalio/testing": "1.20.0"
+  },
+  "engines": {
+    "node": ">= 20.0.0"
+  },
+  "bugs": {
+    "url": "https://github.com/temporalio/sdk-typescript/issues"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/temporalio/sdk-typescript.git",
+    "directory": "contrib/langsmith"
+  },
+  "homepage": "https://github.com/temporalio/sdk-typescript/tree/main/contrib/langsmith",
+  "publishConfig": {
+    "access": "public"
+  },
+  "files": [
+    "src",
+    "lib",
+    "!src/__tests__",
+    "!lib/__tests__"
+  ],
+  "scripts": {
+    "build": "tsc --build",
+    "test": "ava ./lib/__tests__/test-*.js"
+  }
+}

package/src/activity-interceptor.ts

@@ -0,0 +1,139 @@
+/**
+ * Activity-side and Nexus-handler-side LangSmith interceptors. Both run in the
+ * real Node worker process and install the reconstructed run via `withRunTree`
+ * so a user's unchanged body `traceable` calls nest under it.
+ *
+ * @module
+ */
+
+import { isTracingEnabled } from 'langsmith';
+import { RunTree } from 'langsmith/run_trees';
+import { withRunTree } from 'langsmith/traceable';
+import type { Context as ActivityContext } from '@temporalio/activity';
+import type { ActivityInboundCallsInterceptorFactory, NexusInboundCallsInterceptor } from '@temporalio/worker';
+
+import { HEADER_KEY, decodeContextString, readContextHeader, type LangSmithTraceContext } from './propagation';
+import {
+  RUN_TYPE,
+  asOutputs,
+  describeError,
+  buildRunTree,
+  runActivityRunName,
+  runCancelNexusHandlerRunName,
+  runStartNexusHandlerRunName,
+  runTreeFromContext,
+} from './run-tree';
+import type { EmitterConfig } from './sinks';
+
+/**
+ * Reconstruct the propagated parent run from a LangSmith trace context, wired to
+ * the plugin-configured client so descendant runs emit to the right place. The
+ * returned run is the *parent* — never posted itself — used only to parent
+ * the operation run (or directly as ambient when `addTemporalRuns` is off).
+ */
+function reconstructParentRun(config: EmitterConfig, ctx: LangSmithTraceContext | undefined): RunTree | undefined {
+  const parsed = runTreeFromContext(ctx);
+  if (!parsed) {
+    return undefined;
+  }
+  return new RunTree({
+    name: parsed.name || 'parent',
+    run_type: parsed.run_type || RUN_TYPE.CHAIN,
+    id: parsed.id,
+    trace_id: parsed.trace_id,
+    dotted_order: parsed.dotted_order,
+    parent_run_id: parsed.parent_run_id,
+    project_name: config.projectName ?? parsed.project_name,
+    client: config.client,
+    // Force-enable so nested body `traceable` runs emit; tracing gate is `isTracingEnabled()`.
+    tracingEnabled: true,
+  });
+}
+
+/**
+ * Run `fn` with `op` installed as the active LangSmith run, emitting the run
+ * around the call. Shared by activity and Nexus handlers.
+ */
+async function traceOperation<T>(op: RunTree, fn: () => Promise<T>): Promise<T> {
+  await op.postRun();
+  try {
+    const result = await withRunTree(op, fn);
+    await op.end(asOutputs(result));
+    await op.patchRun();
+    return result;
+  } catch (err) {
+    await op.end(undefined, describeError(err));
+    await op.patchRun();
+    throw err;
+  }
+}
+
+/**
+ * Build the activity-inbound interceptor factory. The worker calls the factory
+ * once per activity with that activity's {@link ActivityContext}, from which we
+ * read the activity type for the run name.
+ */
+export function createActivityInboundInterceptor(config: EmitterConfig): ActivityInboundCallsInterceptorFactory {
+  return (ctx: ActivityContext) => ({
+    async execute(input, next) {
+      if (!isTracingEnabled()) {
+        return next(input);
+      }
+      const parent = reconstructParentRun(config, readContextHeader(input.headers));
+      if (!config.addTemporalRuns) {
+        // Propagation only: nest the user's body `traceable` runs under the
+        // reconstructed parent without emitting a Temporal-operation run.
+        // No propagated parent: run the body with no ambient run by design (nothing to attach to).
+        return parent ? withRunTree(parent, () => next(input)) : next(input);
+      }
+      const activityType = ctx.info.activityType;
+      const run = buildRunTree(config, {
+        name: runActivityRunName(activityType),
+        runType: RUN_TYPE.TOOL,
+        parent,
+        inputs: { args: input.args },
+      });
+      return traceOperation(run, () => next(input));
+    },
+  });
+}
+
+function nexusContext(headers: Record<string, string>): LangSmithTraceContext | undefined {
+  return decodeContextString(headers[HEADER_KEY]);
+}
+
+/**
+ * Build the Nexus-handler interceptor. `startOperation` opens a
+ * `RunStartNexusOperationHandler:` run (so workflows the handler starts nest
+ * under it); `cancelOperation` opens `RunCancelNexusOperationHandler:`.
+ */
+export function createNexusInboundInterceptor(config: EmitterConfig): NexusInboundCallsInterceptor {
+  const handle = async <I extends { ctx: { service: string; operation: string; headers: Record<string, string> } }, O>(
+    input: I,
+    next: (input: I) => Promise<O>,
+    nameOf: (s: string, o: string) => string
+  ): Promise<O> => {
+    if (!isTracingEnabled()) {
+      return next(input);
+    }
+    const parent = reconstructParentRun(config, nexusContext(input.ctx.headers));
+    if (!config.addTemporalRuns) {
+      return parent ? withRunTree(parent, () => next(input)) : next(input);
+    }
+    const run = buildRunTree(config, {
+      name: nameOf(input.ctx.service, input.ctx.operation),
+      runType: RUN_TYPE.CHAIN,
+      parent,
+    });
+    return traceOperation(run, () => next(input));
+  };
+
+  return {
+    startOperation(input, next) {
+      return handle(input, next, runStartNexusHandlerRunName);
+    },
+    cancelOperation(input, next) {
+      return handle(input, next, runCancelNexusHandlerRunName);
+    },
+  };
+}