lsd-pi 1.1.4 → 1.1.5

package/README.md

@@ -242,8 +242,9 @@ A few quality-of-life touches in the TUI:
 
 - the footer can show a live cache timer for the current prompt-cache window
 - `/hotkeys` gives you a full shortcut reference on demand
-- `/settings` now includes toggles for Codex rotate, the cache timer, RTK shell-command compression, and a configurable **Main accent** preset
+- `/settings` now includes toggles for Codex rotate, the cache timer, **Pin last prompt**, RTK shell-command compression, and a configurable **Main accent** preset
 - changing the main accent also updates accent-driven UI elements and the text input border across thinking levels
+- when **Pin last prompt** is enabled, LSD keeps your most recent non-command prompt visible above the editor as a lightweight reminder
 
 Some workflow/automation commands still use the legacy namespace:
 

package/dist/resources/extensions/async-jobs/async-bash-tool.js

@@ -6,6 +6,7 @@
  * with await_job.
  */
 import { getShellConfig, sanitizeCommand, DEFAULT_MAX_BYTES, DEFAULT_MAX_LINES, } from "@gsd/pi-coding-agent";
+import { Text } from "@gsd/pi-tui";
 import { Type } from "@sinclair/typebox";
 import { spawn, spawnSync } from "node:child_process";
 import { createWriteStream } from "node:fs";
@@ -70,6 +71,19 @@ export function createAsyncBashTool(getManager, getCwd) {
             "Check /jobs to see all running and recent background jobs.",
         ],
         parameters: schema,
+        renderCall(args, theme, options) {
+            const cmd = args.command ?? "";
+            const display = cmd.length > 80 ? cmd.slice(0, 77) + "..." : cmd;
+            const indicator = options?.statusIndicator ? `${options.statusIndicator} ` : "";
+            let text = indicator + theme.fg("toolTitle", theme.bold("async_bash "));
+            text += theme.fg("bashMode", "$ ");
+            text += theme.fg("accent", display || theme.fg("muted", "..."));
+            if (args.label)
+                text += theme.fg("muted", ` (${args.label})`);
+            if (args.timeout)
+                text += theme.fg("dim", ` timeout:${args.timeout}s`);
+            return new Text(text, 0, 0);
+        },
         async execute(_toolCallId, params, _signal, _onUpdate, _ctx) {
             const manager = getManager();
             const cwd = getCwd();

package/dist/resources/extensions/async-jobs/await-tool.js

@@ -4,6 +4,7 @@
  * If specific job IDs are provided, waits for those jobs.
  * If omitted, waits for any running job to complete.
  */
+import { Text } from "@gsd/pi-tui";
 import { Type } from "@sinclair/typebox";
 const DEFAULT_TIMEOUT_SECONDS = 120;
 const schema = Type.Object({
@@ -21,6 +22,19 @@ export function createAwaitTool(getManager) {
         label: "Await Background Job",
         description: "Wait for background jobs to complete. Provide specific job IDs or omit to wait for the next job that finishes. Returns results of completed jobs.",
         parameters: schema,
+        renderCall(args, theme, options) {
+            const indicator = options?.statusIndicator ? `${options.statusIndicator} ` : "";
+            let text = indicator + theme.fg("toolTitle", theme.bold("await_job"));
+            if (args.jobs && args.jobs.length > 0) {
+                text += " " + theme.fg("accent", args.jobs.join(", "));
+            }
+            else {
+                text += theme.fg("muted", " (any)");
+            }
+            if (args.timeout)
+                text += theme.fg("dim", ` timeout:${args.timeout}s`);
+            return new Text(text, 0, 0);
+        },
         async execute(_toolCallId, params, _signal, _onUpdate, _ctx) {
             const manager = getManager();
             const { jobs: jobIds, timeout } = params;

package/dist/resources/extensions/async-jobs/cancel-job-tool.js

@@ -1,6 +1,7 @@
 /**
  * cancel_job tool — cancel a running background job.
  */
+import { Text } from "@gsd/pi-tui";
 import { Type } from "@sinclair/typebox";
 const schema = Type.Object({
     job_id: Type.String({ description: "The background job ID to cancel (e.g. bg_a1b2c3d4)" }),
@@ -11,6 +12,12 @@ export function createCancelJobTool(getManager) {
         label: "Cancel Background Job",
         description: "Cancel a running background job by its ID.",
         parameters: schema,
+        renderCall(args, theme, options) {
+            const indicator = options?.statusIndicator ? `${options.statusIndicator} ` : "";
+            let text = indicator + theme.fg("toolTitle", theme.bold("cancel_job "));
+            text += theme.fg("accent", args.job_id);
+            return new Text(text, 0, 0);
+        },
         async execute(_toolCallId, params, _signal, _onUpdate, _ctx) {
             const manager = getManager();
             const result = manager.cancel(params.job_id);

package/dist/resources/extensions/cache-timer/index.js

@@ -14,6 +14,7 @@ import { existsSync, readFileSync, writeFileSync } from "node:fs";
 import { join } from "node:path";
 import { getAgentDir } from "@gsd/pi-coding-agent";
 const STATUS_KEY = "cache-timer";
+const IS_MEMORY_MAINTENANCE_WORKER = process.env.LSD_MEMORY_EXTRACT === "1" || process.env.LSD_MEMORY_DREAM === "1";
 // ANSI color codes for timer display
 const ANSI_RESET = "\x1b[0m";
 const ANSI_YELLOW = "\x1b[33m";
@@ -65,6 +66,9 @@ function formatElapsed(ms) {
     return `⏱ ${time}`;
 }
 export default function cacheTimerExtension(pi) {
+    if (IS_MEMORY_MAINTENANCE_WORKER) {
+        return;
+    }
     let timer = null;
     let startTime = null;
     let enabled = readEnabled();

package/dist/resources/extensions/codex-rotate/IMPLEMENTATION.md

@@ -18,6 +18,7 @@ Implemented a Codex OAuth rotation extension for LSD that manages multiple ChatG
    - Writes `openai-codex` credential array with `api_key` type
    - Avoids the `set()` method (which appends duplicates)
    - Maintains stable credential order for index-based backoff
+   - Followed by a live `AuthStorage.reload()` in the extension so the running process sees account changes immediately
 
 3. **Background Refresh Timer** (`index.ts`)
    - Runs every 10 minutes
@@ -42,11 +43,10 @@ Implemented a Codex OAuth rotation extension for LSD that manages multiple ChatG
    - `/codex import-cockpit` - Import from Cockpit
    - `/codex sync` - Force refresh all tokens
 
-6. **Error Detection** (`quota.ts`)
-   - Detects quota/rate limit errors
-   - Detects auth errors (401, expired tokens)
-   - Classifies errors by type
-   - Integrates with `AuthStorage.markUsageLimitReached()`
+6. **Error Detection / Rotation Path**
+   - Quota/rate-limit/auth classification lives in `quota.ts`
+   - Same-turn retry and per-credential backoff are performed by core `RetryHandler`
+   - The extension's responsibility is to keep `auth.json` and the live in-memory auth state in sync
 
 ## File Structure
 
@@ -87,6 +87,7 @@ Using `FileAuthStorageBackend.withLockAsync()`:
 - Prevents race conditions with multiple LSD instances
 - Replaces entire credential array (not just updating keys)
 - Maintains stable order for index-based backoff
+- Requires a follow-up live auth reload in the current process so retry rotation uses the updated credential set
 
 ### 4. Session-Sticky Credential Selection
 
@@ -105,15 +106,15 @@ LSD already uses session-sticky selection by default:
 
 ## Next Steps: Phase 2 (Resilience)
 
-The following features were planned but not yet implemented:
+Remaining improvements that could still be added:
 
-1. **Enhanced error detection in `agent_end`**
-   - Already implemented basic error detection
-   - Could add more sophisticated quota pattern matching
+1. **Richer `/codex status` diagnostics**
+   - Show live backoff / credential-availability state from `AuthStorage`
+   - Make it easier to verify when an account was rotated away after a usage-limit hit
 
-2. **Better credential change listener**
-   - Could listen for external auth.json modifications
-   - Would re-sync on change detection
+2. **External change detection**
+   - Could listen for external auth/account file modifications
+   - Would re-sync / reload on change detection
 
 3. **Migration helpers**
    - Could add `/codex migrate` to assist users moving from Cockpit
@@ -121,8 +122,12 @@ The following features were planned but not yet implemented:
 
 ## Testing
 
-To test the extension:
+Implemented regression coverage now includes:
+- `src/tests/codex-rotate-auth-reload.test.ts`
+- Verifies that codex-rotate syncs require a live auth reload for the running process to see the latest credentials
+- Verifies `quota_exhausted` backoff on one `openai-codex` credential causes the next retry to select the next credential
 
+Manual smoke test:
 1. Add an account: `/codex add`
 2. Check status: `/codex status`
 3. Add another account: `/codex add`

package/dist/resources/extensions/codex-rotate/README.md

@@ -97,14 +97,20 @@ Codex access tokens work identically to API keys:
 - Background timer runs every 10 minutes
 - Tokens are refreshed when expiring within 5 minutes
 - Refreshed tokens are atomically synced to auth.json
+- After every successful sync, the extension reloads live `AuthStorage` so retries and rotation see the latest credentials immediately
 - Failed refreshes disable the account with a reason
 
 ### Error Handling
 
-The extension hooks into `agent_end` events to detect:
+The extension now relies on LSD core retry/backoff handling rather than its own `agent_end` hook:
+- The Codex provider surfaces friendly usage-limit / rate-limit / auth errors
+- Core `RetryHandler` classifies those failures and calls `markUsageLimitReached(...)`
+- If another `openai-codex` credential is available, LSD automatically rotates and retries the same prompt
 - Rate limit errors (429) → 30s backoff
-- Quota exhausted errors → 30min backoff
-- Auth errors (401) → treated as rate limits for immediate rotation
+- Quota exhausted / usage limit errors → 30min backoff
+- Auth errors (401) → treated as immediate credential-rotation failures
+
+To make that work reliably, the extension reloads the live auth state after every successful sync so the retry handler sees the current credential pool immediately.
 
 ## Security
 

package/dist/resources/extensions/codex-rotate/commands.js

@@ -63,6 +63,13 @@ function displayAccounts(ctx, accounts) {
     });
     ctx.ui.notify(lines.join("\n"), "info");
 }
+async function syncAccountsToAuthAndReload(ctx) {
+    const synced = await syncAccountsToAuth(getAllAccounts());
+    if (synced) {
+        ctx.modelRegistry.authStorage.reload();
+    }
+    return synced;
+}
 /**
  * Register the /codex command
  */
@@ -118,7 +125,7 @@ export function registerCodexCommand(pi) {
                             disabled: false,
                         });
                         // Sync to auth.json
-                        const success = await syncAccountsToAuth(getAllAccounts());
+                        const success = await syncAccountsToAuthAndReload(ctx);
                         if (success) {
                             ctx.ui.notify(`Added account: ${email || account.accountId}. Synced to auth.json.`, "success");
                         }
@@ -176,7 +183,7 @@ export function registerCodexCommand(pi) {
                         const confirmed = await ctx.ui.select(`Remove account: ${account.email || account.accountId}?`, ["Yes, remove", "Cancel"], { signal: AbortSignal.timeout(30000) });
                         if (confirmed === "Yes, remove") {
                             removeAccount(account.id);
-                            await syncAccountsToAuth(getAllAccounts());
+                            await syncAccountsToAuthAndReload(ctx);
                             ctx.ui.notify(`Removed account: ${account.email || account.accountId}`, "success");
                         }
                         return;
@@ -200,7 +207,7 @@ export function registerCodexCommand(pi) {
                             return;
                         }
                         updateAccount(account.id, { disabled: false, disabledReason: undefined });
-                        await syncAccountsToAuth(getAllAccounts());
+                        await syncAccountsToAuthAndReload(ctx);
                         ctx.ui.notify(`Enabled account: ${account.email || account.accountId}`, "success");
                         return;
                     }
@@ -223,7 +230,7 @@ export function registerCodexCommand(pi) {
                             return;
                         }
                         updateAccount(account.id, { disabled: true, disabledReason: "manually disabled" });
-                        await syncAccountsToAuth(getAllAccounts());
+                        await syncAccountsToAuthAndReload(ctx);
                         ctx.ui.notify(`Disabled account: ${account.email || account.accountId}`, "success");
                         return;
                     }
@@ -234,7 +241,7 @@ export function registerCodexCommand(pi) {
                             ctx.ui.notify("No account found to import.", "warning");
                             return;
                         }
-                        const account = addAccount({
+                        addAccount({
                             email: imported.email,
                             accountId: imported.accountId,
                             refreshToken: imported.refreshToken,
@@ -243,7 +250,7 @@ export function registerCodexCommand(pi) {
                             lastUsed: undefined,
                             disabled: false,
                         });
-                        await syncAccountsToAuth(getAllAccounts());
+                        await syncAccountsToAuthAndReload(ctx);
                         ctx.ui.notify(`Imported account: ${imported.email || imported.accountId}`, "success");
                         return;
                     }
@@ -257,7 +264,7 @@ export function registerCodexCommand(pi) {
                         for (const acc of imported) {
                             addAccount(acc);
                         }
-                        await syncAccountsToAuth(getAllAccounts());
+                        await syncAccountsToAuthAndReload(ctx);
                         ctx.ui.notify(`Imported ${imported.length} account(s) from Cockpit Tools`, "success");
                         return;
                     }
@@ -278,7 +285,7 @@ export function registerCodexCommand(pi) {
                                 results.failed++;
                             }
                         }
-                        await syncAccountsToAuth(getAllAccounts());
+                        await syncAccountsToAuthAndReload(ctx);
                         if (results.failed === 0) {
                             ctx.ui.notify(`Synced ${results.success} account(s) to auth.json`, "success");
                         }

package/dist/resources/extensions/codex-rotate/index.js

@@ -10,10 +10,13 @@ import { registerCodexCommand } from "./commands.js";
 import { REFRESH_INTERVAL_MS } from "./config.js";
 import { logCodexRotateError } from "./logger.js";
 let refreshTimer = null;
+async function reloadLiveAuthState(ctx) {
+    ctx.modelRegistry.authStorage.reload();
+}
 /**
  * Refresh all accounts that need it
  */
-async function refreshExpiringAccounts() {
+async function refreshExpiringAccounts(ctx) {
     try {
         const accountsNeedingRefresh = getAccountsNeedingRefresh();
         if (accountsNeedingRefresh.length === 0) {
@@ -41,7 +44,10 @@ async function refreshExpiringAccounts() {
         if (successCount > 0) {
             // Sync refreshed accounts to auth.json
             const allAccounts = getAllAccounts();
-            await syncAccountsToAuth(allAccounts);
+            const synced = await syncAccountsToAuth(allAccounts);
+            if (synced) {
+                await reloadLiveAuthState(ctx);
+            }
         }
         if (failCount > 0 && successCount === 0) {
             logCodexRotateError(`Failed to refresh ${failCount} account(s)`);
@@ -54,12 +60,12 @@ async function refreshExpiringAccounts() {
 /**
  * Start the background refresh timer
  */
-function startRefreshTimer() {
+function startRefreshTimer(ctx) {
     if (refreshTimer) {
         clearInterval(refreshTimer);
     }
     refreshTimer = setInterval(() => {
-        void refreshExpiringAccounts();
+        void refreshExpiringAccounts(ctx);
     }, REFRESH_INTERVAL_MS);
 }
 /**
@@ -78,17 +84,20 @@ export default function CodexRotateExtension(pi) {
     // Register commands
     registerCodexCommand(pi);
     // Session start hook
-    pi.on("session_start", async (_event) => {
+    pi.on("session_start", async (_event, ctx) => {
         const accounts = getAllAccounts();
         if (accounts.length === 0) {
             return;
         }
         // Refresh any expiring accounts immediately
-        await refreshExpiringAccounts();
+        await refreshExpiringAccounts(ctx);
         // Sync to auth.json
-        await syncAccountsToAuth(getAllAccounts());
+        const synced = await syncAccountsToAuth(getAllAccounts());
+        if (synced) {
+            await reloadLiveAuthState(ctx);
+        }
         // Start background refresh timer
-        startRefreshTimer();
+        startRefreshTimer(ctx);
     });
     // Session shutdown hook
     pi.on("session_shutdown", () => {