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

package/dist/suspend.d.ts

@@ -0,0 +1,27 @@
+/** LangGraph interrupt validation and suspension output helpers. */
+import type { BaseMessage } from "@langchain/core/messages";
+import type { Interrupt } from "@langchain/langgraph";
+import { type AgentOutput, type StreamEvent } from "@mongodb-js/agent-engine-sdk";
+export interface PublicInterrupt {
+    readonly id: string;
+    readonly value: unknown;
+}
+export declare function validateInterruptValue(id: string, value: unknown): void;
+export declare function interruptSnapshot(captured: readonly Interrupt[]): PublicInterrupt[];
+export declare function resumeSchema(interruptIds: readonly string[]): Record<string, unknown>;
+export declare function invokeSuspendOutput(args: {
+    response: string;
+    threadId: string;
+    checkpointId: string | undefined;
+    interruptValues: readonly unknown[];
+    interrupts?: readonly PublicInterrupt[];
+    messageCount: number;
+    resumed: boolean;
+}): AgentOutput;
+export declare function streamHitlSuspendEvent(args: {
+    interrupts: readonly PublicInterrupt[];
+    checkpointId: string | undefined;
+    resumed: boolean;
+    messages: readonly BaseMessage[];
+}): StreamEvent;
+//# sourceMappingURL=suspend.d.ts.map

package/dist/suspend.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"suspend.d.ts","sourceRoot":"","sources":["../src/suspend.ts"],"names":[],"mappings":"AAAA,oEAAoE;AAEpE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAEL,KAAK,WAAW,EAChB,KAAK,WAAW,EACjB,MAAM,8BAA8B,CAAC;AAItC,MAAM,WAAW,eAAe;IAC9B,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC;CACzB;AAUD,wBAAgB,sBAAsB,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,GAAG,IAAI,CASvE;AAED,wBAAgB,iBAAiB,CAC/B,QAAQ,EAAE,SAAS,SAAS,EAAE,GAC7B,eAAe,EAAE,CAanB;AAED,wBAAgB,YAAY,CAC1B,YAAY,EAAE,SAAS,MAAM,EAAE,GAC9B,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAazB;AAED,wBAAgB,mBAAmB,CAAC,IAAI,EAAE;IACxC,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,MAAM,GAAG,SAAS,CAAC;IACjC,eAAe,EAAE,SAAS,OAAO,EAAE,CAAC;IACpC,UAAU,CAAC,EAAE,SAAS,eAAe,EAAE,CAAC;IACxC,YAAY,EAAE,MAAM,CAAC;IACrB,OAAO,EAAE,OAAO,CAAC;CAClB,GAAG,WAAW,CAsBd;AAED,wBAAgB,sBAAsB,CAAC,IAAI,EAAE;IAC3C,UAAU,EAAE,SAAS,eAAe,EAAE,CAAC;IACvC,YAAY,EAAE,MAAM,GAAG,SAAS,CAAC;IACjC,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,EAAE,SAAS,WAAW,EAAE,CAAC;CAClC,GAAG,WAAW,CAcd"}

package/dist/suspend.js

@@ -0,0 +1,87 @@
+/** LangGraph interrupt validation and suspension output helpers. */
+import { JsonValueSchema, } from "@mongodb-js/agent-engine-sdk";
+import { lcMessagesToPlatform } from "./messages.js";
+function interruptValueType(value) {
+    if (value === null)
+        return "null";
+    if (typeof value !== "object")
+        return typeof value;
+    const constructor = value
+        .constructor;
+    return typeof constructor?.name === "string" ? constructor.name : "object";
+}
+export function validateInterruptValue(id, value) {
+    try {
+        if (JsonValueSchema.safeParse(value).success)
+            return;
+    }
+    catch {
+        // Recursive values can overflow the recursive schema before returning a result.
+    }
+    throw new Error(`LangGraph interrupt "${id}" has a non-JSON-serializable value of type ${interruptValueType(value)}`);
+}
+export function interruptSnapshot(captured) {
+    const interrupts = captured.map((interrupt) => {
+        if (typeof interrupt.id !== "string" || interrupt.id.length === 0) {
+            throw new Error("LangGraph returned an empty interrupt id");
+        }
+        validateInterruptValue(interrupt.id, interrupt.value);
+        return { id: interrupt.id, value: interrupt.value };
+    });
+    const interruptIds = interrupts.map((interrupt) => interrupt.id);
+    if (new Set(interruptIds).size !== interruptIds.length) {
+        throw new Error("LangGraph returned duplicate interrupt ids");
+    }
+    return interrupts;
+}
+export function resumeSchema(interruptIds) {
+    const resumeMap = {
+        type: "object",
+        required: [...interruptIds],
+        properties: Object.fromEntries(interruptIds.map((id) => [id, {}])),
+        additionalProperties: false,
+    };
+    return {
+        type: "object",
+        required: ["resume_map"],
+        properties: { resume_map: resumeMap },
+        additionalProperties: true,
+    };
+}
+export function invokeSuspendOutput(args) {
+    return {
+        response: {
+            response: args.response,
+            execution_id: args.threadId,
+            status: "suspended",
+            suspend_context: {
+                checkpoint_id: args.checkpointId ?? null,
+                interrupt_values: args.interruptValues,
+            },
+            message_count: args.messageCount,
+            resumed: args.resumed,
+            ...(args.interrupts === undefined
+                ? {}
+                : {
+                    interrupts: args.interrupts,
+                    resume_schema: resumeSchema(args.interrupts.map((item) => item.id)),
+                }),
+        },
+    };
+}
+export function streamHitlSuspendEvent(args) {
+    const interruptIds = args.interrupts.map((interrupt) => interrupt.id);
+    return {
+        data: {
+            // Retained so older runners can still consume a new adapter event.
+            suspend_payload: args.interrupts[0]?.value,
+            interrupts: args.interrupts,
+            resume_schema: resumeSchema(interruptIds),
+            resumed: args.resumed,
+            messages: lcMessagesToPlatform(args.messages),
+            metadata: { checkpoint_id: args.checkpointId ?? null },
+        },
+        event: "suspend",
+    };
+}
+//# sourceMappingURL=suspend.js.map

package/dist/suspend.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"suspend.js","sourceRoot":"","sources":["../src/suspend.ts"],"names":[],"mappings":"AAAA,oEAAoE;AAIpE,OAAO,EACL,eAAe,GAGhB,MAAM,8BAA8B,CAAC;AAEtC,OAAO,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAOrD,SAAS,kBAAkB,CAAC,KAAc;IACxC,IAAI,KAAK,KAAK,IAAI;QAAE,OAAO,MAAM,CAAC;IAClC,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,OAAO,KAAK,CAAC;IACnD,MAAM,WAAW,GAAI,KAA8C;SAChE,WAAW,CAAC;IACf,OAAO,OAAO,WAAW,EAAE,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC;AAC7E,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,EAAU,EAAE,KAAc;IAC/D,IAAI,CAAC;QACH,IAAI,eAAe,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,OAAO;YAAE,OAAO;IACvD,CAAC;IAAC,MAAM,CAAC;QACP,gFAAgF;IAClF,CAAC;IACD,MAAM,IAAI,KAAK,CACb,wBAAwB,EAAE,+CAA+C,kBAAkB,CAAC,KAAK,CAAC,EAAE,CACrG,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,QAA8B;IAE9B,MAAM,UAAU,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE;QAC5C,IAAI,OAAO,SAAS,CAAC,EAAE,KAAK,QAAQ,IAAI,SAAS,CAAC,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClE,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC9D,CAAC;QACD,sBAAsB,CAAC,SAAS,CAAC,EAAE,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC;QACtD,OAAO,EAAE,EAAE,EAAE,SAAS,CAAC,EAAE,EAAE,KAAK,EAAE,SAAS,CAAC,KAAK,EAAE,CAAC;IACtD,CAAC,CAAC,CAAC;IACH,MAAM,YAAY,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IACjE,IAAI,IAAI,GAAG,CAAC,YAAY,CAAC,CAAC,IAAI,KAAK,YAAY,CAAC,MAAM,EAAE,CAAC;QACvD,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;IAChE,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,MAAM,UAAU,YAAY,CAC1B,YAA+B;IAE/B,MAAM,SAAS,GAAG;QAChB,IAAI,EAAE,QAAQ;QACd,QAAQ,EAAE,CAAC,GAAG,YAAY,CAAC;QAC3B,UAAU,EAAE,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QAClE,oBAAoB,EAAE,KAAK;KAC5B,CAAC;IACF,OAAO;QACL,IAAI,EAAE,QAAQ;QACd,QAAQ,EAAE,CAAC,YAAY,CAAC;QACxB,UAAU,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE;QACrC,oBAAoB,EAAE,IAAI;KAC3B,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,IAQnC;IACC,OAAO;QACL,QAAQ,EAAE;YACR,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,YAAY,EAAE,IAAI,CAAC,QAAQ;YAC3B,MAAM,EAAE,WAAW;YACnB,eAAe,EAAE;gBACf,aAAa,EAAE,IAAI,CAAC,YAAY,IAAI,IAAI;gBACxC,gBAAgB,EAAE,IAAI,CAAC,eAAwB;aAChD;YACD,aAAa,EAAE,IAAI,CAAC,YAAY;YAChC,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,GAAG,CAAC,IAAI,CAAC,UAAU,KAAK,SAAS;gBAC/B,CAAC,CAAC,EAAE;gBACJ,CAAC,CAAC;oBACE,UAAU,EAAE,IAAI,CAAC,UAAmB;oBACpC,aAAa,EAAE,YAAY,CACzB,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAC9B;iBACX,CAAC;SACP;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,IAKtC;IACC,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IACtE,OAAO;QACL,IAAI,EAAE;YACJ,mEAAmE;YACnE,eAAe,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,KAAc;YACnD,UAAU,EAAE,IAAI,CAAC,UAAmB;YACpC,aAAa,EAAE,YAAY,CAAC,YAAY,CAAU;YAClD,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,QAAQ,EAAE,oBAAoB,CAAC,IAAI,CAAC,QAAQ,CAAU;YACtD,QAAQ,EAAE,EAAE,aAAa,EAAE,IAAI,CAAC,YAAY,IAAI,IAAI,EAAE;SACvD;QACD,KAAK,EAAE,SAAS;KACjB,CAAC;AACJ,CAAC"}

package/dist/thread_id.d.ts

@@ -0,0 +1,30 @@
+/**
+ * Workspace-scoped LangGraph `thread_id` composition.
+ *
+ * Port of Python's `agent_engine_sdk_langgraph/thread_id.py`.
+ */
+import type { RequestContext } from "@mongodb-js/agent-engine-sdk";
+/** Tenant hook mapping a `RequestContext` to a checkpoint `thread_id`. */
+export type ResolveThreadIdHook = (ctx: RequestContext) => string;
+/** Compose the workspace-scoped `thread_id` for a checkpoint key. */
+export declare function scopedThreadId(sessionId?: string | null, workspaceId?: string | null): string;
+/**
+ * Return the `thread_id` keys that may hold checkpoints for a session.
+ *
+ * When a workspace scope is known, only the composite key is queried —
+ * never the bare `session_id`. Bare keys are shared by every workspace on
+ * the same store, so including them would let one tenant read or poison
+ * another's conversation history. Legacy unscoped checkpoints are
+ * intentionally not queried.
+ */
+export declare function threadIdsForQuery(sessionId: string, workspaceId?: string | null): string[];
+/** Flatten `threadIdsForQuery` across many sessions, deduped. */
+export declare function threadIdsForSessionsQuery(sessionIds: string[], workspaceId?: string | null): string[];
+/** Recover the plain `session_id` from a workspace-scoped `thread_id`. */
+export declare function sessionIdFromThreadId(threadId: string, workspaceId?: string | null): string;
+/**
+ * LangGraph `thread_id` for a platform request — the same key invoke uses.
+ * Port of Python's `checkpoint_thread_id`.
+ */
+export declare function checkpointThreadId(ctx: RequestContext, resolveThreadId?: ResolveThreadIdHook | null): string;
+//# sourceMappingURL=thread_id.d.ts.map

package/dist/thread_id.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"thread_id.d.ts","sourceRoot":"","sources":["../src/thread_id.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAEnE,0EAA0E;AAC1E,MAAM,MAAM,mBAAmB,GAAG,CAAC,GAAG,EAAE,cAAc,KAAK,MAAM,CAAC;AAIlE,qEAAqE;AACrE,wBAAgB,cAAc,CAC5B,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,EACzB,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,GAC1B,MAAM,CAIR;AAED;;;;;;;;GAQG;AACH,wBAAgB,iBAAiB,CAC/B,SAAS,EAAE,MAAM,EACjB,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,GAC1B,MAAM,EAAE,CAGV;AAED,iEAAiE;AACjE,wBAAgB,yBAAyB,CACvC,UAAU,EAAE,MAAM,EAAE,EACpB,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,GAC1B,MAAM,EAAE,CAYV;AAED,0EAA0E;AAC1E,wBAAgB,qBAAqB,CACnC,QAAQ,EAAE,MAAM,EAChB,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,GAC1B,MAAM,CAOR;AAED;;;GAGG;AACH,wBAAgB,kBAAkB,CAChC,GAAG,EAAE,cAAc,EACnB,eAAe,CAAC,EAAE,mBAAmB,GAAG,IAAI,GAC3C,MAAM,CAeR"}

package/dist/thread_id.js

@@ -0,0 +1,69 @@
+/**
+ * Workspace-scoped LangGraph `thread_id` composition.
+ *
+ * Port of Python's `agent_engine_sdk_langgraph/thread_id.py`.
+ */
+const DEFAULT_SESSION_ID = "default";
+/** Compose the workspace-scoped `thread_id` for a checkpoint key. */
+export function scopedThreadId(sessionId, workspaceId) {
+    const sid = sessionId || DEFAULT_SESSION_ID;
+    if (!workspaceId)
+        return sid;
+    return `${sid}:${workspaceId}`;
+}
+/**
+ * Return the `thread_id` keys that may hold checkpoints for a session.
+ *
+ * When a workspace scope is known, only the composite key is queried —
+ * never the bare `session_id`. Bare keys are shared by every workspace on
+ * the same store, so including them would let one tenant read or poison
+ * another's conversation history. Legacy unscoped checkpoints are
+ * intentionally not queried.
+ */
+export function threadIdsForQuery(sessionId, workspaceId) {
+    if (!workspaceId)
+        return [sessionId];
+    return [scopedThreadId(sessionId, workspaceId)];
+}
+/** Flatten `threadIdsForQuery` across many sessions, deduped. */
+export function threadIdsForSessionsQuery(sessionIds, workspaceId) {
+    const seen = new Set();
+    const result = [];
+    for (const sessionId of sessionIds) {
+        for (const threadId of threadIdsForQuery(sessionId, workspaceId)) {
+            if (!seen.has(threadId)) {
+                seen.add(threadId);
+                result.push(threadId);
+            }
+        }
+    }
+    return result;
+}
+/** Recover the plain `session_id` from a workspace-scoped `thread_id`. */
+export function sessionIdFromThreadId(threadId, workspaceId) {
+    if (!workspaceId)
+        return threadId;
+    const suffix = `:${workspaceId}`;
+    if (threadId.endsWith(suffix)) {
+        return threadId.slice(0, -suffix.length);
+    }
+    return threadId;
+}
+/**
+ * LangGraph `thread_id` for a platform request — the same key invoke uses.
+ * Port of Python's `checkpoint_thread_id`.
+ */
+export function checkpointThreadId(ctx, resolveThreadId) {
+    if (resolveThreadId != null) {
+        const threadId = resolveThreadId(ctx);
+        if (typeof threadId !== "string" || threadId.trim() === "") {
+            const rendered = typeof threadId === "string"
+                ? JSON.stringify(threadId)
+                : typeof threadId;
+            throw new Error(`App.resolveThreadId must return a non-empty string; got ${rendered}`);
+        }
+        return threadId;
+    }
+    return scopedThreadId(ctx.sessionId, ctx.workspaceId);
+}
+//# sourceMappingURL=thread_id.js.map

package/dist/thread_id.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"thread_id.js","sourceRoot":"","sources":["../src/thread_id.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAOH,MAAM,kBAAkB,GAAG,SAAS,CAAC;AAErC,qEAAqE;AACrE,MAAM,UAAU,cAAc,CAC5B,SAAyB,EACzB,WAA2B;IAE3B,MAAM,GAAG,GAAG,SAAS,IAAI,kBAAkB,CAAC;IAC5C,IAAI,CAAC,WAAW;QAAE,OAAO,GAAG,CAAC;IAC7B,OAAO,GAAG,GAAG,IAAI,WAAW,EAAE,CAAC;AACjC,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,iBAAiB,CAC/B,SAAiB,EACjB,WAA2B;IAE3B,IAAI,CAAC,WAAW;QAAE,OAAO,CAAC,SAAS,CAAC,CAAC;IACrC,OAAO,CAAC,cAAc,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC,CAAC;AAClD,CAAC;AAED,iEAAiE;AACjE,MAAM,UAAU,yBAAyB,CACvC,UAAoB,EACpB,WAA2B;IAE3B,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;QACnC,KAAK,MAAM,QAAQ,IAAI,iBAAiB,CAAC,SAAS,EAAE,WAAW,CAAC,EAAE,CAAC;YACjE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACxB,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;gBACnB,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,0EAA0E;AAC1E,MAAM,UAAU,qBAAqB,CACnC,QAAgB,EAChB,WAA2B;IAE3B,IAAI,CAAC,WAAW;QAAE,OAAO,QAAQ,CAAC;IAClC,MAAM,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;IACjC,IAAI,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9B,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAC3C,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,kBAAkB,CAChC,GAAmB,EACnB,eAA4C;IAE5C,IAAI,eAAe,IAAI,IAAI,EAAE,CAAC;QAC5B,MAAM,QAAQ,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC;QACtC,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,QAAQ,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YAC3D,MAAM,QAAQ,GACZ,OAAO,QAAQ,KAAK,QAAQ;gBAC1B,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC;gBAC1B,CAAC,CAAC,OAAO,QAAQ,CAAC;YACtB,MAAM,IAAI,KAAK,CACb,2DAA2D,QAAQ,EAAE,CACtE,CAAC;QACJ,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,OAAO,cAAc,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,WAAW,CAAC,CAAC;AACxD,CAAC"}

package/dist/workflow_json.d.ts

@@ -0,0 +1,11 @@
+/** JSON and protobuf boundary helpers for durable workflow state. */
+import { type JsonObject } from "@bufbuild/protobuf";
+import { type Value } from "@bufbuild/protobuf/wkt";
+import { type JsonValue } from "@mongodb-js/agent-engine-sdk";
+export declare function serializeJsonValue(value: unknown, path: string): JsonValue;
+export declare function serializeJsonObject(value: unknown, path: string): JsonObject;
+export declare function serializeJsonObjectArray(value: unknown, path: string): JsonObject[];
+export declare function requireJsonObject(value: unknown, path: string): JsonObject;
+export declare function protoValueFromUnknown(value: unknown, path: string): Value;
+export declare function jsonValueFromProto(value: Value | undefined): JsonValue;
+//# sourceMappingURL=workflow_json.d.ts.map

package/dist/workflow_json.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow_json.d.ts","sourceRoot":"","sources":["../src/workflow_json.ts"],"names":[],"mappings":"AAAA,qEAAqE;AAErE,OAAO,EAAoB,KAAK,UAAU,EAAE,MAAM,oBAAoB,CAAC;AACvE,OAAO,EAAe,KAAK,KAAK,EAAE,MAAM,wBAAwB,CAAC;AACjE,OAAO,EAAmB,KAAK,SAAS,EAAE,MAAM,8BAA8B,CAAC;AAE/E,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,SAAS,CAW1E;AAED,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,UAAU,CAU5E;AAED,wBAAgB,wBAAwB,CACtC,KAAK,EAAE,OAAO,EACd,IAAI,EAAE,MAAM,GACX,UAAU,EAAE,CAOd;AAED,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,UAAU,CAW1E;AAED,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,KAAK,CAEzE;AAED,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,KAAK,GAAG,SAAS,GAAG,SAAS,CAEtE"}

package/dist/workflow_json.js

@@ -0,0 +1,49 @@
+/** JSON and protobuf boundary helpers for durable workflow state. */
+import { fromJson, toJson } from "@bufbuild/protobuf";
+import { ValueSchema } from "@bufbuild/protobuf/wkt";
+import { JsonValueSchema } from "@mongodb-js/agent-engine-sdk";
+export function serializeJsonValue(value, path) {
+    let parsed;
+    try {
+        parsed = JsonValueSchema.safeParse(value);
+    }
+    catch (error) {
+        throw new TypeError(`${path} is not JSON-serializable`, { cause: error });
+    }
+    if (!parsed.success) {
+        throw new TypeError(`${path} is not JSON-serializable`);
+    }
+    return parsed.data;
+}
+export function serializeJsonObject(value, path) {
+    const serialized = serializeJsonValue(value, path);
+    if (serialized === null ||
+        Array.isArray(serialized) ||
+        typeof serialized !== "object") {
+        throw new TypeError(`${path} must be an object`);
+    }
+    return serialized;
+}
+export function serializeJsonObjectArray(value, path) {
+    if (!Array.isArray(value)) {
+        throw new TypeError(`${path} must be an array`);
+    }
+    return value.map((item, index) => serializeJsonObject(item, `${path}[${index}]`));
+}
+export function requireJsonObject(value, path) {
+    const parsed = JsonValueSchema.safeParse(value);
+    if (!parsed.success ||
+        parsed.data === null ||
+        Array.isArray(parsed.data) ||
+        typeof parsed.data !== "object") {
+        throw new TypeError(`${path} must be a JSON object`);
+    }
+    return parsed.data;
+}
+export function protoValueFromUnknown(value, path) {
+    return fromJson(ValueSchema, serializeJsonValue(value, path));
+}
+export function jsonValueFromProto(value) {
+    return value === undefined ? null : toJson(ValueSchema, value);
+}
+//# sourceMappingURL=workflow_json.js.map

package/dist/workflow_json.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow_json.js","sourceRoot":"","sources":["../src/workflow_json.ts"],"names":[],"mappings":"AAAA,qEAAqE;AAErE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAmB,MAAM,oBAAoB,CAAC;AACvE,OAAO,EAAE,WAAW,EAAc,MAAM,wBAAwB,CAAC;AACjE,OAAO,EAAE,eAAe,EAAkB,MAAM,8BAA8B,CAAC;AAE/E,MAAM,UAAU,kBAAkB,CAAC,KAAc,EAAE,IAAY;IAC7D,IAAI,MAAM,CAAC;IACX,IAAI,CAAC;QACH,MAAM,GAAG,eAAe,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IAC5C,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,SAAS,CAAC,GAAG,IAAI,2BAA2B,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;IAC5E,CAAC;IACD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACpB,MAAM,IAAI,SAAS,CAAC,GAAG,IAAI,2BAA2B,CAAC,CAAC;IAC1D,CAAC;IACD,OAAO,MAAM,CAAC,IAAI,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,KAAc,EAAE,IAAY;IAC9D,MAAM,UAAU,GAAG,kBAAkB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACnD,IACE,UAAU,KAAK,IAAI;QACnB,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC;QACzB,OAAO,UAAU,KAAK,QAAQ,EAC9B,CAAC;QACD,MAAM,IAAI,SAAS,CAAC,GAAG,IAAI,oBAAoB,CAAC,CAAC;IACnD,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,MAAM,UAAU,wBAAwB,CACtC,KAAc,EACd,IAAY;IAEZ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAC1B,MAAM,IAAI,SAAS,CAAC,GAAG,IAAI,mBAAmB,CAAC,CAAC;IAClD,CAAC;IACD,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAC/B,mBAAmB,CAAC,IAAI,EAAE,GAAG,IAAI,IAAI,KAAK,GAAG,CAAC,CAC/C,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,KAAc,EAAE,IAAY;IAC5D,MAAM,MAAM,GAAG,eAAe,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IAChD,IACE,CAAC,MAAM,CAAC,OAAO;QACf,MAAM,CAAC,IAAI,KAAK,IAAI;QACpB,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC;QAC1B,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,EAC/B,CAAC;QACD,MAAM,IAAI,SAAS,CAAC,GAAG,IAAI,wBAAwB,CAAC,CAAC;IACvD,CAAC;IACD,OAAO,MAAM,CAAC,IAAI,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,KAAc,EAAE,IAAY;IAChE,OAAO,QAAQ,CAAC,WAAW,EAAE,kBAAkB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC;AAChE,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,KAAwB;IACzD,OAAO,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAE,MAAM,CAAC,WAAW,EAAE,KAAK,CAAe,CAAC;AAChF,CAAC"}

package/dist/workflow_message.d.ts

@@ -0,0 +1,11 @@
+/** LangChain message translation for the OE durable workflow contract. */
+import { type BaseMessage } from "@langchain/core/messages";
+import { type WorkflowMessage } from "@mongodb-js/agent-engine-runner-shared";
+/** Owns LangChain-specific serialization and validation at the workflow edge. */
+export declare class WorkflowMessageCodec {
+    static fromLangChainMessage(message: BaseMessage, includeLegacySource?: boolean): WorkflowMessage;
+    static toLangChainMessage(message: WorkflowMessage, path?: string): BaseMessage;
+    private static canonicalSourceMessage;
+    private static fromSourceMessage;
+}
+//# sourceMappingURL=workflow_message.d.ts.map

package/dist/workflow_message.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow_message.d.ts","sourceRoot":"","sources":["../src/workflow_message.ts"],"names":[],"mappings":"AAAA,0EAA0E;AAG1E,OAAO,EAEL,KAAK,WAAW,EAOjB,MAAM,0BAA0B,CAAC;AAElC,OAAO,EAGL,KAAK,eAAe,EACrB,MAAM,wCAAwC,CAAC;AA6ChD,iFAAiF;AACjF,qBAAa,oBAAoB;IAC/B,MAAM,CAAC,oBAAoB,CACzB,OAAO,EAAE,WAAW,EACpB,mBAAmB,UAAQ,GAC1B,eAAe;IAgFlB,MAAM,CAAC,kBAAkB,CACvB,OAAO,EAAE,eAAe,EACxB,IAAI,SAAY,GACf,WAAW;IA8Dd,OAAO,CAAC,MAAM,CAAC,sBAAsB;IA2BrC,OAAO,CAAC,MAAM,CAAC,iBAAiB;CA+CjC"}

package/dist/workflow_message.js

@@ -0,0 +1,179 @@
+/** LangChain message translation for the OE durable workflow contract. */
+import { create } from "@bufbuild/protobuf";
+import { AIMessage, HumanMessage, isBaseMessage, mapStoredMessageToChatMessage, SystemMessage, ToolMessage, } from "@langchain/core/messages";
+import { MessageRole, WorkflowMessageSchema, } from "@mongodb-js/agent-engine-runner-shared";
+import { lcToPlatformMessage } from "./messages.js";
+import { jsonValueFromProto, protoValueFromUnknown, serializeJsonObject, serializeJsonObjectArray, } from "./workflow_json.js";
+const ROLE_TO_WORKFLOW = {
+    user: MessageRole.USER,
+    assistant: MessageRole.ASSISTANT,
+    tool: MessageRole.TOOL,
+    system: MessageRole.SYSTEM,
+};
+const ROLE_FROM_WORKFLOW = {
+    [MessageRole.USER]: "user",
+    [MessageRole.ASSISTANT]: "assistant",
+    [MessageRole.TOOL]: "tool",
+    [MessageRole.SYSTEM]: "system",
+};
+function definedToolCallFields(toolCall) {
+    return Object.fromEntries(Object.entries(toolCall).filter(([, value]) => value !== undefined));
+}
+function durableContent(message, platformContent) {
+    // LangGraph can transiently place a direct ToolMessage result inside another
+    // message's content tuple. That framework-only object is not JSON state, so
+    // keep the adapter's established platform normalization for this one shape.
+    if (Array.isArray(message.content) && message.content.some(isBaseMessage)) {
+        return platformContent;
+    }
+    return message.content;
+}
+/** Owns LangChain-specific serialization and validation at the workflow edge. */
+export class WorkflowMessageCodec {
+    static fromLangChainMessage(message, includeLegacySource = false) {
+        const platformMessage = lcToPlatformMessage(message);
+        const artifactMetadata = message.additional_kwargs["artifacts"];
+        const rawPlatformArtifacts = artifactMetadata === undefined ? [] : artifactMetadata;
+        const rawInvalidToolCalls = message.type === "ai"
+            ? (message.invalid_tool_calls ?? [])
+            : [];
+        const additionalKwargs = { ...platformMessage.additionalKwargs };
+        delete additionalKwargs["artifacts"];
+        delete additionalKwargs["tool_calls"];
+        return create(WorkflowMessageSchema, {
+            role: ROLE_TO_WORKFLOW[platformMessage.role],
+            content: protoValueFromUnknown(durableContent(message, platformMessage.content), "message content"),
+            toolCalls: serializeJsonObjectArray((platformMessage.toolCalls ?? []).map((toolCall) => definedToolCallFields(toolCall)), "message tool_calls"),
+            invalidToolCalls: serializeJsonObjectArray(rawInvalidToolCalls.map((toolCall) => definedToolCallFields(toolCall)), "message invalid_tool_calls"),
+            ...(Object.keys(additionalKwargs).length > 0 && {
+                additionalKwargs: serializeJsonObject(additionalKwargs, "message additional_kwargs"),
+            }),
+            ...(platformMessage.responseMetadata !== undefined && {
+                responseMetadata: serializeJsonObject(platformMessage.responseMetadata, "message response_metadata"),
+            }),
+            platformArtifacts: serializeJsonObjectArray(rawPlatformArtifacts, "message additional_kwargs.artifacts"),
+            ...(includeLegacySource && {
+                sourceMessage: protoValueFromUnknown(this.canonicalSourceMessage(message, platformMessage), "message source"),
+                artifacts: serializeJsonObjectArray(rawPlatformArtifacts, "message artifacts"),
+            }),
+            ...(platformMessage.toolCallId !== undefined && {
+                toolCallId: platformMessage.toolCallId,
+            }),
+            ...(platformMessage.name !== undefined && {
+                name: platformMessage.name,
+            }),
+            ...(platformMessage.id !== undefined && { id: platformMessage.id }),
+            ...(platformMessage.isError !== undefined && {
+                isError: platformMessage.isError,
+            }),
+            // Checkpoint serializers may materialize an omitted artifact as null;
+            // both shapes mean that no machine-readable tool result exists.
+            ...(message instanceof ToolMessage &&
+                message.artifact !== undefined &&
+                message.artifact !== null && {
+                toolArtifact: protoValueFromUnknown(message.artifact, "message tool_artifact"),
+            }),
+        });
+    }
+    static toLangChainMessage(message, path = "message") {
+        const role = ROLE_FROM_WORKFLOW[message.role];
+        if (role === undefined) {
+            throw new TypeError(`${path}: previous workflow message has an unspecified role`);
+        }
+        if (message.sourceMessage !== undefined) {
+            return this.fromSourceMessage(message, role, path);
+        }
+        const content = jsonValueFromProto(message.content);
+        if (typeof content !== "string" && !Array.isArray(content)) {
+            throw new TypeError(`${path}: previous workflow message content is malformed`);
+        }
+        const additionalKwargs = {
+            ...message.additionalKwargs,
+            ...(message.platformArtifacts.length > 0 && {
+                artifacts: message.platformArtifacts,
+            }),
+        };
+        const baseFields = {
+            content: content,
+            additional_kwargs: additionalKwargs,
+            response_metadata: message.responseMetadata ?? {},
+            ...(message.name !== undefined && { name: message.name }),
+            ...(message.id !== undefined && { id: message.id }),
+        };
+        switch (message.role) {
+            case MessageRole.USER:
+                return new HumanMessage(baseFields);
+            case MessageRole.ASSISTANT:
+                return new AIMessage({
+                    ...baseFields,
+                    tool_calls: message.toolCalls,
+                    invalid_tool_calls: message.invalidToolCalls,
+                });
+            case MessageRole.TOOL:
+                if (message.toolCallId === undefined || message.toolCallId === "") {
+                    throw new TypeError(`${path}: previous workflow message cannot be reconstructed`);
+                }
+                return new ToolMessage({
+                    ...baseFields,
+                    tool_call_id: message.toolCallId,
+                    status: message.isError === true ? "error" : "success",
+                    ...(message.toolArtifact !== undefined && {
+                        artifact: jsonValueFromProto(message.toolArtifact),
+                    }),
+                });
+            case MessageRole.SYSTEM:
+                return new SystemMessage(baseFields);
+            default:
+                throw new TypeError(`${path}: previous workflow message has an unspecified role`);
+        }
+    }
+    static canonicalSourceMessage(message, platformMessage) {
+        const storedMessage = message.toDict();
+        if (storedMessage.type !== "ai") {
+            return JSON.parse(JSON.stringify(storedMessage));
+        }
+        const data = { ...storedMessage.data };
+        const additionalKwargs = data["additional_kwargs"] !== null &&
+            typeof data["additional_kwargs"] === "object" &&
+            !Array.isArray(data["additional_kwargs"])
+            ? { ...data["additional_kwargs"] }
+            : {};
+        delete additionalKwargs["tool_calls"];
+        delete data["tool_call_chunks"];
+        delete data["usage_metadata"];
+        data["additional_kwargs"] = additionalKwargs;
+        data["tool_calls"] = (platformMessage.toolCalls ?? []).map((toolCall) => toolCall.toLangchainDict());
+        storedMessage.data = data;
+        return JSON.parse(JSON.stringify(storedMessage));
+    }
+    static fromSourceMessage(message, role, path) {
+        const source = jsonValueFromProto(message.sourceMessage);
+        if (source === null ||
+            Array.isArray(source) ||
+            typeof source !== "object") {
+            throw new TypeError(`${path}: previous workflow message is malformed`);
+        }
+        try {
+            const converted = mapStoredMessageToChatMessage(source);
+            const restored = lcToPlatformMessage(converted);
+            if (restored.role !== role) {
+                throw new TypeError(`${path}: previous workflow message role conflicts with source_message`);
+            }
+            if (message.id !== undefined && restored.id !== message.id) {
+                throw new TypeError(`${path}: previous workflow message id conflicts with source_message`);
+            }
+            if (message.toolCallId !== undefined &&
+                restored.toolCallId !== message.toolCallId) {
+                throw new TypeError(`${path}: previous workflow tool_call_id conflicts with source_message`);
+            }
+            return converted;
+        }
+        catch (error) {
+            if (error instanceof TypeError && error.message.startsWith(`${path}:`)) {
+                throw error;
+            }
+            throw new TypeError(`${path}: previous workflow message cannot be reconstructed`, { cause: error });
+        }
+    }
+}
+//# sourceMappingURL=workflow_message.js.map

package/dist/workflow_message.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow_message.js","sourceRoot":"","sources":["../src/workflow_message.ts"],"names":[],"mappings":"AAAA,0EAA0E;AAE1E,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EACL,SAAS,EAET,YAAY,EACZ,aAAa,EACb,6BAA6B,EAE7B,aAAa,EACb,WAAW,GACZ,MAAM,0BAA0B,CAAC;AAElC,OAAO,EACL,WAAW,EACX,qBAAqB,GAEtB,MAAM,wCAAwC,CAAC;AAEhD,OAAO,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AACpD,OAAO,EACL,kBAAkB,EAClB,qBAAqB,EACrB,mBAAmB,EACnB,wBAAwB,GACzB,MAAM,oBAAoB,CAAC;AAE5B,MAAM,gBAAgB,GAAwC;IAC5D,IAAI,EAAE,WAAW,CAAC,IAAI;IACtB,SAAS,EAAE,WAAW,CAAC,SAAS;IAChC,IAAI,EAAE,WAAW,CAAC,IAAI;IACtB,MAAM,EAAE,WAAW,CAAC,MAAM;CAC3B,CAAC;AAEF,MAAM,kBAAkB,GAAiD;IACvE,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,MAAM;IAC1B,CAAC,WAAW,CAAC,SAAS,CAAC,EAAE,WAAW;IACpC,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,MAAM;IAC1B,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,QAAQ;CAC/B,CAAC;AAEF,SAAS,qBAAqB,CAC5B,QAAiC;IAEjC,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,KAAK,SAAS,CAAC,CACpE,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CACrB,OAAoB,EACpB,eAAwB;IAExB,6EAA6E;IAC7E,4EAA4E;IAC5E,4EAA4E;IAC5E,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC;QAC1E,OAAO,eAAe,CAAC;IACzB,CAAC;IACD,OAAO,OAAO,CAAC,OAAO,CAAC;AACzB,CAAC;AAED,iFAAiF;AACjF,MAAM,OAAO,oBAAoB;IAC/B,MAAM,CAAC,oBAAoB,CACzB,OAAoB,EACpB,mBAAmB,GAAG,KAAK;QAE3B,MAAM,eAAe,GAAG,mBAAmB,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,gBAAgB,GAAG,OAAO,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;QAChE,MAAM,oBAAoB,GACxB,gBAAgB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,gBAAgB,CAAC;QACzD,MAAM,mBAAmB,GACvB,OAAO,CAAC,IAAI,KAAK,IAAI;YACnB,CAAC,CAAC,CAAE,OAAqB,CAAC,kBAAkB,IAAI,EAAE,CAAC;YACnD,CAAC,CAAC,EAAE,CAAC;QACT,MAAM,gBAAgB,GAAG,EAAE,GAAG,eAAe,CAAC,gBAAgB,EAAE,CAAC;QACjE,OAAO,gBAAgB,CAAC,WAAW,CAAC,CAAC;QACrC,OAAO,gBAAgB,CAAC,YAAY,CAAC,CAAC;QAEtC,OAAO,MAAM,CAAC,qBAAqB,EAAE;YACnC,IAAI,EAAE,gBAAgB,CAAC,eAAe,CAAC,IAAI,CAAC;YAC5C,OAAO,EAAE,qBAAqB,CAC5B,cAAc,CAAC,OAAO,EAAE,eAAe,CAAC,OAAO,CAAC,EAChD,iBAAiB,CAClB;YACD,SAAS,EAAE,wBAAwB,CACjC,CAAC,eAAe,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CACjD,qBAAqB,CAAC,QAA8C,CAAC,CACtE,EACD,oBAAoB,CACrB;YACD,gBAAgB,EAAE,wBAAwB,CACxC,mBAAmB,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CACnC,qBAAqB,CAAC,QAA8C,CAAC,CACtE,EACD,4BAA4B,CAC7B;YACD,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,MAAM,GAAG,CAAC,IAAI;gBAC9C,gBAAgB,EAAE,mBAAmB,CACnC,gBAAgB,EAChB,2BAA2B,CAC5B;aACF,CAAC;YACF,GAAG,CAAC,eAAe,CAAC,gBAAgB,KAAK,SAAS,IAAI;gBACpD,gBAAgB,EAAE,mBAAmB,CACnC,eAAe,CAAC,gBAAgB,EAChC,2BAA2B,CAC5B;aACF,CAAC;YACF,iBAAiB,EAAE,wBAAwB,CACzC,oBAAoB,EACpB,qCAAqC,CACtC;YACD,GAAG,CAAC,mBAAmB,IAAI;gBACzB,aAAa,EAAE,qBAAqB,CAClC,IAAI,CAAC,sBAAsB,CAAC,OAAO,EAAE,eAAe,CAAC,EACrD,gBAAgB,CACjB;gBACD,SAAS,EAAE,wBAAwB,CACjC,oBAAoB,EACpB,mBAAmB,CACpB;aACF,CAAC;YACF,GAAG,CAAC,eAAe,CAAC,UAAU,KAAK,SAAS,IAAI;gBAC9C,UAAU,EAAE,eAAe,CAAC,UAAU;aACvC,CAAC;YACF,GAAG,CAAC,eAAe,CAAC,IAAI,KAAK,SAAS,IAAI;gBACxC,IAAI,EAAE,eAAe,CAAC,IAAI;aAC3B,CAAC;YACF,GAAG,CAAC,eAAe,CAAC,EAAE,KAAK,SAAS,IAAI,EAAE,EAAE,EAAE,eAAe,CAAC,EAAE,EAAE,CAAC;YACnE,GAAG,CAAC,eAAe,CAAC,OAAO,KAAK,SAAS,IAAI;gBAC3C,OAAO,EAAE,eAAe,CAAC,OAAO;aACjC,CAAC;YACF,sEAAsE;YACtE,gEAAgE;YAChE,GAAG,CAAC,OAAO,YAAY,WAAW;gBAChC,OAAO,CAAC,QAAQ,KAAK,SAAS;gBAC9B,OAAO,CAAC,QAAQ,KAAK,IAAI,IAAI;gBAC3B,YAAY,EAAE,qBAAqB,CACjC,OAAO,CAAC,QAAQ,EAChB,uBAAuB,CACxB;aACF,CAAC;SACL,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,kBAAkB,CACvB,OAAwB,EACxB,IAAI,GAAG,SAAS;QAEhB,MAAM,IAAI,GAAG,kBAAkB,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC9C,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,MAAM,IAAI,SAAS,CACjB,GAAG,IAAI,qDAAqD,CAC7D,CAAC;QACJ,CAAC;QACD,IAAI,OAAO,CAAC,aAAa,KAAK,SAAS,EAAE,CAAC;YACxC,OAAO,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QACrD,CAAC;QACD,MAAM,OAAO,GAAG,kBAAkB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACpD,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3D,MAAM,IAAI,SAAS,CACjB,GAAG,IAAI,kDAAkD,CAC1D,CAAC;QACJ,CAAC;QACD,MAAM,gBAAgB,GAAG;YACvB,GAAG,OAAO,CAAC,gBAAgB;YAC3B,GAAG,CAAC,OAAO,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,IAAI;gBAC1C,SAAS,EAAE,OAAO,CAAC,iBAAiB;aACrC,CAAC;SACH,CAAC;QACF,MAAM,UAAU,GAAG;YACjB,OAAO,EAAE,OAAgB;YACzB,iBAAiB,EAAE,gBAAgB;YACnC,iBAAiB,EAAE,OAAO,CAAC,gBAAgB,IAAI,EAAE;YACjD,GAAG,CAAC,OAAO,CAAC,IAAI,KAAK,SAAS,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,CAAC;YACzD,GAAG,CAAC,OAAO,CAAC,EAAE,KAAK,SAAS,IAAI,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC;SACpD,CAAC;QAEF,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACrB,KAAK,WAAW,CAAC,IAAI;gBACnB,OAAO,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC;YACtC,KAAK,WAAW,CAAC,SAAS;gBACxB,OAAO,IAAI,SAAS,CAAC;oBACnB,GAAG,UAAU;oBACb,UAAU,EAAE,OAAO,CAAC,SAAkB;oBACtC,kBAAkB,EAAE,OAAO,CAAC,gBAAyB;iBACtD,CAAC,CAAC;YACL,KAAK,WAAW,CAAC,IAAI;gBACnB,IAAI,OAAO,CAAC,UAAU,KAAK,SAAS,IAAI,OAAO,CAAC,UAAU,KAAK,EAAE,EAAE,CAAC;oBAClE,MAAM,IAAI,SAAS,CACjB,GAAG,IAAI,qDAAqD,CAC7D,CAAC;gBACJ,CAAC;gBACD,OAAO,IAAI,WAAW,CAAC;oBACrB,GAAG,UAAU;oBACb,YAAY,EAAE,OAAO,CAAC,UAAU;oBAChC,MAAM,EAAE,OAAO,CAAC,OAAO,KAAK,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;oBACtD,GAAG,CAAC,OAAO,CAAC,YAAY,KAAK,SAAS,IAAI;wBACxC,QAAQ,EAAE,kBAAkB,CAAC,OAAO,CAAC,YAAY,CAAC;qBACnD,CAAC;iBACH,CAAC,CAAC;YACL,KAAK,WAAW,CAAC,MAAM;gBACrB,OAAO,IAAI,aAAa,CAAC,UAAU,CAAC,CAAC;YACvC;gBACE,MAAM,IAAI,SAAS,CACjB,GAAG,IAAI,qDAAqD,CAC7D,CAAC;QACN,CAAC;IACH,CAAC;IAEO,MAAM,CAAC,sBAAsB,CACnC,OAAoB,EACpB,eAAwB;QAExB,MAAM,aAAa,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;QACvC,IAAI,aAAa,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YAChC,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAkB,CAAC;QACpE,CAAC;QAED,MAAM,IAAI,GAAG,EAAE,GAAG,aAAa,CAAC,IAAI,EAA6B,CAAC;QAClE,MAAM,gBAAgB,GACpB,IAAI,CAAC,mBAAmB,CAAC,KAAK,IAAI;YAClC,OAAO,IAAI,CAAC,mBAAmB,CAAC,KAAK,QAAQ;YAC7C,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YACvC,CAAC,CAAC,EAAE,GAAI,IAAI,CAAC,mBAAmB,CAA6B,EAAE;YAC/D,CAAC,CAAC,EAAE,CAAC;QACT,OAAO,gBAAgB,CAAC,YAAY,CAAC,CAAC;QACtC,OAAO,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAChC,OAAO,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC9B,IAAI,CAAC,mBAAmB,CAAC,GAAG,gBAAgB,CAAC;QAC7C,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,eAAe,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CACtE,QAAQ,CAAC,eAAe,EAAE,CAC3B,CAAC;QACF,aAAa,CAAC,IAAI,GAAG,IAAwC,CAAC;QAC9D,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAkB,CAAC;IACpE,CAAC;IAEO,MAAM,CAAC,iBAAiB,CAC9B,OAAwB,EACxB,IAAU,EACV,IAAY;QAEZ,MAAM,MAAM,GAAG,kBAAkB,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACzD,IACE,MAAM,KAAK,IAAI;YACf,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;YACrB,OAAO,MAAM,KAAK,QAAQ,EAC1B,CAAC;YACD,MAAM,IAAI,SAAS,CAAC,GAAG,IAAI,0CAA0C,CAAC,CAAC;QACzE,CAAC;QACD,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,6BAA6B,CAC7C,MAAkC,CACnC,CAAC;YACF,MAAM,QAAQ,GAAG,mBAAmB,CAAC,SAAS,CAAC,CAAC;YAChD,IAAI,QAAQ,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;gBAC3B,MAAM,IAAI,SAAS,CACjB,GAAG,IAAI,gEAAgE,CACxE,CAAC;YACJ,CAAC;YACD,IAAI,OAAO,CAAC,EAAE,KAAK,SAAS,IAAI,QAAQ,CAAC,EAAE,KAAK,OAAO,CAAC,EAAE,EAAE,CAAC;gBAC3D,MAAM,IAAI,SAAS,CACjB,GAAG,IAAI,8DAA8D,CACtE,CAAC;YACJ,CAAC;YACD,IACE,OAAO,CAAC,UAAU,KAAK,SAAS;gBAChC,QAAQ,CAAC,UAAU,KAAK,OAAO,CAAC,UAAU,EAC1C,CAAC;gBACD,MAAM,IAAI,SAAS,CACjB,GAAG,IAAI,gEAAgE,CACxE,CAAC;YACJ,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,SAAS,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,IAAI,GAAG,CAAC,EAAE,CAAC;gBACvE,MAAM,KAAK,CAAC;YACd,CAAC;YACD,MAAM,IAAI,SAAS,CACjB,GAAG,IAAI,qDAAqD,EAC5D,EAAE,KAAK,EAAE,KAAK,EAAE,CACjB,CAAC;QACJ,CAAC;IACH,CAAC;CACF"}

package/dist/workflow_state.d.ts

@@ -0,0 +1,7 @@
+/** LangGraph application-channel translation for OE durable state. */
+import { type StateSnapshot } from "@mongodb-js/agent-engine-runner-shared";
+/** Convert LangGraph application channels into the OE-owned durable snapshot. */
+export declare function channelValuesToStateSnapshot(channelValues: Record<string, unknown>, includeLegacySource?: boolean): StateSnapshot;
+/** Restore OE durable state into the channels expected by LangGraph. */
+export declare function stateSnapshotToChannelValues(snapshot: StateSnapshot): Record<string, unknown>;
+//# sourceMappingURL=workflow_state.d.ts.map

package/dist/workflow_state.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow_state.d.ts","sourceRoot":"","sources":["../src/workflow_state.ts"],"names":[],"mappings":"AAAA,sEAAsE;AAItE,OAAO,EAEL,KAAK,aAAa,EACnB,MAAM,wCAAwC,CAAC;AA4BhD,iFAAiF;AACjF,wBAAgB,4BAA4B,CAC1C,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACtC,mBAAmB,UAAQ,GAC1B,aAAa,CAkBf;AAED,wEAAwE;AACxE,wBAAgB,4BAA4B,CAC1C,QAAQ,EAAE,aAAa,GACtB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAqBzB"}

package/dist/workflow_state.js

@@ -0,0 +1,62 @@
+/** LangGraph application-channel translation for OE durable state. */
+import { create } from "@bufbuild/protobuf";
+import { isBaseMessage } from "@langchain/core/messages";
+import { StateSnapshotSchema, } from "@mongodb-js/agent-engine-runner-shared";
+import { requireJsonObject } from "./workflow_json.js";
+import { WorkflowMessageCodec } from "./workflow_message.js";
+/** LangGraph reserves these channels for execution mechanics, not app state. */
+function isControlChannel(name) {
+    return (name.startsWith("__") ||
+        name.startsWith("branch:") ||
+        name.startsWith("start:"));
+}
+/** Read and validate the conventional LangGraph messages channel. */
+function workflowMessages(value) {
+    if (value === undefined)
+        return [];
+    if (!Array.isArray(value)) {
+        throw new TypeError("LangGraph messages channel must be an array");
+    }
+    return value.map((message, index) => {
+        if (!isBaseMessage(message)) {
+            throw new TypeError(`LangGraph messages[${index}] is not a BaseMessage`);
+        }
+        return message;
+    });
+}
+/** Convert LangGraph application channels into the OE-owned durable snapshot. */
+export function channelValuesToStateSnapshot(channelValues, includeLegacySource = false) {
+    const properties = {};
+    for (const name of Object.keys(channelValues).sort()) {
+        if (name === "messages" || isControlChannel(name))
+            continue;
+        // LangGraph materializes optional channels as undefined until they receive
+        // a value. JSON has no undefined value; omitting the absent channel
+        // reconstructs the same application state.
+        if (channelValues[name] === undefined)
+            continue;
+        properties[name] = channelValues[name];
+    }
+    return create(StateSnapshotSchema, {
+        properties: requireJsonObject(properties, "LangGraph application state"),
+        messageEncodingVersion: 1,
+        messages: workflowMessages(channelValues["messages"]).map((message) => WorkflowMessageCodec.fromLangChainMessage(message, includeLegacySource)),
+    });
+}
+/** Restore OE durable state into the channels expected by LangGraph. */
+export function stateSnapshotToChannelValues(snapshot) {
+    const properties = requireJsonObject(snapshot.properties ?? {}, "previous workflow state");
+    for (const name of Object.keys(properties).sort()) {
+        if (name === "messages" || isControlChannel(name)) {
+            throw new TypeError(`previous workflow state contains reserved LangGraph channel "${name}"`);
+        }
+    }
+    const channelValues = {
+        ...properties,
+    };
+    if (snapshot.messages.length > 0) {
+        channelValues["messages"] = snapshot.messages.map((message, index) => WorkflowMessageCodec.toLangChainMessage(message, `messages[${index}]`));
+    }
+    return channelValues;
+}
+//# sourceMappingURL=workflow_state.js.map

package/dist/workflow_state.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow_state.js","sourceRoot":"","sources":["../src/workflow_state.ts"],"names":[],"mappings":"AAAA,sEAAsE;AAEtE,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAE,aAAa,EAAoB,MAAM,0BAA0B,CAAC;AAC3E,OAAO,EACL,mBAAmB,GAEpB,MAAM,wCAAwC,CAAC;AAEhD,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAE7D,gFAAgF;AAChF,SAAS,gBAAgB,CAAC,IAAY;IACpC,OAAO,CACL,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;QACrB,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC;QAC1B,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAC1B,CAAC;AACJ,CAAC;AAED,qEAAqE;AACrE,SAAS,gBAAgB,CAAC,KAAc;IACtC,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,EAAE,CAAC;IACnC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAC1B,MAAM,IAAI,SAAS,CAAC,6CAA6C,CAAC,CAAC;IACrE,CAAC;IACD,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;QAClC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,SAAS,CAAC,sBAAsB,KAAK,wBAAwB,CAAC,CAAC;QAC3E,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC,CAAC,CAAC;AACL,CAAC;AAED,iFAAiF;AACjF,MAAM,UAAU,4BAA4B,CAC1C,aAAsC,EACtC,mBAAmB,GAAG,KAAK;IAE3B,MAAM,UAAU,GAA4B,EAAE,CAAC;IAC/C,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;QACrD,IAAI,IAAI,KAAK,UAAU,IAAI,gBAAgB,CAAC,IAAI,CAAC;YAAE,SAAS;QAC5D,2EAA2E;QAC3E,oEAAoE;QACpE,2CAA2C;QAC3C,IAAI,aAAa,CAAC,IAAI,CAAC,KAAK,SAAS;YAAE,SAAS;QAChD,UAAU,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;IACzC,CAAC;IAED,OAAO,MAAM,CAAC,mBAAmB,EAAE;QACjC,UAAU,EAAE,iBAAiB,CAAC,UAAU,EAAE,6BAA6B,CAAC;QACxE,sBAAsB,EAAE,CAAC;QACzB,QAAQ,EAAE,gBAAgB,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CACpE,oBAAoB,CAAC,oBAAoB,CAAC,OAAO,EAAE,mBAAmB,CAAC,CACxE;KACF,CAAC,CAAC;AACL,CAAC;AAED,wEAAwE;AACxE,MAAM,UAAU,4BAA4B,CAC1C,QAAuB;IAEvB,MAAM,UAAU,GAAG,iBAAiB,CAClC,QAAQ,CAAC,UAAU,IAAI,EAAE,EACzB,yBAAyB,CAC1B,CAAC;IACF,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;QAClD,IAAI,IAAI,KAAK,UAAU,IAAI,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;YAClD,MAAM,IAAI,SAAS,CACjB,gEAAgE,IAAI,GAAG,CACxE,CAAC;QACJ,CAAC;IACH,CAAC;IACD,MAAM,aAAa,GAA4B;QAC7C,GAAG,UAAU;KACd,CAAC;IACF,IAAI,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACjC,aAAa,CAAC,UAAU,CAAC,GAAG,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,CACnE,oBAAoB,CAAC,kBAAkB,CAAC,OAAO,EAAE,YAAY,KAAK,GAAG,CAAC,CACvE,CAAC;IACJ,CAAC;IACD,OAAO,aAAa,CAAC;AACvB,CAAC"}