pi-oracle 0.2.0 → 0.2.1

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # Changelog
 
+## 0.2.1 - 2026-04-07
+
+### Fixed
+- wake-up guidance now tells receivers to use `oracle_read(jobId)` as the canonical way to consume completion results
+- manual inspection before the first wake-up no longer suppresses the initial reminder attempt
+- wake-up settlement now records provenance so suppressed/settled delivery can be explained in postmortems
+- queued archive cleanup retries now retry queued archive deletion and keep warnings until cleanup succeeds
+- queued archive byte-pressure accounting now includes retained pre-submit archives instead of only currently queued jobs
+
 ## 0.2.0 - 2026-04-06
 
 ### Added

package/README.md

@@ -136,7 +136,9 @@ Detailed design and maintainer docs:
 Completion notification semantics:
 - oracle responses and artifacts are always persisted durably in oracle job state under the configured oracle jobs dir (`${PI_ORACLE_JOBS_DIR:-/tmp}/oracle-<job-id>/` by default)
 - completion delivery into pi sessions is best-effort wake-up based; the extension no longer appends synthetic assistant completion messages into session history
-- manual `oracle_read` or `/oracle-status` inspection settles further reminder retries once the terminal job has been opened
+- wake-up content tells the receiving assistant to call `oracle_read(jobId)` as the canonical completion-consumption path, with saved file paths included as secondary context
+- manual `oracle_read` or `/oracle-status` inspection settles further reminder retries once the terminal job has been opened, and now persists settlement provenance for postmortems
+- manual inspection before the first wake-up attempt is recorded as observation only and does not suppress the first reminder send
 - if a wake-up does not land, the job remains available via its saved response/artifacts and status commands
 
 ## Privacy / local data

package/docs/ORACLE_DESIGN.md

@@ -336,6 +336,14 @@ Important fields include:
 - `wakeupAttemptCount`
 - `wakeupLastRequestedAt`
 - `wakeupSettledAt`
+- `wakeupSettledSource`
+- `wakeupSettledSessionFile`
+- `wakeupSettledSessionKey`
+- `wakeupSettledBeforeFirstAttempt`
+- `wakeupObservedAt`
+- `wakeupObservedSource`
+- `wakeupObservedSessionFile`
+- `wakeupObservedSessionKey`
 - `notifyClaimedAt`
 - `notifyClaimedBy`
 - `artifactFailureCount`
@@ -461,7 +469,9 @@ The extension still uses the same general `pi`-native background completion patt
 - poller scans jobs on an interval
 - completed job durability lives in oracle job state plus saved response/artifact files, not in synthetic session-history assistant messages
 - when a matching job reaches `complete`, `failed`, or `cancelled`, the poller issues bounded best-effort wake-up reminders to whichever matching session is currently live
-- manual `oracle_read` or `/oracle-status` inspection settles further reminder retries once the terminal job has been opened
+- those wake-ups direct the receiver to `oracle_read(jobId)` as the canonical completion-consumption path, while still surfacing saved response/artifact paths as secondary context
+- manual `oracle_read` or `/oracle-status` inspection settles further reminder retries once the terminal job has been opened and persists provenance about which path/session settled the wake-up
+- manual inspection before the first wake-up attempt is recorded separately as observation metadata and does not suppress the first reminder send
 - if no wake-up lands, the job remains available via `/oracle-status`, `oracle_read`, and the saved `${PI_ORACLE_JOBS_DIR:-/tmp}/oracle-<job-id>/` response/artifact files
 - because completion delivery is best-effort, pruning uses explicit terminal-job age policy instead of pretending a durable session notification happened
 - recently sent wake-ups keep response/artifact files retained briefly so follow-up turns do not point at deleted paths if cleanup or pruning races with delivery

package/extensions/oracle/lib/commands.ts

@@ -121,7 +121,11 @@ export function registerOracleCommands(pi: ExtensionAPI, authWorkerPath: string,
         return;
       }
       if (isTerminalOracleJob(job)) {
-        await markWakeupSettled(job.id);
+        await markWakeupSettled(job.id, {
+          source: "oracle_status",
+          sessionFile: ctx.sessionManager.getSessionFile?.(),
+          cwd: ctx.cwd,
+        });
       }
       ctx.ui.notify(summarizeJob(job.id), "info");
     },

package/extensions/oracle/lib/jobs.ts

@@ -25,6 +25,8 @@ export type OracleJobPhase =
   | "failed"
   | "cancelled";
 
+export type OracleWakeupSettlementSource = "oracle_read" | "oracle_status";
+
 export const ACTIVE_ORACLE_JOB_STATUSES: OracleJobStatus[] = ["preparing", "submitted", "waiting"];
 export const OPEN_ORACLE_JOB_STATUSES: OracleJobStatus[] = ["queued", ...ACTIVE_ORACLE_JOB_STATUSES];
 export const TERMINAL_ORACLE_JOB_STATUSES: OracleJobStatus[] = ["complete", "failed", "cancelled"];
@@ -56,6 +58,10 @@ export function shouldAdvanceQueueAfterCancellation(job: Pick<OracleJob, "status
   return job.status === "cancelled" && !job.cleanupPending && !job.cleanupWarnings?.length;
 }
 
+export function hasRetainedPreSubmitArchive(job: Pick<OracleJob, "submittedAt" | "archiveDeletedAfterUpload" | "archivePath">): boolean {
+  return !job.submittedAt && !job.archiveDeletedAfterUpload && typeof job.archivePath === "string" && job.archivePath.length > 0;
+}
+
 export function hasDurableWorkerHandoff(
   job: Pick<OracleJob, "status" | "phase" | "workerPid" | "workerStartedAt" | "heartbeatAt">,
 ): boolean {
@@ -147,6 +153,14 @@ export interface OracleJob {
   wakeupAttemptCount?: number;
   wakeupLastRequestedAt?: string;
   wakeupSettledAt?: string;
+  wakeupSettledSource?: OracleWakeupSettlementSource;
+  wakeupSettledSessionFile?: string;
+  wakeupSettledSessionKey?: string;
+  wakeupSettledBeforeFirstAttempt?: boolean;
+  wakeupObservedAt?: string;
+  wakeupObservedSource?: OracleWakeupSettlementSource;
+  wakeupObservedSessionFile?: string;
+  wakeupObservedSessionKey?: string;
   notifyClaimedAt?: string;
   notifyClaimedBy?: string;
   artifactFailureCount?: number;
@@ -443,17 +457,32 @@ function getTerminalCleanupStaleReason(job: Pick<OracleJob, "status" | "cleanupP
 }
 
 export async function cleanupJobResources(
-  job: Pick<OracleJob, "submittedAt" | "runtimeId" | "runtimeProfileDir" | "runtimeSessionName" | "conversationId">,
+  job: Pick<OracleJob, "submittedAt" | "runtimeId" | "runtimeProfileDir" | "runtimeSessionName" | "conversationId" | "archivePath" | "archiveDeletedAfterUpload">,
 ): Promise<OracleCleanupReport> {
+  const report: OracleCleanupReport = { attempted: [], warnings: [] };
+
+  if (hasRetainedPreSubmitArchive(job)) {
+    report.attempted.push("queuedArchive");
+    await rm(job.archivePath, { force: true }).catch((error: Error) => {
+      report.warnings.push(`Failed to remove queued archive ${job.archivePath}: ${error.message}`);
+    });
+  }
+
   if (!job.submittedAt) {
-    return { attempted: [], warnings: [] };
+    return report;
   }
-  return cleanupRuntimeArtifacts({
+
+  const runtimeReport = await cleanupRuntimeArtifacts({
     runtimeId: job.runtimeId,
     runtimeProfileDir: job.runtimeProfileDir,
     runtimeSessionName: job.runtimeSessionName,
     conversationId: job.conversationId,
   });
+
+  return {
+    attempted: [...report.attempted, ...runtimeReport.attempted],
+    warnings: [...report.warnings, ...runtimeReport.warnings],
+  };
 }
 
 function getCleanupRetentionMs(job: OracleJob): { complete: number; failed: number } {
@@ -774,12 +803,57 @@ export async function noteWakeupRequested(jobId: string, at = new Date().toISOSt
   }
 }
 
-export async function markWakeupSettled(jobId: string, at = new Date().toISOString()): Promise<OracleJob | undefined> {
+function getWakeupSessionKey(sessionFile: string | undefined, cwd: string | undefined): string | undefined {
+  if (!sessionFile || !cwd) return undefined;
+  const projectId = getProjectId(cwd);
+  return `${projectId}::${getSessionId(sessionFile, projectId)}`;
+}
+
+export async function markWakeupSettled(
+  jobId: string,
+  options: {
+    source: OracleWakeupSettlementSource;
+    sessionFile?: string;
+    cwd?: string;
+    at?: string;
+    allowBeforeFirstAttempt?: boolean;
+  },
+): Promise<OracleJob | undefined> {
+  const at = options.at ?? new Date().toISOString();
+  const sessionKey = getWakeupSessionKey(options.sessionFile, options.cwd);
+
   try {
-    return await updateJob(jobId, (job) => ({
-      ...job,
-      wakeupSettledAt: job.wakeupSettledAt ?? at,
-    }));
+    return await updateJob(jobId, (job) => {
+      const beforeFirstAttempt = !job.wakeupLastRequestedAt && (job.wakeupAttemptCount ?? 0) === 0;
+      if (job.wakeupSettledAt) {
+        return {
+          ...job,
+          wakeupSettledSource: job.wakeupSettledSource ?? options.source,
+          wakeupSettledSessionFile: job.wakeupSettledSessionFile ?? options.sessionFile,
+          wakeupSettledSessionKey: job.wakeupSettledSessionKey ?? sessionKey,
+          wakeupSettledBeforeFirstAttempt: job.wakeupSettledBeforeFirstAttempt ?? beforeFirstAttempt,
+        };
+      }
+
+      if (beforeFirstAttempt && !options.allowBeforeFirstAttempt) {
+        return {
+          ...job,
+          wakeupObservedAt: job.wakeupObservedAt ?? at,
+          wakeupObservedSource: job.wakeupObservedSource ?? options.source,
+          wakeupObservedSessionFile: job.wakeupObservedSessionFile ?? options.sessionFile,
+          wakeupObservedSessionKey: job.wakeupObservedSessionKey ?? sessionKey,
+        };
+      }
+
+      return {
+        ...job,
+        wakeupSettledAt: at,
+        wakeupSettledSource: options.source,
+        wakeupSettledSessionFile: options.sessionFile,
+        wakeupSettledSessionKey: sessionKey,
+        wakeupSettledBeforeFirstAttempt: beforeFirstAttempt,
+      };
+    });
   } catch {
     return readJob(jobId);
   }
@@ -805,17 +879,14 @@ export async function cancelOracleJob(id: string, reason = "Cancelled by user"):
         }, now),
       }));
 
-      const cleanupWarnings: string[] = [];
-      await rm(cancelled.archivePath, { force: true }).catch((error: Error) => {
-        cleanupWarnings.push(`Failed to remove queued archive ${cancelled.archivePath}: ${error.message}`);
-      });
-      if (cleanupWarnings.length === 0) return cancelled;
+      const cleanupReport = await cleanupJobResources(cancelled);
+      if (cleanupReport.warnings.length === 0) return cancelled;
 
       return updateJob(id, (job) => ({
         ...job,
-        cleanupWarnings: [...(job.cleanupWarnings || []), ...cleanupWarnings],
+        cleanupWarnings: [...(job.cleanupWarnings || []), ...cleanupReport.warnings],
         lastCleanupAt: now,
-        error: [job.error, ...cleanupWarnings].filter(Boolean).join("\n"),
+        error: [job.error, ...cleanupReport.warnings].filter(Boolean).join("\n"),
       }));
     }
 

package/extensions/oracle/lib/poller.ts

@@ -180,9 +180,10 @@ function buildNotificationContent(job: OraclePollerJob): string {
   const artifactsPath = `${getJobDir(job.id)}/artifacts`;
   return [
     `Oracle job ${job.id} is ${job.status}.`,
-    `Read response: ${responsePath}`,
+    `Use oracle_read with jobId ${job.id} to open the response and settle wake-up retries.`,
+    `Response file: ${responsePath}`,
     `Artifacts: ${artifactsPath}`,
-    job.error ? `Error: ${job.error}` : "Continue from the oracle output.",
+    job.error ? `Error: ${job.error}` : "After oracle_read, continue from the oracle output.",
   ].join("\n");
 }
 

package/extensions/oracle/lib/runtime.ts

@@ -231,7 +231,7 @@ export async function cloneSeedProfileToRuntime(config: OracleConfig, runtimePro
 const AGENT_BROWSER_CLOSE_TIMEOUT_MS = 10_000;
 
 export interface OracleCleanupReport {
-  attempted: Array<"browser" | "runtimeProfileDir" | "conversationLease" | "runtimeLease">;
+  attempted: Array<"browser" | "runtimeProfileDir" | "conversationLease" | "runtimeLease" | "queuedArchive">;
   warnings: string[];
 }
 

package/extensions/oracle/lib/tools.ts

@@ -13,6 +13,7 @@ import {
   getJobDir,
   getSessionFile,
   hasDurableWorkerHandoff,
+  hasRetainedPreSubmitArchive,
   isOpenOracleJob,
   isTerminalOracleJob,
   listOracleJobDirs,
@@ -436,27 +437,60 @@ async function createArchive(cwd: string, files: string[], archivePath: string):
   return createArchiveForTesting(cwd, files, archivePath);
 }
 
-async function getQueuedArchivePressure(): Promise<{ queuedJobs: number; queuedArchiveBytes: number }> {
-  const queuedJobs = listOracleJobDirs()
+export interface QueuedArchivePressure {
+  queuedJobs: number;
+  queuedArchiveBytes: number;
+}
+
+export async function getQueuedArchivePressure(): Promise<QueuedArchivePressure> {
+  const jobs = listOracleJobDirs()
     .map((dir) => readJob(dir))
-    .filter((job): job is NonNullable<typeof job> => Boolean(job && job.status === "queued"));
+    .filter((job): job is NonNullable<typeof job> => Boolean(job));
 
   const queuedArchiveBytes = (await Promise.all(
-    queuedJobs.map(async (job) => {
-      try {
-        return (await stat(job.archivePath)).size;
-      } catch {
-        return 0;
-      }
-    }),
+    jobs
+      .filter((job) => hasRetainedPreSubmitArchive(job))
+      .map(async (job) => {
+        try {
+          return (await stat(job.archivePath)).size;
+        } catch {
+          return 0;
+        }
+      }),
   )).reduce((sum, bytes) => sum + bytes, 0);
 
   return {
-    queuedJobs: queuedJobs.length,
+    queuedJobs: jobs.filter((job) => job.status === "queued").length,
     queuedArchiveBytes,
   };
 }
 
+export function getQueueAdmissionFailure(args: {
+  queuePressure: QueuedArchivePressure;
+  archiveBytes: number;
+  activeJobs: number;
+  maxActiveJobs: number;
+  maxQueuedJobs: number;
+  maxQueuedArchiveBytes: number;
+}): string | undefined {
+  if (args.queuePressure.queuedJobs >= args.maxQueuedJobs) {
+    return (
+      `Oracle is busy (${args.activeJobs}/${args.maxActiveJobs} active, ${args.queuePressure.queuedJobs}/${args.maxQueuedJobs} queued). ` +
+      "Retry later instead of enqueuing more archive state."
+    );
+  }
+
+  const queuedArchiveBytes = args.queuePressure.queuedArchiveBytes + args.archiveBytes;
+  if (queuedArchiveBytes > args.maxQueuedArchiveBytes) {
+    return (
+      `Oracle queued archive storage is full (${queuedArchiveBytes} bytes > ${args.maxQueuedArchiveBytes} bytes across queued jobs and retained pre-submit archives). ` +
+      "Retry later or narrow the archive inputs."
+    );
+  }
+
+  return undefined;
+}
+
 function validateSubmissionOptions(
   params: { effort?: OracleEffort; autoSwitchToThinking?: boolean },
   modelFamily: OracleModelFamily,
@@ -638,18 +672,17 @@ export function registerOracleTools(pi: ExtensionAPI, workerPath: string): void
           if (!runtimeAttempt.acquired) {
             const queuePressure = await getQueuedArchivePressure();
             const maxQueuedJobs = config.browser.maxConcurrentJobs * MAX_QUEUED_JOBS_PER_ACTIVE_RUNTIME;
-            if (queuePressure.queuedJobs >= maxQueuedJobs) {
-              throw new Error(
-                `Oracle is busy (${runtimeAttempt.liveLeases.length}/${config.browser.maxConcurrentJobs} active, ${queuePressure.queuedJobs}/${maxQueuedJobs} queued). ` +
-                  "Retry later instead of enqueuing more archive state.",
-              );
-            }
             const maxQueuedArchiveBytes = config.browser.maxConcurrentJobs * MAX_QUEUED_ARCHIVE_BYTES_PER_ACTIVE_RUNTIME;
-            if (queuePressure.queuedArchiveBytes + currentArchive.archiveBytes > maxQueuedArchiveBytes) {
-              throw new Error(
-                `Oracle queued archive storage is full (${queuePressure.queuedArchiveBytes + currentArchive.archiveBytes} bytes > ${maxQueuedArchiveBytes} bytes across queued jobs). ` +
-                  "Retry later or narrow the archive inputs.",
-              );
+            const queueAdmissionFailure = getQueueAdmissionFailure({
+              queuePressure,
+              archiveBytes: currentArchive.archiveBytes,
+              activeJobs: runtimeAttempt.liveLeases.length,
+              maxActiveJobs: config.browser.maxConcurrentJobs,
+              maxQueuedJobs,
+              maxQueuedArchiveBytes,
+            });
+            if (queueAdmissionFailure) {
+              throw new Error(queueAdmissionFailure);
             }
 
             queued = true;
@@ -854,7 +887,13 @@ export function registerOracleTools(pi: ExtensionAPI, workerPath: string): void
       if (!job || job.projectId !== getProjectId(ctx.cwd)) {
         throw new Error(`Oracle job not found in this project: ${params.jobId}`);
       }
-      const latest = isTerminalOracleJob(job) ? await markWakeupSettled(job.id) : job;
+      const latest = isTerminalOracleJob(job)
+        ? await markWakeupSettled(job.id, {
+          source: "oracle_read",
+          sessionFile: getSessionFile(ctx),
+          cwd: ctx.cwd,
+        })
+        : job;
       const current = latest ?? readJob(job.id) ?? job;
 
       let responsePreview = "";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-oracle",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "ChatGPT web-oracle extension for pi with isolated browser auth, async jobs, and project-context archives.",
   "private": false,
   "license": "MIT",