oira666_pi-subagent 0.2.10 → 0.2.11

package/index.ts

@@ -373,7 +373,7 @@ const RESUME_PROVIDER = "pi-subagent-resume";
 const RESUME_MODEL_ID = "synthetic-tool-call";
 const RESUME_STATE_KEY = "__piSubagentResumeState";
 const SUBAGENT_FALLBACK_MODEL_ENV = "PI_SUBAGENT_FALLBACK_MODEL";
-const RESUME_INTERACTIVE_DELAY_MS = 250;
+const RESUME_INTERACTIVE_DELAY_MS = 50;
 
 type SyntheticResumeState = {
   plan: ResumableSubagentCall | null;
@@ -472,6 +472,7 @@ function getRestorableModel(ctx: any): any | undefined {
 export default function (pi: ExtensionAPI) {
   let resumeModelRegistry: any | undefined;
   let lastRestorableModel: any | undefined;
+  let pendingInteractiveResumePrompt: string | null = null;
 
   async function streamWithRealModelFallback(context: any, options: any, fallback: any) {
     if (!fallback) return null;
@@ -685,20 +686,12 @@ export default function (pi: ExtensionAPI) {
       if (hasCliInitialPrompt(process.argv)) {
         if (ctx.hasUI) ctx.ui.notify(`Resuming ${plan.tasks.length} subagents...`, "info");
       } else {
-        // Do not start the synthetic resume turn synchronously from
-        // session_start. In interactive mode Pi may still be rebuilding the
-        // resumed chat UI; if the tool starts before that finishes, later
-        // renderSessionContext() can clear the live pending tool component and
-        // all real-time tool updates disappear. A short macrotask delay lets
-        // the normal "Resumed session" render settle first.
-        setTimeout(() => {
-          try {
-            pi.sendUserMessage(`Resuming ${plan.tasks.length} subagents...`);
-          } catch (err) {
-            console.error("[pi-subagent] Failed to start deferred resume turn:", err);
-            void restoreModelAfterResumeFailure(ctx);
-          }
-        }, RESUME_INTERACTIVE_DELAY_MS);
+        // Do not start the synthetic resume turn from session_start. Pi renders
+        // the resumed chat only after session_start/resources_discover complete;
+        // starting now lets that render wipe out the live tool component, so no
+        // real-time updates appear. Queue it for resources_discover instead,
+        // which is the last extension hook before the initial chat render.
+        pendingInteractiveResumePrompt = `Resuming ${plan.tasks.length} subagents...`;
       }
     } catch (err) {
       console.error("[pi-subagent] Error in session_start:", err);
@@ -710,6 +703,20 @@ export default function (pi: ExtensionAPI) {
     await restoreModelAfterResumeFailure();
   });
 
+  pi.on("resources_discover", (_event, ctx) => {
+    const prompt = pendingInteractiveResumePrompt;
+    if (!prompt) return;
+    pendingInteractiveResumePrompt = null;
+    setTimeout(() => {
+      try {
+        pi.sendUserMessage(prompt);
+      } catch (err) {
+        console.error("[pi-subagent] Failed to start deferred resume turn:", err);
+        void restoreModelAfterResumeFailure(ctx);
+      }
+    }, RESUME_INTERACTIVE_DELAY_MS);
+  });
+
   // Inject available agents into the system prompt
   pi.on("before_agent_start", async (event) => {
     try {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "oira666_pi-subagent",
-  "version": "0.2.10",
+  "version": "0.2.11",
   "description": "Subagent extension for Pi coding agent. Delegate tasks to specialized agents.",
   "type": "module",
   "main": "index.ts",