@langchain/langgraph-sdk 1.9.22 → 1.9.24

package/dist/stream/controller.js

@@ -14,7 +14,7 @@ import { collectSubgraphHostNamespaces, getHistoryPage, mapSubagentNamespaces, r
 import { MessageMetadataTracker } from "./message-metadata-tracker.js";
 import { LifecycleLoadingTracker } from "./lifecycle-loading-tracker.js";
 import { RootMessageProjection } from "./root-message-projection.js";
-import { prepareOptimisticInput } from "./optimistic-input.js";
+import { prepareOptimisticInput, serializeUpdateMessages } from "./optimistic-input.js";
 import { EMPTY_QUEUE, SubmitCoordinator } from "./submit-coordinator.js";
 import { reconcileToolCallsFromMessages, seedToolCallsFromMessages, upsertToolCall } from "./tool-calls.js";
 import { v7 } from "@langchain/core/utils/uuid";
@@ -961,17 +961,21 @@ var StreamController = class {
 		} : this.#resolveInterruptForResume();
 		if (resolved == null) throw new Error("No pending interrupt to respond to.");
 		const thread = this.#thread;
+		const prepared = options?.update != null ? this.#beginOptimistic(options.update) : void 0;
+		const dispatchUpdate = this.#resolveDispatchUpdate(options?.update, prepared);
 		try {
 			await this.#submitter.dispatchResume(async () => {
 				await thread.respondInput({
 					namespace: resolved.namespace,
 					interrupt_id: resolved.interruptId,
 					response: normalizeHitlResponseForServer(response),
+					...dispatchUpdate != null ? { update: dispatchUpdate } : {},
+					...options?.goto != null ? { goto: options.goto } : {},
 					config: options?.config,
 					metadata: options?.metadata
 				});
 				this.#markInterruptResolvedInRootStore(resolved.interruptId);
-			});
+			}, prepared?.handle);
 		} catch (error) {
 			if (this.#disposed && isAbortLikeError(error)) return;
 			throw error;
@@ -1029,15 +1033,19 @@ var StreamController = class {
 			response: normalizeHitlResponseForServer(response),
 			namespace: pending.find((entry) => entry.interruptId === interruptId)?.namespace ?? [...ROOT_NAMESPACE]
 		}));
+		const prepared = options?.update != null ? this.#beginOptimistic(options.update) : void 0;
+		const dispatchUpdate = this.#resolveDispatchUpdate(options?.update, prepared);
 		try {
 			await this.#submitter.dispatchResume(async () => {
 				await thread.respondInput({
 					responses,
+					...dispatchUpdate != null ? { update: dispatchUpdate } : {},
+					...options?.goto != null ? { goto: options.goto } : {},
 					config: options?.config,
 					metadata: options?.metadata
 				});
 				for (const { interrupt_id: interruptId } of responses) this.#markInterruptResolvedInRootStore(interruptId);
-			});
+			}, prepared?.handle);
 		} catch (error) {
 			if (this.#disposed && isAbortLikeError(error)) return;
 			throw error;
@@ -1538,7 +1546,7 @@ var StreamController = class {
 			prevValue: currentValues[key]
 		}));
 		this.#sawValuesForRun = false;
-		this.#rootMessages.appendOptimistic(prepared.optimisticMessages, prepared.extraValues);
+		this.#rootMessages.appendOptimistic(prepared.optimisticMessages, prepared.extraValues, { sync: true });
 		if (prepared.echoedIds.length > 0) this.#messageMetadata.markPending(prepared.echoedIds);
 		return {
 			dispatchInput: prepared.dispatchInput,
@@ -1549,6 +1557,24 @@ var StreamController = class {
 		};
 	}
 	/**
+	* Pick the `update` payload to dispatch on a resume (`respond` /
+	* `respondAll`).
+	*
+	* When the optimistic path ran ({@link #beginOptimistic} returned a handle),
+	* its `dispatchInput` already carries the minted message ids the server must
+	* echo back, so dispatch that — the echo reconciles the optimistic messages
+	* by id (no duplicate). Otherwise (optimistic UI disabled, or an `update`
+	* with no echoable messages — e.g. the tuple-entry form) fall back to
+	* serializing `BaseMessage` instances to dicts, exactly as before. Returns
+	* `undefined` when there is no `update`, so the server still sees a plain
+	* resume.
+	*/
+	#resolveDispatchUpdate(update, prepared) {
+		if (prepared != null) return prepared.dispatchInput;
+		if (update == null) return void 0;
+		return serializeUpdateMessages(update, this.#messagesKey);
+	}
+	/**
 	* Reconcile optimistic state when a run terminates.
 	*
 	*   - Messages: any echoed id still `"pending"` (never echoed by the