@oh-my-pi/pi-agent-core 15.2.2 → 15.2.3

package/CHANGELOG.md

@@ -2,6 +2,11 @@
 
 ## [Unreleased]
 
+## [15.2.3] - 2026-05-22
+### Added
+
+- Added `onBeforeYield` hook support so user code can run right before the agent loop checks for follow-up messages
+
 ## [15.1.3] - 2026-05-17
 ### Added
 

package/dist/types/agent.d.ts

@@ -265,6 +265,7 @@ export declare class Agent {
     setProviderResponseInterceptor(fn: SimpleStreamOptions["onResponse"] | undefined): void;
     setRawSseEventInterceptor(fn: SimpleStreamOptions["onSseEvent"] | undefined): void;
     setAssistantMessageEventInterceptor(fn: ((message: AssistantMessage, event: AssistantMessageEvent) => void) | undefined): void;
+    setOnBeforeYield(fn: (() => Promise<void> | void) | undefined): void;
     emitExternalEvent(event: AgentEvent): void;
     setSystemPrompt(v: string[]): void;
     setModel(m: Model): void;

package/dist/types/types.d.ts

@@ -94,6 +94,13 @@ export interface AgentLoopConfig extends SimpleStreamOptions {
      * continues with another turn.
      */
     getFollowUpMessages?: () => Promise<AgentMessage[]>;
+    /**
+     * Hook fired right before the loop would exit.
+     *
+     * Called when the agent has no more tool calls and no steering messages,
+     * immediately before polling follow-up messages.
+     */
+    onBeforeYield?: () => Promise<void> | void;
     /**
      * Provides tool execution context, resolved per tool call.
      * Use for late-bound UI or session state access.

package/package.json

@@ -1,7 +1,7 @@
 {
 	"type": "module",
 	"name": "@oh-my-pi/pi-agent-core",
-	"version": "15.2.2",
+	"version": "15.2.3",
 	"description": "General-purpose agent with transport abstraction, state management, and attachment support",
 	"homepage": "https://omp.sh",
 	"author": "Can Boluk",
@@ -35,9 +35,9 @@
 		"fmt": "biome format --write ."
 	},
 	"dependencies": {
-		"@oh-my-pi/pi-ai": "15.2.2",
-		"@oh-my-pi/pi-natives": "15.2.2",
-		"@oh-my-pi/pi-utils": "15.2.2",
+		"@oh-my-pi/pi-ai": "15.2.3",
+		"@oh-my-pi/pi-natives": "15.2.3",
+		"@oh-my-pi/pi-utils": "15.2.3",
 		"@opentelemetry/api": "^1.9.0"
 	},
 	"devDependencies": {

package/src/agent-loop.ts

@@ -589,6 +589,7 @@ async function runLoopBody(
 		}
 
 		// Agent would stop here. Check for follow-up messages.
+		await config.onBeforeYield?.();
 		const followUpMessages = (await config.getFollowUpMessages?.()) || [];
 		if (followUpMessages.length > 0) {
 			// Set as pending so inner loop processes them

package/src/agent.ts

@@ -290,6 +290,7 @@ export class Agent {
 	#onSseEvent?: SimpleStreamOptions["onSseEvent"];
 	#onAssistantMessageEvent?: (message: AssistantMessage, event: AssistantMessageEvent) => void;
 	#onHarmonyLeak?: (event: HarmonyAuditEvent) => void | Promise<void>;
+	#onBeforeYield?: () => Promise<void> | void;
 	#telemetry?: AgentLoopConfig["telemetry"];
 
 	/** Buffered Cursor tool results with text length at time of call (for correct ordering) */
@@ -559,6 +560,10 @@ export class Agent {
 		this.#onAssistantMessageEvent = fn;
 	}
 
+	setOnBeforeYield(fn: (() => Promise<void> | void) | undefined): void {
+		this.#onBeforeYield = fn;
+	}
+
 	emitExternalEvent(event: AgentEvent) {
 		switch (event.type) {
 			case "message_start":
@@ -934,6 +939,7 @@ export class Agent {
 				return this.#dequeueSteeringMessages();
 			},
 			getFollowUpMessages: async () => this.#dequeueFollowUpMessages(),
+			onBeforeYield: () => this.#onBeforeYield?.(),
 			telemetry: this.#telemetry,
 		};
 

package/src/types.ts

@@ -122,6 +122,13 @@ export interface AgentLoopConfig extends SimpleStreamOptions {
 	 * continues with another turn.
 	 */
 	getFollowUpMessages?: () => Promise<AgentMessage[]>;
+	/**
+	 * Hook fired right before the loop would exit.
+	 *
+	 * Called when the agent has no more tool calls and no steering messages,
+	 * immediately before polling follow-up messages.
+	 */
+	onBeforeYield?: () => Promise<void> | void;
 
 	/**
 	 * Provides tool execution context, resolved per tool call.