pi-oracle 0.1.9 → 0.1.11

package/README.md

@@ -35,6 +35,7 @@ An oracle job:
 4. waits in the background
 5. persists the response and any artifacts under `/tmp/oracle-<job-id>/`
    - old terminal jobs are later pruned according to cleanup retention settings
+   - when directory inputs are expanded, project archives automatically skip common bulky generated caches and top-level build outputs such as `node_modules/`, `target/`, virtualenv caches, coverage outputs, and `dist/`/`build/`/`out/`, unless you explicitly pass those directories
 6. wakes the originating `pi` session on completion
 
 ## Example

package/extensions/oracle/lib/jobs.ts

@@ -30,6 +30,9 @@ export const ORACLE_STALE_HEARTBEAT_MS = 3 * 60 * 1000;
 export const ORACLE_NOTIFICATION_CLAIM_TTL_MS = 60_000;
 const ORACLE_COMPLETE_JOB_RETENTION_MS = 14 * 24 * 60 * 60 * 1000;
 const ORACLE_FAILED_JOB_RETENTION_MS = 30 * 24 * 60 * 60 * 1000;
+export const DEFAULT_ORACLE_JOBS_DIR = "/tmp";
+export const ORACLE_JOBS_DIR_ENV = "PI_ORACLE_JOBS_DIR";
+const ORACLE_JOBS_DIR = process.env[ORACLE_JOBS_DIR_ENV]?.trim() || DEFAULT_ORACLE_JOBS_DIR;
 
 export function isActiveOracleJob(job: Pick<OracleJob, "status">): boolean {
   return ACTIVE_ORACLE_JOB_STATUSES.includes(job.status);
@@ -148,20 +151,24 @@ export function getSessionFile(ctx: ExtensionContext): string | undefined {
   return manager.getSessionFile?.();
 }
 
+export function getOracleJobsDir(): string {
+  return ORACLE_JOBS_DIR;
+}
+
 export function getJobDir(id: string): string {
-  return join("/tmp", `oracle-${id}`);
+  return join(ORACLE_JOBS_DIR, `oracle-${id}`);
 }
 
 export function listOracleJobDirs(): string[] {
-  if (!existsSync("/tmp")) return [];
-  return readdirSync("/tmp")
+  if (!existsSync(ORACLE_JOBS_DIR)) return [];
+  return readdirSync(ORACLE_JOBS_DIR)
     .filter((name) => name.startsWith("oracle-"))
-    .map((name) => join("/tmp", name))
+    .map((name) => join(ORACLE_JOBS_DIR, name))
     .filter((path) => existsSync(join(path, "job.json")));
 }
 
 export function readJob(jobDirOrId: string): OracleJob | undefined {
-  const jobDir = jobDirOrId.startsWith("/tmp/oracle-") ? jobDirOrId : getJobDir(jobDirOrId);
+  const jobDir = existsSync(join(jobDirOrId, "job.json")) ? jobDirOrId : getJobDir(jobDirOrId);
   const jobPath = join(jobDir, "job.json");
   if (!existsSync(jobPath)) return undefined;
   try {
@@ -528,6 +535,10 @@ export async function createJob(
   await chmod(promptPath, 0o600).catch(() => undefined);
 
   const now = new Date().toISOString();
+  const normalizedEffort = input.modelFamily === "instant" ? undefined : (input.effort ?? config.defaults.effort);
+  const normalizedAutoSwitchToThinking = input.modelFamily === "instant"
+    ? (input.autoSwitchToThinking ?? config.defaults.autoSwitchToThinking)
+    : false;
   const job: OracleJob = {
     id,
     status: "submitted",
@@ -541,8 +552,8 @@ export async function createJob(
     originSessionFile,
     requestSource: input.requestSource,
     chatModelFamily: input.modelFamily,
-    effort: input.effort,
-    autoSwitchToThinking: input.autoSwitchToThinking,
+    effort: normalizedEffort,
+    autoSwitchToThinking: normalizedAutoSwitchToThinking,
     followUpToJobId: input.followUpToJobId,
     chatUrl: input.followUpToJobId ? input.chatUrl : undefined,
     conversationId,

package/extensions/oracle/lib/locks.ts

@@ -4,7 +4,9 @@ import { mkdirSync, readdirSync, readFileSync } from "node:fs";
 import { mkdir, readFile, rm, writeFile } from "node:fs/promises";
 import { join } from "node:path";
 
-const ORACLE_STATE_DIR = "/tmp/pi-oracle-state";
+export const DEFAULT_ORACLE_STATE_DIR = "/tmp/pi-oracle-state";
+export const ORACLE_STATE_DIR_ENV = "PI_ORACLE_STATE_DIR";
+const ORACLE_STATE_DIR = process.env[ORACLE_STATE_DIR_ENV]?.trim() || DEFAULT_ORACLE_STATE_DIR;
 const LOCKS_DIR = join(ORACLE_STATE_DIR, "locks");
 const LEASES_DIR = join(ORACLE_STATE_DIR, "leases");
 const DEFAULT_WAIT_MS = 30_000;

package/extensions/oracle/lib/runtime.ts

@@ -7,6 +7,8 @@ import type { OracleConfig } from "./config.js";
 import { createLease, listLeaseMetadata, readLeaseMetadata, releaseLease, withAuthLock } from "./locks.js";
 
 const SEED_GENERATION_FILE = ".oracle-seed-generation";
+const DEFAULT_ORACLE_JOBS_DIR = "/tmp";
+const ORACLE_JOBS_DIR = process.env.PI_ORACLE_JOBS_DIR?.trim() || DEFAULT_ORACLE_JOBS_DIR;
 const AGENT_BROWSER_BIN = [process.env.AGENT_BROWSER_PATH, "/opt/homebrew/bin/agent-browser", "/usr/local/bin/agent-browser"].find(
   (candidate) => typeof candidate === "string" && candidate && existsSync(candidate),
 ) || "agent-browser";
@@ -83,7 +85,7 @@ export async function writeSeedGeneration(config: OracleConfig, value = new Date
 }
 
 function activeJobExists(jobId: string): boolean {
-  const path = join("/tmp", `oracle-${jobId}`, "job.json");
+  const path = join(ORACLE_JOBS_DIR, `oracle-${jobId}`, "job.json");
   if (!existsSync(path)) return false;
   try {
     const job = JSON.parse(readFileSync(path, "utf8")) as { status?: string };

package/extensions/oracle/lib/tools.ts

@@ -1,7 +1,7 @@
 import { randomUUID } from "node:crypto";
-import { mkdtemp, rename, rm, stat, writeFile } from "node:fs/promises";
+import { lstat, mkdtemp, readdir, rename, rm, stat, writeFile } from "node:fs/promises";
 import { tmpdir } from "node:os";
-import { join } from "node:path";
+import { basename, join, posix } from "node:path";
 import { StringEnum } from "@mariozechner/pi-ai";
 import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
 import { Type } from "@sinclair/typebox";
@@ -60,11 +60,137 @@ const VALID_EFFORTS: Record<OracleModelFamily, readonly OracleEffort[]> = {
   pro: ["standard", "extended"],
 };
 
-async function createArchive(cwd: string, files: string[], archivePath: string): Promise<string> {
+const DEFAULT_ARCHIVE_EXCLUDED_DIR_NAMES_ANYWHERE = new Set([
+  ".git",
+  ".hg",
+  ".svn",
+  "node_modules",
+  "target",
+  ".venv",
+  "venv",
+  "__pycache__",
+  ".pytest_cache",
+  ".mypy_cache",
+  ".ruff_cache",
+  ".tox",
+  ".nox",
+  ".hypothesis",
+  ".next",
+  ".nuxt",
+  ".svelte-kit",
+  ".turbo",
+  ".parcel-cache",
+  ".cache",
+  ".gradle",
+  ".terraform",
+  "DerivedData",
+  ".build",
+  ".pnpm-store",
+  ".serverless",
+  ".aws-sam",
+]);
+const DEFAULT_ARCHIVE_EXCLUDED_DIR_NAMES_AT_REPO_ROOT = new Set(["coverage", "htmlcov", "tmp", "temp", ".tmp", "dist", "build", "out"]);
+const DEFAULT_ARCHIVE_EXCLUDED_FILES = new Set([".coverage", ".DS_Store", "Thumbs.db"]);
+const DEFAULT_ARCHIVE_EXCLUDED_SUFFIXES = [".pyc", ".pyo", ".pyd", ".tsbuildinfo", ".tfstate"];
+const DEFAULT_ARCHIVE_EXCLUDED_SUBSTRINGS = [".tfstate."];
+const DEFAULT_ARCHIVE_EXCLUDED_PATH_SEQUENCES = [[".yarn", "cache"]] as const;
+
+function pathContainsSequence(relativePath: string, sequence: readonly string[]): boolean {
+  const segments = relativePath.split("/").filter(Boolean);
+  if (sequence.length === 0 || segments.length < sequence.length) return false;
+  for (let index = 0; index <= segments.length - sequence.length; index += 1) {
+    if (sequence.every((segment, offset) => segments[index + offset] === segment)) return true;
+  }
+  return false;
+}
+
+function getRelativeDepth(relativePath: string): number {
+  return relativePath.split("/").filter(Boolean).length;
+}
+
+function shouldExcludeArchivePath(relativePath: string, isDirectory: boolean, options?: { forceInclude?: boolean }): boolean {
+  const normalized = posix.normalize(relativePath).replace(/^\.\//, "");
+  if (!normalized || normalized === ".") return false;
+  if (options?.forceInclude) return false;
+  const name = basename(normalized);
+  if (DEFAULT_ARCHIVE_EXCLUDED_PATH_SEQUENCES.some((sequence) => pathContainsSequence(normalized, sequence))) return true;
+  if (isDirectory) {
+    if (DEFAULT_ARCHIVE_EXCLUDED_DIR_NAMES_ANYWHERE.has(name)) return true;
+    if (getRelativeDepth(normalized) === 1 && DEFAULT_ARCHIVE_EXCLUDED_DIR_NAMES_AT_REPO_ROOT.has(name)) return true;
+    return false;
+  }
+  if (DEFAULT_ARCHIVE_EXCLUDED_FILES.has(name)) return true;
+  if (DEFAULT_ARCHIVE_EXCLUDED_SUFFIXES.some((suffix) => name.endsWith(suffix))) return true;
+  if (DEFAULT_ARCHIVE_EXCLUDED_SUBSTRINGS.some((needle) => name.includes(needle))) return true;
+  return false;
+}
+
+async function isSymlinkToDirectory(path: string): Promise<boolean> {
+  try {
+    return (await stat(path)).isDirectory();
+  } catch {
+    return false;
+  }
+}
+
+async function shouldExcludeArchiveChild(
+  absolutePath: string,
+  relativePath: string,
+  child: { isDirectory(): boolean; isSymbolicLink(): boolean },
+  options?: { forceInclude?: boolean },
+): Promise<boolean> {
+  const isDirectoryLike = child.isDirectory() || (child.isSymbolicLink() && await isSymlinkToDirectory(absolutePath));
+  return shouldExcludeArchivePath(relativePath, isDirectoryLike, options);
+}
+
+async function expandArchiveEntries(cwd: string, relativePath: string, options?: { forceIncludeSubtree?: boolean }): Promise<string[]> {
+  const normalized = posix.normalize(relativePath).replace(/^\.\//, "");
+  if (normalized === ".") {
+    const children = await readdir(cwd, { withFileTypes: true });
+    const results: string[] = [];
+    for (const child of children.sort((a, b) => a.name.localeCompare(b.name))) {
+      const childRelative = child.name;
+      if (await shouldExcludeArchiveChild(join(cwd, childRelative), childRelative, child)) continue;
+      if (child.isDirectory()) results.push(...await expandArchiveEntries(cwd, childRelative));
+      else results.push(childRelative);
+    }
+    return results;
+  }
+
+  const absolute = join(cwd, normalized);
+  const entry = await lstat(absolute);
+  if (!entry.isDirectory()) return [normalized];
+  if (shouldExcludeArchivePath(normalized, true, { forceInclude: options?.forceIncludeSubtree })) return [];
+
+  const children = await readdir(absolute, { withFileTypes: true });
+  const results: string[] = [];
+  for (const child of children.sort((a, b) => a.name.localeCompare(b.name))) {
+    const childRelative = posix.join(normalized, child.name);
+    if (await shouldExcludeArchiveChild(join(cwd, childRelative), childRelative, child, { forceInclude: options?.forceIncludeSubtree })) continue;
+    if (child.isDirectory()) results.push(...await expandArchiveEntries(cwd, childRelative, { forceIncludeSubtree: options?.forceIncludeSubtree }));
+    else results.push(childRelative);
+  }
+  return results;
+}
+
+export async function resolveExpandedArchiveEntries(cwd: string, files: string[]): Promise<string[]> {
   const entries = resolveArchiveInputs(cwd, files);
+  return Array.from(new Set((await Promise.all(entries.map(async (entry) => {
+    const absolute = join(cwd, entry.relative);
+    const statEntry = await lstat(absolute);
+    const forceIncludeSubtree = statEntry.isDirectory() && entry.relative !== "." && shouldExcludeArchivePath(entry.relative, true);
+    return expandArchiveEntries(cwd, entry.relative, { forceIncludeSubtree });
+  }))).flat())).sort();
+}
+
+async function createArchive(cwd: string, files: string[], archivePath: string): Promise<string> {
+  const expandedEntries = await resolveExpandedArchiveEntries(cwd, files);
+  if (expandedEntries.length === 0) {
+    throw new Error("Oracle archive inputs are empty after default exclusions");
+  }
   const listDir = await mkdtemp(join(tmpdir(), "oracle-filelist-"));
   const listPath = join(listDir, "files.list");
-  await writeFile(listPath, Buffer.from(`${entries.map((entry) => entry.relative).join("\0")}\0`), { mode: 0o600 });
+  await writeFile(listPath, Buffer.from(`${expandedEntries.join("\0")}\0`), { mode: 0o600 });
 
   try {
     const { spawn } = await import("node:child_process");

package/extensions/oracle/worker/artifact-heuristics.mjs

@@ -0,0 +1,76 @@
+export const FILE_LABEL_PATTERN_SOURCE = String.raw`(?:^|[^\w])[^\n]*\.[A-Za-z0-9]{1,12}(?:$|[^\w])`;
+const FILE_LABEL_PATTERN = new RegExp(FILE_LABEL_PATTERN_SOURCE);
+export const GENERIC_ARTIFACT_LABELS = ["ATTACHED", "DONE"];
+const GENERIC_ARTIFACT_LABEL_SET = new Set(GENERIC_ARTIFACT_LABELS);
+
+export function parseSnapshotEntries(snapshot) {
+  return String(snapshot || "")
+    .split("\n")
+    .map((line, lineIndex) => {
+      const refMatch = line.match(/\bref=(e\d+)\b/);
+      if (!refMatch) return undefined;
+      const kindMatch = line.match(/^\s*-\s*([^\s]+)/);
+      const quotedMatch = line.match(/"([^"]*)"/);
+      const valueMatch = line.match(/:\s*(.+)$/);
+      return {
+        line,
+        lineIndex,
+        ref: `@${refMatch[1]}`,
+        kind: kindMatch ? kindMatch[1] : undefined,
+        label: quotedMatch ? quotedMatch[1] : undefined,
+        value: valueMatch ? valueMatch[1].trim() : undefined,
+        disabled: /\bdisabled\b/.test(line),
+      };
+    })
+    .filter(Boolean);
+}
+
+function normalizeText(value) {
+  return String(value || "").replace(/\s+/g, " ").trim();
+}
+
+export function isLikelyArtifactLabel(label) {
+  const normalized = normalizeText(label);
+  if (!normalized) return false;
+  if (GENERIC_ARTIFACT_LABEL_SET.has(normalized.toUpperCase())) return true;
+  return FILE_LABEL_PATTERN.test(normalized);
+}
+
+export function isStructuralArtifactCandidate(candidate) {
+  const label = normalizeText(candidate?.label);
+  if (!isLikelyArtifactLabel(label)) return false;
+
+  const listItemText = normalizeText(candidate?.listItemText);
+  const listItemFileButtonCount = Number(candidate?.listItemFileButtonCount || 0);
+  const paragraphFileButtonCount = Number(candidate?.paragraphFileButtonCount || 0);
+  const paragraphOtherTextLength = Number(candidate?.paragraphOtherTextLength ?? Number.POSITIVE_INFINITY);
+  const focusableFileButtonCount = Number(candidate?.focusableFileButtonCount || 0);
+  const focusableOtherTextLength = Number(candidate?.focusableOtherTextLength ?? Number.POSITIVE_INFINITY);
+
+  if (listItemText === label && listItemFileButtonCount === 1) {
+    return true;
+  }
+
+  if (paragraphFileButtonCount === 1 && paragraphOtherTextLength <= 32) {
+    return true;
+  }
+
+  if (focusableFileButtonCount >= 1 && focusableOtherTextLength <= 64) {
+    return true;
+  }
+
+  return false;
+}
+
+export function filterStructuralArtifactCandidates(candidates) {
+  const seen = new Set();
+  const filtered = [];
+  for (const candidate of candidates || []) {
+    const label = normalizeText(candidate?.label);
+    if (!label || seen.has(label)) continue;
+    if (!isStructuralArtifactCandidate(candidate)) continue;
+    seen.add(label);
+    filtered.push({ label });
+  }
+  return filtered;
+}

package/extensions/oracle/worker/auth-bootstrap.mjs

@@ -33,7 +33,8 @@ const SNAPSHOT_PATH = "/tmp/oracle-auth.snapshot.txt";
 const BODY_PATH = "/tmp/oracle-auth.body.txt";
 const SCREENSHOT_PATH = "/tmp/oracle-auth.png";
 const REAL_CHROME_USER_DATA_DIR = resolve(homedir(), "Library", "Application Support", "Google", "Chrome");
-const ORACLE_STATE_DIR = "/tmp/pi-oracle-state";
+const DEFAULT_ORACLE_STATE_DIR = "/tmp/pi-oracle-state";
+const ORACLE_STATE_DIR = process.env.PI_ORACLE_STATE_DIR?.trim() || DEFAULT_ORACLE_STATE_DIR;
 const LOCKS_DIR = join(ORACLE_STATE_DIR, "locks");
 const STALE_STAGING_PROFILE_MAX_AGE_MS = 24 * 60 * 60 * 1000;
 const AGENT_BROWSER_BIN = [process.env.AGENT_BROWSER_PATH, "/opt/homebrew/bin/agent-browser", "/usr/local/bin/agent-browser"].find(

package/extensions/oracle/worker/run-job.mjs

@@ -3,6 +3,7 @@ import { existsSync } from "node:fs";
 import { appendFile, chmod, mkdir, readFile, rename, rm, stat, writeFile } from "node:fs/promises";
 import { basename, dirname, join } from "node:path";
 import { spawn } from "node:child_process";
+import { FILE_LABEL_PATTERN_SOURCE, filterStructuralArtifactCandidates, GENERIC_ARTIFACT_LABELS, parseSnapshotEntries } from "./artifact-heuristics.mjs";
 
 const jobId = process.argv[2];
 if (!jobId) {
@@ -10,7 +11,9 @@ if (!jobId) {
   process.exit(1);
 }
 
-const jobDir = `/tmp/oracle-${jobId}`;
+const DEFAULT_ORACLE_JOBS_DIR = "/tmp";
+const ORACLE_JOBS_DIR = process.env.PI_ORACLE_JOBS_DIR?.trim() || DEFAULT_ORACLE_JOBS_DIR;
+const jobDir = join(ORACLE_JOBS_DIR, `oracle-${jobId}`);
 const jobPath = `${jobDir}/job.json`;
 const CHATGPT_LABELS = {
   composer: "Chat with ChatGPT",
@@ -26,7 +29,8 @@ const MODEL_FAMILY_PREFIX = {
   pro: "Pro ",
 };
 
-const ORACLE_STATE_DIR = "/tmp/pi-oracle-state";
+const DEFAULT_ORACLE_STATE_DIR = "/tmp/pi-oracle-state";
+const ORACLE_STATE_DIR = process.env.PI_ORACLE_STATE_DIR?.trim() || DEFAULT_ORACLE_STATE_DIR;
 const LOCKS_DIR = join(ORACLE_STATE_DIR, "locks");
 const LEASES_DIR = join(ORACLE_STATE_DIR, "leases");
 const SEED_GENERATION_FILE = ".oracle-seed-generation";
@@ -37,6 +41,9 @@ const ARTIFACT_DOWNLOAD_HEARTBEAT_MS = 10_000;
 const ARTIFACT_DOWNLOAD_TIMEOUT_MS = 90_000;
 const ARTIFACT_DOWNLOAD_MAX_ATTEMPTS = 2;
 const AGENT_BROWSER_CLOSE_TIMEOUT_MS = 10_000;
+const MODEL_CONFIGURATION_SETTLE_TIMEOUT_MS = 20_000;
+const MODEL_CONFIGURATION_SETTLE_POLL_MS = 250;
+const MODEL_CONFIGURATION_CLOSE_RETRY_MS = 1_000;
 const AGENT_BROWSER_BIN = [process.env.AGENT_BROWSER_PATH, "/opt/homebrew/bin/agent-browser", "/usr/local/bin/agent-browser"].find(
   (candidate) => typeof candidate === "string" && candidate && existsSync(candidate),
 ) || "agent-browser";
@@ -539,27 +546,6 @@ function buildLoginProbeScript(timeoutMs) {
   `);
 }
 
-function parseSnapshotEntries(snapshot) {
-  return snapshot
-    .split("\n")
-    .map((line) => {
-      const refMatch = line.match(/\bref=(e\d+)\b/);
-      if (!refMatch) return undefined;
-      const kindMatch = line.match(/^\s*-\s*([^\s]+)/);
-      const quotedMatch = line.match(/"([^"]*)"/);
-      const valueMatch = line.match(/:\s*(.+)$/);
-      return {
-        line,
-        ref: `@${refMatch[1]}`,
-        kind: kindMatch ? kindMatch[1] : undefined,
-        label: quotedMatch ? quotedMatch[1] : undefined,
-        value: valueMatch ? valueMatch[1].trim() : undefined,
-        disabled: /\bdisabled\b/.test(line),
-      };
-    })
-    .filter(Boolean);
-}
-
 function findEntry(snapshot, predicate) {
   return parseSnapshotEntries(snapshot).find(predicate);
 }
@@ -572,14 +558,21 @@ function findLastEntry(snapshot, predicate) {
   return undefined;
 }
 
-function matchesModelFamilyButton(candidate, family) {
-  return candidate.kind === "button" && typeof candidate.label === "string" && candidate.label.startsWith(MODEL_FAMILY_PREFIX[family]) && !candidate.disabled;
-}
-
 function titleCase(value) {
   return value ? `${value[0].toUpperCase()}${value.slice(1)}` : value;
 }
 
+function matchesModelFamilyLabel(label, family) {
+  const normalized = String(label || "");
+  const prefix = MODEL_FAMILY_PREFIX[family];
+  const exact = prefix.trim();
+  return normalized === exact || normalized.startsWith(prefix) || normalized.startsWith(`${exact},`);
+}
+
+function matchesModelFamilyButton(candidate, family) {
+  return candidate.kind === "button" && typeof candidate.label === "string" && matchesModelFamilyLabel(candidate.label, family) && !candidate.disabled;
+}
+
 function requestedEffortLabel(job) {
   return job.effort ? titleCase(job.effort) : undefined;
 }
@@ -612,9 +605,9 @@ function snapshotHasModelConfigurationUi(snapshot) {
     entries
       .filter((entry) => entry.kind === "button" && typeof entry.label === "string")
       .flatMap((entry) =>
-        Object.entries(MODEL_FAMILY_PREFIX)
-          .filter(([, prefix]) => entry.label.startsWith(prefix))
-          .map(([family]) => family),
+        Object.keys(MODEL_FAMILY_PREFIX)
+          .filter((family) => matchesModelFamilyLabel(entry.label, family))
+          .map((family) => family),
       ),
   );
   const hasCloseButton = entries.some((entry) => entry.kind === "button" && entry.label === CHATGPT_LABELS.close && !entry.disabled);
@@ -627,18 +620,35 @@ function snapshotHasModelConfigurationUi(snapshot) {
 function snapshotStronglyMatchesRequestedModel(snapshot, job) {
   const entries = parseSnapshotEntries(snapshot);
   const familyMatched = entries.some((entry) => matchesModelFamilyButton(entry, job.chatModelFamily));
+  const effortLabel = requestedEffortLabel(job);
   if (job.chatModelFamily === "thinking") {
-    return familyMatched || effortSelectionVisible(snapshot, requestedEffortLabel(job));
+    return familyMatched || effortSelectionVisible(snapshot, effortLabel);
   }
   if (job.chatModelFamily === "pro") {
-    return familyMatched;
+    return effortLabel ? familyMatched && effortSelectionVisible(snapshot, effortLabel) : familyMatched;
   }
   return familyMatched;
 }
 
+function thinkingSelectionVisible(snapshot) {
+  const entries = parseSnapshotEntries(snapshot);
+  return entries.some((entry) => !entry.disabled && entry.kind === "button" && matchesModelFamilyLabel(entry.label, "thinking"));
+}
+
+function composerControlsVisible(snapshot) {
+  const entries = parseSnapshotEntries(snapshot);
+  const hasComposer = entries.some(
+    (entry) => entry.kind === "textbox" && entry.label === CHATGPT_LABELS.composer && !entry.disabled,
+  );
+  const hasAddFiles = entries.some(
+    (entry) => entry.kind === "button" && entry.label === CHATGPT_LABELS.addFiles && !entry.disabled,
+  );
+  return hasComposer && hasAddFiles;
+}
+
 function snapshotWeaklyMatchesRequestedModel(snapshot, job) {
   if (job.chatModelFamily === "thinking") {
-    return effortSelectionVisible(snapshot, requestedEffortLabel(job));
+    return effortSelectionVisible(snapshot, requestedEffortLabel(job)) || thinkingSelectionVisible(snapshot);
   }
   if (job.chatModelFamily === "pro") {
     return !thinkingChipVisible(snapshot);
@@ -955,6 +965,54 @@ async function openModelConfiguration(job) {
   throw new Error("Could not open model configuration UI");
 }
 
+async function waitForModelConfigurationToSettle(job, options = {}) {
+  const deadline = Date.now() + MODEL_CONFIGURATION_SETTLE_TIMEOUT_MS;
+  let lastCloseAttemptAt = 0;
+  let fallbackLogged = false;
+  let lastSnapshot = "";
+
+  while (Date.now() < deadline) {
+    const snapshot = await snapshotText(job);
+    lastSnapshot = snapshot;
+    const configurationUiVisible = snapshotHasModelConfigurationUi(snapshot);
+
+    if (!configurationUiVisible) {
+      if (snapshotWeaklyMatchesRequestedModel(snapshot, job)) return;
+      if (options.stronglyVerified) {
+        if (!fallbackLogged) {
+          fallbackLogged = true;
+          await log(`Model configuration closed after strong in-dialog verification for family=${job.chatModelFamily} effort=${job.effort || "(none)"}`);
+        }
+        return;
+      }
+    }
+
+    if (!configurationUiVisible && composerControlsVisible(snapshot) && options.stronglyVerified) {
+      if (!fallbackLogged) {
+        fallbackLogged = true;
+        await log(`Composer became usable after strong in-dialog verification for family=${job.chatModelFamily} effort=${job.effort || "(none)"}`);
+      }
+      return;
+    }
+
+    if (Date.now() - lastCloseAttemptAt >= MODEL_CONFIGURATION_CLOSE_RETRY_MS) {
+      lastCloseAttemptAt = Date.now();
+      if (!(await maybeClickLabeledEntry(job, CHATGPT_LABELS.close, { kind: "button" }))) {
+        await agentBrowser(job, "press", "Escape").catch(() => undefined);
+      }
+    }
+
+    await sleep(MODEL_CONFIGURATION_SETTLE_POLL_MS);
+  }
+
+  if (options.stronglyVerified && lastSnapshot && !snapshotHasModelConfigurationUi(lastSnapshot)) {
+    await log(`Model configuration closed only after settle-timeout for family=${job.chatModelFamily} effort=${job.effort || "(none)"}`);
+    return;
+  }
+
+  throw new Error(`Could not verify requested model settings after configuration for ${job.chatModelFamily}`);
+}
+
 async function configureModel(job) {
   const initialSnapshot = await snapshotText(job);
   if (snapshotStronglyMatchesRequestedModel(initialSnapshot, job)) {
@@ -964,6 +1022,7 @@ async function configureModel(job) {
 
   await log(`Configuring model family=${job.chatModelFamily} effort=${job.effort || "(none)"}`);
   let familySnapshot = await openModelConfiguration(job);
+  let verificationSnapshot = familySnapshot;
 
   let familyEntry = findEntry(familySnapshot, (candidate) => matchesModelFamilyButton(candidate, job.chatModelFamily));
   if (!familyEntry && snapshotStronglyMatchesRequestedModel(familySnapshot, job)) {
@@ -977,6 +1036,7 @@ async function configureModel(job) {
     await clickRef(job, familyEntry.ref);
     await agentBrowser(job, "wait", "800");
     familySnapshot = await snapshotText(job);
+    verificationSnapshot = familySnapshot;
   }
 
   if (job.chatModelFamily === "thinking" || job.chatModelFamily === "pro") {
@@ -990,6 +1050,7 @@ async function configureModel(job) {
       await clickLabeledEntry(job, effortLabel, { kind: "option" });
       await agentBrowser(job, "wait", "400");
       const effortSnapshot = await snapshotText(job);
+      verificationSnapshot = effortSnapshot;
       const selectedEffort = findEntry(
         effortSnapshot,
         (candidate) => candidate.kind === "combobox" && candidate.value === effortLabel && !candidate.disabled,
@@ -1002,17 +1063,18 @@ async function configureModel(job) {
 
   if (job.chatModelFamily === "instant" && job.autoSwitchToThinking) {
     await maybeClickLabeledEntry(job, CHATGPT_LABELS.autoSwitchToThinking);
+    verificationSnapshot = await snapshotText(job);
   }
 
-  if (!(await maybeClickLabeledEntry(job, CHATGPT_LABELS.close, { kind: "button" }))) {
-    await agentBrowser(job, "press", "Escape").catch(() => undefined);
+  const stronglyVerified = snapshotStronglyMatchesRequestedModel(verificationSnapshot, job);
+  if (!stronglyVerified) {
+    throw new Error(`Could not verify requested model settings in configuration UI for ${job.chatModelFamily}`);
   }
-  await agentBrowser(job, "wait", "500");
 
-  const postCloseSnapshot = await snapshotText(job);
-  if (!snapshotWeaklyMatchesRequestedModel(postCloseSnapshot, job)) {
-    throw new Error(`Could not verify requested model settings after configuration for ${job.chatModelFamily}`);
+  if (!(await maybeClickLabeledEntry(job, CHATGPT_LABELS.close, { kind: "button" }))) {
+    await agentBrowser(job, "press", "Escape").catch(() => undefined);
   }
+  await waitForModelConfigurationToSettle(job, { stronglyVerified });
 }
 
 async function uploadArchive(job) {
@@ -1192,14 +1254,6 @@ async function detectType(path) {
   return result.stdout || "unknown";
 }
 
-function isLikelyArtifactLabel(label) {
-  const normalized = String(label || "").trim();
-  if (!normalized) return false;
-  const upper = normalized.toUpperCase();
-  if (upper === "ATTACHED" || upper === "DONE") return true;
-  return /(?:^|[^\w])[^\n]*\.[A-Za-z0-9]{1,12}(?:$|[^\w])/.test(normalized);
-}
-
 function preferredArtifactName(label, index) {
   const normalized = String(label || "").trim();
   const fileNameMatch = normalized.match(/([A-Za-z0-9._-]+\.[A-Za-z0-9]{1,12})(?!.*[A-Za-z0-9._-]+\.[A-Za-z0-9]{1,12})/);
@@ -1207,57 +1261,82 @@ function preferredArtifactName(label, index) {
   return `artifact-${String(index + 1).padStart(2, "0")}`;
 }
 
-function artifactCandidatesFromEntries(entries) {
-  const excluded = new Set([
-    "Copy response",
-    "Good response",
-    "Bad response",
-    "Share",
-    "Switch model",
-    "More actions",
-    CHATGPT_LABELS.addFiles,
-    "Start dictation",
-    "Start Voice",
-    "Model selector",
-    "Open conversation options",
-    "Scroll to bottom",
-    CHATGPT_LABELS.close,
-  ]);
-
-  const seen = new Set();
-  const candidates = [];
-  for (const entry of entries) {
-    if (!entry.label) continue;
-    if (excluded.has(entry.label)) continue;
-    if (entry.label.startsWith("Thought for ")) continue;
-    if (entry.kind !== "button" && entry.kind !== "link") continue;
-    if (!isLikelyArtifactLabel(entry.label)) continue;
-    if (seen.has(entry.label)) continue;
-    seen.add(entry.label);
-    candidates.push({ label: entry.label });
-  }
-  return candidates;
-}
-
-async function collectArtifactCandidates(job, responseIndex) {
+async function collectArtifactCandidates(job, responseIndex, responseText = "") {
   const snapshot = await snapshotText(job);
   const targetSlice = assistantSnapshotSlice(snapshot, responseIndex);
   if (!targetSlice) return { snapshot, targetSlice, candidates: [] };
+
+  const structural = await evalPage(
+    job,
+    toJsonScript(`
+      const normalize = (value) => String(value || '').replace(/\s+/g, ' ').trim();
+      const genericArtifactLabels = new Set(${JSON.stringify(GENERIC_ARTIFACT_LABELS)});
+      const fileLabelPattern = new RegExp(${JSON.stringify(FILE_LABEL_PATTERN_SOURCE)});
+      const isFileLabel = (value) => {
+        const normalized = normalize(value);
+        if (!normalized) return false;
+        if (genericArtifactLabels.has(normalized.toUpperCase())) return true;
+        return fileLabelPattern.test(normalized);
+      };
+      const headings = Array.from(document.querySelectorAll('h1,h2,h3,h4,h5,h6,[role="heading"]'))
+        .filter((el) => normalize(el.textContent) === 'ChatGPT said:');
+      const host = headings[${responseIndex}]?.nextElementSibling;
+      if (!host) return { candidates: [] };
+
+      const fileButtons = (node) => node
+        ? Array.from(node.querySelectorAll('button, a')).map((candidate) => normalize(candidate.textContent)).filter(isFileLabel)
+        : [];
+      const otherTextLength = (text, labels) => {
+        let remaining = normalize(text);
+        for (const label of labels || []) {
+          remaining = normalize(remaining.replaceAll(label, ' '));
+        }
+        remaining = normalize(remaining.replaceAll('Coding Citation', ' '));
+        return remaining.length;
+      };
+      const focusableFor = (node) => node?.closest('[tabindex]');
+
+      const candidates = Array.from(host.querySelectorAll('button, a'))
+        .map((button) => {
+          const label = normalize(button.textContent);
+          if (!isFileLabel(label)) return null;
+          const paragraph = button.closest('p');
+          const listItem = button.closest('li');
+          const focusable = focusableFor(button);
+          const paragraphFileLabels = fileButtons(paragraph);
+          const focusableFileLabels = fileButtons(focusable);
+          return {
+            label,
+            paragraphText: normalize(paragraph?.textContent),
+            listItemText: normalize(listItem?.textContent),
+            paragraphFileButtonCount: paragraphFileLabels.length,
+            paragraphOtherTextLength: otherTextLength(paragraph?.textContent, paragraphFileLabels),
+            listItemFileButtonCount: fileButtons(listItem).length,
+            focusableFileButtonCount: focusableFileLabels.length,
+            focusableOtherTextLength: otherTextLength(focusable?.textContent, focusableFileLabels),
+          };
+        })
+        .filter(Boolean);
+
+      return { candidates };
+    `),
+  );
+
   return {
     snapshot,
     targetSlice,
-    candidates: artifactCandidatesFromEntries(parseSnapshotEntries(targetSlice)),
+    candidates: filterStructuralArtifactCandidates(structural?.candidates || []),
   };
 }
 
-async function waitForStableArtifactCandidates(job, responseIndex) {
+async function waitForStableArtifactCandidates(job, responseIndex, responseText = "") {
   const deadline = Date.now() + ARTIFACT_CANDIDATE_STABILITY_TIMEOUT_MS;
   let lastSignature;
   let stablePolls = 0;
   let latest = { snapshot: "", targetSlice: undefined, candidates: [] };
 
   while (Date.now() < deadline) {
-    latest = await collectArtifactCandidates(job, responseIndex);
+    latest = await collectArtifactCandidates(job, responseIndex, responseText);
     const signature = latest.candidates.map((candidate) => candidate.label).join("\n");
     if (signature === lastSignature) stablePolls += 1;
     else {
@@ -1272,12 +1351,12 @@ async function waitForStableArtifactCandidates(job, responseIndex) {
   return latest;
 }
 
-async function reopenConversationForArtifacts(job, responseIndex, reason) {
+async function reopenConversationForArtifacts(job, responseIndex, responseText, reason) {
   const targetUrl = job.chatUrl || stripQuery(await currentUrl(job));
   await log(`Reopening conversation before artifact capture (${reason}): ${targetUrl}`);
   await agentBrowser(job, "open", targetUrl);
   await agentBrowser(job, "wait", "1500");
-  return waitForStableArtifactCandidates(job, responseIndex);
+  return waitForStableArtifactCandidates(job, responseIndex, responseText);
 }
 
 async function withHeartbeatWhile(task, intervalMs = ARTIFACT_DOWNLOAD_HEARTBEAT_MS) {
@@ -1309,14 +1388,14 @@ async function flushArtifactsState(artifacts) {
   }));
 }
 
-async function downloadArtifacts(job, responseIndex) {
+async function downloadArtifacts(job, responseIndex, responseText = "") {
   if (!job.config.artifacts.capture) {
     await secureWriteText(`${jobDir}/artifacts.json`, "[]\n");
     await mutateJob((current) => ({ ...current, artifactPaths: [] }));
     return [];
   }
 
-  const { targetSlice, candidates } = await reopenConversationForArtifacts(job, responseIndex, "initial");
+  const { targetSlice, candidates } = await reopenConversationForArtifacts(job, responseIndex, responseText, "initial");
   if (!targetSlice) {
     await log(`No assistant response found in snapshot for response index ${responseIndex}`);
     await secureWriteText(`${jobDir}/artifacts.json`, "[]\n");
@@ -1378,7 +1457,7 @@ async function downloadArtifacts(job, responseIndex) {
         if (attempt >= ARTIFACT_DOWNLOAD_MAX_ATTEMPTS) {
           artifacts.push({ displayName: candidate.label, unconfirmed: true, error: message });
         } else {
-          await reopenConversationForArtifacts(job, responseIndex, `retry ${attempt + 1} for ${candidate.label}`);
+          await reopenConversationForArtifacts(job, responseIndex, responseText, `retry ${attempt + 1} for ${candidate.label}`);
           await sleep(1_000);
         }
       } finally {
@@ -1443,7 +1522,7 @@ async function run() {
     currentJob = await mutateJob((job) => ({ ...job, ...phasePatch("extracting_response", { heartbeatAt: new Date().toISOString() }) }));
     await secureWriteText(currentJob.responsePath, `${completion.responseText.trim()}\n`);
     currentJob = await mutateJob((job) => ({ ...job, ...phasePatch("downloading_artifacts", { heartbeatAt: new Date().toISOString() }) }));
-    const artifacts = await downloadArtifacts(currentJob, completion.responseIndex);
+    const artifacts = await downloadArtifacts(currentJob, completion.responseIndex, completion.responseText);
     const artifactFailureCount = artifacts.filter((artifact) => artifact.unconfirmed || artifact.error).length;
 
     await heartbeat(

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-oracle",
-  "version": "0.1.9",
+  "version": "0.1.11",
   "description": "ChatGPT web-oracle extension for pi with isolated browser auth, async jobs, and project-context archives.",
   "private": false,
   "license": "MIT",
@@ -40,8 +40,8 @@
     ]
   },
   "scripts": {
-    "check:oracle-extension": "node --check extensions/oracle/worker/run-job.mjs && node --check extensions/oracle/worker/auth-cookie-policy.mjs && node --check extensions/oracle/worker/auth-bootstrap.mjs && esbuild extensions/oracle/index.ts --bundle --platform=node --format=esm --external:@mariozechner/pi-coding-agent --external:@mariozechner/pi-ai --external:@sinclair/typebox --outfile=/tmp/pi-oracle-extension-check.js",
-    "sanity:oracle": "tsx scripts/oracle-sanity.ts",
+    "check:oracle-extension": "node --check extensions/oracle/worker/run-job.mjs && node --check extensions/oracle/worker/artifact-heuristics.mjs && node --check extensions/oracle/worker/auth-cookie-policy.mjs && node --check extensions/oracle/worker/auth-bootstrap.mjs && esbuild extensions/oracle/index.ts --bundle --platform=node --format=esm --external:@mariozechner/pi-coding-agent --external:@mariozechner/pi-ai --external:@sinclair/typebox --outfile=/tmp/pi-oracle-extension-check.js",
+    "sanity:oracle": "node scripts/oracle-sanity-runner.mjs",
     "pack:check": "npm pack --dry-run"
   },
   "dependencies": {