pi-agent-extensions 0.4.2 → 0.4.3

package/docs/dev/handoff/implementation-log.md

@@ -80,7 +80,7 @@ Implementing the `/handoff` command as specified in `docs/spec_handoff.md`. Foll
 | 3 | LLM extraction | Done | - | extraction.ts with retry logic |
 | 3 | Loader UI | Done | - | progress.ts with ProgressLoader + BorderedLoader fallback |
 | 4 | Editor review | Done | - | Using ctx.ui.editor() |
-| 4 | Session creation | Done | - | Using ctx.newSession() with parent tracking |
+| 4 | Session creation | Done | - | Using ctx.newSession() with parent tracking and withSession |
 | 5 | Non-UI mode | Done | - | Print to stdout |
 | 5 | Documentation | Done | - | docs/handoff.md |
 | 5 | Edge cases | Done | - | Error handling, model resolution |

package/docs/dev/handoff/spec.md

@@ -124,8 +124,8 @@ When goal is too short or vague (e.g., “continue”, “fix”):
    - Optionally prepend `/skill:<name>` if a skill was used last.
    - Optionally prepend a short handoff preamble to set expectations in the new thread.
 5. **Create new session**:
-   - `ctx.newSession({ parentSession: currentSessionFile })`
-   - `ctx.ui.setEditorText(compiledPrompt)`
+   - `ctx.newSession({ parentSession: currentSessionFile, withSession })`
+   - inside `withSession(newCtx)`, call `newCtx.ui.setEditorText(compiledPrompt)`
 
 ## Output Schema (LLM → Extension)
 ```json
@@ -441,8 +441,8 @@ The following decisions were made during implementation planning:
 - **Why**: Provides valuable context about the codebase state at handoff time
 
 #### Session Creation Flow
-- **Decision**: Use `ctx.newSession({ parentSession })` then `ctx.ui.setEditorText()`
-- **Why**: Creates proper session lineage tracking; setEditorText prefills the prompt for user to review and submit when ready
+- **Decision**: Use `ctx.newSession({ parentSession, withSession })` and perform editor setup inside `withSession(newCtx)`
+- **Why**: Creates proper session lineage tracking and avoids using a stale command context after session replacement; `newCtx.ui.setEditorText()` prefills the prompt for user review and submission
 
 #### Documentation
 - **Decision**: Create `docs/handoff.md` with usage and examples

package/extensions/handoff/index.ts

@@ -196,19 +196,21 @@ async function runHandoffCommand(
     return;
   }
 
-  // Create new session with parent tracking
+  // Create new session with parent tracking.
+  // Any post-session-replacement work must happen inside withSession,
+  // using the fresh ctx passed by Pi.
   const newSessionResult = await ctx.newSession({
     parentSession: currentSessionFile,
+    withSession: async (newCtx) => {
+      newCtx.ui.setEditorText(editedPrompt);
+      newCtx.ui.notify("Handoff ready. Press Enter to send.", "info");
+    },
   });
 
   if (newSessionResult.cancelled) {
     ctx.ui.notify("New session cancelled", "info");
     return;
   }
-
-  // Set the edited prompt in the editor for submission
-  ctx.ui.setEditorText(editedPrompt);
-  ctx.ui.notify("Handoff ready. Press Enter to send.", "info");
 }
 
 /**

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-agent-extensions",
-  "version": "0.4.2",
+  "version": "0.4.3",
   "description": "Collection of extensions for pi coding agent",
   "type": "module",
   "repository": {