@minpeter/pss-runtime 0.1.0-next.0 → 0.1.0-next.1

package/dist/session/session.js

@@ -1,43 +1,36 @@
-import { runWithAgentPluginScope } from "../plugins/scope.js";
 import { runAgentLoop } from "../agent-loop.js";
-import { ModelMessageHistory } from "./history.js";
-import { createRuntimeInputStepLifecycle, runPluginAfterTurnHandlers, runPluginBeforeTurnHandlers } from "./lifecycle.js";
-import { addRuntimeInput, closeRuntimeInput, createRuntimeInputState, normalizeAgentInput, shiftRuntimeInput } from "./runtime-input.js";
-import { SessionOverlayState } from "./overlay.js";
+import { normalizeAgentInput } from "./input-normalization.js";
 import { BufferedAgentRun } from "./run.js";
-import { decodeStoredSessionSnapshot, encodeSessionSnapshot } from "./snapshot.js";
+import { addSteeringInput, closeRuntimeInput, createRuntimeInputState, hooksForRuntimeInput, withRuntimeInputWindow, withSteeringPlacement } from "./runtime-input.js";
+import { errorMessage, runAfterTurnHook, sessionKilledError, sessionTerminalError } from "./session-errors.js";
+import { closeKilledRuntimeInputs } from "./session-kill.js";
+import { drainRuntimeInput } from "./session-runtime-drain.js";
+import { SessionState } from "./session-state.js";
+import { emitTurnErrorAfterRecovery } from "./session-turn-error.js";
 //#region src/session/session.ts
 var AgentSession = class {
+	#hooks;
 	#inputQueue = [];
-	#internalLlm;
 	#llm;
-	#onPluginError;
-	#persistence;
-	#plugins;
+	#pendingRuntimeInputs = [];
+	#state;
 	#activeAbort;
 	#activeRun;
 	#activeRuntimeInput;
-	#history = new ModelMessageHistory();
+	#deletePromise;
 	#killed = false;
-	#loadPromise;
-	#loaded = false;
-	#pluginState = {};
-	#compactions = [];
-	#overlays = new SessionOverlayState();
 	#running = false;
-	#storeVersion;
-	constructor(llm, persistence, plugins, internalLlm = llm, onPluginError) {
-		this.#internalLlm = internalLlm;
-		this.#onPluginError = onPluginError;
-		this.#persistence = persistence;
-		this.#plugins = plugins;
+	#runToCloseOnKill;
+	constructor(llm, persistence, hooks) {
+		this.#hooks = hooks;
 		this.#llm = llm;
+		this.#state = new SessionState(persistence);
 	}
 	async send(input) {
-		if (this.#killed) throw sessionKilledError();
-		await this.#ensureLoaded();
-		if (this.#killed) throw sessionKilledError();
-		const runtimeInput = createRuntimeInputState();
+		if (this.#killed || this.#deletePromise) throw sessionTerminalError(this.#killed);
+		await this.#state.ensureLoaded();
+		if (this.#killed || this.#deletePromise) throw sessionTerminalError(this.#killed);
+		const runtimeInput = createRuntimeInputState(this.#pendingRuntimeInputs.splice(0));
 		const acceptedInput = normalizeAgentInput(input);
 		const run = new BufferedAgentRun();
 		run.emit(acceptedInput);
@@ -56,80 +49,68 @@ var AgentSession = class {
 		return run;
 	}
 	async steer(input) {
-		if (this.#killed) throw sessionKilledError();
+		if (this.#killed || this.#deletePromise) throw sessionTerminalError(this.#killed);
 		const runtimeInput = this.#activeRuntimeInput;
 		const run = this.#activeRun;
 		if (!(runtimeInput && run)) return this.send(input);
-		await addRuntimeInput(runtimeInput, input);
-		return run;
-	}
-	async overlay(input) {
-		if (this.#killed) throw sessionKilledError();
-		await this.#ensureLoaded();
-		if (this.#killed) throw sessionKilledError();
-		const activeRun = this.#activeRun;
-		if (activeRun) {
-			const placement = this.#activeRuntimeInput?.steerPlacement ?? this.#activeRuntimeInput?.placement ?? "step-end";
-			const entry = this.#overlays.appendActiveOverlay(input, placement);
-			if (!entry) throw new Error("Active overlay frame is unavailable.");
-			activeRun.emit({
-				input: entry.summary,
-				placement: entry.placement,
-				type: "overlay-accepted"
-			});
-			return activeRun;
-		}
-		const entry = this.#overlays.appendPendingOverlay(input);
-		const run = new BufferedAgentRun();
-		run.emit({
-			input: entry.summary,
-			placement: entry.placement,
-			type: "overlay-accepted"
-		});
-		run.close();
+		await addSteeringInput(runtimeInput, input);
 		return run;
 	}
 	interrupt() {
 		this.#activeAbort?.abort();
 	}
-	kill() {
+	delete() {
+		this.#deletePromise ??= this.#state.delete().then(() => this.kill(), (error) => {
+			this.#deletePromise = void 0;
+			throw error;
+		});
+		return this.#deletePromise;
+	}
+	enqueueRuntimeInput(input, placement = "turn-start") {
 		if (this.#killed) return;
-		this.#killed = true;
-		this.#expireActiveOverlays("kill");
-		this.#activeAbort?.abort();
-		closeRuntimeInput(this.#activeRuntimeInput, sessionKilledError().message);
-		while (this.#inputQueue.length > 0) {
-			const item = this.#inputQueue.shift();
-			closeRuntimeInput(item?.runtimeInput, sessionKilledError().message);
-			item?.run.emit({
-				type: "turn-error",
-				message: sessionKilledError().message
+		const runtimeInput = this.#activeRuntimeInput;
+		if (runtimeInput && !runtimeInput.closedReason) {
+			if (placement === "turn-start" && runtimeInput.placement !== placement) {
+				this.#enqueuePendingRuntimeInput({
+					input,
+					placement
+				});
+				return;
+			}
+			runtimeInput.queue.push({
+				input,
+				placement
 			});
-			item?.run.close(void 0, sessionKilledError().message);
+			return;
 		}
+		this.#enqueuePendingRuntimeInput({
+			input,
+			placement
+		});
 	}
-	async #ensureLoaded() {
-		if (this.#loaded) return;
-		this.#loadPromise ??= this.#loadSessionState();
-		try {
-			await this.#loadPromise;
-		} catch (error) {
-			this.#loadPromise = void 0;
-			throw error;
-		}
+	emitObserverEvent(event) {
+		this.#activeRun?.emit(event);
 	}
-	async #loadSessionState() {
-		if (this.#loaded) return;
-		await this.#replaceWithStoredSession();
-		this.#loaded = true;
+	#enqueuePendingRuntimeInput(input) {
+		const queuedTurn = this.#inputQueue[0];
+		if (input.placement === "turn-start" && queuedTurn) {
+			queuedTurn.runtimeInput.queue.push(input);
+			return;
+		}
+		this.#pendingRuntimeInputs.push(input);
 	}
-	async #replaceWithStoredSession() {
-		const stored = await this.#persistence.store.load(this.#persistence.key);
-		this.#storeVersion = stored?.version;
-		const snapshot = decodeStoredSessionSnapshot(stored);
-		this.#history = new ModelMessageHistory(snapshot.history);
-		this.#pluginState = structuredClone(snapshot.pluginState);
-		this.#compactions = structuredClone(snapshot.compactions);
+	kill() {
+		if (this.#killed) return;
+		this.#killed = true;
+		const killedError = sessionKilledError();
+		this.#pendingRuntimeInputs.length = 0;
+		this.#activeAbort?.abort();
+		closeKilledRuntimeInputs({
+			activeRuntimeInput: this.#activeRuntimeInput,
+			inputQueue: this.#inputQueue,
+			message: killedError.message,
+			runToClose: this.#runToCloseOnKill ?? this.#activeRun
+		});
 	}
 	async #drainInputQueue() {
 		if (this.#running) return;
@@ -146,242 +127,79 @@ var AgentSession = class {
 	async #processQueuedInput({ input, run, runtimeInput }) {
 		const activeAbort = new AbortController();
 		this.#activeAbort = activeAbort;
-		const historySnapshot = this.#history.modelSnapshot();
-		this.#overlays.startTurn(input, historySnapshot);
 		this.#activeRun = run;
 		this.#activeRuntimeInput = runtimeInput;
+		this.#runToCloseOnKill = run;
+		const historySnapshot = this.#state.modelSnapshot();
 		try {
-			await this.#withSteeringPlacement(runtimeInput, "turn-start", async () => {
-				await runPluginBeforeTurnHandlers(this.#pluginLifecycle(), {
+			await withSteeringPlacement(runtimeInput, "turn-start", async () => {
+				await this.#hooks?.beforeTurn?.({
+					history: this.#state.modelSnapshot(),
 					input,
-					runtimeInput,
 					signal: activeAbort.signal
 				});
 			});
-			await this.#withRuntimeInputWindow(runtimeInput, "turn-start", async () => {
+			await withRuntimeInputWindow(runtimeInput, "turn-start", async () => {
 				await run.emitBoundary({ type: "turn-start" });
 			});
-			this.#history.appendUserInput(input);
-			await this.#commitHistory();
-			await this.#drainRuntimeInput(run, runtimeInput, "turn-start");
+			this.#state.appendUserInput(input);
+			await this.#state.commit();
+			await drainRuntimeInput({
+				placement: "turn-start",
+				run,
+				runtimeInput,
+				state: this.#state
+			});
 			const result = await runAgentLoop({
 				emit: async (event) => {
 					if (event.type === "step-start" || event.type === "step-end") {
-						await this.#withRuntimeInputWindow(runtimeInput, event.type, async () => {
+						await withRuntimeInputWindow(runtimeInput, event.type, async () => {
 							await run.emitBoundary(event);
 						});
-						const overlayInputAdded = event.type === "step-end" && this.#overlays.consumeStepEndOverlayInputAdded();
-						const runtimeInputAdded = await this.#drainRuntimeInput(run, runtimeInput, event.type);
-						if (event.type === "step-end") return {
-							overlayInputAdded,
-							runtimeInputAdded
-						};
-						return;
+						const runtimeInputAdded = await drainRuntimeInput({
+							placement: event.type,
+							run,
+							runtimeInput,
+							state: this.#state
+						});
+						return event.type === "step-end" ? { runtimeInputAdded } : void 0;
 					}
 					run.emit(event);
 				},
-				history: this.#history,
-				stepLifecycle: createRuntimeInputStepLifecycle({
-					lifecycle: this.#pluginLifecycle(),
-					runtimeInput,
-					withSteeringPlacement: (placement, callback) => this.#withSteeringPlacement(runtimeInput, placement, callback)
-				}),
-				llm: (context) => this.#invokeLlm(context),
+				history: this.#state.history,
+				hooks: hooksForRuntimeInput(this.#hooks, runtimeInput),
+				llm: this.#llm,
 				signal: activeAbort.signal
 			});
-			await this.#commitHistory();
+			await this.#state.commit();
 			const terminalEvent = result === "aborted" ? "turn-abort" : "turn-end";
 			closeRuntimeInput(runtimeInput, terminalEvent);
-			this.#expireActiveOverlays(terminalEvent);
 			this.#activeRuntimeInput = void 0;
 			this.#activeRun = void 0;
-			if (await runPluginAfterTurnHandlers(this.#pluginLifecycle(), {
+			await runAfterTurnHook(this.#hooks, {
+				history: this.#state.modelSnapshot(),
 				input,
 				result,
 				signal: activeAbort.signal
-			})) await this.#commitHistory();
+			});
 			run.emit({ type: terminalEvent });
 		} catch (error) {
-			if (error instanceof SessionCommitConflictError) {
-				run.emit({
-					type: "turn-error",
-					message: error.message
-				});
-				closeRuntimeInput(runtimeInput, "a session commit conflict");
-				this.#expireActiveOverlays("turn-error");
-				this.#activeAbort = void 0;
-				return;
-			}
-			this.#history.rollback(historySnapshot);
-			try {
-				await this.#commitHistory();
-			} catch (rollbackError) {
-				run.emit({
-					type: "turn-error",
-					message: `${errorMessage(error)}; history rollback persistence failed: ${errorMessage(rollbackError)}`
-				});
-				closeRuntimeInput(runtimeInput, "turn-error");
-				this.#expireActiveOverlays("turn-error");
-				this.#activeAbort = void 0;
-				return;
-			}
-			run.emit({
-				type: "turn-error",
-				message: errorMessage(error)
+			await emitTurnErrorAfterRecovery({
+				error: error instanceof Error ? error : new Error(String(error)),
+				historySnapshot,
+				run,
+				runtimeInput,
+				state: this.#state
 			});
-			closeRuntimeInput(runtimeInput, "turn-error");
-			this.#expireActiveOverlays("turn-error");
 		} finally {
 			closeRuntimeInput(runtimeInput);
 			this.#activeAbort = void 0;
 			this.#activeRun = void 0;
 			this.#activeRuntimeInput = void 0;
-			this.#overlays.resetActiveTurn();
+			this.#runToCloseOnKill = void 0;
 			run.close(void 0, runtimeInput.closedReason);
 		}
 	}
-	async #commitHistory() {
-		const result = await this.#persistence.store.commit(this.#persistence.key, { state: encodeSessionSnapshot({
-			compactions: this.#compactions,
-			history: this.#history.modelSnapshot(),
-			pluginState: this.#pluginState
-		}) }, { expectedVersion: this.#storeVersion ?? null });
-		if (!result.ok) {
-			await this.#replaceWithStoredSession();
-			throw new SessionCommitConflictError(this.#persistence.key);
-		}
-		this.#storeVersion = result.version;
-	}
-	#createPluginScope(signal) {
-		return {
-			eventHandlers: this.#plugins?.eventHandlers,
-			getCompactions: () => structuredClone(this.#compactions),
-			getPluginState: (pluginName) => this.#pluginState[pluginName],
-			history: () => this.#history.modelSnapshot(),
-			sessionKey: this.#persistence.key,
-			overlay: (input) => this.overlay(input),
-			setCompactions: (compactions) => {
-				this.#compactions = structuredClone([...compactions]);
-			},
-			setPluginState: (pluginName, state) => {
-				this.#pluginState = {
-					...this.#pluginState,
-					[pluginName]: structuredClone(state)
-				};
-			},
-			signal,
-			steer: (input) => this.steer(input),
-			summarize: (messages) => this.#summarizeForPlugins(messages, signal)
-		};
-	}
-	#pluginLifecycle() {
-		return {
-			createScope: (signal) => this.#createPluginScope(signal),
-			history: () => this.#history.modelSnapshot(),
-			onPluginError: this.#onPluginError,
-			plugins: this.#plugins,
-			sessionKey: this.#persistence.key,
-			overlaySession: (input) => this.overlay(input),
-			steerCurrentRun: async (runtimeInput, input) => {
-				await addRuntimeInput(runtimeInput, input);
-				if (!this.#activeRun) throw new Error("Agent plugin steering requires an active run.");
-				return this.#activeRun;
-			},
-			steerSession: (input) => this.steer(input)
-		};
-	}
-	#invokeLlm({ history, signal }) {
-		const activeSignal = signal ?? new AbortController().signal;
-		const invoke = async () => {
-			let transformedHistory = history;
-			for (const transform of this.#plugins?.contextTransforms ?? []) transformedHistory = await transform({
-				history: transformedHistory,
-				sessionKey: this.#persistence.key,
-				signal: activeSignal
-			});
-			const modelHistory = this.#overlays.compose(transformedHistory, history);
-			this.#overlays.markInferenceStarted();
-			return this.#llm({
-				history: modelHistory,
-				signal: activeSignal
-			});
-		};
-		return runWithAgentPluginScope(this.#createPluginScope(activeSignal), invoke);
-	}
-	async #summarizeForPlugins(messages, signal) {
-		return outputToText(await this.#internalLlm({
-			history: [{
-				content: "Summarize these earlier session messages for future model context.",
-				role: "system"
-			}, ...messages],
-			signal
-		})) || `Summarized ${messages.length} messages.`;
-	}
-	async #withRuntimeInputWindow(runtimeInput, placement, callback) {
-		const previousSteerPlacement = runtimeInput.steerPlacement;
-		runtimeInput.placement = placement;
-		runtimeInput.steerPlacement = placement;
-		try {
-			return await callback();
-		} finally {
-			runtimeInput.placement = void 0;
-			runtimeInput.steerPlacement = previousSteerPlacement;
-		}
-	}
-	async #withSteeringPlacement(runtimeInput, placement, callback) {
-		const previousSteerPlacement = runtimeInput.steerPlacement;
-		runtimeInput.steerPlacement = placement;
-		try {
-			return await callback();
-		} finally {
-			runtimeInput.steerPlacement = previousSteerPlacement;
-		}
-	}
-	async #drainRuntimeInput(run, runtimeInput, placement) {
-		let added = false;
-		let next = shiftRuntimeInput(runtimeInput, placement);
-		while (next) {
-			added = true;
-			run.emit({
-				type: "runtime-input",
-				input: next.input,
-				placement
-			});
-			this.#history.appendUserInput(next.input);
-			await this.#commitHistory();
-			next = shiftRuntimeInput(runtimeInput, placement);
-		}
-		return added;
-	}
-	#expireActiveOverlays(reason) {
-		const event = this.#overlays.expireActiveFrame(reason);
-		if (!event) return;
-		this.#activeRun?.emit(event);
-	}
-};
-function errorMessage(error) {
-	if (error instanceof Error) return error.message;
-	return String(error);
-}
-function outputToText(output) {
-	const parts = [];
-	for (const message of output) {
-		if (message.role !== "assistant") continue;
-		if (typeof message.content === "string") {
-			parts.push(message.content);
-			continue;
-		}
-		for (const part of message.content) if (part.type === "text") parts.push(part.text);
-	}
-	return parts.join("\n").trim();
-}
-function sessionKilledError() {
-	return /* @__PURE__ */ new Error("Session killed");
-}
-var SessionCommitConflictError = class extends Error {
-	constructor(key) {
-		super(`Session ${JSON.stringify(key)} commit conflict`);
-	}
 };
 //#endregion
 export { AgentSession };

package/dist/session/session.js.map

@@ -1 +1 @@
-{"version":3,"file":"session.js","names":["#inputQueue","#internalLlm","#llm","#onPluginError","#persistence","#plugins","#overlays","#killed","#ensureLoaded","#drainInputQueue","#activeRuntimeInput","#activeRun","#activeAbort","#expireActiveOverlays","#loaded","#loadPromise","#loadSessionState","#replaceWithStoredSession","#storeVersion","#history","#pluginState","#compactions","#running","#processQueuedInput","#withSteeringPlacement","#pluginLifecycle","#withRuntimeInputWindow","#commitHistory","#drainRuntimeInput","#invokeLlm","#summarizeForPlugins","#createPluginScope"],"sources":["../../src/session/session.ts"],"sourcesContent":["import type { ModelMessage } from \"ai\";\nimport { runAgentLoop } from \"../agent-loop\";\nimport type { Llm, LlmOutput } from \"../llm\";\nimport type { ResolvedAgentPlugins } from \"../plugins/runner\";\nimport {\n  type AgentPluginScope,\n  runWithAgentPluginScope,\n} from \"../plugins/scope\";\nimport type { UserInput } from \"./events\";\nimport { ModelMessageHistory } from \"./history\";\nimport type { AgentInput } from \"./input\";\nimport {\n  type AgentPluginErrorHandler,\n  type AgentSessionLifecycle,\n  createRuntimeInputStepLifecycle,\n  runPluginAfterTurnHandlers,\n  runPluginBeforeTurnHandlers,\n} from \"./lifecycle\";\nimport { SessionOverlayState } from \"./overlay\";\nimport type { AgentRun } from \"./run\";\nimport { BufferedAgentRun } from \"./run\";\nimport {\n  addRuntimeInput,\n  closeRuntimeInput,\n  createRuntimeInputState,\n  normalizeAgentInput,\n  type RuntimeInputPlacement,\n  type RuntimeInputState,\n  shiftRuntimeInput,\n} from \"./runtime-input\";\nimport {\n  type AgentCompactionOverlay,\n  decodeStoredSessionSnapshot,\n  encodeSessionSnapshot,\n} from \"./snapshot\";\nimport type { SessionStore } from \"./store/types\";\n\nexport type { AgentInput, SessionInput, UserInput } from \"./input\";\nexport type { AgentRun } from \"./run\";\n\ninterface SessionPersistenceOptions {\n  readonly key: string;\n  readonly store: SessionStore;\n}\n\ninterface QueuedInput {\n  readonly input: UserInput;\n  readonly run: BufferedAgentRun;\n  readonly runtimeInput: RuntimeInputState;\n}\n\nexport class AgentSession {\n  readonly #inputQueue: QueuedInput[] = [];\n  readonly #internalLlm: Llm;\n  readonly #llm: Llm;\n  readonly #onPluginError?: AgentPluginErrorHandler;\n  readonly #persistence: SessionPersistenceOptions;\n  readonly #plugins?: ResolvedAgentPlugins;\n  #activeAbort?: AbortController;\n  #activeRun?: BufferedAgentRun;\n  #activeRuntimeInput?: RuntimeInputState;\n  #history = new ModelMessageHistory();\n  #killed = false;\n  #loadPromise?: Promise<void>;\n  #loaded = false;\n  #pluginState: Record<string, unknown> = {};\n  #compactions: AgentCompactionOverlay[] = [];\n  readonly #overlays = new SessionOverlayState();\n  #running = false;\n  #storeVersion: string | undefined;\n\n  constructor(\n    llm: Llm,\n    persistence: SessionPersistenceOptions,\n    plugins?: ResolvedAgentPlugins,\n    internalLlm: Llm = llm,\n    onPluginError?: AgentPluginErrorHandler\n  ) {\n    this.#internalLlm = internalLlm;\n    this.#onPluginError = onPluginError;\n    this.#persistence = persistence;\n    this.#plugins = plugins;\n    this.#llm = llm;\n  }\n\n  async send(input: AgentInput): Promise<AgentRun> {\n    if (this.#killed) {\n      throw sessionKilledError();\n    }\n\n    await this.#ensureLoaded();\n\n    if (this.#killed) {\n      throw sessionKilledError();\n    }\n\n    const runtimeInput = createRuntimeInputState();\n    const acceptedInput = normalizeAgentInput(input);\n    const run = new BufferedAgentRun();\n    run.emit(acceptedInput);\n    this.#inputQueue.push({\n      input: structuredClone(acceptedInput),\n      run,\n      runtimeInput,\n    });\n    this.#drainInputQueue().catch((error: unknown) => {\n      run.emit({ type: \"turn-error\", message: errorMessage(error) });\n      run.close();\n    });\n    return run;\n  }\n\n  async steer(input: AgentInput): Promise<AgentRun> {\n    if (this.#killed) {\n      throw sessionKilledError();\n    }\n\n    const runtimeInput = this.#activeRuntimeInput;\n    const run = this.#activeRun;\n    if (!(runtimeInput && run)) {\n      return this.send(input);\n    }\n\n    await addRuntimeInput(runtimeInput, input);\n    return run;\n  }\n\n  async overlay(input: AgentInput): Promise<AgentRun> {\n    if (this.#killed) {\n      throw sessionKilledError();\n    }\n\n    await this.#ensureLoaded();\n\n    if (this.#killed) {\n      throw sessionKilledError();\n    }\n\n    const activeRun = this.#activeRun;\n    if (activeRun) {\n      const placement =\n        this.#activeRuntimeInput?.steerPlacement ??\n        this.#activeRuntimeInput?.placement ??\n        \"step-end\";\n      const entry = this.#overlays.appendActiveOverlay(input, placement);\n      if (!entry) {\n        throw new Error(\"Active overlay frame is unavailable.\");\n      }\n      activeRun.emit({\n        input: entry.summary,\n        placement: entry.placement,\n        type: \"overlay-accepted\",\n      });\n      return activeRun;\n    }\n\n    const entry = this.#overlays.appendPendingOverlay(input);\n    const run = new BufferedAgentRun();\n    run.emit({\n      input: entry.summary,\n      placement: entry.placement,\n      type: \"overlay-accepted\",\n    });\n    run.close();\n    return run;\n  }\n\n  interrupt(): void {\n    this.#activeAbort?.abort();\n  }\n\n  kill(): void {\n    if (this.#killed) {\n      return;\n    }\n\n    this.#killed = true;\n    this.#expireActiveOverlays(\"kill\");\n    this.#activeAbort?.abort();\n    closeRuntimeInput(this.#activeRuntimeInput, sessionKilledError().message);\n\n    while (this.#inputQueue.length > 0) {\n      const item = this.#inputQueue.shift();\n      closeRuntimeInput(item?.runtimeInput, sessionKilledError().message);\n      item?.run.emit({\n        type: \"turn-error\",\n        message: sessionKilledError().message,\n      });\n      item?.run.close(undefined, sessionKilledError().message);\n    }\n  }\n\n  async #ensureLoaded(): Promise<void> {\n    if (this.#loaded) {\n      return;\n    }\n\n    this.#loadPromise ??= this.#loadSessionState();\n    try {\n      await this.#loadPromise;\n    } catch (error) {\n      this.#loadPromise = undefined;\n      throw error;\n    }\n  }\n\n  async #loadSessionState(): Promise<void> {\n    if (this.#loaded) {\n      return;\n    }\n\n    await this.#replaceWithStoredSession();\n    this.#loaded = true;\n  }\n\n  async #replaceWithStoredSession(): Promise<void> {\n    const stored = await this.#persistence.store.load(this.#persistence.key);\n    this.#storeVersion = stored?.version;\n    const snapshot = decodeStoredSessionSnapshot(stored);\n    this.#history = new ModelMessageHistory(snapshot.history);\n    this.#pluginState = structuredClone(snapshot.pluginState);\n    this.#compactions = structuredClone(snapshot.compactions);\n  }\n\n  async #drainInputQueue(): Promise<void> {\n    if (this.#running) {\n      return;\n    }\n\n    this.#running = true;\n    try {\n      while (!this.#killed && this.#inputQueue.length > 0) {\n        const item = this.#inputQueue.shift();\n        if (item) {\n          await this.#processQueuedInput(item);\n        }\n      }\n    } finally {\n      this.#running = false;\n    }\n  }\n\n  async #processQueuedInput({\n    input,\n    run,\n    runtimeInput,\n  }: QueuedInput): Promise<void> {\n    const activeAbort = new AbortController();\n    this.#activeAbort = activeAbort;\n    const historySnapshot = this.#history.modelSnapshot();\n    this.#overlays.startTurn(input, historySnapshot);\n    this.#activeRun = run;\n    this.#activeRuntimeInput = runtimeInput;\n\n    try {\n      await this.#withSteeringPlacement(\n        runtimeInput,\n        \"turn-start\",\n        async () => {\n          await runPluginBeforeTurnHandlers(this.#pluginLifecycle(), {\n            input,\n            runtimeInput,\n            signal: activeAbort.signal,\n          });\n        }\n      );\n      await this.#withRuntimeInputWindow(\n        runtimeInput,\n        \"turn-start\",\n        async () => {\n          await run.emitBoundary({ type: \"turn-start\" });\n        }\n      );\n      this.#history.appendUserInput(input);\n      await this.#commitHistory();\n      await this.#drainRuntimeInput(run, runtimeInput, \"turn-start\");\n\n      const result = await runAgentLoop({\n        emit: async (event) => {\n          if (event.type === \"step-start\" || event.type === \"step-end\") {\n            await this.#withRuntimeInputWindow(\n              runtimeInput,\n              event.type,\n              async () => {\n                await run.emitBoundary(event);\n              }\n            );\n            const overlayInputAdded =\n              event.type === \"step-end\" &&\n              this.#overlays.consumeStepEndOverlayInputAdded();\n            const runtimeInputAdded = await this.#drainRuntimeInput(\n              run,\n              runtimeInput,\n              event.type\n            );\n\n            if (event.type === \"step-end\") {\n              return {\n                overlayInputAdded,\n                runtimeInputAdded,\n              };\n            }\n            return;\n          }\n\n          run.emit(event);\n        },\n        history: this.#history,\n        stepLifecycle: createRuntimeInputStepLifecycle({\n          lifecycle: this.#pluginLifecycle(),\n          runtimeInput,\n          withSteeringPlacement: (placement, callback) =>\n            this.#withSteeringPlacement(runtimeInput, placement, callback),\n        }),\n        llm: (context) => this.#invokeLlm(context),\n        signal: activeAbort.signal,\n      });\n\n      await this.#commitHistory();\n      const terminalEvent = result === \"aborted\" ? \"turn-abort\" : \"turn-end\";\n      closeRuntimeInput(runtimeInput, terminalEvent);\n      this.#expireActiveOverlays(terminalEvent);\n      this.#activeRuntimeInput = undefined;\n      this.#activeRun = undefined;\n      const pluginHandlersRan = await runPluginAfterTurnHandlers(\n        this.#pluginLifecycle(),\n        {\n          input,\n          result,\n          signal: activeAbort.signal,\n        }\n      );\n      if (pluginHandlersRan) {\n        await this.#commitHistory();\n      }\n      run.emit({ type: terminalEvent });\n    } catch (error) {\n      if (error instanceof SessionCommitConflictError) {\n        run.emit({ type: \"turn-error\", message: error.message });\n        closeRuntimeInput(runtimeInput, \"a session commit conflict\");\n        this.#expireActiveOverlays(\"turn-error\");\n        this.#activeAbort = undefined;\n        return;\n      }\n\n      this.#history.rollback(historySnapshot);\n      try {\n        await this.#commitHistory();\n      } catch (rollbackError) {\n        run.emit({\n          type: \"turn-error\",\n          message: `${errorMessage(error)}; history rollback persistence failed: ${errorMessage(\n            rollbackError\n          )}`,\n        });\n        closeRuntimeInput(runtimeInput, \"turn-error\");\n        this.#expireActiveOverlays(\"turn-error\");\n        this.#activeAbort = undefined;\n        return;\n      }\n      run.emit({ type: \"turn-error\", message: errorMessage(error) });\n      closeRuntimeInput(runtimeInput, \"turn-error\");\n      this.#expireActiveOverlays(\"turn-error\");\n    } finally {\n      closeRuntimeInput(runtimeInput);\n      this.#activeAbort = undefined;\n      this.#activeRun = undefined;\n      this.#activeRuntimeInput = undefined;\n      this.#overlays.resetActiveTurn();\n      run.close(undefined, runtimeInput.closedReason);\n    }\n  }\n\n  async #commitHistory(): Promise<void> {\n    const result = await this.#persistence.store.commit(\n      this.#persistence.key,\n      {\n        state: encodeSessionSnapshot({\n          compactions: this.#compactions,\n          history: this.#history.modelSnapshot(),\n          pluginState: this.#pluginState,\n        }),\n      },\n      { expectedVersion: this.#storeVersion ?? null }\n    );\n\n    if (!result.ok) {\n      await this.#replaceWithStoredSession();\n      throw new SessionCommitConflictError(this.#persistence.key);\n    }\n\n    this.#storeVersion = result.version;\n  }\n\n  #createPluginScope(signal: AbortSignal): AgentPluginScope {\n    return {\n      eventHandlers: this.#plugins?.eventHandlers,\n      getCompactions: () => structuredClone(this.#compactions),\n      getPluginState: (pluginName) => this.#pluginState[pluginName],\n      history: () => this.#history.modelSnapshot(),\n      sessionKey: this.#persistence.key,\n      overlay: (input) => this.overlay(input),\n      setCompactions: (compactions) => {\n        this.#compactions = structuredClone([...compactions]);\n      },\n      setPluginState: (pluginName, state) => {\n        this.#pluginState = {\n          ...this.#pluginState,\n          [pluginName]: structuredClone(state),\n        };\n      },\n      signal,\n      steer: (input) => this.steer(input),\n      summarize: (messages) => this.#summarizeForPlugins(messages, signal),\n    };\n  }\n\n  #pluginLifecycle(): AgentSessionLifecycle {\n    return {\n      createScope: (signal) => this.#createPluginScope(signal),\n      history: () => this.#history.modelSnapshot(),\n      onPluginError: this.#onPluginError,\n      plugins: this.#plugins,\n      sessionKey: this.#persistence.key,\n      overlaySession: (input) => this.overlay(input),\n      steerCurrentRun: async (runtimeInput, input) => {\n        await addRuntimeInput(runtimeInput, input);\n        if (!this.#activeRun) {\n          throw new Error(\"Agent plugin steering requires an active run.\");\n        }\n        return this.#activeRun;\n      },\n      steerSession: (input) => this.steer(input),\n    };\n  }\n\n  #invokeLlm({\n    history,\n    signal,\n  }: {\n    readonly history: readonly ModelMessage[];\n    readonly signal?: AbortSignal;\n  }): Promise<LlmOutput> {\n    const activeSignal = signal ?? new AbortController().signal;\n    const invoke = async () => {\n      let transformedHistory = history;\n      for (const transform of this.#plugins?.contextTransforms ?? []) {\n        transformedHistory = await transform({\n          history: transformedHistory,\n          sessionKey: this.#persistence.key,\n          signal: activeSignal,\n        });\n      }\n\n      const modelHistory = this.#overlays.compose(transformedHistory, history);\n      this.#overlays.markInferenceStarted();\n      return this.#llm({\n        history: modelHistory,\n        signal: activeSignal,\n      });\n    };\n\n    return runWithAgentPluginScope(\n      this.#createPluginScope(activeSignal),\n      invoke\n    );\n  }\n\n  async #summarizeForPlugins(\n    messages: readonly ModelMessage[],\n    signal: AbortSignal\n  ): Promise<string> {\n    const output = await this.#internalLlm({\n      history: [\n        {\n          content:\n            \"Summarize these earlier session messages for future model context.\",\n          role: \"system\",\n        },\n        ...messages,\n      ],\n      signal,\n    });\n    return outputToText(output) || `Summarized ${messages.length} messages.`;\n  }\n\n  async #withRuntimeInputWindow<T>(\n    runtimeInput: RuntimeInputState,\n    placement: RuntimeInputPlacement,\n    callback: () => Promise<T>\n  ): Promise<T> {\n    const previousSteerPlacement = runtimeInput.steerPlacement;\n    runtimeInput.placement = placement;\n    runtimeInput.steerPlacement = placement;\n    try {\n      return await callback();\n    } finally {\n      runtimeInput.placement = undefined;\n      runtimeInput.steerPlacement = previousSteerPlacement;\n    }\n  }\n\n  async #withSteeringPlacement<T>(\n    runtimeInput: RuntimeInputState,\n    placement: RuntimeInputPlacement,\n    callback: () => Promise<T>\n  ): Promise<T> {\n    const previousSteerPlacement = runtimeInput.steerPlacement;\n    runtimeInput.steerPlacement = placement;\n    try {\n      return await callback();\n    } finally {\n      runtimeInput.steerPlacement = previousSteerPlacement;\n    }\n  }\n\n  async #drainRuntimeInput(\n    run: BufferedAgentRun,\n    runtimeInput: RuntimeInputState,\n    placement: RuntimeInputPlacement\n  ): Promise<boolean> {\n    let added = false;\n    let next = shiftRuntimeInput(runtimeInput, placement);\n    while (next) {\n      added = true;\n      run.emit({ type: \"runtime-input\", input: next.input, placement });\n      this.#history.appendUserInput(next.input);\n      await this.#commitHistory();\n      next = shiftRuntimeInput(runtimeInput, placement);\n    }\n\n    return added;\n  }\n\n  #expireActiveOverlays(\n    reason: \"kill\" | \"turn-abort\" | \"turn-end\" | \"turn-error\"\n  ): void {\n    const event = this.#overlays.expireActiveFrame(reason);\n    if (!event) {\n      return;\n    }\n\n    this.#activeRun?.emit(event);\n  }\n}\n\nfunction errorMessage(error: unknown): string {\n  if (error instanceof Error) {\n    return error.message;\n  }\n\n  return String(error);\n}\n\nfunction outputToText(output: LlmOutput): string {\n  const parts: string[] = [];\n  for (const message of output) {\n    if (message.role !== \"assistant\") {\n      continue;\n    }\n\n    if (typeof message.content === \"string\") {\n      parts.push(message.content);\n      continue;\n    }\n\n    for (const part of message.content) {\n      if (part.type === \"text\") {\n        parts.push(part.text);\n      }\n    }\n  }\n\n  return parts.join(\"\\n\").trim();\n}\n\nfunction sessionKilledError(): Error {\n  return new Error(\"Session killed\");\n}\n\nclass SessionCommitConflictError extends Error {\n  constructor(key: string) {\n    super(`Session ${JSON.stringify(key)} commit conflict`);\n  }\n}\n"],"mappings":";;;;;;;;;AAmDA,IAAa,eAAb,MAA0B;CACxB,cAAsC,CAAC;CACvC;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA,WAAW,IAAI,oBAAoB;CACnC,UAAU;CACV;CACA,UAAU;CACV,eAAwC,CAAC;CACzC,eAAyC,CAAC;CAC1C,YAAqB,IAAI,oBAAoB;CAC7C,WAAW;CACX;CAEA,YACE,KACA,aACA,SACA,cAAmB,KACnB,eACA;EACA,KAAKC,eAAe;EACpB,KAAKE,iBAAiB;EACtB,KAAKC,eAAe;EACpB,KAAKC,WAAW;EAChB,KAAKH,OAAO;CACd;CAEA,MAAM,KAAK,OAAsC;EAC/C,IAAI,KAAKK,SACP,MAAM,mBAAmB;EAG3B,MAAM,KAAKC,cAAc;EAEzB,IAAI,KAAKD,SACP,MAAM,mBAAmB;EAG3B,MAAM,eAAe,wBAAwB;EAC7C,MAAM,gBAAgB,oBAAoB,KAAK;EAC/C,MAAM,MAAM,IAAI,iBAAiB;EACjC,IAAI,KAAK,aAAa;EACtB,KAAKP,YAAY,KAAK;GACpB,OAAO,gBAAgB,aAAa;GACpC;GACA;EACF,CAAC;EACD,KAAKS,iBAAiB,EAAE,OAAO,UAAmB;GAChD,IAAI,KAAK;IAAE,MAAM;IAAc,SAAS,aAAa,KAAK;GAAE,CAAC;GAC7D,IAAI,MAAM;EACZ,CAAC;EACD,OAAO;CACT;CAEA,MAAM,MAAM,OAAsC;EAChD,IAAI,KAAKF,SACP,MAAM,mBAAmB;EAG3B,MAAM,eAAe,KAAKG;EAC1B,MAAM,MAAM,KAAKC;EACjB,IAAI,EAAE,gBAAgB,MACpB,OAAO,KAAK,KAAK,KAAK;EAGxB,MAAM,gBAAgB,cAAc,KAAK;EACzC,OAAO;CACT;CAEA,MAAM,QAAQ,OAAsC;EAClD,IAAI,KAAKJ,SACP,MAAM,mBAAmB;EAG3B,MAAM,KAAKC,cAAc;EAEzB,IAAI,KAAKD,SACP,MAAM,mBAAmB;EAG3B,MAAM,YAAY,KAAKI;EACvB,IAAI,WAAW;GACb,MAAM,YACJ,KAAKD,qBAAqB,kBAC1B,KAAKA,qBAAqB,aAC1B;GACF,MAAM,QAAQ,KAAKJ,UAAU,oBAAoB,OAAO,SAAS;GACjE,IAAI,CAAC,OACH,MAAM,IAAI,MAAM,sCAAsC;GAExD,UAAU,KAAK;IACb,OAAO,MAAM;IACb,WAAW,MAAM;IACjB,MAAM;GACR,CAAC;GACD,OAAO;EACT;EAEA,MAAM,QAAQ,KAAKA,UAAU,qBAAqB,KAAK;EACvD,MAAM,MAAM,IAAI,iBAAiB;EACjC,IAAI,KAAK;GACP,OAAO,MAAM;GACb,WAAW,MAAM;GACjB,MAAM;EACR,CAAC;EACD,IAAI,MAAM;EACV,OAAO;CACT;CAEA,YAAkB;EAChB,KAAKM,cAAc,MAAM;CAC3B;CAEA,OAAa;EACX,IAAI,KAAKL,SACP;EAGF,KAAKA,UAAU;EACf,KAAKM,sBAAsB,MAAM;EACjC,KAAKD,cAAc,MAAM;EACzB,kBAAkB,KAAKF,qBAAqB,mBAAmB,EAAE,OAAO;EAExE,OAAO,KAAKV,YAAY,SAAS,GAAG;GAClC,MAAM,OAAO,KAAKA,YAAY,MAAM;GACpC,kBAAkB,MAAM,cAAc,mBAAmB,EAAE,OAAO;GAClE,MAAM,IAAI,KAAK;IACb,MAAM;IACN,SAAS,mBAAmB,EAAE;GAChC,CAAC;GACD,MAAM,IAAI,MAAM,KAAA,GAAW,mBAAmB,EAAE,OAAO;EACzD;CACF;CAEA,MAAMQ,gBAA+B;EACnC,IAAI,KAAKM,SACP;EAGF,KAAKC,iBAAiB,KAAKC,kBAAkB;EAC7C,IAAI;GACF,MAAM,KAAKD;EACb,SAAS,OAAO;GACd,KAAKA,eAAe,KAAA;GACpB,MAAM;EACR;CACF;CAEA,MAAMC,oBAAmC;EACvC,IAAI,KAAKF,SACP;EAGF,MAAM,KAAKG,0BAA0B;EACrC,KAAKH,UAAU;CACjB;CAEA,MAAMG,4BAA2C;EAC/C,MAAM,SAAS,MAAM,KAAKb,aAAa,MAAM,KAAK,KAAKA,aAAa,GAAG;EACvE,KAAKc,gBAAgB,QAAQ;EAC7B,MAAM,WAAW,4BAA4B,MAAM;EACnD,KAAKC,WAAW,IAAI,oBAAoB,SAAS,OAAO;EACxD,KAAKC,eAAe,gBAAgB,SAAS,WAAW;EACxD,KAAKC,eAAe,gBAAgB,SAAS,WAAW;CAC1D;CAEA,MAAMZ,mBAAkC;EACtC,IAAI,KAAKa,UACP;EAGF,KAAKA,WAAW;EAChB,IAAI;GACF,OAAO,CAAC,KAAKf,WAAW,KAAKP,YAAY,SAAS,GAAG;IACnD,MAAM,OAAO,KAAKA,YAAY,MAAM;IACpC,IAAI,MACF,MAAM,KAAKuB,oBAAoB,IAAI;GAEvC;EACF,UAAU;GACR,KAAKD,WAAW;EAClB;CACF;CAEA,MAAMC,oBAAoB,EACxB,OACA,KACA,gBAC6B;EAC7B,MAAM,cAAc,IAAI,gBAAgB;EACxC,KAAKX,eAAe;EACpB,MAAM,kBAAkB,KAAKO,SAAS,cAAc;EACpD,KAAKb,UAAU,UAAU,OAAO,eAAe;EAC/C,KAAKK,aAAa;EAClB,KAAKD,sBAAsB;EAE3B,IAAI;GACF,MAAM,KAAKc,uBACT,cACA,cACA,YAAY;IACV,MAAM,4BAA4B,KAAKC,iBAAiB,GAAG;KACzD;KACA;KACA,QAAQ,YAAY;IACtB,CAAC;GACH,CACF;GACA,MAAM,KAAKC,wBACT,cACA,cACA,YAAY;IACV,MAAM,IAAI,aAAa,EAAE,MAAM,aAAa,CAAC;GAC/C,CACF;GACA,KAAKP,SAAS,gBAAgB,KAAK;GACnC,MAAM,KAAKQ,eAAe;GAC1B,MAAM,KAAKC,mBAAmB,KAAK,cAAc,YAAY;GAE7D,MAAM,SAAS,MAAM,aAAa;IAChC,MAAM,OAAO,UAAU;KACrB,IAAI,MAAM,SAAS,gBAAgB,MAAM,SAAS,YAAY;MAC5D,MAAM,KAAKF,wBACT,cACA,MAAM,MACN,YAAY;OACV,MAAM,IAAI,aAAa,KAAK;MAC9B,CACF;MACA,MAAM,oBACJ,MAAM,SAAS,cACf,KAAKpB,UAAU,gCAAgC;MACjD,MAAM,oBAAoB,MAAM,KAAKsB,mBACnC,KACA,cACA,MAAM,IACR;MAEA,IAAI,MAAM,SAAS,YACjB,OAAO;OACL;OACA;MACF;MAEF;KACF;KAEA,IAAI,KAAK,KAAK;IAChB;IACA,SAAS,KAAKT;IACd,eAAe,gCAAgC;KAC7C,WAAW,KAAKM,iBAAiB;KACjC;KACA,wBAAwB,WAAW,aACjC,KAAKD,uBAAuB,cAAc,WAAW,QAAQ;IACjE,CAAC;IACD,MAAM,YAAY,KAAKK,WAAW,OAAO;IACzC,QAAQ,YAAY;GACtB,CAAC;GAED,MAAM,KAAKF,eAAe;GAC1B,MAAM,gBAAgB,WAAW,YAAY,eAAe;GAC5D,kBAAkB,cAAc,aAAa;GAC7C,KAAKd,sBAAsB,aAAa;GACxC,KAAKH,sBAAsB,KAAA;GAC3B,KAAKC,aAAa,KAAA;GASlB,IAAI,MAR4B,2BAC9B,KAAKc,iBAAiB,GACtB;IACE;IACA;IACA,QAAQ,YAAY;GACtB,CACF,GAEE,MAAM,KAAKE,eAAe;GAE5B,IAAI,KAAK,EAAE,MAAM,cAAc,CAAC;EAClC,SAAS,OAAO;GACd,IAAI,iBAAiB,4BAA4B;IAC/C,IAAI,KAAK;KAAE,MAAM;KAAc,SAAS,MAAM;IAAQ,CAAC;IACvD,kBAAkB,cAAc,2BAA2B;IAC3D,KAAKd,sBAAsB,YAAY;IACvC,KAAKD,eAAe,KAAA;IACpB;GACF;GAEA,KAAKO,SAAS,SAAS,eAAe;GACtC,IAAI;IACF,MAAM,KAAKQ,eAAe;GAC5B,SAAS,eAAe;IACtB,IAAI,KAAK;KACP,MAAM;KACN,SAAS,GAAG,aAAa,KAAK,EAAE,yCAAyC,aACvE,aACF;IACF,CAAC;IACD,kBAAkB,cAAc,YAAY;IAC5C,KAAKd,sBAAsB,YAAY;IACvC,KAAKD,eAAe,KAAA;IACpB;GACF;GACA,IAAI,KAAK;IAAE,MAAM;IAAc,SAAS,aAAa,KAAK;GAAE,CAAC;GAC7D,kBAAkB,cAAc,YAAY;GAC5C,KAAKC,sBAAsB,YAAY;EACzC,UAAU;GACR,kBAAkB,YAAY;GAC9B,KAAKD,eAAe,KAAA;GACpB,KAAKD,aAAa,KAAA;GAClB,KAAKD,sBAAsB,KAAA;GAC3B,KAAKJ,UAAU,gBAAgB;GAC/B,IAAI,MAAM,KAAA,GAAW,aAAa,YAAY;EAChD;CACF;CAEA,MAAMqB,iBAAgC;EACpC,MAAM,SAAS,MAAM,KAAKvB,aAAa,MAAM,OAC3C,KAAKA,aAAa,KAClB,EACE,OAAO,sBAAsB;GAC3B,aAAa,KAAKiB;GAClB,SAAS,KAAKF,SAAS,cAAc;GACrC,aAAa,KAAKC;EACpB,CAAC,EACH,GACA,EAAE,iBAAiB,KAAKF,iBAAiB,KAAK,CAChD;EAEA,IAAI,CAAC,OAAO,IAAI;GACd,MAAM,KAAKD,0BAA0B;GACrC,MAAM,IAAI,2BAA2B,KAAKb,aAAa,GAAG;EAC5D;EAEA,KAAKc,gBAAgB,OAAO;CAC9B;CAEA,mBAAmB,QAAuC;EACxD,OAAO;GACL,eAAe,KAAKb,UAAU;GAC9B,sBAAsB,gBAAgB,KAAKgB,YAAY;GACvD,iBAAiB,eAAe,KAAKD,aAAa;GAClD,eAAe,KAAKD,SAAS,cAAc;GAC3C,YAAY,KAAKf,aAAa;GAC9B,UAAU,UAAU,KAAK,QAAQ,KAAK;GACtC,iBAAiB,gBAAgB;IAC/B,KAAKiB,eAAe,gBAAgB,CAAC,GAAG,WAAW,CAAC;GACtD;GACA,iBAAiB,YAAY,UAAU;IACrC,KAAKD,eAAe;KAClB,GAAG,KAAKA;MACP,aAAa,gBAAgB,KAAK;IACrC;GACF;GACA;GACA,QAAQ,UAAU,KAAK,MAAM,KAAK;GAClC,YAAY,aAAa,KAAKU,qBAAqB,UAAU,MAAM;EACrE;CACF;CAEA,mBAA0C;EACxC,OAAO;GACL,cAAc,WAAW,KAAKC,mBAAmB,MAAM;GACvD,eAAe,KAAKZ,SAAS,cAAc;GAC3C,eAAe,KAAKhB;GACpB,SAAS,KAAKE;GACd,YAAY,KAAKD,aAAa;GAC9B,iBAAiB,UAAU,KAAK,QAAQ,KAAK;GAC7C,iBAAiB,OAAO,cAAc,UAAU;IAC9C,MAAM,gBAAgB,cAAc,KAAK;IACzC,IAAI,CAAC,KAAKO,YACR,MAAM,IAAI,MAAM,+CAA+C;IAEjE,OAAO,KAAKA;GACd;GACA,eAAe,UAAU,KAAK,MAAM,KAAK;EAC3C;CACF;CAEA,WAAW,EACT,SACA,UAIqB;EACrB,MAAM,eAAe,UAAU,IAAI,gBAAgB,EAAE;EACrD,MAAM,SAAS,YAAY;GACzB,IAAI,qBAAqB;GACzB,KAAK,MAAM,aAAa,KAAKN,UAAU,qBAAqB,CAAC,GAC3D,qBAAqB,MAAM,UAAU;IACnC,SAAS;IACT,YAAY,KAAKD,aAAa;IAC9B,QAAQ;GACV,CAAC;GAGH,MAAM,eAAe,KAAKE,UAAU,QAAQ,oBAAoB,OAAO;GACvE,KAAKA,UAAU,qBAAqB;GACpC,OAAO,KAAKJ,KAAK;IACf,SAAS;IACT,QAAQ;GACV,CAAC;EACH;EAEA,OAAO,wBACL,KAAK6B,mBAAmB,YAAY,GACpC,MACF;CACF;CAEA,MAAMD,qBACJ,UACA,QACiB;EAYjB,OAAO,aAAa,MAXC,KAAK7B,aAAa;GACrC,SAAS,CACP;IACE,SACE;IACF,MAAM;GACR,GACA,GAAG,QACL;GACA;EACF,CAAC,CACyB,KAAK,cAAc,SAAS,OAAO;CAC/D;CAEA,MAAMyB,wBACJ,cACA,WACA,UACY;EACZ,MAAM,yBAAyB,aAAa;EAC5C,aAAa,YAAY;EACzB,aAAa,iBAAiB;EAC9B,IAAI;GACF,OAAO,MAAM,SAAS;EACxB,UAAU;GACR,aAAa,YAAY,KAAA;GACzB,aAAa,iBAAiB;EAChC;CACF;CAEA,MAAMF,uBACJ,cACA,WACA,UACY;EACZ,MAAM,yBAAyB,aAAa;EAC5C,aAAa,iBAAiB;EAC9B,IAAI;GACF,OAAO,MAAM,SAAS;EACxB,UAAU;GACR,aAAa,iBAAiB;EAChC;CACF;CAEA,MAAMI,mBACJ,KACA,cACA,WACkB;EAClB,IAAI,QAAQ;EACZ,IAAI,OAAO,kBAAkB,cAAc,SAAS;EACpD,OAAO,MAAM;GACX,QAAQ;GACR,IAAI,KAAK;IAAE,MAAM;IAAiB,OAAO,KAAK;IAAO;GAAU,CAAC;GAChE,KAAKT,SAAS,gBAAgB,KAAK,KAAK;GACxC,MAAM,KAAKQ,eAAe;GAC1B,OAAO,kBAAkB,cAAc,SAAS;EAClD;EAEA,OAAO;CACT;CAEA,sBACE,QACM;EACN,MAAM,QAAQ,KAAKrB,UAAU,kBAAkB,MAAM;EACrD,IAAI,CAAC,OACH;EAGF,KAAKK,YAAY,KAAK,KAAK;CAC7B;AACF;AAEA,SAAS,aAAa,OAAwB;CAC5C,IAAI,iBAAiB,OACnB,OAAO,MAAM;CAGf,OAAO,OAAO,KAAK;AACrB;AAEA,SAAS,aAAa,QAA2B;CAC/C,MAAM,QAAkB,CAAC;CACzB,KAAK,MAAM,WAAW,QAAQ;EAC5B,IAAI,QAAQ,SAAS,aACnB;EAGF,IAAI,OAAO,QAAQ,YAAY,UAAU;GACvC,MAAM,KAAK,QAAQ,OAAO;GAC1B;EACF;EAEA,KAAK,MAAM,QAAQ,QAAQ,SACzB,IAAI,KAAK,SAAS,QAChB,MAAM,KAAK,KAAK,IAAI;CAG1B;CAEA,OAAO,MAAM,KAAK,IAAI,EAAE,KAAK;AAC/B;AAEA,SAAS,qBAA4B;CACnC,uBAAO,IAAI,MAAM,gBAAgB;AACnC;AAEA,IAAM,6BAAN,cAAyC,MAAM;CAC7C,YAAY,KAAa;EACvB,MAAM,WAAW,KAAK,UAAU,GAAG,EAAE,iBAAiB;CACxD;AACF"}
+{"version":3,"file":"session.js","names":["#hooks","#inputQueue","#llm","#pendingRuntimeInputs","#state","#killed","#deletePromise","#drainInputQueue","#activeRuntimeInput","#activeRun","#activeAbort","#enqueuePendingRuntimeInput","#runToCloseOnKill","#running","#processQueuedInput"],"sources":["../../src/session/session.ts"],"sourcesContent":["import { runAgentLoop } from \"../agent-loop\";\nimport type { AgentHooks } from \"../hooks\";\nimport type { Llm } from \"../llm\";\nimport type { AgentEvent } from \"./events\";\nimport type { AgentInput, UserInput } from \"./input\";\nimport { normalizeAgentInput } from \"./input-normalization\";\nimport { type AgentRun, BufferedAgentRun } from \"./run\";\nimport {\n  addSteeringInput,\n  closeRuntimeInput,\n  createRuntimeInputState,\n  hooksForRuntimeInput,\n  type QueuedInput,\n  type QueuedRuntimeInput,\n  type RuntimeInputPlacement,\n  type RuntimeInputState,\n  withRuntimeInputWindow,\n  withSteeringPlacement,\n} from \"./runtime-input\";\nimport {\n  errorMessage,\n  runAfterTurnHook,\n  sessionKilledError,\n  sessionTerminalError,\n} from \"./session-errors\";\nimport { closeKilledRuntimeInputs } from \"./session-kill\";\nimport { drainRuntimeInput } from \"./session-runtime-drain\";\nimport { type SessionPersistenceOptions, SessionState } from \"./session-state\";\nimport { emitTurnErrorAfterRecovery } from \"./session-turn-error\";\n\nexport type { AgentInput, SessionInput, UserInput } from \"./input\";\nexport type { AgentRun } from \"./run\";\n\nexport class AgentSession {\n  readonly #hooks?: AgentHooks;\n  readonly #inputQueue: QueuedInput[] = [];\n  readonly #llm: Llm;\n  readonly #pendingRuntimeInputs: QueuedRuntimeInput[] = [];\n  readonly #state: SessionState;\n  #activeAbort?: AbortController;\n  #activeRun?: BufferedAgentRun;\n  #activeRuntimeInput?: RuntimeInputState;\n  #deletePromise?: Promise<void>;\n  #killed = false;\n  #running = false;\n  #runToCloseOnKill?: BufferedAgentRun;\n\n  constructor(\n    llm: Llm,\n    persistence: SessionPersistenceOptions,\n    hooks?: AgentHooks\n  ) {\n    this.#hooks = hooks;\n    this.#llm = llm;\n    this.#state = new SessionState(persistence);\n  }\n\n  async send(input: AgentInput): Promise<AgentRun> {\n    if (this.#killed || this.#deletePromise) {\n      throw sessionTerminalError(this.#killed);\n    }\n\n    await this.#state.ensureLoaded();\n\n    if (this.#killed || this.#deletePromise) {\n      throw sessionTerminalError(this.#killed);\n    }\n\n    const runtimeInput = createRuntimeInputState(\n      this.#pendingRuntimeInputs.splice(0)\n    );\n    const acceptedInput = normalizeAgentInput(input);\n    const run = new BufferedAgentRun();\n    run.emit(acceptedInput);\n    this.#inputQueue.push({\n      input: structuredClone(acceptedInput),\n      run,\n      runtimeInput,\n    });\n    this.#drainInputQueue().catch((error: unknown) => {\n      run.emit({ type: \"turn-error\", message: errorMessage(error) });\n      run.close();\n    });\n    return run;\n  }\n\n  async steer(input: AgentInput): Promise<AgentRun> {\n    if (this.#killed || this.#deletePromise) {\n      throw sessionTerminalError(this.#killed);\n    }\n\n    const runtimeInput = this.#activeRuntimeInput;\n    const run = this.#activeRun;\n    if (!(runtimeInput && run)) {\n      return this.send(input);\n    }\n\n    await addSteeringInput(runtimeInput, input);\n    return run;\n  }\n\n  interrupt(): void {\n    this.#activeAbort?.abort();\n  }\n\n  delete(): Promise<void> {\n    this.#deletePromise ??= this.#state.delete().then(\n      () => this.kill(),\n      (error: unknown) => {\n        this.#deletePromise = undefined;\n        throw error;\n      }\n    );\n    return this.#deletePromise;\n  }\n\n  enqueueRuntimeInput(\n    input: UserInput,\n    placement: RuntimeInputPlacement = \"turn-start\"\n  ): void {\n    if (this.#killed) {\n      return;\n    }\n\n    const runtimeInput = this.#activeRuntimeInput;\n    if (runtimeInput && !runtimeInput.closedReason) {\n      if (placement === \"turn-start\" && runtimeInput.placement !== placement) {\n        this.#enqueuePendingRuntimeInput({ input, placement });\n        return;\n      }\n\n      runtimeInput.queue.push({ input, placement });\n      return;\n    }\n\n    this.#enqueuePendingRuntimeInput({ input, placement });\n  }\n\n  emitObserverEvent(event: AgentEvent): void {\n    this.#activeRun?.emit(event);\n  }\n\n  #enqueuePendingRuntimeInput(input: QueuedRuntimeInput): void {\n    const queuedTurn = this.#inputQueue[0];\n    if (input.placement === \"turn-start\" && queuedTurn) {\n      queuedTurn.runtimeInput.queue.push(input);\n      return;\n    }\n\n    this.#pendingRuntimeInputs.push(input);\n  }\n\n  kill(): void {\n    if (this.#killed) {\n      return;\n    }\n\n    this.#killed = true;\n    const killedError = sessionKilledError();\n    this.#pendingRuntimeInputs.length = 0;\n    this.#activeAbort?.abort();\n    closeKilledRuntimeInputs({\n      activeRuntimeInput: this.#activeRuntimeInput,\n      inputQueue: this.#inputQueue,\n      message: killedError.message,\n      runToClose: this.#runToCloseOnKill ?? this.#activeRun,\n    });\n  }\n\n  async #drainInputQueue(): Promise<void> {\n    if (this.#running) {\n      return;\n    }\n\n    this.#running = true;\n    try {\n      while (!this.#killed && this.#inputQueue.length > 0) {\n        const item = this.#inputQueue.shift();\n        if (item) {\n          await this.#processQueuedInput(item);\n        }\n      }\n    } finally {\n      this.#running = false;\n    }\n  }\n\n  async #processQueuedInput({\n    input,\n    run,\n    runtimeInput,\n  }: QueuedInput): Promise<void> {\n    const activeAbort = new AbortController();\n    this.#activeAbort = activeAbort;\n    this.#activeRun = run;\n    this.#activeRuntimeInput = runtimeInput;\n    this.#runToCloseOnKill = run;\n    const historySnapshot = this.#state.modelSnapshot();\n\n    try {\n      await withSteeringPlacement(runtimeInput, \"turn-start\", async () => {\n        await this.#hooks?.beforeTurn?.({\n          history: this.#state.modelSnapshot(),\n          input,\n          signal: activeAbort.signal,\n        });\n      });\n      await withRuntimeInputWindow(runtimeInput, \"turn-start\", async () => {\n        await run.emitBoundary({ type: \"turn-start\" });\n      });\n      this.#state.appendUserInput(input);\n      await this.#state.commit();\n      await drainRuntimeInput({\n        placement: \"turn-start\",\n        run,\n        runtimeInput,\n        state: this.#state,\n      });\n\n      const result = await runAgentLoop({\n        emit: async (event) => {\n          if (event.type === \"step-start\" || event.type === \"step-end\") {\n            await withRuntimeInputWindow(runtimeInput, event.type, async () => {\n              await run.emitBoundary(event);\n            });\n            const runtimeInputAdded = await drainRuntimeInput({\n              placement: event.type,\n              run,\n              runtimeInput,\n              state: this.#state,\n            });\n\n            return event.type === \"step-end\"\n              ? { runtimeInputAdded }\n              : undefined;\n          }\n\n          run.emit(event);\n        },\n        history: this.#state.history,\n        hooks: hooksForRuntimeInput(this.#hooks, runtimeInput),\n        llm: this.#llm,\n        signal: activeAbort.signal,\n      });\n\n      await this.#state.commit();\n      const terminalEvent = result === \"aborted\" ? \"turn-abort\" : \"turn-end\";\n      closeRuntimeInput(runtimeInput, terminalEvent);\n      this.#activeRuntimeInput = undefined;\n      this.#activeRun = undefined;\n      await runAfterTurnHook(this.#hooks, {\n        history: this.#state.modelSnapshot(),\n        input,\n        result,\n        signal: activeAbort.signal,\n      });\n      run.emit({ type: terminalEvent });\n    } catch (error) {\n      const turnError =\n        error instanceof Error ? error : new Error(String(error));\n      await emitTurnErrorAfterRecovery({\n        error: turnError,\n        historySnapshot,\n        run,\n        runtimeInput,\n        state: this.#state,\n      });\n    } finally {\n      closeRuntimeInput(runtimeInput);\n      this.#activeAbort = undefined;\n      this.#activeRun = undefined;\n      this.#activeRuntimeInput = undefined;\n      this.#runToCloseOnKill = undefined;\n      run.close(undefined, runtimeInput.closedReason);\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;AAiCA,IAAa,eAAb,MAA0B;CACxB;CACA,cAAsC,CAAC;CACvC;CACA,wBAAuD,CAAC;CACxD;CACA;CACA;CACA;CACA;CACA,UAAU;CACV,WAAW;CACX;CAEA,YACE,KACA,aACA,OACA;EACA,KAAKA,SAAS;EACd,KAAKE,OAAO;EACZ,KAAKE,SAAS,IAAI,aAAa,WAAW;CAC5C;CAEA,MAAM,KAAK,OAAsC;EAC/C,IAAI,KAAKC,WAAW,KAAKC,gBACvB,MAAM,qBAAqB,KAAKD,OAAO;EAGzC,MAAM,KAAKD,OAAO,aAAa;EAE/B,IAAI,KAAKC,WAAW,KAAKC,gBACvB,MAAM,qBAAqB,KAAKD,OAAO;EAGzC,MAAM,eAAe,wBACnB,KAAKF,sBAAsB,OAAO,CAAC,CACrC;EACA,MAAM,gBAAgB,oBAAoB,KAAK;EAC/C,MAAM,MAAM,IAAI,iBAAiB;EACjC,IAAI,KAAK,aAAa;EACtB,KAAKF,YAAY,KAAK;GACpB,OAAO,gBAAgB,aAAa;GACpC;GACA;EACF,CAAC;EACD,KAAKM,iBAAiB,EAAE,OAAO,UAAmB;GAChD,IAAI,KAAK;IAAE,MAAM;IAAc,SAAS,aAAa,KAAK;GAAE,CAAC;GAC7D,IAAI,MAAM;EACZ,CAAC;EACD,OAAO;CACT;CAEA,MAAM,MAAM,OAAsC;EAChD,IAAI,KAAKF,WAAW,KAAKC,gBACvB,MAAM,qBAAqB,KAAKD,OAAO;EAGzC,MAAM,eAAe,KAAKG;EAC1B,MAAM,MAAM,KAAKC;EACjB,IAAI,EAAE,gBAAgB,MACpB,OAAO,KAAK,KAAK,KAAK;EAGxB,MAAM,iBAAiB,cAAc,KAAK;EAC1C,OAAO;CACT;CAEA,YAAkB;EAChB,KAAKC,cAAc,MAAM;CAC3B;CAEA,SAAwB;EACtB,KAAKJ,mBAAmB,KAAKF,OAAO,OAAO,EAAE,WACrC,KAAK,KAAK,IACf,UAAmB;GAClB,KAAKE,iBAAiB,KAAA;GACtB,MAAM;EACR,CACF;EACA,OAAO,KAAKA;CACd;CAEA,oBACE,OACA,YAAmC,cAC7B;EACN,IAAI,KAAKD,SACP;EAGF,MAAM,eAAe,KAAKG;EAC1B,IAAI,gBAAgB,CAAC,aAAa,cAAc;GAC9C,IAAI,cAAc,gBAAgB,aAAa,cAAc,WAAW;IACtE,KAAKG,4BAA4B;KAAE;KAAO;IAAU,CAAC;IACrD;GACF;GAEA,aAAa,MAAM,KAAK;IAAE;IAAO;GAAU,CAAC;GAC5C;EACF;EAEA,KAAKA,4BAA4B;GAAE;GAAO;EAAU,CAAC;CACvD;CAEA,kBAAkB,OAAyB;EACzC,KAAKF,YAAY,KAAK,KAAK;CAC7B;CAEA,4BAA4B,OAAiC;EAC3D,MAAM,aAAa,KAAKR,YAAY;EACpC,IAAI,MAAM,cAAc,gBAAgB,YAAY;GAClD,WAAW,aAAa,MAAM,KAAK,KAAK;GACxC;EACF;EAEA,KAAKE,sBAAsB,KAAK,KAAK;CACvC;CAEA,OAAa;EACX,IAAI,KAAKE,SACP;EAGF,KAAKA,UAAU;EACf,MAAM,cAAc,mBAAmB;EACvC,KAAKF,sBAAsB,SAAS;EACpC,KAAKO,cAAc,MAAM;EACzB,yBAAyB;GACvB,oBAAoB,KAAKF;GACzB,YAAY,KAAKP;GACjB,SAAS,YAAY;GACrB,YAAY,KAAKW,qBAAqB,KAAKH;EAC7C,CAAC;CACH;CAEA,MAAMF,mBAAkC;EACtC,IAAI,KAAKM,UACP;EAGF,KAAKA,WAAW;EAChB,IAAI;GACF,OAAO,CAAC,KAAKR,WAAW,KAAKJ,YAAY,SAAS,GAAG;IACnD,MAAM,OAAO,KAAKA,YAAY,MAAM;IACpC,IAAI,MACF,MAAM,KAAKa,oBAAoB,IAAI;GAEvC;EACF,UAAU;GACR,KAAKD,WAAW;EAClB;CACF;CAEA,MAAMC,oBAAoB,EACxB,OACA,KACA,gBAC6B;EAC7B,MAAM,cAAc,IAAI,gBAAgB;EACxC,KAAKJ,eAAe;EACpB,KAAKD,aAAa;EAClB,KAAKD,sBAAsB;EAC3B,KAAKI,oBAAoB;EACzB,MAAM,kBAAkB,KAAKR,OAAO,cAAc;EAElD,IAAI;GACF,MAAM,sBAAsB,cAAc,cAAc,YAAY;IAClE,MAAM,KAAKJ,QAAQ,aAAa;KAC9B,SAAS,KAAKI,OAAO,cAAc;KACnC;KACA,QAAQ,YAAY;IACtB,CAAC;GACH,CAAC;GACD,MAAM,uBAAuB,cAAc,cAAc,YAAY;IACnE,MAAM,IAAI,aAAa,EAAE,MAAM,aAAa,CAAC;GAC/C,CAAC;GACD,KAAKA,OAAO,gBAAgB,KAAK;GACjC,MAAM,KAAKA,OAAO,OAAO;GACzB,MAAM,kBAAkB;IACtB,WAAW;IACX;IACA;IACA,OAAO,KAAKA;GACd,CAAC;GAED,MAAM,SAAS,MAAM,aAAa;IAChC,MAAM,OAAO,UAAU;KACrB,IAAI,MAAM,SAAS,gBAAgB,MAAM,SAAS,YAAY;MAC5D,MAAM,uBAAuB,cAAc,MAAM,MAAM,YAAY;OACjE,MAAM,IAAI,aAAa,KAAK;MAC9B,CAAC;MACD,MAAM,oBAAoB,MAAM,kBAAkB;OAChD,WAAW,MAAM;OACjB;OACA;OACA,OAAO,KAAKA;MACd,CAAC;MAED,OAAO,MAAM,SAAS,aAClB,EAAE,kBAAkB,IACpB,KAAA;KACN;KAEA,IAAI,KAAK,KAAK;IAChB;IACA,SAAS,KAAKA,OAAO;IACrB,OAAO,qBAAqB,KAAKJ,QAAQ,YAAY;IACrD,KAAK,KAAKE;IACV,QAAQ,YAAY;GACtB,CAAC;GAED,MAAM,KAAKE,OAAO,OAAO;GACzB,MAAM,gBAAgB,WAAW,YAAY,eAAe;GAC5D,kBAAkB,cAAc,aAAa;GAC7C,KAAKI,sBAAsB,KAAA;GAC3B,KAAKC,aAAa,KAAA;GAClB,MAAM,iBAAiB,KAAKT,QAAQ;IAClC,SAAS,KAAKI,OAAO,cAAc;IACnC;IACA;IACA,QAAQ,YAAY;GACtB,CAAC;GACD,IAAI,KAAK,EAAE,MAAM,cAAc,CAAC;EAClC,SAAS,OAAO;GAGd,MAAM,2BAA2B;IAC/B,OAFA,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC;IAGxD;IACA;IACA;IACA,OAAO,KAAKA;GACd,CAAC;EACH,UAAU;GACR,kBAAkB,YAAY;GAC9B,KAAKM,eAAe,KAAA;GACpB,KAAKD,aAAa,KAAA;GAClB,KAAKD,sBAAsB,KAAA;GAC3B,KAAKI,oBAAoB,KAAA;GACzB,IAAI,MAAM,KAAA,GAAW,aAAa,YAAY;EAChD;CACF;AACF"}

package/dist/session/snapshot.js

@@ -1,45 +1,19 @@
 //#region src/session/snapshot.ts
-function encodeSessionSnapshot(state) {
+function encodeSessionSnapshot(history) {
 	return {
-		compactions: structuredClone([...state.compactions ?? []]),
-		history: structuredClone([...state.history]),
-		pluginState: structuredClone({ ...state.pluginState ?? {} }),
-		schemaVersion: 2
+		schemaVersion: 1,
+		history: structuredClone(history)
 	};
 }
 function decodeStoredSessionSnapshot(stored) {
-	if (!stored) return emptySnapshotState();
+	if (!stored) return [];
 	const snapshot = stored.state;
-	if (isSessionSnapshotV1(snapshot)) return {
-		...emptySnapshotState(),
-		history: structuredClone(snapshot.history)
-	};
-	if (isSessionSnapshotV2(snapshot)) return {
-		compactions: structuredClone(snapshot.compactions),
-		history: structuredClone(snapshot.history),
-		pluginState: structuredClone(snapshot.pluginState)
-	};
+	if (isSessionSnapshotV1(snapshot)) return structuredClone(snapshot.history);
 	throw new Error("Unsupported stored session state");
 }
-function emptySnapshotState() {
-	return {
-		compactions: [],
-		history: [],
-		pluginState: {}
-	};
-}
 function isSessionSnapshotV1(value) {
 	return value !== null && typeof value === "object" && "schemaVersion" in value && value.schemaVersion === 1 && "history" in value && Array.isArray(value.history);
 }
-function isSessionSnapshotV2(value) {
-	return value !== null && typeof value === "object" && "schemaVersion" in value && value.schemaVersion === 2 && "history" in value && Array.isArray(value.history) && "pluginState" in value && isRecord(value.pluginState) && "compactions" in value && Array.isArray(value.compactions) && value.compactions.every(isCompactionOverlay);
-}
-function isCompactionOverlay(value) {
-	return value !== null && typeof value === "object" && "createdAt" in value && typeof value.createdAt === "string" && "endIndex" in value && typeof value.endIndex === "number" && "id" in value && typeof value.id === "string" && "startIndex" in value && typeof value.startIndex === "number" && "summary" in value && typeof value.summary === "string";
-}
-function isRecord(value) {
-	return value !== null && typeof value === "object" && !Array.isArray(value);
-}
 //#endregion
 export { decodeStoredSessionSnapshot, encodeSessionSnapshot };
 

package/dist/session/snapshot.js.map

@@ -1 +1 @@
-{"version":3,"file":"snapshot.js","names":[],"sources":["../../src/session/snapshot.ts"],"sourcesContent":["import type { ModelMessage } from \"ai\";\nimport type { StoredSession } from \"./store/types\";\n\nexport interface AgentSessionSnapshotV1 {\n  readonly history: ModelMessage[];\n  readonly schemaVersion: 1;\n}\n\nexport interface AgentCompactionOverlay {\n  readonly createdAt: string;\n  readonly endIndex: number;\n  readonly id: string;\n  readonly startIndex: number;\n  readonly summary: string;\n}\n\nexport interface AgentSessionSnapshotV2 {\n  readonly compactions: AgentCompactionOverlay[];\n  readonly history: ModelMessage[];\n  readonly pluginState: Record<string, unknown>;\n  readonly schemaVersion: 2;\n}\n\nexport interface AgentSessionSnapshotState {\n  readonly compactions?: readonly AgentCompactionOverlay[];\n  readonly history: readonly ModelMessage[];\n  readonly pluginState?: Readonly<Record<string, unknown>>;\n}\n\nexport type AgentSessionSnapshot = AgentSessionSnapshotV2;\n\nexport interface DecodedSessionSnapshot {\n  readonly compactions: AgentCompactionOverlay[];\n  readonly history: ModelMessage[];\n  readonly pluginState: Record<string, unknown>;\n}\n\nexport function encodeSessionSnapshot(\n  state: AgentSessionSnapshotState\n): AgentSessionSnapshot {\n  return {\n    compactions: structuredClone([...(state.compactions ?? [])]),\n    history: structuredClone([...state.history]),\n    pluginState: structuredClone({ ...(state.pluginState ?? {}) }),\n    schemaVersion: 2,\n  };\n}\n\nexport function decodeStoredSessionSnapshot(\n  stored: StoredSession | null\n): DecodedSessionSnapshot {\n  if (!stored) {\n    return emptySnapshotState();\n  }\n\n  const snapshot = stored.state;\n  if (isSessionSnapshotV1(snapshot)) {\n    return {\n      ...emptySnapshotState(),\n      history: structuredClone(snapshot.history),\n    };\n  }\n\n  if (isSessionSnapshotV2(snapshot)) {\n    return {\n      compactions: structuredClone(snapshot.compactions),\n      history: structuredClone(snapshot.history),\n      pluginState: structuredClone(snapshot.pluginState),\n    };\n  }\n\n  throw new Error(\"Unsupported stored session state\");\n}\n\nfunction emptySnapshotState(): DecodedSessionSnapshot {\n  return { compactions: [], history: [], pluginState: {} };\n}\n\nfunction isSessionSnapshotV1(value: unknown): value is AgentSessionSnapshotV1 {\n  return (\n    value !== null &&\n    typeof value === \"object\" &&\n    \"schemaVersion\" in value &&\n    value.schemaVersion === 1 &&\n    \"history\" in value &&\n    Array.isArray(value.history)\n  );\n}\n\nfunction isSessionSnapshotV2(value: unknown): value is AgentSessionSnapshotV2 {\n  return (\n    value !== null &&\n    typeof value === \"object\" &&\n    \"schemaVersion\" in value &&\n    value.schemaVersion === 2 &&\n    \"history\" in value &&\n    Array.isArray(value.history) &&\n    \"pluginState\" in value &&\n    isRecord(value.pluginState) &&\n    \"compactions\" in value &&\n    Array.isArray(value.compactions) &&\n    value.compactions.every(isCompactionOverlay)\n  );\n}\n\nfunction isCompactionOverlay(value: unknown): value is AgentCompactionOverlay {\n  return (\n    value !== null &&\n    typeof value === \"object\" &&\n    \"createdAt\" in value &&\n    typeof value.createdAt === \"string\" &&\n    \"endIndex\" in value &&\n    typeof value.endIndex === \"number\" &&\n    \"id\" in value &&\n    typeof value.id === \"string\" &&\n    \"startIndex\" in value &&\n    typeof value.startIndex === \"number\" &&\n    \"summary\" in value &&\n    typeof value.summary === \"string\"\n  );\n}\n\nfunction isRecord(value: unknown): value is Record<string, unknown> {\n  return value !== null && typeof value === \"object\" && !Array.isArray(value);\n}\n"],"mappings":";AAqCA,SAAgB,sBACd,OACsB;CACtB,OAAO;EACL,aAAa,gBAAgB,CAAC,GAAI,MAAM,eAAe,CAAC,CAAE,CAAC;EAC3D,SAAS,gBAAgB,CAAC,GAAG,MAAM,OAAO,CAAC;EAC3C,aAAa,gBAAgB,EAAE,GAAI,MAAM,eAAe,CAAC,EAAG,CAAC;EAC7D,eAAe;CACjB;AACF;AAEA,SAAgB,4BACd,QACwB;CACxB,IAAI,CAAC,QACH,OAAO,mBAAmB;CAG5B,MAAM,WAAW,OAAO;CACxB,IAAI,oBAAoB,QAAQ,GAC9B,OAAO;EACL,GAAG,mBAAmB;EACtB,SAAS,gBAAgB,SAAS,OAAO;CAC3C;CAGF,IAAI,oBAAoB,QAAQ,GAC9B,OAAO;EACL,aAAa,gBAAgB,SAAS,WAAW;EACjD,SAAS,gBAAgB,SAAS,OAAO;EACzC,aAAa,gBAAgB,SAAS,WAAW;CACnD;CAGF,MAAM,IAAI,MAAM,kCAAkC;AACpD;AAEA,SAAS,qBAA6C;CACpD,OAAO;EAAE,aAAa,CAAC;EAAG,SAAS,CAAC;EAAG,aAAa,CAAC;CAAE;AACzD;AAEA,SAAS,oBAAoB,OAAiD;CAC5E,OACE,UAAU,QACV,OAAO,UAAU,YACjB,mBAAmB,SACnB,MAAM,kBAAkB,KACxB,aAAa,SACb,MAAM,QAAQ,MAAM,OAAO;AAE/B;AAEA,SAAS,oBAAoB,OAAiD;CAC5E,OACE,UAAU,QACV,OAAO,UAAU,YACjB,mBAAmB,SACnB,MAAM,kBAAkB,KACxB,aAAa,SACb,MAAM,QAAQ,MAAM,OAAO,KAC3B,iBAAiB,SACjB,SAAS,MAAM,WAAW,KAC1B,iBAAiB,SACjB,MAAM,QAAQ,MAAM,WAAW,KAC/B,MAAM,YAAY,MAAM,mBAAmB;AAE/C;AAEA,SAAS,oBAAoB,OAAiD;CAC5E,OACE,UAAU,QACV,OAAO,UAAU,YACjB,eAAe,SACf,OAAO,MAAM,cAAc,YAC3B,cAAc,SACd,OAAO,MAAM,aAAa,YAC1B,QAAQ,SACR,OAAO,MAAM,OAAO,YACpB,gBAAgB,SAChB,OAAO,MAAM,eAAe,YAC5B,aAAa,SACb,OAAO,MAAM,YAAY;AAE7B;AAEA,SAAS,SAAS,OAAkD;CAClE,OAAO,UAAU,QAAQ,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,KAAK;AAC5E"}
+{"version":3,"file":"snapshot.js","names":[],"sources":["../../src/session/snapshot.ts"],"sourcesContent":["import type { ModelMessage } from \"ai\";\nimport type { StoredSession } from \"./store/types\";\n\nexport interface AgentSessionSnapshotV1 {\n  readonly history: ModelMessage[];\n  readonly schemaVersion: 1;\n}\n\nexport type AgentSessionSnapshot = AgentSessionSnapshotV1;\n\nexport function encodeSessionSnapshot(\n  history: ModelMessage[]\n): AgentSessionSnapshot {\n  return { schemaVersion: 1, history: structuredClone(history) };\n}\n\nexport function decodeStoredSessionSnapshot(\n  stored: StoredSession | null\n): ModelMessage[] {\n  if (!stored) {\n    return [];\n  }\n\n  const snapshot = stored.state;\n  if (isSessionSnapshotV1(snapshot)) {\n    return structuredClone(snapshot.history);\n  }\n\n  throw new Error(\"Unsupported stored session state\");\n}\n\nfunction isSessionSnapshotV1(value: unknown): value is AgentSessionSnapshotV1 {\n  return (\n    value !== null &&\n    typeof value === \"object\" &&\n    \"schemaVersion\" in value &&\n    value.schemaVersion === 1 &&\n    \"history\" in value &&\n    Array.isArray(value.history)\n  );\n}\n"],"mappings":";AAUA,SAAgB,sBACd,SACsB;CACtB,OAAO;EAAE,eAAe;EAAG,SAAS,gBAAgB,OAAO;CAAE;AAC/D;AAEA,SAAgB,4BACd,QACgB;CAChB,IAAI,CAAC,QACH,OAAO,CAAC;CAGV,MAAM,WAAW,OAAO;CACxB,IAAI,oBAAoB,QAAQ,GAC9B,OAAO,gBAAgB,SAAS,OAAO;CAGzC,MAAM,IAAI,MAAM,kCAAkC;AACpD;AAEA,SAAS,oBAAoB,OAAiD;CAC5E,OACE,UAAU,QACV,OAAO,UAAU,YACjB,mBAAmB,SACnB,MAAM,kBAAkB,KACxB,aAAa,SACb,MAAM,QAAQ,MAAM,OAAO;AAE/B"}

package/dist/session/store/file.d.ts

@@ -8,6 +8,7 @@ declare class FileSessionStore implements SessionStore {
   commit(key: string, next: SessionStoreCommit, options: {
     expectedVersion: string | null;
   }): Promise<CommitResult>;
+  delete(key: string): Promise<void>;
 }
 //#endregion
 export { FileSessionStore };

package/dist/session/store/file.js

@@ -57,6 +57,20 @@ var FileSessionStore = class {
 			});
 		}
 	}
+	async delete(key) {
+		const file = this.#fileForKey(key);
+		const lockDirectory = `${file}.lock`;
+		await mkdir(dirname(file), { recursive: true });
+		await acquireFileLock(lockDirectory);
+		try {
+			await rm(file, { force: true });
+		} finally {
+			await rm(lockDirectory, {
+				force: true,
+				recursive: true
+			});
+		}
+	}
 	#fileForKey(key) {
 		return join(this.#directory, `${Buffer.from(key).toString("base64url")}.json`);
 	}