@oh-my-pi/pi-agent-core 15.5.7 → 15.5.8

package/dist/types/agent.d.ts

@@ -1,6 +1,3 @@
-/** Agent class that uses the agent-loop directly.
- * No transport abstraction - calls streamSimple via the loop.
- */
 import { type AssistantMessage, type AssistantMessageEvent, type CursorExecHandlers, type CursorToolResultHandler, type Effort, type ImageContent, type Message, type Model, type ProviderSessionState, type ServiceTier, type SimpleStreamOptions, type ThinkingBudgets, type ToolChoice } from "@oh-my-pi/pi-ai";
 import type { AppendOnlyContextManager } from "./append-only-context";
 import type { HarmonyAuditEvent } from "./harmony-leak";

package/dist/types/utils/yield.d.ts

@@ -24,22 +24,10 @@
  * sleep loops) are retained for the agent-loop hot-path where Promises
  * resolve frequently and the keepalive alone is insufficient.
  */
-/**
- * Await a Promise with an event-loop keepalive active.
- *
- * This is the primary fix for Bun's busy-wait on unresolved Promises.
- * Use it wherever you `await` a Promise that may remain unresolved for
- * an extended period (user input, long-running subprocess, etc.).
- *
- * ```ts
- * // Before (100% CPU while waiting):
- * const input = await mode.getUserInput();
- *
- * // After (proper epoll_wait sleep):
- * const input = await keepaliveWhile(mode.getUserInput());
- * ```
- */
-export declare function keepaliveWhile<T>(promise: Promise<T>): Promise<T>;
+export declare class EventLoopKeepalive {
+    #private;
+    [Symbol.dispose](): void;
+}
 /**
  * Yield to the Bun event loop, sleeping for at least 20 ms — but at most
  * once every {@link YIELD_INTERVAL_MS}. Callers in hot paths can invoke

package/package.json

@@ -1,7 +1,7 @@
 {
 	"type": "module",
 	"name": "@oh-my-pi/pi-agent-core",
-	"version": "15.5.7",
+	"version": "15.5.8",
 	"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.5.7",
-		"@oh-my-pi/pi-natives": "15.5.7",
-		"@oh-my-pi/pi-utils": "15.5.7",
+		"@oh-my-pi/pi-ai": "15.5.8",
+		"@oh-my-pi/pi-natives": "15.5.8",
+		"@oh-my-pi/pi-utils": "15.5.8",
 		"@opentelemetry/api": "^1.9.0"
 	},
 	"devDependencies": {

package/src/agent.ts

@@ -1,7 +1,6 @@
 /** Agent class that uses the agent-loop directly.
  * No transport abstraction - calls streamSimple via the loop.
  */
-
 import { isPromise } from "node:util/types";
 import {
 	type AssistantMessage,
@@ -36,6 +35,7 @@ import type {
 	StreamFn,
 	ToolCallContext,
 } from "./types";
+import { EventLoopKeepalive } from "./utils/yield";
 
 /**
  * Default convertToLlm: Keep only LLM-compatible messages, convert attachments.
@@ -859,7 +859,7 @@ export class Agent {
 		if (!model) throw new Error("No model configured");
 
 		let skipInitialSteeringPoll = options?.skipInitialSteeringPoll === true;
-
+		using _ = new EventLoopKeepalive();
 		const { promise, resolve } = Promise.withResolvers<void>();
 		this.#runningPrompt = promise;
 		this.#resolveRunningPrompt = resolve;

package/src/utils/yield.ts

@@ -31,30 +31,10 @@ import { scheduler } from "node:timers/promises";
 // EventLoopKeepalive — the primary fix for idle-state busy-wait
 // ---------------------------------------------------------------------------
 
-const KEEPALIVE_INTERVAL_MS = 86_400_000; // 24 hours — re-armed each interval
-
-/**
- * Await a Promise with an event-loop keepalive active.
- *
- * This is the primary fix for Bun's busy-wait on unresolved Promises.
- * Use it wherever you `await` a Promise that may remain unresolved for
- * an extended period (user input, long-running subprocess, etc.).
- *
- * ```ts
- * // Before (100% CPU while waiting):
- * const input = await mode.getUserInput();
- *
- * // After (proper epoll_wait sleep):
- * const input = await keepaliveWhile(mode.getUserInput());
- * ```
- */
-export async function keepaliveWhile<T>(promise: Promise<T>): Promise<T> {
-	const ka = setInterval(() => {}, KEEPALIVE_INTERVAL_MS);
-	try {
-		ka.unref();
-		return await promise;
-	} finally {
-		clearInterval(ka);
+export class EventLoopKeepalive {
+	#tmr = setInterval(() => {}, 86_400_000).unref();
+	[Symbol.dispose](): void {
+		clearInterval(this.#tmr);
 	}
 }