pi-better-harness 0.29.0 → 0.29.1

package/node_modules/pi-better-goal/README.md

@@ -55,6 +55,8 @@ metadata:
 
 Invoking the alias (`/skill:resolve-issues`, or `/goal /skill:resolve-issues task`) records the target coordinator, with the target's registered path, as the workflow owner, so its task plan, continuation instructions, and plan tools behave exactly as if the target had been invoked. The target must be a registered `workflow-role: coordinator` skill; an unregistered target, a non-coordinator target, or a chain of aliases is refused with a notice.
 
+After every successful context compaction (Pi's `session_compact` event) while a coordinator owns the session, the goal extension re-injects the skill so the model sees it before its next step; the compaction summary keeps run state but not the skill's rules, and the summary itself is left unchanged. If the skill directory ships `scripts/rehydrate.mjs` (checked first) or `rehydrate.mjs`, it runs as `node <script> --cwd <session cwd>` (no shell, 10-second timeout, stderr ignored) and its trimmed stdout becomes the message; empty output injects nothing, so the script decides whether a run is in progress. Without a script, or when it fails or times out, the message is: "Your context was just compacted while you coordinate the `<name>` workflow. Before your next step, read `<SKILL.md path>` in full, with no line limit." The message is a `workflow-rehydrate` custom message: inside a running agent loop (or before an overflow retry) it steers the next model call; otherwise it is appended without starting a turn.
+
 When `pi-better-plan` is installed, it defers its prompt, checklist, and `update_plan` tool to a skill-owned task plan. Skills without this metadata retain normal goal and plan behavior. The skill itself owns its planning format and worker policy; the harness does not enumerate skills or impose a shared workflow schema. The former `pi-better-plan-workflow: coordinator` metadata remains supported for installed skills.
 
 ## Install

package/node_modules/pi-better-goal/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-better-goal",
-  "version": "0.8.4",
+  "version": "0.9.0",
   "description": "Pi extension for goal tracking with background-aware continuation.",
   "license": "MIT",
   "type": "module",

package/node_modules/pi-better-goal/src/index.ts

@@ -40,6 +40,7 @@ import {
 import { createRenderScheduler } from "./shared-render-scheduler.js";
 import { collectSubagentActivity } from "./subagents.js";
 import { currentWorkflowOwner, skillCommandName, workflowEntry, workflowOwnerFromSkill, WORKFLOW_ENTRY_TYPE } from "./workflow.js";
+import { workflowRehydrateMessage, WORKFLOW_REHYDRATE_MESSAGE_TYPE } from "./rehydrate.js";
 import {
   EVENT_ACTIVITY,
   EVENT_READY,
@@ -312,6 +313,8 @@ export default function (pi: ExtensionAPI): void {
   let snapshotSequence = 0;
   let lastAppliedSnapshotSequence = 0;
   let foregroundRunning = false;
+  /** True between agent_start and agent_end: the agent loop will call the model again without a new prompt. */
+  let agentLoopActive = false;
   let pollTimer: ReturnType<typeof setInterval> | undefined;
   let collecting = false;
   let collectionPending = false;
@@ -1186,6 +1189,7 @@ export default function (pi: ExtensionAPI): void {
     collectionPending = false;
     currentCtx = ctx;
     foregroundRunning = !ctx.isIdle();
+    agentLoopActive = false;
     try {
       preferences = readGoalPreferences();
     } catch (error) {
@@ -1334,6 +1338,7 @@ export default function (pi: ExtensionAPI): void {
     agentGeneration = executionGeneration;
     currentCtx = ctx;
     foregroundRunning = true;
+    agentLoopActive = true;
     clearIdleContinuation();
     continuationQueuedFor = null;
     lastAgentEvidence = null;
@@ -1355,6 +1360,7 @@ export default function (pi: ExtensionAPI): void {
   });
 
   pi.on("agent_end", async (event, ctx) => {
+    agentLoopActive = false;
     lastAgentEvidence = agentGeneration === executionGeneration ? continuationEvidence(event.messages) : null;
     // Only an observed editor Escape with pause disabled exempts this abort.
     // Unknown interrupt sources retain the safety fallback on every host.
@@ -1483,6 +1489,29 @@ export default function (pi: ExtensionAPI): void {
     scheduleIdleContinuation(goal, ctx, "continuation", snapshot);
   });
 
+  // A compaction summary keeps run state but drops the coordinator skill's rules. Re-inject them
+  // (or the skill's own rehydrate script output) so the model sees them before its next step.
+  pi.on("session_compact", async (event, ctx) => {
+    const owner = getWorkflow(ctx);
+    if (!owner) return;
+    const content = await workflowRehydrateMessage(owner, ctx.cwd);
+    if (!content) return;
+    // Inside a running agent loop (mid-run compaction or an overflow retry) steer it into the next
+    // model call. Otherwise append it without triggering a turn: the next turn, whatever starts it,
+    // sees it. Steering after the loop ended would start an unprompted continuation.
+    const options = agentLoopActive || event.willRetry ? { deliverAs: "steer" as const } : { triggerTurn: false };
+    try {
+      pi.sendMessage({
+        customType: WORKFLOW_REHYDRATE_MESSAGE_TYPE,
+        content,
+        display: true,
+        details: { workflow: owner.name, path: owner.path, reason: event.reason },
+      }, options);
+    } catch {
+      // Best effort: never fail or block the session over a rehydrate notice.
+    }
+  });
+
   pi.on("session_shutdown", async (_event, ctx) => {
     if (typeof unsubscribeSettingsRequest === "function") unsubscribeSettingsRequest();
 

package/node_modules/pi-better-goal/src/rehydrate.ts

@@ -0,0 +1,57 @@
+import { execFile } from "node:child_process";
+import { statSync } from "node:fs";
+import { dirname, join } from "node:path";
+
+import type { WorkflowOwner } from "./workflow.js";
+
+export const WORKFLOW_REHYDRATE_MESSAGE_TYPE = "workflow-rehydrate";
+export const REHYDRATE_TIMEOUT_MS = 10_000;
+
+/** Script locations a coordinator skill may ship, in precedence order, relative to its SKILL.md directory. */
+const REHYDRATE_SCRIPTS = [join("scripts", "rehydrate.mjs"), "rehydrate.mjs"];
+
+export function rehydrateFallbackMessage(owner: WorkflowOwner): string {
+  return `Your context was just compacted while you coordinate the ${owner.name} workflow. ` +
+    `Before your next step, read ${owner.path} in full, with no line limit.`;
+}
+
+/** The first rehydrate script the owner's skill ships, or undefined when it ships none. */
+export function findRehydrateScript(owner: WorkflowOwner): string | undefined {
+  const skillDir = dirname(owner.path);
+  for (const relative of REHYDRATE_SCRIPTS) {
+    const candidate = join(skillDir, relative);
+    try {
+      if (statSync(candidate).isFile()) return candidate;
+    } catch {
+      // Missing or unreadable: try the next location.
+    }
+  }
+  return undefined;
+}
+
+function runScript(script: string, cwd: string, timeoutMs: number): Promise<string> {
+  return new Promise((resolve, reject) => {
+    execFile("node", [script, "--cwd", cwd], { cwd, timeout: timeoutMs, encoding: "utf8", windowsHide: true },
+      (error, stdout) => (error ? reject(error) : resolve(stdout)));
+  });
+}
+
+/**
+ * The message to inject after a compaction while `owner` coordinates the session, or null for none.
+ * A skill's rehydrate script decides the text (empty output means no run is in progress); without
+ * a script, or when it fails or times out, the model is told to re-read SKILL.md. Never throws.
+ */
+export async function workflowRehydrateMessage(
+  owner: WorkflowOwner,
+  cwd: string,
+  options: { timeoutMs?: number } = {},
+): Promise<string | null> {
+  try {
+    const script = findRehydrateScript(owner);
+    if (!script) return rehydrateFallbackMessage(owner);
+    const stdout = (await runScript(script, cwd, options.timeoutMs ?? REHYDRATE_TIMEOUT_MS)).trim();
+    return stdout.length > 0 ? stdout : null;
+  } catch {
+    return rehydrateFallbackMessage(owner);
+  }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-better-harness",
-  "version": "0.29.0",
+  "version": "0.29.1",
   "description": "Pi extension bundle for a write sandbox, subagents, background tasks, SSH, goals, and structured plans.",
   "license": "MIT",
   "type": "module",
@@ -54,7 +54,7 @@
   },
   "dependencies": {
     "pi-better-background-tasks": "0.9.0",
-    "pi-better-goal": "0.8.4",
+    "pi-better-goal": "0.9.0",
     "pi-better-plan": "0.9.0",
     "pi-better-sandbox": "0.12.0",
     "pi-better-ssh": "0.1.2",