feat: auto-resume workflows after a provider usage limit resets (closes #27)

extensions/workflow.ts

@@ -13,6 +13,7 @@ import {
   registerWorkflowCommands,
   registerWorkflowModelsCommand,
   saveWorkflowSettingsForCwd,
+  UsageLimitScheduler,
   WorkflowManager,
 } from "../src/index.js";
 
@@ -33,6 +34,15 @@ export default function extension(pi: ExtensionAPI) {
 
   const workflowTool = createWorkflowTool({ cwd, manager, storage });
   pi.registerTool(workflowTool);
+  // Auto-resume runs that paused on a provider usage limit once the quota is
+  // likely refilled. Standalone: only consumes the manager's public surface, so
+  // it stays decoupled from manager/persistence internals. Its constructor also
+  // re-arms any run that was already paused-on-usage_limit before this process
+  // started (cold start), so restarting pi doesn't strand a paused run.
+  const usageLimitScheduler = new UsageLimitScheduler(manager);
+  pi.on("session_shutdown", () => {
+    usageLimitScheduler.dispose();
+  });
   // Standing /effort opt-in (off|high|ultra): auto-arms a workflow for substantive
   // messages, like CC's ultracode. Shared with the editor's input hook below and
   // with the explicit /workflows run <prompt> manual trigger.

src/index.ts

@@ -77,6 +77,13 @@ export { createSharedStoreTools, SharedStore } from "./shared-store.js";
 export type { StructuredOutputCapture, StructuredOutputToolOptions } from "./structured-output.js";
 export { createStructuredOutputTool } from "./structured-output.js";
 export { deliverText, installResultDelivery, installTaskPanel, type TaskPanelOptions } from "./task-panel.js";
+export type {
+  AutoResumeDelayParams,
+  SchedulableWorkflowManager,
+  TimerHandle,
+  UsageLimitSchedulerOptions,
+} from "./usage-limit-scheduler.js";
+export { computeAutoResumeDelayMs, parseResetHintMs, UsageLimitScheduler } from "./usage-limit-scheduler.js";
 export { createWebFetchTool, createWebSearchTool, createWebTools } from "./web-tools.js";
 export type {
   AgentOptions,

src/run-persistence.ts

@@ -64,6 +64,18 @@ export interface PersistedRunState {
   };
   /** Cached agent results for resume, keyed by deterministic call index. */
   journal?: Array<{ index: number; hash: string; result: unknown }>;
+  /**
+   * Opt-out of auto-resume for this run (default true, i.e. eligible unless
+   * explicitly set to false via ExecOptions.autoResume). Set once at run start
+   * and carried through resumes; see UsageLimitScheduler.
+   */
+  autoResume?: boolean;
+  /**
+   * Auto-resume attempt counter for the current usage_limit pause-cycle, owned
+   * and persisted by UsageLimitScheduler (best-effort). Absent/0 means no
+   * auto-resume attempt has been recorded yet.
+   */
+  autoResumeAttempts?: number;
 }
 
 export interface RunPersistence {