tau-coding-agent 0.1.4 → 0.1.6

package/extensions/insights.ts

@@ -1,4 +1,5 @@
-import { complete, type Api, type Model, type UserMessage } from "@earendil-works/pi-ai";
+import { complete } from "@earendil-works/pi-ai/compat";
+import type { Api, Model, UserMessage } from "@earendil-works/pi-ai";
 import {
   BorderedLoader,
   SessionManager,
@@ -26,6 +27,7 @@ import {
 } from "@earendil-works/pi-tui";
 import { createHash, randomBytes } from "node:crypto";
 import fs from "node:fs/promises";
+import os from "node:os";
 import path from "node:path";
 
 // --- Constants ---
@@ -140,6 +142,7 @@ type ConfiguredModelSelection = {
   model: Model<Api>;
   apiKey?: string;
   headers?: Record<string, string>;
+  env?: Record<string, string>;
 };
 
 type ReadonlySessionManager = Pick<
@@ -480,8 +483,8 @@ async function runInsightsCommand(
     return;
   }
 
-  if (!ctx.hasUI) {
-    ctx.ui.notify("/insights currently requires interactive mode", "warning");
+  if (ctx.mode !== "tui") {
+    ctx.ui.notify("/insights currently requires TUI mode", "warning");
     return;
   }
 
@@ -561,9 +564,37 @@ async function runInsightsCommand(
     return;
   }
 
+  let reportPath: string | undefined;
+  try {
+    reportPath = await saveInsightsReport(result.reportMarkdown, result.generatedAt);
+  } catch (error) {
+    const message = error instanceof Error ? error.message : "Unknown error";
+    ctx.ui.notify(`Failed to save insights report: ${message}`, "warning");
+  }
+
   await ctx.ui.custom<void>((tui, theme, _kb, done) => {
     return new InsightsReportComponent(result, tui, theme, done);
   });
+
+  if (reportPath) {
+    ctx.ui.notify(`Insights report saved to ${reportPath}`, "info");
+  }
+}
+
+async function saveInsightsReport(reportMarkdown: string, generatedAt: string): Promise<string> {
+  const timestamp = generatedAt.replace(/[:.]/g, "-");
+  const reportPath = path.join(os.tmpdir(), `tau-insights-${timestamp}.md`);
+  const reportFile = await fs.open(reportPath, "wx", 0o600);
+
+  try {
+    await reportFile.writeFile(`${reportMarkdown.trimEnd()}\n`, "utf8");
+    await reportFile.close();
+    return reportPath;
+  } catch (error) {
+    await reportFile.close().catch(() => undefined);
+    await fs.rm(reportPath, { force: true }).catch(() => undefined);
+    throw error;
+  }
 }
 
 async function runInsightsPipeline(
@@ -748,7 +779,12 @@ async function getConfiguredModelSelection(
 
   const auth = await ctx.modelRegistry.getApiKeyAndHeaders(ctx.model);
   if (!auth.ok) return null;
-  return { model: ctx.model, apiKey: auth.apiKey, headers: auth.headers };
+  return {
+    model: ctx.model,
+    apiKey: auth.apiKey,
+    headers: auth.headers,
+    env: auth.env,
+  };
 }
 
 async function listTargets(
@@ -1341,6 +1377,7 @@ async function extractFacet(
     {
       apiKey: selection.apiKey,
       headers: selection.headers,
+      env: selection.env,
       signal,
     },
   );
@@ -1513,6 +1550,7 @@ async function synthesizeReport(
     {
       apiKey: selection.apiKey,
       headers: selection.headers,
+      env: selection.env,
       signal,
     },
   );

package/extensions/loop.ts

@@ -7,7 +7,8 @@
  */
 
 import { Type } from "typebox";
-import { complete, type Api, type Model, type UserMessage } from "@earendil-works/pi-ai";
+import { complete } from "@earendil-works/pi-ai/compat";
+import type { Api, Model, UserMessage } from "@earendil-works/pi-ai";
 import {
   compact,
   defineTool,
@@ -53,6 +54,7 @@ type ModelRequestAuth = {
   model: Model<Api>;
   apiKey?: string;
   headers?: Record<string, string>;
+  env?: Record<string, string>;
 };
 
 async function withPromptSignal<T>(pi: ExtensionAPI, run: () => Promise<T>): Promise<T> {
@@ -136,13 +138,20 @@ async function selectSummaryModel(ctx: ExtensionContext): Promise<ModelRequestAu
 
       const auth = await ctx.modelRegistry.getApiKeyAndHeaders(candidate);
       if (auth.ok) {
-        return { model: candidate, apiKey: auth.apiKey, headers: auth.headers };
+        return {
+          model: candidate,
+          apiKey: auth.apiKey,
+          headers: auth.headers,
+          env: auth.env,
+        };
       }
     }
   }
 
   const auth = await ctx.modelRegistry.getApiKeyAndHeaders(ctx.model);
-  return auth.ok ? { model: ctx.model, apiKey: auth.apiKey, headers: auth.headers } : null;
+  return auth.ok
+    ? { model: ctx.model, apiKey: auth.apiKey, headers: auth.headers, env: auth.env }
+    : null;
 }
 
 async function summarizeBreakoutCondition(
@@ -164,7 +173,7 @@ async function summarizeBreakoutCondition(
   const response = await complete(
     selection.model,
     { systemPrompt: SUMMARY_SYSTEM_PROMPT, messages: [userMessage] },
-    { apiKey: selection.apiKey, headers: selection.headers },
+    { apiKey: selection.apiKey, headers: selection.headers, env: selection.env },
   );
 
   if (response.stopReason === "aborted" || response.stopReason === "error") {
@@ -213,6 +222,8 @@ async function loadState(ctx: ExtensionContext): Promise<LoopStateData> {
 
 export default function loopExtension(pi: ExtensionAPI): void {
   let loopState: LoopStateData = { active: false };
+  let lastAgentEndMessages: Array<{ role?: string; stopReason?: string }> = [];
+  let pendingLoopPrompt: ReturnType<typeof setImmediate> | undefined;
 
   function persistState(state: LoopStateData): void {
     pi.appendEntry(LOOP_STATE_ENTRY, state);
@@ -271,6 +282,23 @@ export default function loopExtension(pi: ExtensionAPI): void {
     );
   }
 
+  function scheduleLoopPrompt(ctx: ExtensionContext): void {
+    if (pendingLoopPrompt) return;
+
+    pendingLoopPrompt = setImmediate(() => {
+      pendingLoopPrompt = undefined;
+      if (!loopState.active || !ctx.isIdle()) return;
+      triggerLoopPrompt(ctx);
+    });
+    pendingLoopPrompt.unref?.();
+  }
+
+  function cancelScheduledLoopPrompt(): void {
+    if (!pendingLoopPrompt) return;
+    clearImmediate(pendingLoopPrompt);
+    pendingLoopPrompt = undefined;
+  }
+
   async function showLoopSelector(ctx: ExtensionContext): Promise<LoopStateData | null> {
     const items: SelectItem[] = LOOP_PRESETS.map((preset) => ({
       value: preset.value,
@@ -385,11 +413,11 @@ export default function loopExtension(pi: ExtensionAPI): void {
       name: "signal_loop_success",
       label: "Signal Loop Success",
       description:
-        "Stop the active loop when the breakout condition is satisfied. Only call this tool when explicitly instructed to do so by the user, tool or system prompt.",
+        "Stop the active loop when the breakout condition is satisfied. Only call signal_loop_success when explicitly instructed to do so by the user, tool, or system prompt.",
       promptSnippet: "Signal that the active /loop breakout condition has been satisfied",
       promptGuidelines: [
-        "Call this tool only when the active loop's breakout condition is actually satisfied.",
-        "Do not call this tool unless a loop is currently active.",
+        "Call signal_loop_success only when the active loop's breakout condition is actually satisfied.",
+        "Do not call signal_loop_success unless a loop is currently active.",
       ],
       parameters: Type.Object({}),
       async execute(_toolCallId, _params, _signal, _onUpdate, ctx) {
@@ -416,7 +444,7 @@ export default function loopExtension(pi: ExtensionAPI): void {
     handler: async (args, ctx) => {
       let nextState = parseArgs(args);
       if (!nextState) {
-        if (!ctx.hasUI) {
+        if (ctx.mode !== "tui") {
           ctx.ui.notify("Usage: /loop tests | /loop custom <condition> | /loop self", "warning");
           return;
         }
@@ -458,10 +486,14 @@ export default function loopExtension(pi: ExtensionAPI): void {
     },
   });
 
-  pi.on("agent_end", async (event, ctx) => {
+  pi.on("agent_end", async (event) => {
+    lastAgentEndMessages = event.messages;
+  });
+
+  pi.on("agent_settled", async (_event, ctx) => {
     if (!loopState.active) return;
 
-    if (ctx.hasUI && wasLastAssistantAborted(event.messages)) {
+    if (ctx.hasUI && wasLastAssistantAborted(lastAgentEndMessages)) {
       const confirm = await withPromptSignal(pi, () =>
         ctx.ui.confirm("Break active loop?", "Operation aborted. Break out of the loop?"),
       );
@@ -471,13 +503,13 @@ export default function loopExtension(pi: ExtensionAPI): void {
       }
     }
 
-    triggerLoopPrompt(ctx);
+    scheduleLoopPrompt(ctx);
   });
 
   pi.on("session_before_compact", async (event, ctx) => {
     if (!loopState.active || !loopState.mode || !ctx.model) return;
     const auth = await ctx.modelRegistry.getApiKeyAndHeaders(ctx.model);
-    if (!auth.ok || !auth.apiKey) return;
+    if (!auth.ok) return;
 
     const instructionParts = [
       event.customInstructions,
@@ -494,6 +526,9 @@ export default function loopExtension(pi: ExtensionAPI): void {
         auth.headers,
         instructionParts,
         event.signal,
+        ctx.thinkingLevel,
+        undefined, // Use Pi's default summarization stream.
+        auth.env,
       );
       return { compaction };
     } catch (error) {
@@ -526,4 +561,8 @@ export default function loopExtension(pi: ExtensionAPI): void {
   pi.on("session_start", async (_event, ctx) => {
     await restoreLoopState(ctx);
   });
+
+  pi.on("session_shutdown", async () => {
+    cancelScheduledLoopPrompt();
+  });
 }

package/extensions/memory.ts

@@ -1,8 +1,9 @@
+import { complete } from "@earendil-works/pi-ai/compat";
 import {
-  complete,
   StringEnum,
   type Api,
   type Model,
+  type Usage,
   type UserMessage,
 } from "@earendil-works/pi-ai";
 import {
@@ -176,6 +177,11 @@ type DreamResult = {
   summaryPath: string | null;
 };
 
+type DreamExecution = {
+  result: DreamResult;
+  usage?: Usage;
+};
+
 type AutoDreamStatus = {
   coreLines: number;
   coreChars: number;
@@ -194,6 +200,7 @@ type ModelSelection = {
   model: Model<Api>;
   apiKey?: string;
   headers?: Record<string, string>;
+  env?: Record<string, string>;
 };
 
 class MemoryReadmeMissingError extends Error {
@@ -209,12 +216,17 @@ async function toolDream(
   cwd: string,
   ctx: ExtensionContext,
   reason?: string,
-): Promise<{ content: Array<{ type: "text"; text: string }>; details: DreamResult }> {
+): Promise<{
+  content: Array<{ type: "text"; text: string }>;
+  details: DreamResult;
+  usage?: Usage;
+}> {
   try {
-    const result = await runMemoryDream(cwd, ctx, reason);
+    const execution = await runMemoryDream(cwd, ctx, reason);
     return {
-      content: [{ type: "text", text: result.summary }],
-      details: result,
+      content: [{ type: "text", text: execution.result.summary }],
+      details: execution.result,
+      usage: execution.usage,
     };
   } catch (error) {
     if (error instanceof MemoryReadmeMissingError) {
@@ -374,7 +386,7 @@ async function runMemoryDream(
   cwd: string,
   ctx: ExtensionContext,
   reason?: string,
-): Promise<DreamResult> {
+): Promise<DreamExecution> {
   const paths = getMemoryPaths(cwd);
   const lockPaths = [
     paths.readmeFile,
@@ -399,13 +411,15 @@ async function runMemoryDream(
     const summary = "Memory dream: nothing to consolidate.";
     notify(ctx, summary, "info");
     return {
-      summary,
-      updatedBlocks: [],
-      consumedLogs: 0,
-      consumedCompactions: 0,
-      lastDreamAt: snapshot.state?.lastDreamAt ?? nowIso(),
-      lastDreamedLogAt: snapshot.state?.lastDreamedLogAt ?? null,
-      summaryPath: null,
+      result: {
+        summary,
+        updatedBlocks: [],
+        consumedLogs: 0,
+        consumedCompactions: 0,
+        lastDreamAt: snapshot.state?.lastDreamAt ?? nowIso(),
+        lastDreamedLogAt: snapshot.state?.lastDreamedLogAt ?? null,
+        summaryPath: null,
+      },
     };
   }
 
@@ -419,6 +433,7 @@ async function runMemoryDream(
     {
       apiKey: selection.apiKey,
       headers: selection.headers,
+      env: selection.env,
       signal: ctx.signal,
     },
   );
@@ -469,7 +484,7 @@ async function runMemoryDream(
     );
   }
 
-  return withMemoryMutationQueue(lockPaths, async () => {
+  const result = await withMemoryMutationQueue(lockPaths, async () => {
     ensureMemoryInitialized(await pathExists(paths.memoryRoot));
 
     const currentReplay = await collectDreamReplay(cwd);
@@ -517,6 +532,8 @@ async function runMemoryDream(
       summaryPath,
     };
   });
+
+  return { result, usage: response.usage };
 }
 
 async function loadMemoryPrompt(cwd: string): Promise<string | undefined> {
@@ -967,6 +984,7 @@ async function selectDreamModel(ctx: ExtensionContext): Promise<ModelSelection>
     model: ctx.model,
     apiKey: auth.apiKey,
     headers: auth.headers,
+    env: auth.env,
   };
 }
 
@@ -1710,7 +1728,7 @@ export default function memoryExtension(pi: ExtensionAPI): void {
         description: `Update one .agents/memory/core block while enforcing the shared ${CORE_LINE_CAP}-line and ${CORE_CHAR_CAP}-character caps`,
         promptSnippet: `Update one core memory block in .agents/memory/core with ${CORE_LINE_CAP}-line and ${CORE_CHAR_CAP}-character cap enforcement`,
         promptGuidelines: [
-          "Use this tool when updating .agents/memory/core/directives.md, context.md, focus.md, or pending.md.",
+          "Use memory_update_block when updating .agents/memory/core/directives.md, context.md, focus.md, or pending.md.",
           `If a write would exceed the ${CORE_LINE_CAP}-line or ${CORE_CHAR_CAP}-character core cap, run memory_dream or remove content first.`,
         ],
         parameters: MEMORY_UPDATE_BLOCK_PARAMS,
@@ -1741,7 +1759,7 @@ export default function memoryExtension(pi: ExtensionAPI): void {
         description: "Append an entry to .agents/memory/log.md using the repo memory log format",
         promptSnippet: "Append an entry to the repo memory log at .agents/memory/log.md",
         promptGuidelines: [
-          "Use this tool for important decisions, discoveries, plans, experiments, prompt ingests, and attachment additions worth remembering.",
+          "Use memory_append_log for important decisions, discoveries, plans, experiments, prompt ingests, and attachment additions worth remembering.",
           "Use importance labels high, medium, or low.",
           "If this entry replaces or corrects prior memory, set supersedes and/or invalidates links explicitly.",
           "The memory log is append-only. Do not rewrite or truncate older entries.",
@@ -1779,7 +1797,7 @@ export default function memoryExtension(pi: ExtensionAPI): void {
         promptSnippet: "Consolidate repo memory with dream-based core compression",
         promptGuidelines: [
           "Dream is the only consolidation mechanism for .agents/memory/core.",
-          "Use this tool when logs have accumulated, core needs compression, or recent compaction context should be folded into memory.",
+          "Use memory_dream when logs have accumulated, core needs compression, or recent compaction context should be folded into memory.",
         ],
         parameters: MEMORY_DREAM_PARAMS,
         async execute(_toolCallId, params, _signal, _onUpdate, ctx) {

package/extensions/notify.ts

@@ -10,6 +10,7 @@
  * - Windows toast: Windows Terminal (WSL)
  */
 
+import { execFile } from "node:child_process";
 import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
 
 const REVIEW_EVENT_START = "review:start";
@@ -39,7 +40,6 @@ function notifyOSC99(title: string, body: string): void {
 }
 
 function notifyWindows(title: string, body: string): void {
-  const { execFile } = require("child_process");
   execFile("powershell.exe", ["-NoProfile", "-Command", windowsToastScript(title, body)]);
 }
 
@@ -92,13 +92,33 @@ export default function (pi: ExtensionAPI) {
   let pendingPromptCount = 0;
   const activeReviewSessions = new Set<string>();
   let currentSessionKey: string | undefined;
+  let pendingReadyNotification: ReturnType<typeof setImmediate> | undefined;
 
   const hasCurrentSessionReviewRun = () => {
     if (!currentSessionKey) return false;
     return activeReviewSessions.has(currentSessionKey);
   };
 
+  const cancelReadyNotification = () => {
+    if (!pendingReadyNotification) return;
+    clearImmediate(pendingReadyNotification);
+    pendingReadyNotification = undefined;
+  };
+
+  const scheduleReadyNotification = (ctx: ExtensionContext) => {
+    cancelReadyNotification();
+    pendingReadyNotification = setImmediate(() => {
+      pendingReadyNotification = undefined;
+      if (!ctx.isIdle()) return;
+      if (pendingPromptCount > 0) return;
+      if (hasCurrentSessionReviewRun()) return;
+      notify("Pi", "Ready for input");
+    });
+    pendingReadyNotification.unref?.();
+  };
+
   pi.on("session_start", async (_event, ctx) => {
+    cancelReadyNotification();
     currentSessionKey = getSessionKey(ctx);
   });
 
@@ -143,14 +163,13 @@ export default function (pi: ExtensionAPI) {
     notify("Pi", "Review failed");
   });
 
-  pi.on("agent_end", async (_event, ctx) => {
+  pi.on("agent_settled", async (_event, ctx) => {
     currentSessionKey = getSessionKey(ctx);
-    if (pendingPromptCount > 0) return;
-    if (hasCurrentSessionReviewRun()) return;
-    notify("Pi", "Ready for input");
+    scheduleReadyNotification(ctx);
   });
 
   pi.on("session_shutdown", async (_event, ctx) => {
+    cancelReadyNotification();
     pendingPromptCount = 0;
     const sessionKey = getSessionKey(ctx);
     activeReviewSessions.delete(sessionKey);

package/extensions/openai-fast.ts

@@ -1,7 +1,10 @@
 import { mkdir, readFile, writeFile } from "node:fs/promises";
-import os from "node:os";
 import path from "node:path";
-import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
+import {
+  getAgentDir,
+  type ExtensionAPI,
+  type ExtensionContext,
+} from "@earendil-works/pi-coding-agent";
 
 const STATUS_KEY = "openai-fast";
 const SUPPORTED_APIS = new Set([
@@ -23,8 +26,7 @@ function emptyConfig(): FastConfig {
 }
 
 function getConfigPath(): string {
-  const agentDir = process.env.PI_CODING_AGENT_DIR ?? path.join(os.homedir(), ".pi", "agent");
-  return path.join(agentDir, "openai-fast.json");
+  return path.join(getAgentDir(), "openai-fast.json");
 }
 
 function isObject(value: unknown): value is JsonObject {

package/extensions/openai-verbosity.ts

@@ -1,7 +1,10 @@
 import { mkdir, readFile, writeFile } from "node:fs/promises";
-import os from "node:os";
 import path from "node:path";
-import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
+import {
+  getAgentDir,
+  type ExtensionAPI,
+  type ExtensionContext,
+} from "@earendil-works/pi-coding-agent";
 
 const STATUS_KEY = "openai-verbosity";
 const SUPPORTED_APIS = new Set([
@@ -23,8 +26,7 @@ function emptyConfig(): VerbosityConfig {
 }
 
 function getConfigPath(): string {
-  const agentDir = process.env.PI_CODING_AGENT_DIR ?? path.join(os.homedir(), ".pi", "agent");
-  return path.join(agentDir, "openai-verbosity.json");
+  return path.join(getAgentDir(), "openai-verbosity.json");
 }
 
 function isObject(value: unknown): value is JsonObject {