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

package/dist/execution_session.d.ts

@@ -0,0 +1,63 @@
+/** Shared execution-session contract and native LangGraph implementation. */
+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, RequestContext, StreamEvent } from "@mongodb-js/agent-engine-sdk";
+import { checkpointMetadata } from "./checkpoint_branch.js";
+import type { DurableSubgraphResolver } from "./durable_subgraphs.js";
+export type PrepareAgentInput = (input: AgentInput, ctx: RequestContext) => unknown;
+export type ResolveThreadId = (ctx: RequestContext) => string;
+export interface PreparedRun {
+    readonly graphInput: unknown;
+    readonly config: RunnableConfig;
+}
+type ExecutionMetadata = ReturnType<typeof checkpointMetadata>;
+export interface SessionOptions {
+    readonly callbacks: readonly unknown[];
+    readonly prepareInput: PrepareAgentInput | null;
+    readonly resolveThreadId: ResolveThreadId | null;
+    readonly durableSubgraphs: DurableSubgraphResolver | null;
+}
+export declare function explicitCheckpointId(ctx: RequestContext): string | undefined;
+export declare abstract class ExecutionSession {
+    abstract readonly resumed: boolean;
+    readonly threadId: string;
+    protected readonly graph: CompiledStateGraph<unknown, unknown>;
+    protected readonly ctx: RequestContext;
+    protected readonly input: AgentInput;
+    protected readonly callbacks: readonly unknown[];
+    protected readonly prepareInput: PrepareAgentInput | null;
+    protected readonly durableSubgraphs: DurableSubgraphResolver | null;
+    protected constructor(graph: CompiledStateGraph<unknown, unknown>, ctx: RequestContext, input: AgentInput, options: SessionOptions);
+    protected baseConfig(): RunnableConfig;
+    protected freshGraphInput(ctx?: RequestContext): unknown;
+    abstract prepareRun(): Promise<PreparedRun>;
+    abstract invokeInterrupt(config: RunnableConfig, response: string, messages: readonly unknown[]): Promise<Command | AgentOutput | null>;
+    abstract streamInterrupt(config: RunnableConfig, captured: readonly Interrupt[], messages: readonly BaseMessage[]): Promise<Command | StreamEvent | null>;
+    abstract complete(): Promise<ExecutionMetadata | void>;
+    abstract close(): Promise<void>;
+    runInGraphScope<T>(fn: () => T): T;
+    iterateInGraphScope<T>(iterable: AsyncIterable<T>): AsyncIterable<T>;
+}
+export declare class NativeSession extends ExecutionSession {
+    readonly resumed: boolean;
+    private readonly config;
+    private readonly branchSource;
+    private readonly resumeCommand;
+    constructor(graph: CompiledStateGraph<unknown, unknown>, ctx: RequestContext, input: AgentInput, options: SessionOptions);
+    prepareRun(): Promise<PreparedRun>;
+    /**
+     * After an in-app fork, the destination's first continue carries no user
+     * message; invoke(null) so LangGraph resumes from the copied checkpoint
+     * instead of appending an empty HumanMessage (docs/session-fork.md).
+     */
+    private firstDestContinueInput;
+    invokeInterrupt(config: RunnableConfig, response: string, messages: readonly unknown[]): Promise<AgentOutput | null>;
+    streamInterrupt(_config: RunnableConfig, captured: readonly Interrupt[], messages: readonly BaseMessage[]): Promise<StreamEvent | null>;
+    complete(): Promise<ExecutionMetadata>;
+    close(): Promise<void>;
+    private getCheckpointId;
+    private fenceCancelledPredecessorWrites;
+}
+export {};
+//# sourceMappingURL=execution_session.d.ts.map

package/dist/execution_session.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"execution_session.d.ts","sourceRoot":"","sources":["../src/execution_session.ts"],"names":[],"mappings":"AAAA,6EAA6E;AAE7E,OAAO,EAAgB,KAAK,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAC1E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAChE,OAAO,EACL,OAAO,EACP,KAAK,kBAAkB,EACvB,KAAK,SAAS,EACf,MAAM,sBAAsB,CAAC;AAE9B,OAAO,KAAK,EACV,UAAU,EACV,WAAW,EACX,cAAc,EACd,WAAW,EACZ,MAAM,8BAA8B,CAAC;AAGtC,OAAO,EAGL,kBAAkB,EAEnB,MAAM,wBAAwB,CAAC;AAChC,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,wBAAwB,CAAC;AAatE,MAAM,MAAM,iBAAiB,GAAG,CAC9B,KAAK,EAAE,UAAU,EACjB,GAAG,EAAE,cAAc,KAChB,OAAO,CAAC;AAEb,MAAM,MAAM,eAAe,GAAG,CAAC,GAAG,EAAE,cAAc,KAAK,MAAM,CAAC;AAE9D,MAAM,WAAW,WAAW;IAC1B,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC;IAC7B,QAAQ,CAAC,MAAM,EAAE,cAAc,CAAC;CACjC;AAED,KAAK,iBAAiB,GAAG,UAAU,CAAC,OAAO,kBAAkB,CAAC,CAAC;AAE/D,MAAM,WAAW,cAAc;IAC7B,QAAQ,CAAC,SAAS,EAAE,SAAS,OAAO,EAAE,CAAC;IACvC,QAAQ,CAAC,YAAY,EAAE,iBAAiB,GAAG,IAAI,CAAC;IAChD,QAAQ,CAAC,eAAe,EAAE,eAAe,GAAG,IAAI,CAAC;IACjD,QAAQ,CAAC,gBAAgB,EAAE,uBAAuB,GAAG,IAAI,CAAC;CAC3D;AAoFD,wBAAgB,oBAAoB,CAAC,GAAG,EAAE,cAAc,GAAG,MAAM,GAAG,SAAS,CAG5E;AAED,8BAAsB,gBAAgB;IACpC,QAAQ,CAAC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;IACnC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,SAAS,CAAC,QAAQ,CAAC,KAAK,EAAE,kBAAkB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAC/D,SAAS,CAAC,QAAQ,CAAC,GAAG,EAAE,cAAc,CAAC;IACvC,SAAS,CAAC,QAAQ,CAAC,KAAK,EAAE,UAAU,CAAC;IACrC,SAAS,CAAC,QAAQ,CAAC,SAAS,EAAE,SAAS,OAAO,EAAE,CAAC;IACjD,SAAS,CAAC,QAAQ,CAAC,YAAY,EAAE,iBAAiB,GAAG,IAAI,CAAC;IAC1D,SAAS,CAAC,QAAQ,CAAC,gBAAgB,EAAE,uBAAuB,GAAG,IAAI,CAAC;IAEpE,SAAS,aACP,KAAK,EAAE,kBAAkB,CAAC,OAAO,EAAE,OAAO,CAAC,EAC3C,GAAG,EAAE,cAAc,EACnB,KAAK,EAAE,UAAU,EACjB,OAAO,EAAE,cAAc;IAWzB,SAAS,CAAC,UAAU,IAAI,cAAc;IAatC,SAAS,CAAC,eAAe,CAAC,GAAG,GAAE,cAAyB,GAAG,OAAO;IAelE,QAAQ,CAAC,UAAU,IAAI,OAAO,CAAC,WAAW,CAAC;IAE3C,QAAQ,CAAC,eAAe,CACtB,MAAM,EAAE,cAAc,EACtB,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,SAAS,OAAO,EAAE,GAC3B,OAAO,CAAC,OAAO,GAAG,WAAW,GAAG,IAAI,CAAC;IAExC,QAAQ,CAAC,eAAe,CACtB,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;IAExC,QAAQ,CAAC,QAAQ,IAAI,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAEtD,QAAQ,CAAC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAE/B,eAAe,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC;IAI3B,mBAAmB,CAAC,CAAC,EAAE,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC;CAe5E;AAED,qBAAa,aAAc,SAAQ,gBAAgB;IACjD,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;IAC1B,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAiB;IACxC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAkC;IAC/D,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAiB;gBAG7C,KAAK,EAAE,kBAAkB,CAAC,OAAO,EAAE,OAAO,CAAC,EAC3C,GAAG,EAAE,cAAc,EACnB,KAAK,EAAE,UAAU,EACjB,OAAO,EAAE,cAAc;IAgCnB,UAAU,IAAI,OAAO,CAAC,WAAW,CAAC;IAyBxC;;;;OAIG;IACH,OAAO,CAAC,sBAAsB;IAYxB,eAAe,CACnB,MAAM,EAAE,cAAc,EACtB,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,SAAS,OAAO,EAAE,GAC3B,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IAkBxB,eAAe,CACnB,OAAO,EAAE,cAAc,EACvB,QAAQ,EAAE,SAAS,SAAS,EAAE,EAC9B,QAAQ,EAAE,SAAS,WAAW,EAAE,GAC/B,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC;IAYxB,QAAQ,IAAI,OAAO,CAAC,iBAAiB,CAAC;IAS5C,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;YAIR,eAAe;YA8Bf,+BAA+B;CAgD9C"}

package/dist/execution_session.js

@@ -0,0 +1,300 @@
+/** Shared execution-session contract and native LangGraph implementation. */
+import { HumanMessage } from "@langchain/core/messages";
+import { Command, } from "@langchain/langgraph";
+import { isDeepStrictEqual } from "node:util";
+import { getLogger } from "@mongodb-js/agent-engine-runner-shared";
+import { initializeCheckpointBranch, branchPointFromMetadata, checkpointMetadata, } from "./checkpoint_branch.js";
+import { PlatformCheckpointer } from "./platform_checkpointer.js";
+import { takeContinueWithoutUserMessage } from "./session_fork.js";
+import { interruptSnapshot, invokeSuspendOutput, streamHitlSuspendEvent, validateInterruptValue, } from "./suspend.js";
+import { checkpointThreadId } from "./thread_id.js";
+const logger = getLogger("agent_engine_sdk_langgraph.execution_session");
+function interruptValuesFromState(state) {
+    const values = [];
+    const seen = new Map();
+    for (const task of state.tasks ?? []) {
+        for (const interrupt of task.interrupts ?? []) {
+            validateInterruptValue(interrupt.id ?? "<unknown>", interrupt.value);
+            const id = typeof interrupt.id === "string" ? interrupt.id : "";
+            // A nested subgraph's interrupt can repeat under its parent task with
+            // an identical value; project each native id once. A repeated id with
+            // a different value is conflicting framework state — fail closed the
+            // same way the stream and durable paths do.
+            if (id) {
+                const previous = seen.get(id);
+                if (previous !== undefined) {
+                    if (isDeepStrictEqual(previous, interrupt.value))
+                        continue;
+                    throw new Error("LangGraph returned duplicate interrupt ids");
+                }
+                seen.set(id, interrupt.value);
+            }
+            values.push(interrupt.value);
+        }
+    }
+    return values;
+}
+function payloadDict(input) {
+    if (input.payload === null ||
+        typeof input.payload !== "object" ||
+        Array.isArray(input.payload)) {
+        throw new TypeError(`Expected dict payload, got ${typeof input.payload}`);
+    }
+    return input.payload;
+}
+function resolveCheckpointThreadId(ctx, resolveThreadId) {
+    return checkpointThreadId(ctx, resolveThreadId);
+}
+function normalizeCheckpointId(value) {
+    if (value === null || value === undefined)
+        return undefined;
+    const checkpointId = String(value).trim();
+    if (checkpointId === "" || checkpointId.toLowerCase() === "none") {
+        return undefined;
+    }
+    return checkpointId;
+}
+export function explicitCheckpointId(ctx) {
+    const checkpointId = normalizeCheckpointId(ctx.metadata?.["checkpoint_id"]);
+    return checkpointId?.startsWith("thread:") ? undefined : checkpointId;
+}
+export class ExecutionSession {
+    threadId;
+    graph;
+    ctx;
+    input;
+    callbacks;
+    prepareInput;
+    durableSubgraphs;
+    constructor(graph, ctx, input, options) {
+        this.graph = graph;
+        this.ctx = ctx;
+        this.input = input;
+        this.callbacks = options.callbacks;
+        this.prepareInput = options.prepareInput;
+        this.durableSubgraphs = options.durableSubgraphs;
+        this.threadId = resolveCheckpointThreadId(ctx, options.resolveThreadId);
+    }
+    baseConfig() {
+        const config = {
+            configurable: {
+                thread_id: this.threadId,
+                request_context: this.ctx,
+            },
+        };
+        if (this.callbacks.length > 0) {
+            config.callbacks = [...this.callbacks];
+        }
+        return config;
+    }
+    freshGraphInput(ctx = this.ctx) {
+        if (this.prepareInput !== null) {
+            return this.prepareInput(this.input, ctx);
+        }
+        const payload = payloadDict(this.input);
+        const graphInput = {
+            messages: [
+                new HumanMessage({ content: String(payload["message"] ?? "") }),
+            ],
+        };
+        if (ctx.userId)
+            graphInput["user_id"] = ctx.userId;
+        if (ctx.sessionId)
+            graphInput["session_id"] = ctx.sessionId;
+        return graphInput;
+    }
+    runInGraphScope(fn) {
+        return fn();
+    }
+    async *iterateInGraphScope(iterable) {
+        const iterator = iterable[Symbol.asyncIterator]();
+        try {
+            while (true) {
+                const result = await this.runInGraphScope(() => iterator.next());
+                if (result.done)
+                    return;
+                yield result.value;
+            }
+        }
+        finally {
+            const close = iterator.return;
+            if (close !== undefined) {
+                await this.runInGraphScope(() => close.call(iterator));
+            }
+        }
+    }
+}
+export class NativeSession extends ExecutionSession {
+    resumed;
+    config;
+    branchSource;
+    resumeCommand;
+    constructor(graph, ctx, input, options) {
+        super(graph, ctx, input, options);
+        this.resumed = ctx.resume === true;
+        const resumeData = ctx.resumeData;
+        if (ctx.resume === true &&
+            (resumeData === undefined || resumeData === null)) {
+            throw new Error("ctx.resume is true but resume_data is missing. " +
+                "Provide resume_data to resume a suspended graph.");
+        }
+        this.resumeCommand =
+            ctx.resume === true && resumeData !== undefined && resumeData !== null
+                ? new Command({ resume: resumeData })
+                : null;
+        this.branchSource = this.resumed
+            ? undefined
+            : branchPointFromMetadata(ctx.metadata);
+        const config = this.baseConfig();
+        const checkpointId = explicitCheckpointId(ctx);
+        if (checkpointId !== undefined) {
+            config.configurable = {
+                ...config.configurable,
+                checkpoint_id: checkpointId,
+            };
+        }
+        this.config = config;
+    }
+    async prepareRun() {
+        // A null graph input is meaningful (continue the copied checkpoint without
+        // a user turn), so the branches stay explicit: `??` would fall through it.
+        let graphInput;
+        if (this.resumeCommand !== null) {
+            graphInput = this.resumeCommand;
+        }
+        else {
+            graphInput =
+                this.firstDestContinueInput() === null ? null : this.freshGraphInput();
+        }
+        // HITL resume continues a thread that already received any branch copy;
+        // clients and OE dispatch may still send the source branch point.
+        if (!this.resumed && this.branchSource !== undefined) {
+            await initializeCheckpointBranch(this.graph, this.config, this.branchSource);
+        }
+        if (this.ctx.previousExecutionCancelled === true && !this.resumed) {
+            await this.fenceCancelledPredecessorWrites();
+        }
+        return { graphInput, config: this.config };
+    }
+    /**
+     * After an in-app fork, the destination's first continue carries no user
+     * message; invoke(null) so LangGraph resumes from the copied checkpoint
+     * instead of appending an empty HumanMessage (docs/session-fork.md).
+     */
+    firstDestContinueInput() {
+        if (this.resumed)
+            return undefined;
+        if (!takeContinueWithoutUserMessage(this.threadId))
+            return undefined;
+        const raw = this.input.payload;
+        const payload = raw !== null && typeof raw === "object" && !Array.isArray(raw)
+            ? raw
+            : {};
+        if (String(payload["message"] ?? "").trim() !== "")
+            return undefined;
+        return null;
+    }
+    async invokeInterrupt(config, response, messages) {
+        const graph = this.graph;
+        const state = await graph.getState(config);
+        if (!state.next || state.next.length === 0)
+            return null;
+        const interruptValues = interruptValuesFromState(state);
+        const checkpointId = await this.getCheckpointId();
+        return invokeSuspendOutput({
+            response,
+            threadId: this.threadId,
+            checkpointId,
+            interruptValues,
+            messageCount: messages.length,
+            resumed: this.resumed,
+        });
+    }
+    async streamInterrupt(_config, captured, messages) {
+        if (captured.length === 0)
+            return null;
+        const interrupts = interruptSnapshot(captured);
+        const checkpointId = await this.getCheckpointId();
+        return streamHitlSuspendEvent({
+            interrupts,
+            checkpointId,
+            resumed: this.resumed,
+            messages,
+        });
+    }
+    async complete() {
+        const checkpointId = await this.getCheckpointId();
+        return checkpointMetadata(checkpointId === undefined
+            ? undefined
+            : { threadId: this.threadId, checkpointId });
+    }
+    close() {
+        return Promise.resolve();
+    }
+    async getCheckpointId() {
+        const checkpointer = this.graph
+            .checkpointer;
+        if (checkpointer === null || checkpointer === undefined)
+            return undefined;
+        const saver = checkpointer;
+        if (typeof saver.getTuple !== "function")
+            return undefined;
+        try {
+            const tuple = (await saver.getTuple({
+                configurable: { thread_id: this.threadId },
+            }));
+            const checkpoint = tuple?.checkpoint;
+            if (checkpoint !== null &&
+                typeof checkpoint === "object" &&
+                !Array.isArray(checkpoint)) {
+                const id = checkpoint["id"];
+                return typeof id === "string" ? id : undefined;
+            }
+        }
+        catch (exc) {
+            logger.warn(`Failed to get checkpoint ID for thread ${this.threadId} — degrading to undefined ` +
+                `without changing the agent execution outcome: ${exc instanceof Error ? exc.message : String(exc)}`);
+        }
+        return undefined;
+    }
+    async fenceCancelledPredecessorWrites() {
+        // A cancelled pod can leave half-finished writes on the latest checkpoint.
+        // Remove only those writes before a fresh turn can fold them into its state.
+        const graphSaver = this.graph.checkpointer;
+        const saver = (graphSaver instanceof PlatformCheckpointer
+            ? graphSaver.native
+            : graphSaver);
+        if (!saver?.db?.collection ||
+            typeof saver.checkpointCollectionName !== "string" ||
+            typeof saver.checkpointWritesCollectionName !== "string") {
+            return;
+        }
+        try {
+            const checkpoints = saver.db.collection(saver.checkpointCollectionName);
+            const writes = saver.db.collection(saver.checkpointWritesCollectionName);
+            const namespaces = await checkpoints.distinct("checkpoint_ns", {
+                thread_id: this.threadId,
+            });
+            let removed = 0;
+            for (const namespace of namespaces) {
+                const latest = await checkpoints.findOne({ thread_id: this.threadId, checkpoint_ns: namespace }, { sort: { checkpoint_id: -1 }, projection: { checkpoint_id: 1 } });
+                if (!latest)
+                    continue;
+                const result = await writes.deleteMany({
+                    thread_id: this.threadId,
+                    checkpoint_ns: namespace,
+                    checkpoint_id: latest["checkpoint_id"],
+                });
+                removed += result.deletedCount ?? 0;
+            }
+            if (removed > 0) {
+                logger.info(`Discarded ${removed} pending checkpoint writes left by a cancelled run (thread ${this.threadId})`);
+            }
+        }
+        catch (exc) {
+            logger.warn(`Failed to fence a cancelled run's pending writes (thread ${this.threadId}): ` +
+                `${exc instanceof Error ? exc.message : String(exc)}`);
+        }
+    }
+}
+//# sourceMappingURL=execution_session.js.map

package/dist/execution_session.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"execution_session.js","sourceRoot":"","sources":["../src/execution_session.ts"],"names":[],"mappings":"AAAA,6EAA6E;AAE7E,OAAO,EAAE,YAAY,EAAoB,MAAM,0BAA0B,CAAC;AAE1E,OAAO,EACL,OAAO,GAGR,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAO9C,OAAO,EAAE,SAAS,EAAE,MAAM,wCAAwC,CAAC;AAEnE,OAAO,EACL,0BAA0B,EAC1B,uBAAuB,EACvB,kBAAkB,GAEnB,MAAM,wBAAwB,CAAC;AAEhC,OAAO,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AAClE,OAAO,EAAE,8BAA8B,EAAE,MAAM,mBAAmB,CAAC;AACnE,OAAO,EACL,iBAAiB,EACjB,mBAAmB,EACnB,sBAAsB,EACtB,sBAAsB,GACvB,MAAM,cAAc,CAAC;AACtB,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAEpD,MAAM,MAAM,GAAG,SAAS,CAAC,8CAA8C,CAAC,CAAC;AAqDzE,SAAS,wBAAwB,CAAC,KAA0B;IAC1D,MAAM,MAAM,GAAc,EAAE,CAAC;IAC7B,MAAM,IAAI,GAAG,IAAI,GAAG,EAAmB,CAAC;IACxC,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,KAAK,IAAI,EAAE,EAAE,CAAC;QACrC,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,UAAU,IAAI,EAAE,EAAE,CAAC;YAC9C,sBAAsB,CAAC,SAAS,CAAC,EAAE,IAAI,WAAW,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC;YACrE,MAAM,EAAE,GAAG,OAAO,SAAS,CAAC,EAAE,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAChE,sEAAsE;YACtE,sEAAsE;YACtE,qEAAqE;YACrE,4CAA4C;YAC5C,IAAI,EAAE,EAAE,CAAC;gBACP,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC9B,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;oBAC3B,IAAI,iBAAiB,CAAC,QAAQ,EAAE,SAAS,CAAC,KAAK,CAAC;wBAAE,SAAS;oBAC3D,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;gBAChE,CAAC;gBACD,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC;YAChC,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,WAAW,CAAC,KAAiB;IACpC,IACE,KAAK,CAAC,OAAO,KAAK,IAAI;QACtB,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ;QACjC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,EAC5B,CAAC;QACD,MAAM,IAAI,SAAS,CAAC,8BAA8B,OAAO,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;IAC5E,CAAC;IACD,OAAO,KAAK,CAAC,OAAkC,CAAC;AAClD,CAAC;AAED,SAAS,yBAAyB,CAChC,GAAmB,EACnB,eAAuC;IAEvC,OAAO,kBAAkB,CAAC,GAAG,EAAE,eAAe,CAAC,CAAC;AAClD,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAc;IAC3C,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,SAAS,CAAC;IAC5D,MAAM,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;IAC1C,IAAI,YAAY,KAAK,EAAE,IAAI,YAAY,CAAC,WAAW,EAAE,KAAK,MAAM,EAAE,CAAC;QACjE,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,GAAmB;IACtD,MAAM,YAAY,GAAG,qBAAqB,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC;IAC5E,OAAO,YAAY,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC;AACxE,CAAC;AAED,MAAM,OAAgB,gBAAgB;IAE3B,QAAQ,CAAS;IACP,KAAK,CAAuC;IAC5C,GAAG,CAAiB;IACpB,KAAK,CAAa;IAClB,SAAS,CAAqB;IAC9B,YAAY,CAA2B;IACvC,gBAAgB,CAAiC;IAEpE,YACE,KAA2C,EAC3C,GAAmB,EACnB,KAAiB,EACjB,OAAuB;QAEvB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC;QACjD,IAAI,CAAC,QAAQ,GAAG,yBAAyB,CAAC,GAAG,EAAE,OAAO,CAAC,eAAe,CAAC,CAAC;IAC1E,CAAC;IAES,UAAU;QAClB,MAAM,MAAM,GAAmB;YAC7B,YAAY,EAAE;gBACZ,SAAS,EAAE,IAAI,CAAC,QAAQ;gBACxB,eAAe,EAAE,IAAI,CAAC,GAAG;aAC1B;SACF,CAAC;QACF,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,MAAkC,CAAC,SAAS,GAAG,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QACtE,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAES,eAAe,CAAC,MAAsB,IAAI,CAAC,GAAG;QACtD,IAAI,IAAI,CAAC,YAAY,KAAK,IAAI,EAAE,CAAC;YAC/B,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAC5C,CAAC;QACD,MAAM,OAAO,GAAG,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxC,MAAM,UAAU,GAA4B;YAC1C,QAAQ,EAAE;gBACR,IAAI,YAAY,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;aAChE;SACF,CAAC;QACF,IAAI,GAAG,CAAC,MAAM;YAAE,UAAU,CAAC,SAAS,CAAC,GAAG,GAAG,CAAC,MAAM,CAAC;QACnD,IAAI,GAAG,CAAC,SAAS;YAAE,UAAU,CAAC,YAAY,CAAC,GAAG,GAAG,CAAC,SAAS,CAAC;QAC5D,OAAO,UAAU,CAAC;IACpB,CAAC;IAoBD,eAAe,CAAI,EAAW;QAC5B,OAAO,EAAE,EAAE,CAAC;IACd,CAAC;IAED,KAAK,CAAC,CAAC,mBAAmB,CAAI,QAA0B;QACtD,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;QAClD,IAAI,CAAC;YACH,OAAO,IAAI,EAAE,CAAC;gBACZ,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;gBACjE,IAAI,MAAM,CAAC,IAAI;oBAAE,OAAO;gBACxB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC;YAC9B,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,MAAM,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;YACzD,CAAC;QACH,CAAC;IACH,CAAC;CACF;AAED,MAAM,OAAO,aAAc,SAAQ,gBAAgB;IACxC,OAAO,CAAU;IACT,MAAM,CAAiB;IACvB,YAAY,CAAkC;IAC9C,aAAa,CAAiB;IAE/C,YACE,KAA2C,EAC3C,GAAmB,EACnB,KAAiB,EACjB,OAAuB;QAEvB,KAAK,CAAC,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QAClC,IAAI,CAAC,OAAO,GAAG,GAAG,CAAC,MAAM,KAAK,IAAI,CAAC;QACnC,MAAM,UAAU,GAAG,GAAG,CAAC,UAAU,CAAC;QAClC,IACE,GAAG,CAAC,MAAM,KAAK,IAAI;YACnB,CAAC,UAAU,KAAK,SAAS,IAAI,UAAU,KAAK,IAAI,CAAC,EACjD,CAAC;YACD,MAAM,IAAI,KAAK,CACb,iDAAiD;gBAC/C,kDAAkD,CACrD,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,aAAa;YAChB,GAAG,CAAC,MAAM,KAAK,IAAI,IAAI,UAAU,KAAK,SAAS,IAAI,UAAU,KAAK,IAAI;gBACpE,CAAC,CAAC,IAAI,OAAO,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC;gBACrC,CAAC,CAAC,IAAI,CAAC;QACX,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,OAAO;YAC9B,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,uBAAuB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QACjC,MAAM,YAAY,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC;QAC/C,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;YAC/B,MAAM,CAAC,YAAY,GAAG;gBACpB,GAAG,MAAM,CAAC,YAAY;gBACtB,aAAa,EAAE,YAAY;aAC5B,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,UAAU;QACd,2EAA2E;QAC3E,2EAA2E;QAC3E,IAAI,UAAmB,CAAC;QACxB,IAAI,IAAI,CAAC,aAAa,KAAK,IAAI,EAAE,CAAC;YAChC,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC;QAClC,CAAC;aAAM,CAAC;YACN,UAAU;gBACR,IAAI,CAAC,sBAAsB,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;QAC3E,CAAC;QACD,wEAAwE;QACxE,kEAAkE;QAClE,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,YAAY,KAAK,SAAS,EAAE,CAAC;YACrD,MAAM,0BAA0B,CAC9B,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,YAAY,CAClB,CAAC;QACJ,CAAC;QACD,IAAI,IAAI,CAAC,GAAG,CAAC,0BAA0B,KAAK,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClE,MAAM,IAAI,CAAC,+BAA+B,EAAE,CAAC;QAC/C,CAAC;QACD,OAAO,EAAE,UAAU,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;IAC7C,CAAC;IAED;;;;OAIG;IACK,sBAAsB;QAC5B,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO,SAAS,CAAC;QACnC,IAAI,CAAC,8BAA8B,CAAC,IAAI,CAAC,QAAQ,CAAC;YAAE,OAAO,SAAS,CAAC;QACrE,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;QAC/B,MAAM,OAAO,GACX,GAAG,KAAK,IAAI,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC;YAC5D,CAAC,CAAE,GAA+B;YAClC,CAAC,CAAC,EAAE,CAAC;QACT,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE;YAAE,OAAO,SAAS,CAAC;QACrE,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,MAAsB,EACtB,QAAgB,EAChB,QAA4B;QAE5B,MAAM,KAAK,GAAG,IAAI,CAAC,KAElB,CAAC;QACF,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QAC3C,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QACxD,MAAM,eAAe,GAAG,wBAAwB,CAAC,KAAK,CAAC,CAAC;QACxD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,OAAO,mBAAmB,CAAC;YACzB,QAAQ;YACR,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,YAAY;YACZ,eAAe;YACf,YAAY,EAAE,QAAQ,CAAC,MAAM;YAC7B,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,OAAuB,EACvB,QAA8B,EAC9B,QAAgC;QAEhC,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QACvC,MAAM,UAAU,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,OAAO,sBAAsB,CAAC;YAC5B,UAAU;YACV,YAAY;YACZ,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,QAAQ;SACT,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,OAAO,kBAAkB,CACvB,YAAY,KAAK,SAAS;YACxB,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,YAAY,EAAE,CAC9C,CAAC;IACJ,CAAC;IAED,KAAK;QACH,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;IAC3B,CAAC;IAEO,KAAK,CAAC,eAAe;QAC3B,MAAM,YAAY,GAAI,IAAI,CAAC,KAAoC;aAC5D,YAAY,CAAC;QAChB,IAAI,YAAY,KAAK,IAAI,IAAI,YAAY,KAAK,SAAS;YAAE,OAAO,SAAS,CAAC;QAC1E,MAAM,KAAK,GAAG,YAEb,CAAC;QACF,IAAI,OAAO,KAAK,CAAC,QAAQ,KAAK,UAAU;YAAE,OAAO,SAAS,CAAC;QAC3D,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,CAAC,MAAM,KAAK,CAAC,QAAQ,CAAC;gBAClC,YAAY,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,QAAQ,EAAE;aAC3C,CAAC,CAAoC,CAAC;YACvC,MAAM,UAAU,GAAG,KAAK,EAAE,UAAU,CAAC;YACrC,IACE,UAAU,KAAK,IAAI;gBACnB,OAAO,UAAU,KAAK,QAAQ;gBAC9B,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,EAC1B,CAAC;gBACD,MAAM,EAAE,GAAI,UAAsC,CAAC,IAAI,CAAC,CAAC;gBACzD,OAAO,OAAO,EAAE,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;YACjD,CAAC;QACH,CAAC;QAAC,OAAO,GAAY,EAAE,CAAC;YACtB,MAAM,CAAC,IAAI,CACT,0CAA0C,IAAI,CAAC,QAAQ,4BAA4B;gBACjF,iDAAiD,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CACtG,CAAC;QACJ,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,KAAK,CAAC,+BAA+B;QAC3C,2EAA2E;QAC3E,6EAA6E;QAC7E,MAAM,UAAU,GAAI,IAAI,CAAC,KAAoC,CAAC,YAAY,CAAC;QAC3E,MAAM,KAAK,GAAG,CACZ,UAAU,YAAY,oBAAoB;YACxC,CAAC,CAAC,UAAU,CAAC,MAAM;YACnB,CAAC,CAAC,UAAU,CACU,CAAC;QAC3B,IACE,CAAC,KAAK,EAAE,EAAE,EAAE,UAAU;YACtB,OAAO,KAAK,CAAC,wBAAwB,KAAK,QAAQ;YAClD,OAAO,KAAK,CAAC,8BAA8B,KAAK,QAAQ,EACxD,CAAC;YACD,OAAO;QACT,CAAC;QACD,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,KAAK,CAAC,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,wBAAwB,CAAC,CAAC;YACxE,MAAM,MAAM,GAAG,KAAK,CAAC,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;YACzE,MAAM,UAAU,GAAG,MAAM,WAAW,CAAC,QAAQ,CAAC,eAAe,EAAE;gBAC7D,SAAS,EAAE,IAAI,CAAC,QAAQ;aACzB,CAAC,CAAC;YACH,IAAI,OAAO,GAAG,CAAC,CAAC;YAChB,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;gBACnC,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,OAAO,CACtC,EAAE,SAAS,EAAE,IAAI,CAAC,QAAQ,EAAE,aAAa,EAAE,SAAS,EAAE,EACtD,EAAE,IAAI,EAAE,EAAE,aAAa,EAAE,CAAC,CAAC,EAAE,EAAE,UAAU,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,EAAE,CAClE,CAAC;gBACF,IAAI,CAAC,MAAM;oBAAE,SAAS;gBACtB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC;oBACrC,SAAS,EAAE,IAAI,CAAC,QAAQ;oBACxB,aAAa,EAAE,SAAS;oBACxB,aAAa,EAAE,MAAM,CAAC,eAAe,CAAC;iBACvC,CAAC,CAAC;gBACH,OAAO,IAAI,MAAM,CAAC,YAAY,IAAI,CAAC,CAAC;YACtC,CAAC;YACD,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;gBAChB,MAAM,CAAC,IAAI,CACT,aAAa,OAAO,8DAA8D,IAAI,CAAC,QAAQ,GAAG,CACnG,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,OAAO,GAAY,EAAE,CAAC;YACtB,MAAM,CAAC,IAAI,CACT,4DAA4D,IAAI,CAAC,QAAQ,KAAK;gBAC5E,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CACxD,CAAC;QACJ,CAAC;IACH,CAAC;CACF"}

package/dist/index.d.ts

@@ -0,0 +1,19 @@
+/**
+ * LangChain SDK for Atlas Agent Engine.
+ *
+ * Port of `agent_engine_sdk_langgraph/__init__.py`.
+ */
+export { LangGraphBaseAgent } from "./agent.js";
+export type { PrepareAgentInput, ResolveThreadId } from "./agent.js";
+export { LangGraphQueryPlugin } from "./query.js";
+export type { LangGraphQueryPluginParams } from "./query.js";
+export { App } from "./runtime.js";
+export type { DeepAgentOptions } from "./runtime.js";
+export type { SessionFinishStatus } from "@mongodb-js/agent-engine-runner-shared";
+export { getCallAbortSignal } from "@mongodb-js/agent-engine-runner-shared";
+export { withCallInterruptSupport } from "./call_interrupt.js";
+export { createAgentEngineDeepAgent, type CreateAgentEngineDeepAgentOptions, } from "./deep_agent.js";
+export { validateSubagentTree, MAX_SUBAGENT_NESTING_DEPTH, } from "./subagents.js";
+export { AgentEngineToolPodBackend } from "./backends/toolpod.js";
+export { AgentEngineToolSandboxBackend } from "./backends/tool_sandbox.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAGA;;;;GAIG;AAEH,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAChD,YAAY,EAAE,iBAAiB,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AACrE,OAAO,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAClD,YAAY,EAAE,0BAA0B,EAAE,MAAM,YAAY,CAAC;AAC7D,OAAO,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AACnC,YAAY,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AACrD,YAAY,EAAE,mBAAmB,EAAE,MAAM,wCAAwC,CAAC;AAClF,OAAO,EAAE,kBAAkB,EAAE,MAAM,wCAAwC,CAAC;AAC5E,OAAO,EAAE,wBAAwB,EAAE,MAAM,qBAAqB,CAAC;AAC/D,OAAO,EACL,0BAA0B,EAC1B,KAAK,iCAAiC,GACvC,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACL,oBAAoB,EACpB,0BAA0B,GAC3B,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,yBAAyB,EAAE,MAAM,uBAAuB,CAAC;AAClE,OAAO,EAAE,6BAA6B,EAAE,MAAM,4BAA4B,CAAC"}

package/dist/index.js

@@ -0,0 +1,17 @@
+// Copyright 2026 MongoDB, Inc.
+// SPDX-License-Identifier: Apache-2.0
+/**
+ * LangChain SDK for Atlas Agent Engine.
+ *
+ * Port of `agent_engine_sdk_langgraph/__init__.py`.
+ */
+export { LangGraphBaseAgent } from "./agent.js";
+export { LangGraphQueryPlugin } from "./query.js";
+export { App } from "./runtime.js";
+export { getCallAbortSignal } from "@mongodb-js/agent-engine-runner-shared";
+export { withCallInterruptSupport } from "./call_interrupt.js";
+export { createAgentEngineDeepAgent, } from "./deep_agent.js";
+export { validateSubagentTree, MAX_SUBAGENT_NESTING_DEPTH, } from "./subagents.js";
+export { AgentEngineToolPodBackend } from "./backends/toolpod.js";
+export { AgentEngineToolSandboxBackend } from "./backends/tool_sandbox.js";
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,+BAA+B;AAC/B,sCAAsC;AAEtC;;;;GAIG;AAEH,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAEhD,OAAO,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAElD,OAAO,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AAGnC,OAAO,EAAE,kBAAkB,EAAE,MAAM,wCAAwC,CAAC;AAC5E,OAAO,EAAE,wBAAwB,EAAE,MAAM,qBAAqB,CAAC;AAC/D,OAAO,EACL,0BAA0B,GAE3B,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACL,oBAAoB,EACpB,0BAA0B,GAC3B,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,yBAAyB,EAAE,MAAM,uBAAuB,CAAC;AAClE,OAAO,EAAE,6BAA6B,EAAE,MAAM,4BAA4B,CAAC"}

package/dist/llm_adapter.d.ts

@@ -0,0 +1,41 @@
+/**
+ * Lightweight adapter wrapping a LangChain `BaseChatModel` as `BaseLLM`.
+ *
+ * Used by the Tool Pod (`server/tool.ts`) to invoke LLMs through the
+ * framework-neutral `BaseLLM` protocol while the underlying provider is
+ * still a LangChain ChatModel.
+ *
+ * Eventually runtime `createLlm` will return `BaseLLM` directly,
+ * and this adapter will only be used internally by agent-engine-sdk-langgraph.
+ *
+ * Port of `agent_engine_sdk_langgraph/llm_adapter.py`.
+ *
+ * **Sync vs async note:** Python's `BaseLLM` exposes both `invoke` (sync) and
+ * `ainvoke` (async). JavaScript has no synchronous network I/O, so the TS
+ * port collapses both into a single async `invoke`.
+ */
+import type { BaseMessage, BaseMessageChunk } from "@langchain/core/messages";
+import { type BaseLLM, type JsonValue, type LLMResponse, type LLMStreamChunk, type LLMToolSchema, type Message } from "@mongodb-js/agent-engine-sdk";
+/**
+ * Structural contract for the LangChain chat model wrapped by the adapter.
+ *
+ * Avoids tying the public adapter to a specific LangChain class — any object
+ * with `invoke`/`stream`/`bindTools` (or a subset of those) will satisfy it,
+ * which keeps tests trivial to write.
+ */
+export interface LangChainChatModelLike {
+    invoke(input: readonly BaseMessage[], config?: Record<string, unknown>): Promise<BaseMessage>;
+    stream?(input: readonly BaseMessage[], config?: Record<string, unknown>): AsyncIterable<BaseMessageChunk> | Promise<AsyncIterable<BaseMessageChunk>>;
+    bindTools?(tools: ReadonlyArray<Record<string, JsonValue>>, kwargs?: {
+        tool_choice?: unknown;
+    }): LangChainChatModelLike;
+}
+export declare class LangChainLLMAdapter implements BaseLLM {
+    private readonly llm;
+    constructor(llm: LangChainChatModelLike, tools?: readonly LLMToolSchema[], toolChoice?: unknown);
+    /** LLM invocation. */
+    invoke(messages: readonly Message[], kwargs?: Readonly<Record<string, unknown>>): Promise<LLMResponse>;
+    /** Streaming LLM invocation. Matches `BaseLLM.stream` from agent-engine-sdk. */
+    stream(messages: readonly Message[], kwargs?: Readonly<Record<string, unknown>>): AsyncIterable<LLMStreamChunk>;
+}
+//# sourceMappingURL=llm_adapter.d.ts.map

package/dist/llm_adapter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm_adapter.d.ts","sourceRoot":"","sources":["../src/llm_adapter.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAE9E,OAAO,EACL,KAAK,OAAO,EACZ,KAAK,SAAS,EACd,KAAK,WAAW,EAChB,KAAK,cAAc,EAEnB,KAAK,aAAa,EAClB,KAAK,OAAO,EAEb,MAAM,8BAA8B,CAAC;AActC;;;;;;GAMG;AACH,MAAM,WAAW,sBAAsB;IACrC,MAAM,CACJ,KAAK,EAAE,SAAS,WAAW,EAAE,EAC7B,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC/B,OAAO,CAAC,WAAW,CAAC,CAAC;IACxB,MAAM,CAAC,CACL,KAAK,EAAE,SAAS,WAAW,EAAE,EAC7B,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC/B,aAAa,CAAC,gBAAgB,CAAC,GAAG,OAAO,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC,CAAC;IAC9E,SAAS,CAAC,CACR,KAAK,EAAE,aAAa,CAAC,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,EAC/C,MAAM,CAAC,EAAE;QAAE,WAAW,CAAC,EAAE,OAAO,CAAA;KAAE,GACjC,sBAAsB,CAAC;CAC3B;AAMD,qBAAa,mBAAoB,YAAW,OAAO;IACjD,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAyB;gBAG3C,GAAG,EAAE,sBAAsB,EAC3B,KAAK,CAAC,EAAE,SAAS,aAAa,EAAE,EAChC,UAAU,CAAC,EAAE,OAAO;IA0BtB,sBAAsB;IAChB,MAAM,CACV,QAAQ,EAAE,SAAS,OAAO,EAAE,EAC5B,MAAM,GAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAM,GAC7C,OAAO,CAAC,WAAW,CAAC;IAMvB,gFAAgF;IACzE,MAAM,CACX,QAAQ,EAAE,SAAS,OAAO,EAAE,EAC5B,MAAM,GAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAM,GAC7C,aAAa,CAAC,cAAc,CAAC;CAqBjC"}

package/dist/llm_adapter.js

@@ -0,0 +1,177 @@
+/**
+ * Lightweight adapter wrapping a LangChain `BaseChatModel` as `BaseLLM`.
+ *
+ * Used by the Tool Pod (`server/tool.ts`) to invoke LLMs through the
+ * framework-neutral `BaseLLM` protocol while the underlying provider is
+ * still a LangChain ChatModel.
+ *
+ * Eventually runtime `createLlm` will return `BaseLLM` directly,
+ * and this adapter will only be used internally by agent-engine-sdk-langgraph.
+ *
+ * Port of `agent_engine_sdk_langgraph/llm_adapter.py`.
+ *
+ * **Sync vs async note:** Python's `BaseLLM` exposes both `invoke` (sync) and
+ * `ainvoke` (async). JavaScript has no synchronous network I/O, so the TS
+ * port collapses both into a single async `invoke`.
+ */
+import {} from "@mongodb-js/agent-engine-sdk";
+import { LLMResult, accumulateStreamUsage, jsonSafeMetadata, normalizeContent, normalizeToolCallArgs, } from "@mongodb-js/agent-engine-runner-shared";
+import { platformMessagesToLc } from "./messages.js";
+// ---------------------------------------------------------------------------
+// LangChainLLMAdapter
+// ---------------------------------------------------------------------------
+export class LangChainLLMAdapter {
+    llm;
+    constructor(llm, tools, toolChoice) {
+        const serializedTools = tools && tools.length > 0
+            ? tools.map((tool) => tool.toLangchainDict())
+            : undefined;
+        if (serializedTools) {
+            if (typeof llm.bindTools !== "function") {
+                // Match Python's behavior: raise loudly when tools are supplied but
+                // the underlying model has no `bind_tools` / `bindTools` method.
+                throw new Error("LLM does not support tool binding (missing `bindTools` method).");
+            }
+            // tool_choice (e.g. a function name from withStructuredOutput) is the
+            // force point: LangChain translates it against the bound tools to
+            // require the call. A choice without tools cannot bind to anything.
+            this.llm = llm.bindTools(serializedTools, toolChoice != null ? { tool_choice: toolChoice } : undefined);
+        }
+        else {
+            this.llm = llm;
+        }
+    }
+    /** LLM invocation. */
+    async invoke(messages, kwargs = {}) {
+        const lcMessages = platformMessagesToLc(messages);
+        const response = await this.llm.invoke(lcMessages, kwargs);
+        return responseToLlmResponse(response);
+    }
+    /** Streaming LLM invocation. Matches `BaseLLM.stream` from agent-engine-sdk. */
+    async *stream(messages, kwargs = {}) {
+        const lcMessages = platformMessagesToLc(messages);
+        if (typeof this.llm.stream !== "function") {
+            throw new Error("Underlying LLM does not support streaming (no `stream` method).");
+        }
+        const streamResult = this.llm.stream(lcMessages, kwargs);
+        // LangChain.js streams are AsyncIterable directly; some adapters return a Promise<AsyncIterable>.
+        const stream = isPromise(streamResult) ? await streamResult : streamResult;
+        let running;
+        for await (const chunk of stream) {
+            const streamChunk = chunkToLlmStreamChunk(chunk);
+            running = accumulateStreamUsage(running, streamChunk.usage);
+            yield running !== undefined
+                ? { ...streamChunk, usage: running }
+                : streamChunk;
+        }
+    }
+}
+// ---------------------------------------------------------------------------
+// Conversion helpers
+// ---------------------------------------------------------------------------
+/** Convert a LangChain response to sdk-core `LLMResponse`. */
+function responseToLlmResponse(response) {
+    return LLMResult.fromResponse(response).toResponse();
+}
+function optionalStrAttr(message, attr) {
+    const value = message[attr];
+    return typeof value === "string" ? value : undefined;
+}
+function optionalJsonDictAttr(message, attr) {
+    const value = message[attr];
+    return jsonSafeMetadata(value);
+}
+/** Convert a LangChain stream chunk to sdk-core `LLMStreamChunk`. */
+function chunkToLlmStreamChunk(chunk) {
+    const rawContent = chunk.content;
+    const content = normalizeContent(rawContent);
+    let toolCalls;
+    const rawChunks = chunk["tool_call_chunks"];
+    if (Array.isArray(rawChunks) && rawChunks.length > 0) {
+        toolCalls = [];
+        for (const tc of rawChunks) {
+            const tcDict = coerceToDict(tc);
+            if (tcDict === undefined)
+                continue;
+            toolCalls.push(buildToolCallChunk(tcDict, /* includeIndex */ true));
+        }
+    }
+    // Fall back to merged tool_calls when no tool_call_chunks are present.
+    if (!toolCalls || toolCalls.length === 0) {
+        const rawCalls = chunk["tool_calls"];
+        if (Array.isArray(rawCalls) && rawCalls.length > 0) {
+            toolCalls = [];
+            for (const tc of rawCalls) {
+                if (isPlainObject(tc)) {
+                    toolCalls.push(buildToolCallChunk(tc, 
+                    /* includeIndex */ false));
+                }
+            }
+        }
+    }
+    const usage = LLMResult.extractUsage(chunk);
+    const id = optionalStrAttr(chunk, "id");
+    const name = optionalStrAttr(chunk, "name");
+    const responseMetadata = optionalJsonDictAttr(chunk, "response_metadata");
+    const additionalKwargs = optionalJsonDictAttr(chunk, "additional_kwargs");
+    // Emit camelCase keys to match the sdk-core `LLMStreamChunk` contract.
+    // (Python's LLMStreamChunk uses snake_case fields, but the TS model is
+    // camelCase — the port must translate, not copy Python's key names.)
+    return {
+        ...(content !== "" ? { content } : {}),
+        ...(toolCalls && toolCalls.length > 0 ? { toolCalls } : {}),
+        ...(usage !== undefined ? { usage } : {}),
+        ...(id !== undefined ? { id } : {}),
+        ...(name !== undefined ? { name } : {}),
+        ...(responseMetadata !== undefined ? { responseMetadata } : {}),
+        ...(additionalKwargs !== undefined ? { additionalKwargs } : {}),
+    };
+}
+function buildToolCallChunk(data, includeIndex) {
+    const id = data["id"];
+    const name = data["name"];
+    const type = data["type"];
+    const index = data["index"];
+    const argsRaw = data["args"];
+    const args = argsRaw !== undefined && argsRaw !== null
+        ? normalizeToolCallArgs(argsRaw)
+        : null;
+    return {
+        ...(typeof id === "string" ? { id } : {}),
+        ...(typeof name === "string" ? { name } : {}),
+        ...(args !== null ? { args } : {}),
+        ...(typeof type === "string" ? { type } : {}),
+        ...(includeIndex && typeof index === "number" ? { index } : {}),
+    };
+}
+function coerceToDict(tc) {
+    if (isPlainObject(tc))
+        return tc;
+    // LangChain.js sometimes ships Pydantic-style class instances; honour them
+    // when they expose a `toDict()` / `toJSON()` accessor like the Python original.
+    if (tc !== null && typeof tc === "object") {
+        const obj = tc;
+        if (typeof obj.toDict === "function") {
+            const result = obj.toDict();
+            return isPlainObject(result)
+                ? result
+                : undefined;
+        }
+        if (typeof obj.toJSON === "function") {
+            const result = obj.toJSON();
+            return isPlainObject(result)
+                ? result
+                : undefined;
+        }
+    }
+    return undefined;
+}
+function isPlainObject(value) {
+    return value !== null && typeof value === "object" && !Array.isArray(value);
+}
+function isPromise(value) {
+    return (value !== null &&
+        typeof value === "object" &&
+        typeof value.then === "function");
+}
+//# sourceMappingURL=llm_adapter.js.map