@oh-my-pi/pi-coding-agent 16.1.0 → 16.1.2

package/src/extensibility/extensions/runner.ts

@@ -595,7 +595,11 @@ export class ExtensionRunner {
 	}
 
 	async emit<TEvent extends RunnerEmitEvent>(event: TEvent): Promise<RunnerEmitResult<TEvent>> {
-		const ctx = this.createContext();
+		// Defer the per-event context allocation (and the Promise.race/Bun.sleep
+		// timeout machinery) to the first matching handler. Streaming sessions emit
+		// message_update / tool_execution_* per delta with usually no extension
+		// subscribed; building `ctx` for a zero-handler event is pure waste.
+		let ctx: ExtensionContext | undefined;
 		let result: SessionBeforeEventResult | SessionCompactingResult | SessionStopEventResult | undefined;
 
 		if (this.#isSessionShutdownEvent(event)) {
@@ -604,17 +608,19 @@ export class ExtensionRunner {
 			for (const ext of this.extensions) {
 				const handlers = ext.handlers.get(event.type);
 				if (!handlers || handlers.length === 0) continue;
+				ctx ??= this.createContext();
 				for (const handler of handlers) {
 					promises.push(this.#runHandlerWithTimeout(handler, event, ctx, ext, timeoutMs));
 				}
 			}
-			await Promise.all(promises);
+			if (promises.length > 0) await Promise.all(promises);
 			return result as RunnerEmitResult<TEvent>;
 		}
 
 		for (const ext of this.extensions) {
 			const handlers = ext.handlers.get(event.type);
 			if (!handlers || handlers.length === 0) continue;
+			ctx ??= this.createContext();
 
 			for (const handler of handlers) {
 				const handlerResult = await this.#runHandlerWithTimeout(