@oh-my-pi/pi-agent-core 18.0.3 → 18.0.4

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # Changelog
 
+## [Unreleased]
+
+## [18.0.4] - 2026-08-24
+
+### Changed
+
+- Improved performance in append-only context mode by memoizing message serialization, keeping per-call sync overhead flat as conversations grow.
+
+### Fixed
+
+- Fixed an issue where `onTurnEnd` was skipped for turns ended by a terminal tool result (such as a subagent's final `yield`).
+
 ## [18.0.0] - 2026-08-22
 
 ### Fixed

package/package.json

@@ -1,7 +1,7 @@
 {
 	"type": "module",
 	"name": "@oh-my-pi/pi-agent-core",
-	"version": "18.0.3",
+	"version": "18.0.4",
 	"description": "General-purpose agent with transport abstraction, state management, and attachment support",
 	"homepage": "https://omp.sh",
 	"author": "Stencil Labs, Inc.",
@@ -35,16 +35,16 @@
 		"fmt": "biome format --write ."
 	},
 	"dependencies": {
-		"@oh-my-pi/pi-ai": "18.0.3",
-		"@oh-my-pi/pi-catalog": "18.0.3",
-		"@oh-my-pi/pi-natives": "18.0.3",
-		"@oh-my-pi/pi-utils": "18.0.3",
-		"@oh-my-pi/pi-wire": "18.0.3",
-		"@oh-my-pi/snapcompact": "18.0.3",
+		"@oh-my-pi/pi-ai": "18.0.4",
+		"@oh-my-pi/pi-catalog": "18.0.4",
+		"@oh-my-pi/pi-natives": "18.0.4",
+		"@oh-my-pi/pi-utils": "18.0.4",
+		"@oh-my-pi/pi-wire": "18.0.4",
+		"@oh-my-pi/snapcompact": "18.0.4",
 		"@opentelemetry/api": "^1.9.1"
 	},
 	"devDependencies": {
-		"@oh-my-pi/omptype": "18.0.3",
+		"@oh-my-pi/omptype": "18.0.4",
 		"@opentelemetry/context-async-hooks": "^2.9.0",
 		"@opentelemetry/sdk-trace-base": "^2.9.0",
 		"@types/bun": "^1.3.14"

package/src/agent-loop.ts

@@ -623,6 +623,15 @@ function buildAgentEndEvent(
  * Push a `turn_end` event and run the awaited per-turn hook when the run is
  * still healthy. The hook is skipped for externally aborted or errored turns so
  * a user interrupt does not hang on a background backlog wait.
+ *
+ * A {@link TERMINAL_TOOL_RESULT_ABORT_REASON} abort is the exception: it is a
+ * graceful yield (e.g. a subagent's final `yield` tool), not a user interrupt.
+ * The completed tool batch is persisted and the turn must still reach
+ * `onTurnEnd` so per-turn bookkeeping — notably advisor review of the yield
+ * delta (#9505) — runs exactly as it does for a plain end-of-turn message. The
+ * hook receives no signal in that case so downstream waits (advisor catch-up)
+ * behave identically to a normal final turn instead of short-circuiting on the
+ * spent abort.
  */
 async function emitTurnEnd(
 	stream: EventStream<AgentEvent, AgentMessage[]>,
@@ -635,10 +644,16 @@ async function emitTurnEnd(
 	runHookOnAbortedMessage = false,
 ): Promise<void> {
 	stream.push({ type: "turn_end", message, toolResults });
+	const terminalYield = signal?.reason === TERMINAL_TOOL_RESULT_ABORT_REASON;
 	const isAbortedOrError =
 		message.role === "assistant" && (message.stopReason === "aborted" || message.stopReason === "error");
-	if (signal?.aborted || (isAbortedOrError && !runHookOnAbortedMessage)) return;
-	await config.onTurnEnd?.(currentContext.messages, signal, { message, toolResults, willContinue: false, ...context });
+	if ((signal?.aborted && !terminalYield) || (isAbortedOrError && !runHookOnAbortedMessage)) return;
+	await config.onTurnEnd?.(currentContext.messages, terminalYield ? undefined : signal, {
+		message,
+		toolResults,
+		willContinue: false,
+		...context,
+	});
 }
 
 function createGateStopMessage(model: Model, reason: string | undefined): AssistantMessage {

package/src/append-only-context.ts

@@ -16,7 +16,8 @@
 
 import type { Context, Message, Tool } from "@oh-my-pi/pi-ai";
 import { normalizeTools } from "./agent-loop";
-import type { AgentContext } from "./types";
+import { messageEstimateVersion } from "./compaction/message-cache";
+import type { AgentContext, AgentMessage } from "./types";
 
 // ---------------------------------------------------------------------------
 // StablePrefix (formerly ImmutablePrefix)
@@ -177,6 +178,26 @@ export class AppendOnlyContextManager {
 	 * point on every subsequent turn.
 	 */
 	#messageDigests: number[] = [];
+	/**
+	 * Digests memoized by message object identity, validated by the message's
+	 * estimate version ({@link messageEstimateVersion}). Synced message objects
+	 * are stable between calls: converted fragments are cached per session
+	 * message identity and handed back unchanged on every call, so an unchanged
+	 * history re-hits this memo instead of re-serializing every previously-
+	 * synced message on each LLM call.
+	 *
+	 * Owner-side rewrites (prune/shake/strip-images) mutate messages IN PLACE
+	 * under stable identity — for assistant pass-through fragments the log
+	 * aliases the very object being mutated — so a bare identity memo would
+	 * serve pre-mutation bytes forever. Those owners MUST call
+	 * `invalidateMessageCache`, which bumps the symbol-keyed version tag
+	 * this memo validates before every hit; a version mismatch recomputes from
+	 * actual bytes and the sync diverges exactly as if a fresh object had
+	 * arrived. Mutating a synced message without that bump violates the
+	 * cache-coherence contract shared with the tokenizer and convert caches
+	 * (see `compaction/message-cache.ts`) and is unsupported.
+	 */
+	#digestMemo = new WeakMap<object, { version: number; digest: number }>();
 
 	build(context: AgentContext, options: BuildOptions): Context {
 		this.prefix.build(context, options);
@@ -290,6 +311,9 @@ export class AppendOnlyContextManager {
 	#messageDigest(msg: unknown): number {
 		if (!msg || typeof msg !== "object") return 0;
 		const m = msg as Record<string, unknown>;
+		const version = messageEstimateVersion(msg as AgentMessage);
+		const cached = this.#digestMemo.get(m);
+		if (cached !== undefined && cached.version === version) return cached.digest;
 		const payload = JSON.stringify({
 			r: m.role ?? null,
 			c: m.content ?? null,
@@ -304,7 +328,9 @@ export class AppendOnlyContextManager {
 		for (let j = 0; j < payload.length; j++) {
 			hash = ((hash << 5) - hash + payload.charCodeAt(j)) | 0;
 		}
-		return hash >>> 0;
+		const hash32 = hash >>> 0;
+		this.#digestMemo.set(m, { version, digest: hash32 });
+		return hash32;
 	}
 }