From 2db10fb1d4043b3d508d46c03628ec0978af124a Mon Sep 17 00:00:00 2001
From: Vincent Koc <vincentkoc@ieee.org>
Date: Sun, 22 Mar 2026 18:32:16 -0700
Subject: [PATCH] perf(reply): lazy-load embedded queue steering

---
 src/agents/pi-embedded-queue.runtime.ts |  1 +
 src/auto-reply/reply/agent-runner.ts    | 10 +++++++++-
 2 files changed, 10 insertions(+), 1 deletion(-)
 create mode 100644 src/agents/pi-embedded-queue.runtime.ts

diff --git a/src/agents/pi-embedded-queue.runtime.ts b/src/agents/pi-embedded-queue.runtime.ts
new file mode 100644
index 00000000000..15cd91cd987
--- /dev/null
+++ b/src/agents/pi-embedded-queue.runtime.ts
@@ -0,0 +1 @@
+export { queueEmbeddedPiMessage } from "./pi-embedded.js";
diff --git a/src/auto-reply/reply/agent-runner.ts b/src/auto-reply/reply/agent-runner.ts
index 61b9869bef4..cabef3edef5 100644
--- a/src/auto-reply/reply/agent-runner.ts
+++ b/src/auto-reply/reply/agent-runner.ts
@@ -3,7 +3,6 @@ import { lookupContextTokens } from "../../agents/context.js";
 import { DEFAULT_CONTEXT_TOKENS } from "../../agents/defaults.js";
 import { resolveModelAuthMode } from "../../agents/model-auth.js";
 import { isCliProvider } from "../../agents/model-selection.js";
-import { queueEmbeddedPiMessage } from "../../agents/pi-embedded.js";
 import { hasNonzeroUsage } from "../../agents/usage.js";
 import {
   resolveSessionFilePath,
@@ -58,6 +57,14 @@ import { createTypingSignaler } from "./typing-mode.js";
 import type { TypingController } from "./typing.js";
 
 const BLOCK_REPLY_SEND_TIMEOUT_MS = 15_000;
+let piEmbeddedQueueRuntimePromise: Promise<
+  typeof import("../../agents/pi-embedded-queue.runtime.js")
+> | null = null;
+
+function loadPiEmbeddedQueueRuntime() {
+  piEmbeddedQueueRuntimePromise ??= import("../../agents/pi-embedded-queue.runtime.js");
+  return piEmbeddedQueueRuntimePromise;
+}
 
 export async function runReplyAgent(params: {
   commandBody: string;
@@ -194,6 +201,7 @@ export async function runReplyAgent(params: {
   };
 
   if (shouldSteer && isStreaming) {
+    const { queueEmbeddedPiMessage } = await loadPiEmbeddedQueueRuntime();
     const steered = queueEmbeddedPiMessage(followupRun.run.sessionId, followupRun.prompt);
     if (steered && !shouldFollowup) {
       await touchActiveSessionEntry();