@mongodb-js/agent-engine-sdk-langgraph 0.11.3

package/dist/durable_deep_agent.js

@@ -0,0 +1,176 @@
+/**
+ * Durable operation-path boundaries for Deep Agent task delegation.
+ *
+ * Deep Agent exposes subagent dispatch through the public `task` tool and
+ * middleware hooks, rather than as a compiled child graph:
+ * https://docs.langchain.com/oss/javascript/deepagents/subagents
+ *
+ * Same-step `task` siblings stamp ToolCall-id ordinals after FinalizeStep.
+ * Stamping in `afterModel` would key those ordinals to the model step, which
+ * FinalizeStep then leaves behind.
+ *
+ * Port of `durable_deep_agent.py`.
+ */
+import { AsyncLocalStorage } from "node:async_hooks";
+import { createMiddleware } from "langchain";
+import { isHumanMessage, isToolMessage, } from "@langchain/core/messages";
+import { childOperationBoundaryScope, currentAttemptContext, preallocateChildOperationOrdinals, currentPendingChildOperationBatch, setPendingChildOperationBatch, } from "@mongodb-js/agent-engine-runner-shared";
+import { parseTaskCall } from "./deep_agent_task.js";
+import { UnsupportedDurableGraphError } from "./platform_checkpointer.js";
+const activeTaskCallId = new AsyncLocalStorage();
+function messagesFromState(state) {
+    if (state === null || typeof state !== "object")
+        return [];
+    const messages = state.messages;
+    return Array.isArray(messages) ? messages : [];
+}
+/**
+ * Assign a deterministic id to one message. LangChain's serializer emits only
+ * keys present in `lc_kwargs`, so a late-added `id` must be mirrored there or
+ * checkpointed messages silently lose it (same as `assignBaseMessageId`).
+ */
+function assignMessageId(message, id) {
+    message.id = id;
+    const kwargs = message.lc_kwargs;
+    if (kwargs !== undefined && kwargs !== null && typeof kwargs === "object") {
+        kwargs["id"] = id;
+    }
+}
+/**
+ * Replace Deep Agent's generated message ids before LangGraph persists them.
+ * No-op outside a task scope, so native sessions keep provider ids.
+ */
+export function stampDeepAgentMessageIds(messages) {
+    const taskCallId = activeTaskCallId.getStore();
+    if (taskCallId === undefined)
+        return;
+    messages.forEach((message, index) => {
+        // isHumanMessage/isToolMessage key off getType(), matching messages a
+        // nested graph created through its own @langchain/core module copy —
+        // instanceof would miss those.
+        if (isHumanMessage(message)) {
+            assignMessageId(message, `durable-deep-agent-input:${taskCallId}:${index}`);
+        }
+        else if (isToolMessage(message)) {
+            assignMessageId(message, `durable-deep-agent-tool-result:${taskCallId}:${message.tool_call_id}`);
+        }
+    });
+}
+function stampTaskResultMessageId(result, toolCallId) {
+    if (isToolMessage(result)) {
+        assignMessageId(result, `durable-deep-agent-task-result:${toolCallId}`);
+        return;
+    }
+    const update = result.update;
+    if (update === null || typeof update !== "object" || Array.isArray(update)) {
+        return;
+    }
+    const messages = update["messages"];
+    if (!Array.isArray(messages))
+        return;
+    for (const message of messages) {
+        if (isToolMessage(message) && message.tool_call_id === toolCallId) {
+            assignMessageId(message, `durable-deep-agent-task-result:${toolCallId}`);
+        }
+    }
+}
+export function createDurableDeepAgentMiddleware(options) {
+    const unsupportedSubagentNames = options.unsupportedSubagentNames ?? new Set();
+    const boundaryFromTask = (toolCall) => {
+        const taskCall = parseTaskCall(toolCall);
+        if (taskCall === null)
+            return null;
+        if (taskCall.toolCallId === "") {
+            throw new Error("durable Deep Agent task requires a non-empty tool call ID");
+        }
+        if (taskCall.subagentName === "") {
+            throw new Error("durable Deep Agent task requires a non-empty subagent type");
+        }
+        if (unsupportedSubagentNames.has(taskCall.subagentName)) {
+            throw new UnsupportedDurableGraphError(`compiled Deep Agent subagent '${taskCall.subagentName}' must use ` +
+                "checkpointer=None; independent child checkpointers are not supported " +
+                "during durable execution");
+        }
+        return {
+            name: taskCall.subagentName,
+            occurrenceKey: taskCall.toolCallId,
+        };
+    };
+    const taskBoundary = (toolCall) => {
+        if (currentAttemptContext() === null)
+            return null;
+        return boundaryFromTask(toolCall);
+    };
+    const taskBoundariesFromState = (state) => {
+        const messages = messagesFromState(state);
+        if (messages.length === 0)
+            return [];
+        const last = messages[messages.length - 1];
+        const toolCalls = last?.tool_calls ?? [];
+        if (!Array.isArray(toolCalls))
+            return [];
+        const boundaries = [];
+        for (const toolCall of toolCalls) {
+            const candidate = boundaryFromTask(toolCall);
+            if (candidate !== null)
+                boundaries.push(candidate);
+        }
+        return boundaries;
+    };
+    const pendingBoundaries = (state) => {
+        const fromState = taskBoundariesFromState(state);
+        if (fromState.length > 0)
+            return fromState;
+        return currentPendingChildOperationBatch();
+    };
+    const preallocatePendingTaskBatch = (toolCall, state) => {
+        // Phase 2: stamp on the ToolNode superstep, after FinalizeStep advanced
+        // the root step counter. Skip if this wrap is not in the stashed batch
+        // (sequential later-step task calls must first-seen-allocate).
+        if (currentAttemptContext() === null)
+            return;
+        const current = boundaryFromTask(toolCall);
+        if (current === null)
+            return;
+        const pending = pendingBoundaries(state);
+        if (pending.length === 0 ||
+            !pending.some((candidate) => candidate.name === current.name &&
+                candidate.occurrenceKey === current.occurrenceKey)) {
+            return;
+        }
+        preallocateChildOperationOrdinals(pending);
+        setPendingChildOperationBatch([]);
+    };
+    return createMiddleware({
+        name: "durableDeepAgentMiddleware",
+        afterModel: (state) => {
+            // Phase 1: validate and stash. Stamping here would key ordinals to the
+            // model step, which FinalizeStep then leaves behind. Mutate the
+            // attempt-scoped holder so Pregel worker copies see the batch.
+            if (currentAttemptContext() === null)
+                return undefined;
+            setPendingChildOperationBatch(taskBoundariesFromState(state));
+            return undefined;
+        },
+        wrapToolCall: (request, handler) => {
+            preallocatePendingTaskBatch(request.toolCall, request.state);
+            const boundary = taskBoundary(request.toolCall);
+            if (boundary === null)
+                return handler(request);
+            return activeTaskCallId.run(boundary.occurrenceKey, () => {
+                const result = childOperationBoundaryScope(boundary, () => handler(request));
+                // Both branches may be a promise; stamp after settlement so the
+                // returned Command/ToolMessage carries the deterministic id.
+                if (result instanceof Promise) {
+                    return result.then((settled) => {
+                        stampTaskResultMessageId(settled, boundary.occurrenceKey);
+                        return settled;
+                    });
+                }
+                stampTaskResultMessageId(result, boundary.occurrenceKey);
+                return result;
+            });
+        },
+    });
+}
+//# sourceMappingURL=durable_deep_agent.js.map

package/dist/durable_deep_agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"durable_deep_agent.js","sourceRoot":"","sources":["../src/durable_deep_agent.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAEH,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAE7C,OAAO,EAEL,cAAc,EACd,aAAa,GAEd,MAAM,0BAA0B,CAAC;AAClC,OAAO,EACL,2BAA2B,EAC3B,qBAAqB,EACrB,iCAAiC,EACjC,iCAAiC,EACjC,6BAA6B,GAE9B,MAAM,wCAAwC,CAAC;AAEhD,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,4BAA4B,EAAE,MAAM,4BAA4B,CAAC;AAI1E,MAAM,gBAAgB,GAAG,IAAI,iBAAiB,EAAU,CAAC;AAEzD,SAAS,iBAAiB,CAAC,KAAc;IACvC,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,EAAE,CAAC;IAC3D,MAAM,QAAQ,GAAI,KAAgC,CAAC,QAAQ,CAAC;IAC5D,OAAO,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAE,QAA0B,CAAC,CAAC,CAAC,EAAE,CAAC;AACpE,CAAC;AAED;;;;GAIG;AACH,SAAS,eAAe,CAAC,OAAoB,EAAE,EAAU;IACvD,OAAO,CAAC,EAAE,GAAG,EAAE,CAAC;IAChB,MAAM,MAAM,GAAI,OAAmD,CAAC,SAAS,CAAC;IAC9E,IAAI,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,IAAI,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC1E,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;IACpB,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,wBAAwB,CAAC,QAAuB;IAC9D,MAAM,UAAU,GAAG,gBAAgB,CAAC,QAAQ,EAAE,CAAC;IAC/C,IAAI,UAAU,KAAK,SAAS;QAAE,OAAO;IACrC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;QAClC,sEAAsE;QACtE,qEAAqE;QACrE,+BAA+B;QAC/B,IAAI,cAAc,CAAC,OAAO,CAAC,EAAE,CAAC;YAC5B,eAAe,CACb,OAAO,EACP,4BAA4B,UAAU,IAAI,KAAK,EAAE,CAClD,CAAC;QACJ,CAAC;aAAM,IAAI,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC;YAClC,eAAe,CACb,OAAO,EACP,kCAAkC,UAAU,IAAI,OAAO,CAAC,YAAY,EAAE,CACvE,CAAC;QACJ,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,wBAAwB,CAC/B,MAAsB,EACtB,UAAkB;IAElB,IAAI,aAAa,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,eAAe,CAAC,MAAM,EAAE,kCAAkC,UAAU,EAAE,CAAC,CAAC;QACxE,OAAO;IACT,CAAC;IACD,MAAM,MAAM,GAAI,MAA+B,CAAC,MAAM,CAAC;IACvD,IAAI,MAAM,KAAK,IAAI,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAC3E,OAAO;IACT,CAAC;IACD,MAAM,QAAQ,GAAI,MAAkC,CAAC,UAAU,CAAC,CAAC;IACjE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC;QAAE,OAAO;IACrC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,aAAa,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,YAAY,KAAK,UAAU,EAAE,CAAC;YAClE,eAAe,CAAC,OAAO,EAAE,kCAAkC,UAAU,EAAE,CAAC,CAAC;QAC3E,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,UAAU,gCAAgC,CAAC,OAEhD;IACC,MAAM,wBAAwB,GAC5B,OAAO,CAAC,wBAAwB,IAAI,IAAI,GAAG,EAAE,CAAC;IAEhD,MAAM,gBAAgB,GAAG,CACvB,QAAiB,EACc,EAAE;QACjC,MAAM,QAAQ,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;QACzC,IAAI,QAAQ,KAAK,IAAI;YAAE,OAAO,IAAI,CAAC;QACnC,IAAI,QAAQ,CAAC,UAAU,KAAK,EAAE,EAAE,CAAC;YAC/B,MAAM,IAAI,KAAK,CACb,2DAA2D,CAC5D,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,CAAC,YAAY,KAAK,EAAE,EAAE,CAAC;YACjC,MAAM,IAAI,KAAK,CACb,4DAA4D,CAC7D,CAAC;QACJ,CAAC;QACD,IAAI,wBAAwB,CAAC,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;YACxD,MAAM,IAAI,4BAA4B,CACpC,iCAAiC,QAAQ,CAAC,YAAY,aAAa;gBACjE,uEAAuE;gBACvE,0BAA0B,CAC7B,CAAC;QACJ,CAAC;QACD,OAAO;YACL,IAAI,EAAE,QAAQ,CAAC,YAAY;YAC3B,aAAa,EAAE,QAAQ,CAAC,UAAU;SACnC,CAAC;IACJ,CAAC,CAAC;IAEF,MAAM,YAAY,GAAG,CAAC,QAAiB,EAAiC,EAAE;QACxE,IAAI,qBAAqB,EAAE,KAAK,IAAI;YAAE,OAAO,IAAI,CAAC;QAClD,OAAO,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC,CAAC;IAEF,MAAM,uBAAuB,GAAG,CAC9B,KAAc,EACY,EAAE;QAC5B,MAAM,QAAQ,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAC1C,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QACrC,MAAM,IAAI,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC3C,MAAM,SAAS,GACZ,IAA+C,EAAE,UAAU,IAAI,EAAE,CAAC;QACrE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC;YAAE,OAAO,EAAE,CAAC;QACzC,MAAM,UAAU,GAA6B,EAAE,CAAC;QAChD,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YACjC,MAAM,SAAS,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YAC7C,IAAI,SAAS,KAAK,IAAI;gBAAE,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACrD,CAAC;QACD,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;IAEF,MAAM,iBAAiB,GAAG,CACxB,KAAc,EACqB,EAAE;QACrC,MAAM,SAAS,GAAG,uBAAuB,CAAC,KAAK,CAAC,CAAC;QACjD,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC;YAAE,OAAO,SAAS,CAAC;QAC3C,OAAO,iCAAiC,EAAE,CAAC;IAC7C,CAAC,CAAC;IAEF,MAAM,2BAA2B,GAAG,CAClC,QAAiB,EACjB,KAAc,EACR,EAAE;QACR,wEAAwE;QACxE,uEAAuE;QACvE,+DAA+D;QAC/D,IAAI,qBAAqB,EAAE,KAAK,IAAI;YAAE,OAAO;QAC7C,MAAM,OAAO,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC3C,IAAI,OAAO,KAAK,IAAI;YAAE,OAAO;QAC7B,MAAM,OAAO,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;QACzC,IACE,OAAO,CAAC,MAAM,KAAK,CAAC;YACpB,CAAC,OAAO,CAAC,IAAI,CACX,CAAC,SAAS,EAAE,EAAE,CACZ,SAAS,CAAC,IAAI,KAAK,OAAO,CAAC,IAAI;gBAC/B,SAAS,CAAC,aAAa,KAAK,OAAO,CAAC,aAAa,CACpD,EACD,CAAC;YACD,OAAO;QACT,CAAC;QACD,iCAAiC,CAAC,OAAO,CAAC,CAAC;QAC3C,6BAA6B,CAAC,EAAE,CAAC,CAAC;IACpC,CAAC,CAAC;IAEF,OAAO,gBAAgB,CAAC;QACtB,IAAI,EAAE,4BAA4B;QAClC,UAAU,EAAE,CAAC,KAAK,EAAE,EAAE;YACpB,uEAAuE;YACvE,gEAAgE;YAChE,+DAA+D;YAC/D,IAAI,qBAAqB,EAAE,KAAK,IAAI;gBAAE,OAAO,SAAS,CAAC;YACvD,6BAA6B,CAAC,uBAAuB,CAAC,KAAK,CAAC,CAAC,CAAC;YAC9D,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,YAAY,EAAE,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE;YACjC,2BAA2B,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;YAC7D,MAAM,QAAQ,GAAG,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAChD,IAAI,QAAQ,KAAK,IAAI;gBAAE,OAAO,OAAO,CAAC,OAAO,CAAC,CAAC;YAC/C,OAAO,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;gBACvD,MAAM,MAAM,GAAG,2BAA2B,CAAC,QAAQ,EAAE,GAAG,EAAE,CACxD,OAAO,CAAC,OAAO,CAAC,CACjB,CAAC;gBACF,gEAAgE;gBAChE,6DAA6D;gBAC7D,IAAI,MAAM,YAAY,OAAO,EAAE,CAAC;oBAC9B,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE;wBAC7B,wBAAwB,CAAC,OAAO,EAAE,QAAQ,CAAC,aAAa,CAAC,CAAC;wBAC1D,OAAO,OAAO,CAAC;oBACjB,CAAC,CAAC,CAAC;gBACL,CAAC;gBACD,wBAAwB,CAAC,MAAM,EAAE,QAAQ,CAAC,aAAa,CAAC,CAAC;gBACzD,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC,CAAC;QACL,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}

package/dist/durable_message_identity.d.ts

@@ -0,0 +1,5 @@
+/** Stable identities for messages emitted by durable LangGraph nodes. */
+import { type CompiledStateGraph } from "@langchain/langgraph";
+/** Stamp missing durable message IDs before named LangGraph reducers run. */
+export declare function installDurableMessageIdentity(graph: CompiledStateGraph<unknown, unknown>): void;
+//# sourceMappingURL=durable_message_identity.d.ts.map

package/dist/durable_message_identity.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"durable_message_identity.d.ts","sourceRoot":"","sources":["../src/durable_message_identity.ts"],"names":[],"mappings":"AAAA,yEAAyE;AAWzE,OAAO,EAKL,KAAK,kBAAkB,EACxB,MAAM,sBAAsB,CAAC;AAsI9B,6EAA6E;AAC7E,wBAAgB,6BAA6B,CAC3C,KAAK,EAAE,kBAAkB,CAAC,OAAO,EAAE,OAAO,CAAC,GAC1C,IAAI,CAgCN"}

package/dist/durable_message_identity.js

@@ -0,0 +1,106 @@
+/** Stable identities for messages emitted by durable LangGraph nodes. */
+import { createHash } from "node:crypto";
+import { RemoveMessage, coerceMessageLikeToMessage, } from "@langchain/core/messages";
+import { RunnableLambda } from "@langchain/core/runnables";
+import { Overwrite, isCommand, messagesStateReducer, } from "@langchain/langgraph";
+import { currentAttemptContext, currentOperationPath, currentStepOrdinal, } from "@mongodb-js/agent-engine-runner-shared";
+const installedNodes = new WeakSet();
+function durableMessageId(nodeName, outputPath) {
+    const attempt = currentAttemptContext();
+    if (attempt === null) {
+        throw new Error("durable message identity requires an active attempt");
+    }
+    const { executionId } = attempt.workflowIdentity;
+    const operationPath = currentOperationPath().segments.map((segment) => [
+        segment.name,
+        segment.ordinal.toString(),
+    ]);
+    const producer = JSON.stringify([operationPath, nodeName, outputPath]);
+    const producerHash = createHash("sha256").update(producer).digest("hex");
+    return `durable-message:${executionId}:${currentStepOrdinal()}:${producerHash}`;
+}
+function assignBaseMessageId(message, nodeName, path) {
+    if ((message.id === null || message.id === undefined) &&
+        !RemoveMessage.isInstance(message)) {
+        const id = durableMessageId(nodeName, path);
+        message.id = id;
+        message.lc_kwargs.id = id;
+    }
+}
+function assignMessageId(value, nodeName, path) {
+    const message = coerceMessageLikeToMessage(value);
+    assignBaseMessageId(message, nodeName, path);
+    return message;
+}
+function assignMessageUpdateIds(value, nodeName, path) {
+    if (Overwrite.isInstance(value) && "__overwrite__" in value) {
+        const overwrite = value;
+        const normalized = assignMessageUpdateIds(overwrite.__overwrite__, nodeName, [...path, "__overwrite__"]);
+        return value instanceof Overwrite
+            ? new Overwrite(normalized)
+            : { ...overwrite, __overwrite__: normalized };
+    }
+    if (Array.isArray(value)) {
+        return value.map((item, index) => assignMessageId(item, nodeName, [...path, index]));
+    }
+    return assignMessageId(value, nodeName, path);
+}
+function assignStateUpdateIds(value, nodeName, path, messageChannels) {
+    if (value === null || typeof value !== "object")
+        return value;
+    const update = value;
+    for (const channel of messageChannels) {
+        if (channel in update) {
+            update[channel] = assignMessageUpdateIds(update[channel], nodeName, [
+                ...path,
+                channel,
+            ]);
+        }
+    }
+    return update;
+}
+function withDurableMessageIdentity(output, nodeName, messageChannels) {
+    if (currentAttemptContext() === null)
+        return output;
+    if (isCommand(output)) {
+        const command = output;
+        if (command.update !== null &&
+            typeof command.update === "object" &&
+            !Array.isArray(command.update)) {
+            assignStateUpdateIds(command.update, nodeName, ["update"], messageChannels);
+        }
+        // Tuple-pair and implicit-root updates remain LangGraph-owned shapes.
+        return command;
+    }
+    return assignStateUpdateIds(output, nodeName, [], messageChannels);
+}
+/** Stamp missing durable message IDs before named LangGraph reducers run. */
+export function installDurableMessageIdentity(graph) {
+    const source = graph;
+    const discovered = source.getSubgraphs?.call(graph, undefined, true);
+    const graphs = [
+        ...(discovered === undefined
+            ? []
+            : [...discovered].map(([, subgraph]) => subgraph)),
+        graph,
+    ];
+    for (const compiled of graphs) {
+        const { channels, nodes } = compiled;
+        if (nodes === undefined)
+            continue;
+        const messageChannels = new Set(Object.entries(channels ?? {})
+            .filter(([name, channel]) => name !== "__root__" && channel.operator === messagesStateReducer)
+            .map(([name]) => name));
+        if (messageChannels.size === 0)
+            continue;
+        for (const [nodeName, node] of Object.entries(nodes)) {
+            if (nodeName.startsWith("__") || installedNodes.has(node))
+                continue;
+            node.bound = node.bound.pipe(new RunnableLambda({
+                func: (output) => withDurableMessageIdentity(output, nodeName, messageChannels),
+            }));
+            installedNodes.add(node);
+        }
+    }
+}
+//# sourceMappingURL=durable_message_identity.js.map

package/dist/durable_message_identity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"durable_message_identity.js","sourceRoot":"","sources":["../src/durable_message_identity.ts"],"names":[],"mappings":"AAAA,yEAAyE;AAEzE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,OAAO,EACL,aAAa,EACb,0BAA0B,GAG3B,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,cAAc,EAAiB,MAAM,2BAA2B,CAAC;AAC1E,OAAO,EACL,SAAS,EACT,SAAS,EACT,oBAAoB,GAGrB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EACL,qBAAqB,EACrB,oBAAoB,EACpB,kBAAkB,GACnB,MAAM,wCAAwC,CAAC;AAEhD,MAAM,cAAc,GAAG,IAAI,OAAO,EAAU,CAAC;AAE7C,SAAS,gBAAgB,CACvB,QAAgB,EAChB,UAAwC;IAExC,MAAM,OAAO,GAAG,qBAAqB,EAAE,CAAC;IACxC,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;QACrB,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;IACzE,CAAC;IACD,MAAM,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,gBAA2C,CAAC;IAC5E,MAAM,aAAa,GAAG,oBAAoB,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC;QACrE,OAAO,CAAC,IAAI;QACZ,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE;KAC3B,CAAC,CAAC;IACH,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,aAAa,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC;IACvE,MAAM,YAAY,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACzE,OAAO,mBAAmB,WAAW,IAAI,kBAAkB,EAAE,IAAI,YAAY,EAAE,CAAC;AAClF,CAAC;AAED,SAAS,mBAAmB,CAC1B,OAAoB,EACpB,QAAgB,EAChB,IAAkC;IAElC,IACE,CAAC,OAAO,CAAC,EAAE,KAAK,IAAI,IAAI,OAAO,CAAC,EAAE,KAAK,SAAS,CAAC;QACjD,CAAC,aAAa,CAAC,UAAU,CAAC,OAAO,CAAC,EAClC,CAAC;QACD,MAAM,EAAE,GAAG,gBAAgB,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC5C,OAAO,CAAC,EAAE,GAAG,EAAE,CAAC;QAChB,OAAO,CAAC,SAAS,CAAC,EAAE,GAAG,EAAE,CAAC;IAC5B,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CACtB,KAAc,EACd,QAAgB,EAChB,IAAkC;IAElC,MAAM,OAAO,GAAG,0BAA0B,CAAC,KAAwB,CAAC,CAAC;IACrE,mBAAmB,CAAC,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC;IAC7C,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,sBAAsB,CAC7B,KAAc,EACd,QAAgB,EAChB,IAAkC;IAElC,IAAI,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,eAAe,IAAI,KAAK,EAAE,CAAC;QAC5D,MAAM,SAAS,GAAG,KAEjB,CAAC;QACF,MAAM,UAAU,GAAG,sBAAsB,CACvC,SAAS,CAAC,aAAa,EACvB,QAAQ,EACR,CAAC,GAAG,IAAI,EAAE,eAAe,CAAC,CAC3B,CAAC;QACF,OAAO,KAAK,YAAY,SAAS;YAC/B,CAAC,CAAC,IAAI,SAAS,CAAC,UAAU,CAAC;YAC3B,CAAC,CAAC,EAAE,GAAG,SAAS,EAAE,aAAa,EAAE,UAAU,EAAE,CAAC;IAClD,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAC/B,eAAe,CAAC,IAAI,EAAE,QAAQ,EAAE,CAAC,GAAG,IAAI,EAAE,KAAK,CAAC,CAAC,CAClD,CAAC;IACJ,CAAC;IACD,OAAO,eAAe,CAAC,KAAK,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC;AAChD,CAAC;AAED,SAAS,oBAAoB,CAC3B,KAAc,EACd,QAAgB,EAChB,IAAkC,EAClC,eAAoC;IAEpC,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IAE9D,MAAM,MAAM,GAAG,KAAgC,CAAC;IAChD,KAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CAAC;QACtC,IAAI,OAAO,IAAI,MAAM,EAAE,CAAC;YACtB,MAAM,CAAC,OAAO,CAAC,GAAG,sBAAsB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE;gBAClE,GAAG,IAAI;gBACP,OAAO;aACR,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,0BAA0B,CACjC,MAAe,EACf,QAAgB,EAChB,eAAoC;IAEpC,IAAI,qBAAqB,EAAE,KAAK,IAAI;QAAE,OAAO,MAAM,CAAC;IACpD,IAAI,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;QACtB,MAAM,OAAO,GAAG,MAAiB,CAAC;QAClC,IACE,OAAO,CAAC,MAAM,KAAK,IAAI;YACvB,OAAO,OAAO,CAAC,MAAM,KAAK,QAAQ;YAClC,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,EAC9B,CAAC;YACD,oBAAoB,CAClB,OAAO,CAAC,MAAM,EACd,QAAQ,EACR,CAAC,QAAQ,CAAC,EACV,eAAe,CAChB,CAAC;QACJ,CAAC;QACD,sEAAsE;QACtE,OAAO,OAAO,CAAC;IACjB,CAAC;IACD,OAAO,oBAAoB,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE,EAAE,eAAe,CAAC,CAAC;AACrE,CAAC;AAYD,6EAA6E;AAC7E,MAAM,UAAU,6BAA6B,CAC3C,KAA2C;IAE3C,MAAM,MAAM,GAAG,KAAkC,CAAC;IAClD,MAAM,UAAU,GAAG,MAAM,CAAC,YAAY,EAAE,IAAI,CAAC,KAAK,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;IACrE,MAAM,MAAM,GAAc;QACxB,GAAG,CAAC,UAAU,KAAK,SAAS;YAC1B,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,CAAC;QACpD,KAAK;KACN,CAAC;IACF,KAAK,MAAM,QAAQ,IAAI,MAAM,EAAE,CAAC;QAC9B,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,QAA0B,CAAC;QACvD,IAAI,KAAK,KAAK,SAAS;YAAE,SAAS;QAClC,MAAM,eAAe,GAAG,IAAI,GAAG,CAC7B,MAAM,CAAC,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC;aAC3B,MAAM,CACL,CAAC,CAAC,IAAI,EAAE,OAAO,CAAC,EAAE,EAAE,CAClB,IAAI,KAAK,UAAU,IAAI,OAAO,CAAC,QAAQ,KAAK,oBAAoB,CACnE;aACA,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CACzB,CAAC;QACF,IAAI,eAAe,CAAC,IAAI,KAAK,CAAC;YAAE,SAAS;QACzC,KAAK,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACrD,IAAI,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC;gBAAE,SAAS;YACpE,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAC1B,IAAI,cAAc,CAAC;gBACjB,IAAI,EAAE,CAAC,MAAe,EAAE,EAAE,CACxB,0BAA0B,CAAC,MAAM,EAAE,QAAQ,EAAE,eAAe,CAAC;aAChE,CAAC,CACH,CAAC;YACF,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;AACH,CAAC"}

package/dist/durable_session.d.ts

@@ -0,0 +1,30 @@
+/** OE-attempt-scoped LangGraph session behavior. */
+import type { BaseMessage } from "@langchain/core/messages";
+import type { RunnableConfig } from "@langchain/core/runnables";
+import { Command, type CompiledStateGraph, type Interrupt } from "@langchain/langgraph";
+import type { AgentInput, AgentOutput, StreamEvent, RequestContext } from "@mongodb-js/agent-engine-sdk";
+import { type AttemptContext } from "@mongodb-js/agent-engine-runner-shared";
+import { ExecutionSession, type PreparedRun, type SessionOptions } from "./execution_session.js";
+import type { PlatformCheckpointer } from "./platform_checkpointer.js";
+export interface DurableExecution {
+    readonly attempt: AttemptContext;
+    readonly checkpointer: PlatformCheckpointer;
+    readonly executionId: string;
+}
+export declare class DurableSession extends ExecutionSession {
+    readonly resumed: boolean;
+    private readonly attempt;
+    private readonly checkpointer;
+    private readonly executionId;
+    private readonly resumeActivityIds;
+    private resumeFrontierSettled;
+    constructor(graph: CompiledStateGraph<unknown, unknown>, ctx: RequestContext, input: AgentInput, options: SessionOptions, execution: DurableExecution);
+    prepareRun(): Promise<PreparedRun>;
+    invokeInterrupt(config: RunnableConfig, response: string, messages: readonly unknown[]): Promise<Command | AgentOutput | null>;
+    streamInterrupt(config: RunnableConfig, captured: readonly Interrupt[], messages: readonly BaseMessage[]): Promise<Command | StreamEvent | null>;
+    private settlePendingInterrupts;
+    complete(): Promise<void>;
+    close(): Promise<void>;
+    runInGraphScope<T>(fn: () => T): T;
+}
+//# sourceMappingURL=durable_session.d.ts.map

package/dist/durable_session.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"durable_session.d.ts","sourceRoot":"","sources":["../src/durable_session.ts"],"names":[],"mappings":"AAAA,oDAAoD;AAIpD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAChE,OAAO,EACL,OAAO,EAIP,KAAK,kBAAkB,EACvB,KAAK,SAAS,EACf,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,UAAU,EACV,WAAW,EACX,WAAW,EACX,cAAc,EACf,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAuBL,KAAK,cAAc,EAKpB,MAAM,wCAAwC,CAAC;AAEhD,OAAO,EACL,gBAAgB,EAChB,KAAK,WAAW,EAChB,KAAK,cAAc,EACpB,MAAM,wBAAwB,CAAC;AAQhC,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AA0LvE,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,CAAC,OAAO,EAAE,cAAc,CAAC;IACjC,QAAQ,CAAC,YAAY,EAAE,oBAAoB,CAAC;IAC5C,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;CAC9B;AAiGD,qBAAa,cAAe,SAAQ,gBAAgB;IAClD,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;IAC1B,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAiB;IACzC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAuB;IACpD,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAsB;IACxD,OAAO,CAAC,qBAAqB,CAAS;gBAGpC,KAAK,EAAE,kBAAkB,CAAC,OAAO,EAAE,OAAO,CAAC,EAC3C,GAAG,EAAE,cAAc,EACnB,KAAK,EAAE,UAAU,EACjB,OAAO,EAAE,cAAc,EACvB,SAAS,EAAE,gBAAgB;IAiCvB,UAAU,IAAI,OAAO,CAAC,WAAW,CAAC;IAiClC,eAAe,CACnB,MAAM,EAAE,cAAc,EACtB,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,SAAS,OAAO,EAAE,GAC3B,OAAO,CAAC,OAAO,GAAG,WAAW,GAAG,IAAI,CAAC;IAclC,eAAe,CACnB,MAAM,EAAE,cAAc,EACtB,QAAQ,EAAE,SAAS,SAAS,EAAE,EAC9B,QAAQ,EAAE,SAAS,WAAW,EAAE,GAC/B,OAAO,CAAC,OAAO,GAAG,WAAW,GAAG,IAAI,CAAC;YAW1B,uBAAuB;IA0JrC,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IASnB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAInB,eAAe,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC;CAK5C"}