pi-advisor-flow 0.1.6 → 0.1.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-advisor-flow",
-  "version": "0.1.6",
+  "version": "0.1.7",
   "license": "MIT",
   "repository": {
     "type": "git",

package/src/commands.ts

@@ -8,6 +8,7 @@ import {
   contextMaxCharsRef, loadConfig, saveConfig, parseArgs, splitRef,
 } from "./config.js";
 import { AdvisorSettingsSelector, SearchableModelSelector, type AdvisorSettings, type ContextPreset } from "./ui.js";
+import { herdrAdvisorActivity } from "./herdr.js";
 import { adviceForDisplay, consult, renderAdvisorCallBox, resolveAdvisorRequest } from "./tools.js";
 
 const EFFORT_LEVELS = ["Default (Model Default)", "off", "minimal", "low", "medium", "high", "xhigh", "max"];
@@ -44,6 +45,7 @@ export const registerCommands = (pi: ExtensionAPI, dependencies: { consult?: typ
   pi.on("session_shutdown", () => {
     for (const controller of manualConsultations) controller.abort();
     manualConsultations.clear();
+    herdrAdvisorActivity.clear();
   });
 
   pi.registerCommand("advisor-manual", {
@@ -58,6 +60,7 @@ export const registerCommands = (pi: ExtensionAPI, dependencies: { consult?: typ
       manualConsultations.add(controller);
       pi.appendEntry?.("advisor-manual-call", { question });
 
+      herdrAdvisorActivity.start();
       void requestAdvisor(ctx, question, controller.signal)
         .then(({ advice }) => {
           if (controller.signal.aborted) return;
@@ -78,7 +81,10 @@ export const registerCommands = (pi: ExtensionAPI, dependencies: { consult?: typ
           const message = error instanceof Error ? error.message : String(error);
           if (ctx.hasUI) ctx.ui.notify(`Advisor consultation failed: ${message}`, "error");
         })
-        .finally(() => manualConsultations.delete(controller));
+        .finally(() => {
+          manualConsultations.delete(controller);
+          herdrAdvisorActivity.finish();
+        });
     },
   });
 

package/src/herdr.ts

@@ -0,0 +1,87 @@
+import net from "node:net";
+
+const SOURCE = "pi-advisor:advisor-activity";
+const HERDR_PI_SOURCE = "herdr:pi";
+let sequence = Date.now() * 1_000;
+
+const nextSequence = () => ++sequence;
+
+type HerdrRequest = {
+  id: string;
+  method: "pane.report_metadata";
+  params: {
+    pane_id: string;
+    source: string;
+    agent: "pi";
+    applies_to_source: string;
+    state_labels?: { working: string };
+    clear_state_labels?: true;
+    seq: number;
+  };
+};
+
+type Report = (request: HerdrRequest) => void;
+
+const sendToHerdr: Report = (request) => {
+  if (process.env.HERDR_ENV !== "1") return;
+  const paneId = process.env.HERDR_PANE_ID;
+  const socketPath = process.env.HERDR_SOCKET_PATH;
+  if (!paneId || !socketPath) return;
+
+  const endpoint = process.platform === "win32" ? `\\\\.\\pipe\\${socketPath}` : socketPath;
+  const socket = net.createConnection(endpoint);
+  const timeout = setTimeout(() => socket.destroy(), 500);
+  timeout.unref?.();
+  socket.once("connect", () => socket.write(`${JSON.stringify(request)}\n`));
+  socket.once("data", () => socket.destroy());
+  socket.once("error", () => socket.destroy());
+  socket.once("close", () => clearTimeout(timeout));
+};
+
+export class HerdrAdvisorActivity {
+  #activeConsultations = 0;
+
+  constructor(private readonly report: Report = sendToHerdr) {}
+
+  start() {
+    this.#activeConsultations += 1;
+    if (this.#activeConsultations === 1) this.safeReport(false);
+  }
+
+  finish() {
+    if (this.#activeConsultations === 0) return;
+    this.#activeConsultations -= 1;
+    if (this.#activeConsultations === 0) this.safeReport(true);
+  }
+
+  clear() {
+    if (this.#activeConsultations === 0) return;
+    this.#activeConsultations = 0;
+    this.safeReport(true);
+  }
+
+  private safeReport(clear: boolean) {
+    try {
+      this.report(this.request(clear));
+    } catch {
+      // Herdr is optional; an unavailable integration must never affect advice.
+    }
+  }
+
+  private request(clear: boolean): HerdrRequest {
+    return {
+      id: `${SOURCE}:${nextSequence()}`,
+      method: "pane.report_metadata",
+      params: {
+        pane_id: process.env.HERDR_PANE_ID ?? "",
+        source: SOURCE,
+        agent: "pi",
+        applies_to_source: HERDR_PI_SOURCE,
+        ...(clear ? { clear_state_labels: true } : { state_labels: { working: "seeking advice" } }),
+        seq: nextSequence(),
+      },
+    };
+  }
+}
+
+export const herdrAdvisorActivity = new HerdrAdvisorActivity();

package/src/tools.ts

@@ -7,6 +7,7 @@ import {
   advisorRef, advisorEffortRef, contextMaxCharsRef, loadConfig, splitRef,
 } from "./config.js";
 import { recentConversation, textFrom } from "./conversation.js";
+import { herdrAdvisorActivity } from "./herdr.js";
 
 export const SPINNER_FRAMES = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"];
 export const resolveAdvisorRequest = (question?: string) => question?.trim() || undefined;
@@ -159,16 +160,21 @@ export const registerAdvisorTool = (pi: ExtensionAPI) => {
     },
     async execute(_id, params, signal, onUpdate, ctx) {
       const question = resolveAdvisorRequest(params.question);
-      const { advice, thinkingText } = await consult(ctx, question, signal, (t, tx) => {
-        onUpdate?.({
-          content: [{ type: "text", text: tx }],
-          details: { thinking: t, text: tx, advisor: advisorRef, question },
+      herdrAdvisorActivity.start();
+      try {
+        const { advice, thinkingText } = await consult(ctx, question, signal, (t, tx) => {
+          onUpdate?.({
+            content: [{ type: "text", text: tx }],
+            details: { thinking: t, text: tx, advisor: advisorRef, question },
+          });
         });
-      });
-      return {
-        content: [{ type: "text", text: `Advisor (${advisorRef})\n\n${advice}` }],
-        details: { thinking: thinkingText, text: advice, advisor: advisorRef, question },
-      };
+        return {
+          content: [{ type: "text", text: `Advisor (${advisorRef})\n\n${advice}` }],
+          details: { thinking: thinkingText, text: advice, advisor: advisorRef, question },
+        };
+      } finally {
+        herdrAdvisorActivity.finish();
+      }
     },
   });
 };