pi-agent-flow 2.0.1 → 2.0.5

package/dist/flow/warp.js

@@ -0,0 +1,207 @@
+/**
+ * Warp extension — transfer context to a new focused session
+ *
+ * Instead of compacting (which is lossy), warp extracts what matters
+ * for your next task and creates a new session with a generated prompt.
+ *
+ * Usage:
+ *   /flow:warp now implement this for teams as well
+ *   /flow:warp execute phase one of the plan
+ *   /flow:warp check other places that need this fix
+ *   /flow:warp                              (no args = default continuation goal)
+ */
+function isRecord(value) {
+    return typeof value === "object" && value !== null;
+}
+const WARP_TIMEOUT_MS = 5 * 60 * 1000;
+const WARP_POLL_INTERVAL_MS = 25;
+const WARP_INSTRUCTIONS = `You are writing a warp note for another AI agent with NO access to this chat.
+
+Extract only task-relevant context from this conversation.
+
+Rules:
+- Be tight and token-efficient.
+- Use only concrete facts from this conversation.
+- Prefer specifics: file paths, symbols, commands, errors, outputs, decisions.
+- Include constraints/invariants only when explicit, non-negotiable, and task-relevant.
+- Include line numbers only if known from this conversation.
+- Omit irrelevant history and broad retrospectives.
+- Do not invent missing details.
+- If a critical detail is unknown, say so briefly and include the smallest verification step.
+- Do not write a plan unless one already exists in this conversation.
+- Do not call tools.
+
+Your output MUST use this exact format:
+
+FRONTMATTER (YAML between --- delimiters):
+  context       — 1-2 sentence orientation summary
+  end_goal      — The finish line, not the next step
+  decisions     — Key choices already made (list)
+  files         — Files touched with what changed (list)
+  open_items    — Unresolved work or questions (list)
+  watch_out     — Edge cases, gotchas, fragile assumptions (list)
+  context_gathering:
+    aim         — What the initial scout/discovery should accomplish
+    scope       — Specific things to explore or map (list)
+  execution_plan:
+    - phase     — Phase name
+      parallel  — true/false, can this run alongside other phases?
+      group     — If parallel, which execution group (A, B, C...)
+      flow      — Which flow type to use (scout, build, audit, craft...)
+      flows     — OR multiple flows if parallel within the phase
+      task      — Clear, actionable task for this phase
+      depends_on — Phase(s) that must complete first
+      produces  — What "done" means for this phase
+  success_criteria — How to know the overall work is complete (list)
+
+BODY (after the closing ---):
+  ## Context
+  A concise summary of the conversation context.
+
+RULES:
+1. Always start with a context_gathering phase — the new session has no context yet, so discovery comes first.
+2. Mark phases parallel:true when they have no data dependencies on each other. Use group labels (A, B, C) to cluster parallel work.
+3. Each phase should produce a concrete artifact, evidence of completion before moving on to the next.
+4. Respect the given plan scaffold.
+5. Use flow types from: scout, build, audit, craft, debug, ideas.
+6. Success criteria should be the final state, i.e. integration test pass, code coverage with verified output, etc.
+7. If an active goal from the prior session exists, include it in the frontmatter context.
+8. Preserve unresolved blockers, open questions, or "not done" items from prior flow results in open_items.
+9. Flag any uncertain areas — parts of the codebase, design decisions, or assumptions that may have shifted since the conversation and need re-assessment via a scout or audit flow before committing to a plan.
+10. No tool calls; all attempts that you need to discover, note them to the watch_out list or execution_plan.
+11. Your entire response must be the warp prompt starting with '---' (YAML frontmatter opening). No preamble, no explanations, no tool calls.
+
+IMPORTANT: You are a text generation assistant, not an agent. Do NOT attempt tool calls, file operations, code execution, or any actions. Output ONLY the structured prompt text.`;
+function getRoleMessage(entry, role) {
+    if (entry.type !== "message" || !isRecord(entry.message)) {
+        return null;
+    }
+    const message = entry.message;
+    if (message.role !== role || !("content" in message)) {
+        return null;
+    }
+    return message;
+}
+function getAssistantMessage(entry) {
+    const message = getRoleMessage(entry, "assistant");
+    if (!message || !isRecord(message)) {
+        return null;
+    }
+    return message;
+}
+function getMessageText(entry) {
+    if (entry.type !== "message" || !isRecord(entry.message) || !("content" in entry.message)) {
+        return "";
+    }
+    const content = entry.message.content;
+    if (typeof content === "string") {
+        return content.trim();
+    }
+    if (!Array.isArray(content)) {
+        return "";
+    }
+    return content
+        .filter((block) => isRecord(block) && block.type === "text" && typeof block.text === "string")
+        .map((block) => block.text)
+        .join("\n")
+        .trim();
+}
+function findUserMessageIndex(entries, fromIndex, text) {
+    for (let i = entries.length - 1; i >= fromIndex; i--) {
+        const entry = entries[i];
+        if (!getRoleMessage(entry, "user")) {
+            continue;
+        }
+        if (getMessageText(entry) === text) {
+            return i;
+        }
+    }
+    return -1;
+}
+function hasAssistantAfterIndex(entries, fromIndex) {
+    for (let i = entries.length - 1; i > fromIndex; i--) {
+        const entry = entries[i];
+        if (getRoleMessage(entry, "assistant")) {
+            return true;
+        }
+    }
+    return false;
+}
+async function waitForWarpTurn(ctx, startIndex, warpRequest, timeoutMs = WARP_TIMEOUT_MS) {
+    const deadline = Date.now() + timeoutMs;
+    while (Date.now() < deadline) {
+        const branch = ctx.sessionManager.getBranch();
+        const warpUserIndex = findUserMessageIndex(branch, startIndex, warpRequest);
+        if (warpUserIndex !== -1 && hasAssistantAfterIndex(branch, warpUserIndex) && ctx.isIdle()) {
+            return { branch, warpUserIndex };
+        }
+        await new Promise((resolve) => setTimeout(resolve, WARP_POLL_INTERVAL_MS));
+    }
+    return null;
+}
+function getAssistantText(entries, fromIndex) {
+    for (let i = entries.length - 1; i >= fromIndex; i--) {
+        const entry = entries[i];
+        const message = getAssistantMessage(entry);
+        if (!message) {
+            continue;
+        }
+        if (message.stopReason !== "stop") {
+            return null;
+        }
+        const text = getMessageText(entry);
+        if (text.length > 0) {
+            return text;
+        }
+    }
+    return null;
+}
+const DEFAULT_GOAL = "Continue where we left off — summarize what we have done, where we are, and what the natural next step is.";
+export default function (pi) {
+    pi.registerCommand("flow:warp", {
+        description: "Transfer context to a new focused session",
+        handler: async (args, ctx) => {
+            const notify = (message, level) => {
+                ctx.ui?.notify?.(message, level);
+            };
+            if (!ctx.model) {
+                notify("No model selected", "error");
+                return;
+            }
+            const task = args.trim() || DEFAULT_GOAL;
+            const currentSessionFile = ctx.sessionManager.getSessionFile();
+            const startIndex = ctx.sessionManager.getBranch().length;
+            const warpRequest = `${WARP_INSTRUCTIONS}\n\nTask for the next agent:\n${task}`;
+            if (ctx.isIdle()) {
+                pi.sendUserMessage(warpRequest);
+            }
+            else {
+                pi.sendUserMessage(warpRequest, { deliverAs: "followUp" });
+            }
+            notify("Generating warp note...", "info");
+            const warpTurn = await waitForWarpTurn(ctx, startIndex, warpRequest);
+            if (!warpTurn) {
+                notify("Timed out waiting for warp note", "error");
+                return;
+            }
+            const warpNote = getAssistantText(warpTurn.branch, warpTurn.warpUserIndex + 1);
+            if (!warpNote) {
+                notify("Failed to capture warp note from the assistant response", "error");
+                return;
+            }
+            const promptForNewSession = `${warpNote}\n\n---\n\n## Task\n${task}`;
+            const newSessionResult = await ctx.newSession({
+                parentSession: currentSessionFile,
+                withSession: async (newCtx) => {
+                    await newCtx.sendUserMessage(promptForNewSession);
+                    newCtx.ui?.notify?.("Warp ready...", "info");
+                },
+            });
+            if (newSessionResult.cancelled) {
+                notify("New session cancelled", "info");
+                return;
+            }
+        },
+    });
+}
+//# sourceMappingURL=warp.js.map

package/dist/flow/warp.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"warp.js","sourceRoot":"","sources":["../../src/flow/warp.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAIH,SAAS,QAAQ,CAAC,KAAc;IAC/B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,CAAC;AACpD,CAAC;AAYD,MAAM,eAAe,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;AACtC,MAAM,qBAAqB,GAAG,EAAE,CAAC;AAEjC,MAAM,iBAAiB,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;kLAwDwJ,CAAC;AAEnL,SAAS,cAAc,CAAC,KAAmB,EAAE,IAA0B;IACtE,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;QAC1D,OAAO,IAAI,CAAC;IACb,CAAC;IACD,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;IAC9B,IAAI,OAAO,CAAC,IAAI,KAAK,IAAI,IAAI,CAAC,CAAC,SAAS,IAAI,OAAO,CAAC,EAAE,CAAC;QACtD,OAAO,IAAI,CAAC;IACb,CAAC;IACD,OAAO,OAAsB,CAAC;AAC/B,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAmB;IAC/C,MAAM,OAAO,GAAG,cAAc,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;IACnD,IAAI,CAAC,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QACpC,OAAO,IAAI,CAAC;IACb,CAAC;IACD,OAAO,OAA+B,CAAC;AACxC,CAAC;AAED,SAAS,cAAc,CAAC,KAAmB;IAC1C,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,SAAS,IAAI,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;QAC3F,OAAO,EAAE,CAAC;IACX,CAAC;IACD,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;IACtC,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QACjC,OAAO,OAAO,CAAC,IAAI,EAAE,CAAC;IACvB,CAAC;IACD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC7B,OAAO,EAAE,CAAC;IACX,CAAC;IACD,OAAO,OAAO;SACZ,MAAM,CAAC,CAAC,KAAK,EAA2C,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC;SACtI,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC;SAC1B,IAAI,CAAC,IAAI,CAAC;SACV,IAAI,EAAE,CAAC;AACV,CAAC;AAED,SAAS,oBAAoB,CAAC,OAAuB,EAAE,SAAiB,EAAE,IAAY;IACrF,KAAK,IAAI,CAAC,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;QACtD,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACzB,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,CAAC;YACpC,SAAS;QACV,CAAC;QACD,IAAI,cAAc,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC;YACpC,OAAO,CAAC,CAAC;QACV,CAAC;IACF,CAAC;IACD,OAAO,CAAC,CAAC,CAAC;AACX,CAAC;AAED,SAAS,sBAAsB,CAAC,OAAuB,EAAE,SAAiB;IACzE,KAAK,IAAI,CAAC,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;QACrD,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACzB,IAAI,cAAc,CAAC,KAAK,EAAE,WAAW,CAAC,EAAE,CAAC;YACxC,OAAO,IAAI,CAAC;QACb,CAAC;IACF,CAAC;IACD,OAAO,KAAK,CAAC;AACd,CAAC;AAOD,KAAK,UAAU,eAAe,CAC7B,GAA4B,EAC5B,UAAkB,EAClB,WAAmB,EACnB,SAAS,GAAG,eAAe;IAE3B,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;IACxC,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,GAAG,CAAC,cAAc,CAAC,SAAS,EAAoB,CAAC;QAChE,MAAM,aAAa,GAAG,oBAAoB,CAAC,MAAM,EAAE,UAAU,EAAE,WAAW,CAAC,CAAC;QAC5E,IAAI,aAAa,KAAK,CAAC,CAAC,IAAI,sBAAsB,CAAC,MAAM,EAAE,aAAa,CAAC,IAAI,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3F,OAAO,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;QAClC,CAAC;QACD,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,qBAAqB,CAAC,CAAC,CAAC;IAClF,CAAC;IACD,OAAO,IAAI,CAAC;AACb,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAuB,EAAE,SAAiB;IACnE,KAAK,IAAI,CAAC,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;QACtD,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACzB,MAAM,OAAO,GAAG,mBAAmB,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,OAAO,EAAE,CAAC;YACd,SAAS;QACV,CAAC;QACD,IAAI,OAAO,CAAC,UAAU,KAAK,MAAM,EAAE,CAAC;YACnC,OAAO,IAAI,CAAC;QACb,CAAC;QACD,MAAM,IAAI,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;QACnC,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrB,OAAO,IAAI,CAAC;QACb,CAAC;IACF,CAAC;IACD,OAAO,IAAI,CAAC;AACb,CAAC;AAED,MAAM,YAAY,GAAG,4GAA4G,CAAC;AAElI,MAAM,CAAC,OAAO,WAAW,EAAgB;IACxC,EAAE,CAAC,eAAe,CAAC,WAAW,EAAE;QAC/B,WAAW,EAAE,2CAA2C;QACxD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;YAC5B,MAAM,MAAM,GAAG,CAAC,OAAe,EAAE,KAAmC,EAAE,EAAE;gBACvE,GAAG,CAAC,EAAE,EAAE,MAAM,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YAClC,CAAC,CAAC;YAEF,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;gBAChB,MAAM,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;gBACrC,OAAO;YACR,CAAC;YAED,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE,IAAI,YAAY,CAAC;YAEzC,MAAM,kBAAkB,GAAG,GAAG,CAAC,cAAc,CAAC,cAAc,EAAE,CAAC;YAC/D,MAAM,UAAU,GAAG,GAAG,CAAC,cAAc,CAAC,SAAS,EAAE,CAAC,MAAM,CAAC;YAEzD,MAAM,WAAW,GAAG,GAAG,iBAAiB,iCAAiC,IAAI,EAAE,CAAC;YAEhF,IAAI,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC;gBAClB,EAAE,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;YACjC,CAAC;iBAAM,CAAC;gBACP,EAAE,CAAC,eAAe,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,CAAC;YAC5D,CAAC;YAED,MAAM,CAAC,yBAAyB,EAAE,MAAM,CAAC,CAAC;YAC1C,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,GAAG,EAAE,UAAU,EAAE,WAAW,CAAC,CAAC;YAErE,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACf,MAAM,CAAC,iCAAiC,EAAE,OAAO,CAAC,CAAC;gBACnD,OAAO;YACR,CAAC;YAED,MAAM,QAAQ,GAAG,gBAAgB,CAAC,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC,aAAa,GAAG,CAAC,CAAC,CAAC;YAE/E,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACf,MAAM,CAAC,yDAAyD,EAAE,OAAO,CAAC,CAAC;gBAC3E,OAAO;YACR,CAAC;YAED,MAAM,mBAAmB,GAAG,GAAG,QAAQ,uBAAuB,IAAI,EAAE,CAAC;YAErE,MAAM,gBAAgB,GAAG,MAAM,GAAG,CAAC,UAAU,CAAC;gBAC7C,aAAa,EAAE,kBAAkB;gBACjC,WAAW,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;oBAC7B,MAAM,MAAM,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;oBAClD,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;gBAC9C,CAAC;aACD,CAAC,CAAC;YAEH,IAAI,gBAAgB,CAAC,SAAS,EAAE,CAAC;gBAChC,MAAM,CAAC,uBAAuB,EAAE,MAAM,CAAC,CAAC;gBACxC,OAAO;YACR,CAAC;QACF,CAAC;KACD,CAAC,CAAC;AACJ,CAAC"}

package/dist/index.d.ts

@@ -4,10 +4,9 @@
  * Dives into specialized flow states running as isolated pi processes.
  * Each flow receives a forked snapshot of the current session context.
  */
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import { logWarn, logError } from "./config/log.js";
 export { logWarn, logError };
-export { compressToolResults, stripBatchReadToolCalls } from "./snapshot/snapshot.js";
 export { type FlowColorConfig } from "./tui/flow-colors.js";
 export default function (pi: ExtensionAPI): void;
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAoB,MAAM,+BAA+B,CAAC;AA+CpF,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;AAoF7B,OAAO,EAAE,mBAAmB,EAAE,uBAAuB,EAAE,MAAM,wBAAwB,CAAC;AACtF,OAAO,EAAE,KAAK,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAM5D,MAAM,CAAC,OAAO,WAAW,EAAE,EAAE,YAAY,QAiaxC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAoB,MAAM,iCAAiC,CAAC;AAkDtF,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;AAsL7B,OAAO,EAAE,KAAK,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAM5D,MAAM,CAAC,OAAO,WAAW,EAAE,EAAE,YAAY,QAwexC"}