pi-all-notification 1.0.4 → 1.0.5

package/extensions/index.ts

@@ -18,7 +18,7 @@ import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
 import { join, dirname } from "node:path";
 import { homedir } from "node:os";
 import { execSync } from "node:child_process";
-import { describeTask, settleNotification } from "./settle.ts";
+import { describeBeforeAgentStart, settleNotification } from "./settle.ts";
 
 // ─── Types ────────────────────────────────────────────────────
 
@@ -763,7 +763,7 @@ export default function (pi: ExtensionAPI) {
   // Capture the prompt for the completion notification: `agent_settled` carries
   // no messages, and this is the last point the prompt text is available.
   pi.on("before_agent_start", (event, _ctx) => {
-    lastTaskDescription = describeTask((event as { prompt?: unknown }).prompt);
+    lastTaskDescription = describeBeforeAgentStart(event);
   });
 
   pi.on("agent_start", (_event, ctx) => {

package/extensions/settle.ts

@@ -22,6 +22,32 @@ export function describeTask(prompt: unknown): string | null {
   return firstLine.length > 40 ? `${firstLine.slice(0, 37)}...` : firstLine;
 }
 
+/**
+ * Task text from `before_agent_start`.
+ *
+ * Pi 1.3 also fires this event for runs started by a custom message, where the
+ * event carries `message` instead of `prompt`.
+ */
+export function describeBeforeAgentStart(event: unknown): string | null {
+  if (!event || typeof event !== "object") return null;
+  const { prompt, message } = event as { prompt?: unknown; message?: unknown };
+  const fromPrompt = describeTask(prompt);
+  if (fromPrompt) return fromPrompt;
+  if (!message || typeof message !== "object") return null;
+  const content = (message as { content?: unknown }).content;
+  if (typeof content === "string") return describeTask(content);
+  if (!Array.isArray(content)) return null;
+  const text = content
+    .map((item) =>
+      item && typeof item === "object" && (item as { type?: unknown }).type === "text"
+        ? (item as { text?: unknown }).text
+        : undefined,
+    )
+    .filter((item): item is string => typeof item === "string")
+    .join("\n");
+  return describeTask(text);
+}
+
 /**
  * The notification for a settled run.
  *

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-all-notification",
-  "version": "1.0.4",
+  "version": "1.0.5",
   "description": "Multi-channel notification for pi: macOS desktop, Bark push, custom webhook",
   "keywords": [
     "pi-package",

package/tests/settle.test.ts

@@ -1,6 +1,6 @@
 import { describe, it } from "node:test";
 import assert from "node:assert/strict";
-import { describeTask, settleNotification } from "../extensions/settle.ts";
+import { describeBeforeAgentStart, describeTask, settleNotification } from "../extensions/settle.ts";
 
 describe("describeTask", () => {
   it("takes the first line of the prompt", () => {
@@ -22,6 +22,38 @@ describe("describeTask", () => {
   });
 });
 
+describe("describeBeforeAgentStart", () => {
+  it("reads the prompt from a user run", () => {
+    assert.equal(describeBeforeAgentStart({ messageType: "user", prompt: "Fix the build" }), "Fix the build");
+  });
+
+  it("reads text content from a custom-message run", () => {
+    const event = {
+      messageType: "custom",
+      message: { role: "custom", content: [{ type: "text", text: "Review the release\nand tag it" }] },
+    };
+    assert.equal(describeBeforeAgentStart(event), "Review the release");
+  });
+
+  it("reads string content from a custom-message run", () => {
+    const event = {
+      messageType: "custom",
+      message: { role: "custom", content: "Review the release" },
+    };
+    assert.equal(describeBeforeAgentStart(event), "Review the release");
+  });
+
+  it("ignores non-text custom content", () => {
+    assert.equal(
+      describeBeforeAgentStart({
+        messageType: "custom",
+        message: { role: "custom", content: [{ type: "image", data: "..." }] },
+      }),
+      null,
+    );
+  });
+});
+
 describe("settleNotification", () => {
   it("reports a natural finish as complete", () => {
     const note = settleNotification({ aborted: false }, "pi-plugins", "Run the tests");