oira666_pi-subagent 0.2.9 → 0.2.11

package/index.ts

@@ -373,6 +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 = 50;
 
 type SyntheticResumeState = {
   plan: ResumableSubagentCall | null;
@@ -471,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;
@@ -684,7 +686,12 @@ export default function (pi: ExtensionAPI) {
       if (hasCliInitialPrompt(process.argv)) {
         if (ctx.hasUI) ctx.ui.notify(`Resuming ${plan.tasks.length} subagents...`, "info");
       } else {
-        pi.sendUserMessage(`Resuming ${plan.tasks.length} subagents...`);
+        // 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);
@@ -696,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.9",
+  "version": "0.2.11",
   "description": "Subagent extension for Pi coding agent. Delegate tasks to specialized agents.",
   "type": "module",
   "main": "index.ts",