pi-agent-flow 1.8.1 → 1.8.3

package/dist/types.js

@@ -0,0 +1,143 @@
+/**
+ * Shared type definitions for the flow-state delegation extension.
+ */
+import { getFlowFinalText } from "./runner-events.js";
+/** Create an empty UsageStats object. */
+export function emptyFlowUsage() {
+    return { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, cost: 0, contextTokens: 0, turns: 0, toolCalls: 0, smoothedTps: 0 };
+}
+/** Sum usage across multiple results. */
+export function aggregateFlowUsage(results) {
+    const total = emptyFlowUsage();
+    for (const r of results) {
+        total.input += r.usage.input;
+        total.output += r.usage.output;
+        total.cacheRead += r.usage.cacheRead;
+        total.cacheWrite += r.usage.cacheWrite;
+        total.cost += r.usage.cost;
+        total.turns += r.usage.turns;
+        total.toolCalls += r.usage.toolCalls;
+        if ((r.usage.smoothedTps ?? 0) > (total.smoothedTps ?? 0)) {
+            total.smoothedTps = r.usage.smoothedTps;
+        }
+    }
+    return total;
+}
+/** Whether the child emitted a final assistant text response. */
+export function hasFlowOutput(r) {
+    return getFlowFinalText(r.messages).trim().length > 0;
+}
+/** Whether the child semantically completed the run. */
+export function isFlowComplete(r) {
+    return Boolean(r.sawAgentEnd) && hasFlowOutput(r);
+}
+/** Whether a result should be treated as successful by the wrapper/UI. */
+export function isFlowSuccess(r) {
+    if (r.exitCode === -1)
+        return false;
+    if (isFlowComplete(r))
+        return true;
+    return r.exitCode === 0 && r.stopReason !== "error" && r.stopReason !== "aborted" && r.stopReason !== "timeout";
+}
+/** Whether a result represents an error. */
+export function isFlowError(r) {
+    if (r.exitCode === -1)
+        return false;
+    return !isFlowSuccess(r);
+}
+/** Reconcile process exit status with semantic completion observed from Pi's event stream. */
+export function normalizeFlowResult(result, wasAborted) {
+    const hasSemanticSuccess = isFlowComplete(result);
+    if (wasAborted) {
+        if (hasSemanticSuccess) {
+            result.exitCode = 0;
+            if (result.stopReason === "aborted")
+                result.stopReason = undefined;
+            if (result.errorMessage === "Flow was aborted.") {
+                result.errorMessage = undefined;
+            }
+        }
+        else {
+            result.exitCode = 130;
+            result.stopReason = "aborted";
+            result.errorMessage = "Flow was aborted.";
+            if (!result.stderr.trim())
+                result.stderr = "Flow was aborted.";
+        }
+        return result;
+    }
+    if (result.exitCode > 0) {
+        if (hasSemanticSuccess) {
+            result.exitCode = 0;
+            if (result.stopReason === "error")
+                result.stopReason = undefined;
+            if (result.errorMessage === result.stderr.trim()) {
+                result.errorMessage = undefined;
+            }
+        }
+        else {
+            if (!result.stopReason)
+                result.stopReason = "error";
+            if (!result.errorMessage && result.stderr.trim()) {
+                result.errorMessage = result.stderr.trim();
+            }
+        }
+    }
+    return result;
+}
+/** Extract the last assistant text from a message history. */
+export function getFlowOutput(messages) {
+    return getFlowFinalText(messages);
+}
+/** Extract all display-worthy items from a message history. */
+export function getFlowDisplayItems(messages) {
+    const items = [];
+    for (const msg of messages) {
+        if (msg.role === "assistant") {
+            for (const part of msg.content) {
+                if (part.type === "text") {
+                    items.push({ type: "text", text: part.text });
+                }
+                else if (part.type === "toolCall") {
+                    const name = part.name ?? part.toolName ?? "unknown";
+                    const args = (part.arguments ?? part.input ?? {});
+                    items.push({ type: "toolCall", name, args });
+                }
+            }
+        }
+    }
+    return items;
+}
+/** Extract the last tool call from message history. */
+export function getLastToolCall(messages) {
+    for (let i = messages.length - 1; i >= 0; i--) {
+        const msg = messages[i];
+        if (msg.role === "assistant") {
+            for (let j = msg.content.length - 1; j >= 0; j--) {
+                const part = msg.content[j];
+                if (part.type === "toolCall") {
+                    const name = part.name ?? part.toolName ?? "unknown";
+                    const args = (part.arguments ?? part.input ?? {});
+                    return { type: "toolCall", name, args };
+                }
+            }
+        }
+    }
+    return undefined;
+}
+/** Extract the last assistant text from message history. */
+export function getLastAssistantText(messages) {
+    for (let i = messages.length - 1; i >= 0; i--) {
+        const msg = messages[i];
+        if (msg.role === "assistant") {
+            for (let j = msg.content.length - 1; j >= 0; j--) {
+                const part = msg.content[j];
+                if (part.type === "text" && part.text.trim()) {
+                    return part.text.trim();
+                }
+            }
+        }
+    }
+    return "";
+}
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AA8ItD,yCAAyC;AACzC,MAAM,UAAU,cAAc;IAC7B,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;AAChI,CAAC;AAED,yCAAyC;AACzC,MAAM,UAAU,kBAAkB,CAAC,OAAuB;IACzD,MAAM,KAAK,GAAG,cAAc,EAAE,CAAC;IAC/B,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;QACzB,KAAK,CAAC,KAAK,IAAI,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC;QAC7B,KAAK,CAAC,MAAM,IAAI,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC;QAC/B,KAAK,CAAC,SAAS,IAAI,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC;QACrC,KAAK,CAAC,UAAU,IAAI,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC;QACvC,KAAK,CAAC,IAAI,IAAI,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC;QAC3B,KAAK,CAAC,KAAK,IAAI,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC;QAC7B,KAAK,CAAC,SAAS,IAAI,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC;QACrC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,IAAI,CAAC,CAAC,EAAE,CAAC;YAC3D,KAAK,CAAC,WAAW,GAAG,CAAC,CAAC,KAAK,CAAC,WAAW,CAAC;QACzC,CAAC;IACF,CAAC;IACD,OAAO,KAAK,CAAC;AACd,CAAC;AAED,iEAAiE;AACjE,MAAM,UAAU,aAAa,CAAC,CAAiC;IAC9D,OAAO,gBAAgB,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC;AACvD,CAAC;AAED,wDAAwD;AACxD,MAAM,UAAU,cAAc,CAAC,CAAiD;IAC/E,OAAO,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,IAAI,aAAa,CAAC,CAAC,CAAC,CAAC;AACnD,CAAC;AAED,0EAA0E;AAC1E,MAAM,UAAU,aAAa,CAAC,CAAe;IAC5C,IAAI,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC;QAAE,OAAO,KAAK,CAAC;IACpC,IAAI,cAAc,CAAC,CAAC,CAAC;QAAE,OAAO,IAAI,CAAC;IACnC,OAAO,CAAC,CAAC,QAAQ,KAAK,CAAC,IAAI,CAAC,CAAC,UAAU,KAAK,OAAO,IAAI,CAAC,CAAC,UAAU,KAAK,SAAS,IAAI,CAAC,CAAC,UAAU,KAAK,SAAS,CAAC;AACjH,CAAC;AAED,4CAA4C;AAC5C,MAAM,UAAU,WAAW,CAAC,CAAe;IAC1C,IAAI,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC;QAAE,OAAO,KAAK,CAAC;IACpC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;AAC1B,CAAC;AAED,8FAA8F;AAC9F,MAAM,UAAU,mBAAmB,CAAC,MAAoB,EAAE,UAAmB;IAC5E,MAAM,kBAAkB,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;IAElD,IAAI,UAAU,EAAE,CAAC;QAChB,IAAI,kBAAkB,EAAE,CAAC;YACxB,MAAM,CAAC,QAAQ,GAAG,CAAC,CAAC;YACpB,IAAI,MAAM,CAAC,UAAU,KAAK,SAAS;gBAAE,MAAM,CAAC,UAAU,GAAG,SAAS,CAAC;YACnE,IAAI,MAAM,CAAC,YAAY,KAAK,mBAAmB,EAAE,CAAC;gBACjD,MAAM,CAAC,YAAY,GAAG,SAAS,CAAC;YACjC,CAAC;QACF,CAAC;aAAM,CAAC;YACP,MAAM,CAAC,QAAQ,GAAG,GAAG,CAAC;YACtB,MAAM,CAAC,UAAU,GAAG,SAAS,CAAC;YAC9B,MAAM,CAAC,YAAY,GAAG,mBAAmB,CAAC;YAC1C,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE;gBAAE,MAAM,CAAC,MAAM,GAAG,mBAAmB,CAAC;QAChE,CAAC;QACD,OAAO,MAAM,CAAC;IACf,CAAC;IAED,IAAI,MAAM,CAAC,QAAQ,GAAG,CAAC,EAAE,CAAC;QACzB,IAAI,kBAAkB,EAAE,CAAC;YACxB,MAAM,CAAC,QAAQ,GAAG,CAAC,CAAC;YACpB,IAAI,MAAM,CAAC,UAAU,KAAK,OAAO;gBAAE,MAAM,CAAC,UAAU,GAAG,SAAS,CAAC;YACjE,IAAI,MAAM,CAAC,YAAY,KAAK,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBAClD,MAAM,CAAC,YAAY,GAAG,SAAS,CAAC;YACjC,CAAC;QACF,CAAC;aAAM,CAAC;YACP,IAAI,CAAC,MAAM,CAAC,UAAU;gBAAE,MAAM,CAAC,UAAU,GAAG,OAAO,CAAC;YACpD,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBAClD,MAAM,CAAC,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,CAAC;QACF,CAAC;IACF,CAAC;IAED,OAAO,MAAM,CAAC;AACf,CAAC;AAED,8DAA8D;AAC9D,MAAM,UAAU,aAAa,CAAC,QAAmB;IAChD,OAAO,gBAAgB,CAAC,QAAQ,CAAC,CAAC;AACnC,CAAC;AAED,+DAA+D;AAC/D,MAAM,UAAU,mBAAmB,CAAC,QAAmB;IACtD,MAAM,KAAK,GAAkB,EAAE,CAAC;IAChC,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC5B,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC9B,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;gBAChC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBAC1B,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC/C,CAAC;qBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBACrC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,QAAQ,IAAI,SAAS,CAAC;oBACrD,MAAM,IAAI,GAAG,CAAC,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,KAAK,IAAI,EAAE,CAA4B,CAAC;oBAC7E,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC9C,CAAC;YACF,CAAC;QACF,CAAC;IACF,CAAC;IACD,OAAO,KAAK,CAAC;AACd,CAAC;AAED,uDAAuD;AACvD,MAAM,UAAU,eAAe,CAAC,QAAmB;IAClD,KAAK,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC/C,MAAM,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;QACxB,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC9B,KAAK,IAAI,CAAC,GAAG,GAAG,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBAClD,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBAC5B,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAC9B,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,QAAQ,IAAI,SAAS,CAAC;oBACrD,MAAM,IAAI,GAAG,CAAC,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,KAAK,IAAI,EAAE,CAA4B,CAAC;oBAC7E,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;gBACzC,CAAC;YACF,CAAC;QACF,CAAC;IACF,CAAC;IACD,OAAO,SAAS,CAAC;AAClB,CAAC;AAED,4DAA4D;AAC5D,MAAM,UAAU,oBAAoB,CAAC,QAAmB;IACvD,KAAK,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC/C,MAAM,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;QACxB,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC9B,KAAK,IAAI,CAAC,GAAG,GAAG,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBAClD,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBAC5B,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;oBAC9C,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACzB,CAAC;YACF,CAAC;QACF,CAAC;IACF,CAAC;IACD,OAAO,EAAE,CAAC;AACX,CAAC"}

package/dist/web-tool.d.ts

@@ -0,0 +1,35 @@
+import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
+import { type Static } from "@sinclair/typebox";
+export declare const webSchema: any;
+type WebParams = Static<typeof webSchema>;
+export declare function createWebTool(): {
+    name: string;
+    label: string;
+    description: string;
+    promptSnippet: string;
+    promptGuidelines: string[];
+    parameters: any;
+    execute(_toolCallId: string, params: WebParams, signal: AbortSignal | undefined, _onUpdate: unknown, ctx: ExtensionContext): Promise<{
+        content: {
+            type: "text";
+            text: string;
+        }[];
+        details: {
+            ops: Record<string, unknown>[];
+        };
+    }>;
+};
+export declare function setWebToolsActive(pi: ExtensionAPI, enabled: boolean): void;
+export declare function extractSnippet(raw: string, title: string): string;
+export declare function validateFetchUrl(url: string): void;
+export declare function htmlToMarkdown(html: string, baseUrl: string): {
+    title: string;
+    markdown: string;
+};
+export declare function trimLargeDocument(markdown: string, maxChars: number): string;
+export declare function stripMarkdownFormatting(markdown: string): string;
+export declare function urlToHash(url: string): string;
+export declare function looksLikeUrlPrompt(prompt: string | undefined): boolean;
+export declare function looksLikeWebSearchPrompt(prompt: string | undefined): boolean;
+export {};
+//# sourceMappingURL=web-tool.d.ts.map

package/dist/web-tool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"web-tool.d.ts","sourceRoot":"","sources":["../src/web-tool.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAC;AAEpF,OAAO,EAAQ,KAAK,MAAM,EAAE,MAAM,mBAAmB,CAAC;AA4BtD,eAAO,MAAM,SAAS,KAKpB,CAAC;AAEH,KAAK,SAAS,GAAG,MAAM,CAAC,OAAO,SAAS,CAAC,CAAC;AA8B1C,wBAAgB,aAAa;;;;;;;yBAgBb,MAAM,UACX,SAAS,UACT,WAAW,GAAG,SAAS,aACpB,OAAO,OACb,gBAAgB;;;;;;;;;EAKvB;AAMD,wBAAgB,iBAAiB,CAAC,EAAE,EAAE,YAAY,EAAE,OAAO,EAAE,OAAO,QAKnE;AA6XD,wBAAgB,cAAc,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,CAMjE;AAED,wBAAgB,gBAAgB,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI,CA6BlD;AAED,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAA;CAAE,CAyCjG;AAED,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,CAY5E;AAOD,wBAAgB,uBAAuB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,CA4BhE;AAED,wBAAgB,SAAS,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAE7C;AAgBD,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,GAAG,OAAO,CAGtE;AAED,wBAAgB,wBAAwB,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,GAAG,OAAO,CAkB5E"}