@bastani/atomic 0.9.16-alpha.2 → 0.9.16-alpha.3

package/dist/builtin/web-access/storage.ts

@@ -1,72 +0,0 @@
-import type { ExtensionContext } from "@bastani/atomic";
-import type { ExtractedContent } from "./extract.js";
-import type { SearchResult } from "./perplexity.js";
-
-const CACHE_TTL_MS = 60 * 60 * 1000;
-
-export interface QueryResultData {
-	query: string;
-	answer: string;
-	results: SearchResult[];
-	error: string | null;
-	provider?: string;
-}
-
-export interface StoredSearchData {
-	id: string;
-	type: "search" | "fetch";
-	timestamp: number;
-	queries?: QueryResultData[];
-	urls?: ExtractedContent[];
-}
-
-const storedResults = new Map<string, StoredSearchData>();
-
-export function generateId(): string {
-	return Date.now().toString(36) + Math.random().toString(36).slice(2, 8);
-}
-
-export function storeResult(id: string, data: StoredSearchData): void {
-	storedResults.set(id, data);
-}
-
-export function getResult(id: string): StoredSearchData | null {
-	return storedResults.get(id) ?? null;
-}
-
-export function getAllResults(): StoredSearchData[] {
-	return Array.from(storedResults.values());
-}
-
-export function deleteResult(id: string): boolean {
-	return storedResults.delete(id);
-}
-
-export function clearResults(): void {
-	storedResults.clear();
-}
-
-function isValidStoredData(data: unknown): data is StoredSearchData {
-	if (!data || typeof data !== "object") return false;
-	const d = data as Record<string, unknown>;
-	if (typeof d.id !== "string" || !d.id) return false;
-	if (d.type !== "search" && d.type !== "fetch") return false;
-	if (typeof d.timestamp !== "number") return false;
-	if (d.type === "search" && !Array.isArray(d.queries)) return false;
-	if (d.type === "fetch" && !Array.isArray(d.urls)) return false;
-	return true;
-}
-
-export function restoreFromSession(ctx: ExtensionContext): void {
-	storedResults.clear();
-	const now = Date.now();
-
-	for (const entry of ctx.sessionManager.getBranch()) {
-		if (entry.type === "custom" && entry.customType === "web-search-results") {
-			const data = entry.data;
-			if (isValidStoredData(data) && now - data.timestamp < CACHE_TTL_MS) {
-				storedResults.set(data.id, data);
-			}
-		}
-	}
-}

package/dist/builtin/web-access/subprocess.ts

@@ -1,102 +0,0 @@
-import { createChildProcessEnvironment } from "@bastani/atomic";
-
-export interface BunSubprocessOptions {
-	timeoutMs: number;
-	maxStdoutBytes: number;
-	maxStderrBytes?: number;
-	signal?: AbortSignal;
-	cwd?: string;
-	env?: Record<string, string>;
-}
-
-export interface BunSubprocessResult {
-	exitCode: number;
-	stdout: Buffer;
-	stderr: string;
-}
-
-export class AsyncSubprocessError extends Error {
-	readonly code?: string;
-	readonly stderr: string;
-	readonly killed: boolean;
-	constructor(message: string, options: { code?: string; stderr?: string; killed?: boolean } = {}) {
-		super(message);
-		this.name = "AsyncSubprocessError";
-		this.code = options.code;
-		this.stderr = options.stderr ?? "";
-		this.killed = options.killed === true;
-	}
-}
-
-async function readBounded(
-	stream: ReadableStream<Uint8Array>,
-	maxBytes: number,
-	onOverflow: () => void,
-): Promise<Buffer> {
-	const reader = stream.getReader();
-	const chunks: Uint8Array[] = [];
-	let bytes = 0;
-	try {
-		while (true) {
-			const { done, value } = await reader.read();
-			if (done) break;
-			bytes += value.byteLength;
-			if (bytes > maxBytes) {
-				onOverflow();
-				throw new AsyncSubprocessError(`Subprocess output exceeded ${maxBytes} bytes`, { code: "ENOBUFS", killed: true });
-			}
-			chunks.push(value);
-		}
-	} finally {
-		reader.releaseLock();
-	}
-	return Buffer.concat(chunks.map((chunk) => Buffer.from(chunk)), bytes);
-}
-
-export async function runBunSubprocess(
-	command: string,
-	args: readonly string[],
-	options: BunSubprocessOptions,
-): Promise<BunSubprocessResult> {
-	let proc: ReturnType<typeof Bun.spawn>;
-	try {
-		proc = Bun.spawn([command, ...args], {
-			cwd: options.cwd,
-			env: createChildProcessEnvironment(options.env),
-			stdin: "ignore",
-			stdout: "pipe",
-			stderr: "pipe",
-		});
-	} catch (error) {
-		const failure = error instanceof Error ? error : new Error(String(error));
-		throw new AsyncSubprocessError(failure.message, { code: (failure as Error & { code?: string }).code });
-	}
-	let timedOut = false;
-	let aborted = false;
-	const terminate = (): void => {
-		try { proc.kill("SIGTERM"); } catch {}
-		void Promise.race([proc.exited, Bun.sleep(500)]).then(() => {
-			if (proc.exitCode === null) try { proc.kill("SIGKILL"); } catch {}
-		});
-	};
-	const onAbort = (): void => { aborted = true; terminate(); };
-	options.signal?.addEventListener("abort", onAbort, { once: true });
-	const timeout = setTimeout(() => { timedOut = true; terminate(); }, options.timeoutMs);
-	try {
-		const stdoutPromise = readBounded(proc.stdout as ReadableStream<Uint8Array>, options.maxStdoutBytes, terminate);
-		const stderrPromise = readBounded(proc.stderr as ReadableStream<Uint8Array>, options.maxStderrBytes ?? 256 * 1024, terminate);
-		const [exitCode, stdout, stderrBuffer] = await Promise.all([proc.exited, stdoutPromise, stderrPromise]);
-		const stderr = stderrBuffer.toString("utf8");
-		if (timedOut) throw new AsyncSubprocessError(`${command} timed out`, { code: "ETIMEDOUT", stderr, killed: true });
-		if (aborted) throw new AsyncSubprocessError(`${command} aborted`, { code: "ABORT_ERR", stderr, killed: true });
-		if (exitCode !== 0) throw new AsyncSubprocessError(`${command} exited with code ${exitCode}`, { code: String(exitCode), stderr });
-		return { exitCode, stdout, stderr };
-	} catch (error) {
-		terminate();
-		if (error instanceof AsyncSubprocessError) throw error;
-		throw new AsyncSubprocessError(error instanceof Error ? error.message : String(error), { killed: true });
-	} finally {
-		clearTimeout(timeout);
-		options.signal?.removeEventListener("abort", onAbort);
-	}
-}

package/dist/builtin/web-access/summary-review.ts

@@ -1,287 +0,0 @@
-import type { ProviderHeaders } from "@bastani/pi-ai";
-import { getModel, type Message, type Model } from "@bastani/pi-ai/compat";
-import type { ExtensionContext } from "@bastani/atomic";
-import type { QueryResultData } from "./storage.js";
-
-const PREFERRED_SUMMARY_MODELS = [
-	{ provider: "anthropic", id: "claude-haiku-4-5" },
-	{ provider: "openai-codex", id: "gpt-5.3-codex-spark" },
-] as const;
-
-export interface SummaryMeta {
-	model: string | null;
-	durationMs: number;
-	tokenEstimate: number;
-	fallbackUsed: boolean;
-	fallbackReason?: string;
-	edited?: boolean;
-}
-
-export type SummaryGenerationContext = Pick<ExtensionContext, "model" | "modelRegistry">;
-
-function estimateTokens(text: string): number {
-	const trimmed = text.trim();
-	if (trimmed.length === 0) return 0;
-	return Math.max(1, Math.ceil(trimmed.length / 4));
-}
-
-function summarizeQueryResult(result: QueryResultData): string {
-	if (result.error) {
-		return `Query: ${result.query}\nStatus: Error\nError: ${result.error}`;
-	}
-
-	const lines = [
-		`Query: ${result.query}`,
-		`Provider: ${result.provider ?? "unknown"}`,
-		`Answer: ${result.answer || "(no answer text returned)"}`,
-	];
-
-	if (result.results.length === 0) {
-		lines.push("Sources: none");
-		return lines.join("\n");
-	}
-
-	lines.push("Sources:");
-	for (let i = 0; i < result.results.length; i++) {
-		const source = result.results[i];
-		lines.push(`${i + 1}. ${source.title} — ${source.url}`);
-	}
-
-	return lines.join("\n");
-}
-
-export function buildSummaryPrompt(results: QueryResultData[], feedback?: string): string {
-	const sections = [
-		"You are writing the final web search summary for a coding assistant.",
-		"Write a concise, factual summary using only the provided search results.",
-		"Requirements:",
-		"- Keep it readable and skimmable.",
-		"- Include key findings and caveats.",
-		"- Do not invent sources or claims.",
-		"- If evidence is weak or conflicting, say so explicitly.",
-		"- End with a short \"Sources\" section listing the most relevant URLs.",
-	];
-
-	if (feedback) {
-		sections.push("- Incorporate the user feedback provided below into the summary.");
-	}
-
-	sections.push("");
-	sections.push("<search_results>");
-
-	for (let i = 0; i < results.length; i++) {
-		sections.push(`\n[Result ${i + 1}]`);
-		sections.push(summarizeQueryResult(results[i]));
-	}
-
-	sections.push("\n</search_results>");
-
-	if (feedback) {
-		sections.push("");
-		sections.push("<user_feedback>");
-		sections.push(feedback);
-		sections.push("</user_feedback>");
-	}
-
-	return sections.join("\n");
-}
-
-function buildDeterministicAnswerPreview(answer: string): string {
-	let text = answer.replace(/\s+/g, " ").trim();
-	if (text.length === 0) return "";
-
-	const sourceMarker = text.search(/\bSources?\s*:/i);
-	if (sourceMarker >= 0) text = text.slice(0, sourceMarker).trim();
-	if (text.length === 0) return "";
-
-	return text.length > 240 ? `${text.slice(0, 237)}...` : text;
-}
-
-function buildDeterministicSummaryLines(results: QueryResultData[]): string[] {
-	if (results.length === 0) {
-		return [
-			"No completed search results were available when the curator session finished.",
-			"",
-			"Sources",
-			"- None",
-		];
-	}
-
-	const lines: string[] = [
-		"Summary based on the currently selected search results.",
-		"",
-	];
-
-	const sourceUrls: string[] = [];
-	let successful = 0;
-	let failed = 0;
-
-	for (const result of results) {
-		if (result.error) {
-			failed += 1;
-			lines.push(`- ${result.query}: failed (${result.error})`);
-			continue;
-		}
-
-		successful += 1;
-		const preview = buildDeterministicAnswerPreview(result.answer);
-		if (preview.length > 0) {
-			lines.push(`- ${result.query}: ${preview}`);
-		} else {
-			lines.push(`- ${result.query}: returned ${result.results.length} source${result.results.length === 1 ? "" : "s"} without answer text.`);
-		}
-
-		for (const source of result.results) {
-			if (!sourceUrls.includes(source.url)) {
-				sourceUrls.push(source.url);
-			}
-		}
-	}
-
-	lines.push("");
-	lines.push(`Completed queries: ${results.length}`);
-	lines.push(`Successful: ${successful}`);
-	lines.push(`Failed: ${failed}`);
-	lines.push("");
-	lines.push("Sources");
-
-	if (sourceUrls.length === 0) {
-		lines.push("- None");
-	} else {
-		for (const url of sourceUrls.slice(0, 12)) {
-			lines.push(`- ${url}`);
-		}
-		if (sourceUrls.length > 12) {
-			lines.push(`- ... and ${sourceUrls.length - 12} more`);
-		}
-	}
-
-	return lines;
-}
-
-export function buildDeterministicSummary(results: QueryResultData[]): { summary: string; meta: SummaryMeta } {
-	const summary = buildDeterministicSummaryLines(results).join("\n").trim();
-	const nonEmptySummary = summary.length > 0
-		? summary
-		: "No completed search results were available when the curator session finished.\n\nSources\n- None";
-
-	return {
-		summary: nonEmptySummary,
-		meta: {
-			model: null,
-			durationMs: 0,
-			tokenEstimate: estimateTokens(nonEmptySummary),
-			fallbackUsed: true,
-			fallbackReason: "deterministic-submit-fallback",
-			edited: false,
-		},
-	};
-}
-
-async function resolveSummaryModel(
-	ctx: SummaryGenerationContext,
-	modelOverride?: string,
-): Promise<{ model: Model; apiKey: string; headers?: ProviderHeaders }> {
-	const normalizedOverride = typeof modelOverride === "string" ? modelOverride.trim() : "";
-	if (normalizedOverride.length > 0) {
-		const slashIndex = normalizedOverride.indexOf("/");
-		if (slashIndex <= 0 || slashIndex >= normalizedOverride.length - 1) {
-			throw new Error(`Invalid summary model: ${normalizedOverride}. Use provider/model-id.`);
-		}
-		const provider = normalizedOverride.slice(0, slashIndex);
-		const modelId = normalizedOverride.slice(slashIndex + 1);
-		const selectedModel = ctx.modelRegistry.find(provider, modelId);
-		if (!selectedModel) {
-			throw new Error(`Summary model not found: ${normalizedOverride}`);
-		}
-		const selectedAuth = await ctx.modelRegistry.getApiKeyAndHeaders(selectedModel);
-		if (!selectedAuth.ok || !selectedAuth.apiKey) {
-			throw new Error(`No API key available for summary model ${normalizedOverride}`);
-		}
-		return {
-			model: selectedAuth.baseUrl === undefined ? selectedModel : { ...selectedModel, baseUrl: selectedAuth.baseUrl },
-			apiKey: selectedAuth.apiKey,
-			headers: selectedAuth.headers,
-		};
-	}
-
-	for (const { provider, id } of PREFERRED_SUMMARY_MODELS) {
-		const model = getModel(provider, id);
-		if (!model) continue;
-		const auth = await ctx.modelRegistry.getApiKeyAndHeaders(model);
-		if (auth.ok && auth.apiKey) {
-			return {
-				model: auth.baseUrl === undefined ? model : { ...model, baseUrl: auth.baseUrl },
-				apiKey: auth.apiKey,
-				headers: auth.headers,
-			};
-		}
-	}
-
-	throw new Error(`No API key available for summary models: ${PREFERRED_SUMMARY_MODELS.map(c => `${c.provider}/${c.id}`).join(", ")}`);
-}
-
-function getTextFromContentPart(part: unknown): string {
-	if (!part || typeof part !== "object") return "";
-	const value = part as Record<string, unknown>;
-	if (typeof value.text === "string") return value.text;
-	if (typeof value.refusal === "string") return value.refusal;
-	return "";
-}
-
-function getContentPartType(part: unknown): string {
-	if (!part || typeof part !== "object") return "unknown";
-	const value = part as Record<string, unknown>;
-	return typeof value.type === "string" ? value.type : "unknown";
-}
-
-export async function generateSummaryDraft(
-	results: QueryResultData[],
-	ctx: SummaryGenerationContext,
-	signal?: AbortSignal,
-	modelOverride?: string,
-	feedback?: string,
-): Promise<{ summary: string; meta: SummaryMeta }> {
-	if (!ctx || !ctx.modelRegistry) {
-		throw new Error("Summary generation context unavailable");
-	}
-
-	const startedAt = Date.now();
-	const { model, apiKey, headers } = await resolveSummaryModel(ctx, modelOverride);
-	const prompt = buildSummaryPrompt(results, feedback);
-
-	const userMessage: Message = {
-		role: "user",
-		content: [{ type: "text", text: prompt }],
-		timestamp: Date.now(),
-	};
-
-	const response = await ctx.modelRegistry.complete(model, { messages: [userMessage] }, { apiKey, headers, signal });
-	if (response.stopReason === "aborted") {
-		throw new Error("Aborted");
-	}
-
-	const contentParts = Array.isArray(response.content) ? response.content : [];
-	const summary = contentParts
-		.map(part => getTextFromContentPart(part))
-		.filter(text => text.trim().length > 0)
-		.join("\n")
-		.trim();
-
-	if (summary.length === 0) {
-		const partTypes = contentParts.map(part => getContentPartType(part));
-		const typesLabel = partTypes.length > 0 ? partTypes.join(", ") : "none";
-		throw new Error(`Summary model returned empty response (content parts: ${typesLabel})`);
-	}
-
-	return {
-		summary,
-		meta: {
-			model: `${model.provider}/${model.id}`,
-			durationMs: Math.max(0, Date.now() - startedAt),
-			tokenEstimate: estimateTokens(summary),
-			fallbackUsed: false,
-			edited: false,
-		},
-	};
-}

package/dist/builtin/web-access/utils.ts

@@ -1,44 +0,0 @@
-export function formatSeconds(s: number): string {
-	const h = Math.floor(s / 3600);
-	const m = Math.floor((s % 3600) / 60);
-	const sec = s % 60;
-	if (h > 0) return `${h}:${String(m).padStart(2, "0")}:${String(sec).padStart(2, "0")}`;
-	return `${m}:${String(sec).padStart(2, "0")}`;
-}
-
-export function readExecError(err: unknown): { code?: string; stderr: string; message: string } {
-	if (!err || typeof err !== "object") {
-		return { stderr: "", message: String(err) };
-	}
-	const code = (err as { code?: string }).code;
-	const message = (err as { message?: string }).message ?? "";
-	const stderrRaw = (err as { stderr?: Buffer | string }).stderr;
-	const stderr = Buffer.isBuffer(stderrRaw)
-		? stderrRaw.toString("utf-8")
-		: typeof stderrRaw === "string"
-			? stderrRaw
-			: "";
-	return { code, stderr, message };
-}
-
-export function isTimeoutError(err: unknown): boolean {
-	if (!err || typeof err !== "object") return false;
-	if ((err as { killed?: boolean }).killed) return true;
-	const name = (err as { name?: string }).name;
-	const code = (err as { code?: string }).code;
-	const message = (err as { message?: string }).message ?? "";
-	return name === "AbortError" || code === "ETIMEDOUT" || message.toLowerCase().includes("timed out");
-}
-
-export function trimErrorText(text: string): string {
-	return text.replace(/\s+/g, " ").trim().slice(0, 200);
-}
-
-export function mapFfmpegError(err: unknown): string {
-	const { code, stderr, message } = readExecError(err);
-	if (code === "ENOENT") return "ffmpeg is not installed. Install with: brew install ffmpeg";
-	if (isTimeoutError(err)) return "ffmpeg timed out extracting frame";
-	if (stderr.includes("403")) return "Stream URL returned 403 — may have expired, try again";
-	const snippet = trimErrorText(stderr || message);
-	return snippet ? `ffmpeg failed: ${snippet}` : "ffmpeg failed";
-}