@bdsqqq/pi 0.0.1

package/dist/extensions/web-search.js

@@ -0,0 +1,254 @@
+import { withPromptPatch } from "../core/prompt-patch.js";
+import { boxRendererWindowed, osc8Link } from "../core/box-format.js";
+import { getExtensionConfig } from "../core/config.js";
+import { spawn } from "node:child_process";
+import { Text } from "@mariozechner/pi-tui";
+import { Type } from "@sinclair/typebox";
+//#region packages/extensions/web-search/index.ts
+/**
+* web_search tool — direct HTTP call to Parallel AI's Search API.
+*
+* uses curl (not fetch/SDK) because pi extensions run in a nix-built
+* environment where adding npm deps requires a rebuild. curl is always
+* available and the single-endpoint usage doesn't justify the SDK.
+*
+* cost is derived from the response's usage array, not hardcoded —
+* the API returns UsageItem[] with SKU counts, we multiply by known
+* unit prices. if the API omits usage, we fall back to base search cost.
+*
+* refs:
+*   schema: https://docs.parallel.ai/public-openapi.json (UsageItem)
+*   pricing: https://docs.parallel.ai/pricing (Search API section)
+*/
+const CONFIG_DEFAULTS = {
+	defaultMaxResults: 10,
+	endpoint: "https://api.parallel.ai/v1beta/search",
+	curlTimeoutSecs: 30
+};
+/** per-result excerpts for collapsed display — first 5 visual lines */
+const COLLAPSED_EXCERPTS = [{
+	focus: "head",
+	context: 5
+}];
+/** per-unit pricing by SKU name ($/unit). ref: https://docs.parallel.ai/pricing */
+const SKU_UNIT_COST = {
+	sku_search: .005,
+	sku_search_additional_results: .001
+};
+/** falls back to base search cost when API omits usage (e.g., older API versions). */
+function costFromUsage(usage) {
+	if (!usage?.length) return SKU_UNIT_COST.sku_search ?? 0;
+	let total = 0;
+	for (const item of usage) total += (SKU_UNIT_COST[item.name] ?? 0) * item.count;
+	return total;
+}
+function searchParallel(apiKey, body, endpoint, curlTimeoutSecs, signal) {
+	return new Promise((resolve) => {
+		const payload = JSON.stringify(body);
+		const child = spawn("curl", [
+			"-sL",
+			"-X",
+			"POST",
+			"-H",
+			"Content-Type: application/json",
+			"-H",
+			`x-api-key: ${apiKey}`,
+			"-H",
+			"parallel-beta: search-extract-2025-10-10",
+			"-m",
+			String(curlTimeoutSecs),
+			"-d",
+			payload,
+			endpoint
+		], { stdio: [
+			"ignore",
+			"pipe",
+			"pipe"
+		] });
+		let stdout = "";
+		let stderr = "";
+		let aborted = false;
+		const onAbort = () => {
+			aborted = true;
+			if (!child.killed) child.kill("SIGTERM");
+		};
+		if (signal) if (signal.aborted) onAbort();
+		else signal.addEventListener("abort", onAbort, { once: true });
+		child.stdout?.on("data", (data) => {
+			stdout += data.toString("utf-8");
+		});
+		child.stderr?.on("data", (data) => {
+			stderr += data.toString("utf-8");
+		});
+		child.on("error", (err) => {
+			signal?.removeEventListener("abort", onAbort);
+			resolve({ error: `curl error: ${err.message}` });
+		});
+		child.on("close", (code) => {
+			signal?.removeEventListener("abort", onAbort);
+			if (aborted) {
+				resolve({ error: "search aborted" });
+				return;
+			}
+			if (code !== 0) {
+				resolve({ error: `search failed: ${stderr.trim() || `curl exited with code ${code}`}` });
+				return;
+			}
+			try {
+				resolve({ data: JSON.parse(stdout) });
+			} catch {
+				resolve({ error: `invalid response from Parallel API: ${stdout.slice(0, 200)}` });
+			}
+		});
+	});
+}
+function formatResults(results) {
+	if (results.length === 0) return {
+		text: "(no results found)",
+		headerLineIndices: []
+	};
+	const lines = [];
+	const headerLineIndices = [];
+	for (let i = 0; i < results.length; i++) {
+		const r = results[i];
+		headerLineIndices.push(lines.length);
+		lines.push(`### ${r.title || "(untitled)"}`);
+		lines.push(r.url);
+		if (r.publish_date) lines.push(`*${r.publish_date}*`);
+		if (r.excerpts?.length) {
+			lines.push("");
+			for (let j = 0; j < r.excerpts.length; j++) {
+				const excerptLines = r.excerpts[j].split("\n");
+				lines.push(...excerptLines);
+				if (j < r.excerpts.length - 1) lines.push("");
+			}
+		}
+		if (i < results.length - 1) {
+			lines.push("");
+			lines.push("---");
+			lines.push("");
+		}
+	}
+	return {
+		text: lines.join("\n"),
+		headerLineIndices
+	};
+}
+/** convert raw SearchResult[] into BoxSection[] for box-format rendering. */
+function resultsToSections(results) {
+	return results.map((r) => {
+		const lines = [];
+		lines.push({
+			text: osc8Link(r.url, r.url),
+			highlight: true
+		});
+		if (r.publish_date) lines.push({
+			text: r.publish_date,
+			highlight: true
+		});
+		if (r.excerpts?.length) {
+			lines.push({
+				text: "",
+				highlight: false
+			});
+			for (let j = 0; j < r.excerpts.length; j++) {
+				for (const l of r.excerpts[j].split("\n")) lines.push({
+					text: l,
+					highlight: false
+				});
+				if (j < r.excerpts.length - 1) lines.push({
+					text: "",
+					highlight: false
+				});
+			}
+		}
+		return {
+			header: r.title || "(untitled)",
+			blocks: [{ lines }]
+		};
+	});
+}
+function createWebSearchTool(config = CONFIG_DEFAULTS) {
+	return {
+		name: "web_search",
+		label: "Web Search",
+		description: "Search the web for information relevant to a research objective.\n\nUse when you need up-to-date or precise documentation. Use `read_web_page` to fetch full content from a specific URL.\n\n# Examples\n\nGet API documentation for a specific provider\n```json\n{\"objective\":\"I want to know the request fields for the Stripe billing create customer API. Prefer Stripe's docs site.\"}\n```\n\nSee usage documentation for newly released library features\n```json\n{\"objective\":\"I want to know how to use SvelteKit remote functions, which is a new feature shipped in the last month.\",\"search_queries\":[\"sveltekit\",\"remote function\"]}\n```",
+		parameters: Type.Object({
+			objective: Type.String({ description: "A natural-language description of the broader task or research goal, including any source or freshness guidance." }),
+			search_queries: Type.Optional(Type.Array(Type.String(), { description: "Optional keyword queries to ensure matches for specific terms are prioritized (recommended for best results)." })),
+			max_results: Type.Optional(Type.Number({ description: `The maximum number of results to return (default: ${config.defaultMaxResults}).` }))
+		}),
+		async execute(_toolCallId, params, signal) {
+			const p = params;
+			const apiKey = process.env.PARALLEL_API_KEY;
+			if (!apiKey) return {
+				content: [{
+					type: "text",
+					text: "PARALLEL_API_KEY not set. add it to secrets.yaml and export in shell.nix."
+				}],
+				isError: true
+			};
+			const body = {
+				objective: p.objective,
+				max_results: p.max_results ?? config.defaultMaxResults,
+				excerpts: { max_chars_per_result: 2e3 }
+			};
+			if (p.search_queries?.length) body.search_queries = p.search_queries;
+			const { data, error } = await searchParallel(apiKey, body, config.endpoint, config.curlTimeoutSecs, signal);
+			if (error) return {
+				content: [{
+					type: "text",
+					text: error
+				}],
+				isError: true
+			};
+			if (!data?.results) return { content: [{
+				type: "text",
+				text: "(no results)"
+			}] };
+			const { text, headerLineIndices } = formatResults(data.results);
+			let output = text;
+			if (data.warnings?.length) output += `\n\n**Warnings:** ${data.warnings.join("; ")}`;
+			const resultSections = resultsToSections(data.results);
+			const details = {
+				cost: costFromUsage(data.usage),
+				matchLineIndices: headerLineIndices,
+				resultSections
+			};
+			return {
+				content: [{
+					type: "text",
+					text: output
+				}],
+				details
+			};
+		},
+		renderCall(args, theme) {
+			const objective = args.objective || "...";
+			const short = objective.length > 70 ? `${objective.slice(0, 70)}...` : objective;
+			let text = theme.fg("toolTitle", theme.bold("web_search ")) + theme.fg("dim", short);
+			if (args.search_queries?.length) text += theme.fg("muted", ` [${args.search_queries.join(", ")}]`);
+			return new Text(text, 0, 0);
+		},
+		renderResult(result, { expanded }, _theme) {
+			const sections = result.details?.resultSections;
+			if (!sections?.length) {
+				const text = result.content?.[0];
+				return new Text(text?.type === "text" ? text.text : "(no output)", 0, 0);
+			}
+			return boxRendererWindowed(() => sections, {
+				collapsed: {
+					maxSections: 3,
+					excerpts: COLLAPSED_EXCERPTS
+				},
+				expanded: {}
+			}, void 0, expanded);
+		}
+	};
+}
+function web_search_default(pi) {
+	const cfg = getExtensionConfig("@bds_pi/web-search", CONFIG_DEFAULTS);
+	pi.registerTool(withPromptPatch(createWebSearchTool(cfg)));
+}
+//#endregion
+export { createWebSearchTool, web_search_default as default };

package/dist/extensions.d.ts

@@ -0,0 +1,29 @@
+import export_default from "./extensions/bash.js";
+import export_default$1 from "./extensions/code-review.js";
+import commandPaletteExtension from "./extensions/command-palette.js";
+import export_default$2 from "./extensions/create-file.js";
+import export_default$3 from "./extensions/edit-file.js";
+import export_default$4 from "./extensions/editor.js";
+import export_default$5 from "./extensions/finder.js";
+import export_default$6 from "./extensions/format-file.js";
+import export_default$7 from "./extensions/github.js";
+import export_default$8 from "./extensions/glob.js";
+import export_default$9 from "./extensions/grep.js";
+import export_default$10 from "./extensions/handoff.js";
+import export_default$11 from "./extensions/librarian.js";
+import export_default$12 from "./extensions/look-at.js";
+import export_default$15 from "./extensions/read.js";
+import export_default$13 from "./extensions/ls.js";
+import mermaidInlineExtension from "./extensions/mermaid.js";
+import export_default$14 from "./extensions/oracle.js";
+import export_default$16 from "./extensions/read-session.js";
+import export_default$17 from "./extensions/read-web-page.js";
+import export_default$18 from "./extensions/search-sessions.js";
+import export_default$19 from "./extensions/session-name.js";
+import export_default$20 from "./extensions/skill.js";
+import export_default$21 from "./extensions/system-prompt.js";
+import export_default$22 from "./extensions/task.js";
+import export_default$23 from "./extensions/tool-harness.js";
+import export_default$24 from "./extensions/undo-edit.js";
+import export_default$25 from "./extensions/web-search.js";
+export { export_default as bash, export_default$1 as codeReview, commandPaletteExtension as commandPalette, export_default$2 as createFile, export_default$3 as editFile, export_default$4 as editor, export_default$5 as finder, export_default$6 as formatFile, export_default$7 as github, export_default$8 as glob, export_default$9 as grep, export_default$10 as handoff, export_default$11 as librarian, export_default$12 as lookAt, export_default$13 as ls, mermaidInlineExtension as mermaid, export_default$14 as oracle, export_default$15 as read, export_default$16 as readSession, export_default$17 as readWebPage, export_default$18 as searchSessions, export_default$19 as sessionName, export_default$20 as skill, export_default$21 as systemPrompt, export_default$22 as task, export_default$23 as toolHarness, export_default$24 as undoEdit, export_default$25 as webSearch };

package/dist/extensions.js

@@ -0,0 +1,29 @@
+import read_default from "./extensions/read.js";
+import bash_default from "./extensions/bash.js";
+import code_review_default from "./extensions/code-review.js";
+import { t as commandPaletteExtension } from "./command-palette-B97yKqne.js";
+import create_file_default from "./extensions/create-file.js";
+import edit_file_default from "./extensions/edit-file.js";
+import { t as editor_default } from "./editor-BljciPdx.js";
+import finder_default from "./extensions/finder.js";
+import format_file_default from "./extensions/format-file.js";
+import github_default from "./extensions/github.js";
+import glob_default from "./extensions/glob.js";
+import grep_default from "./extensions/grep.js";
+import handoff_default from "./extensions/handoff.js";
+import librarian_default from "./extensions/librarian.js";
+import look_at_default from "./extensions/look-at.js";
+import ls_default from "./extensions/ls.js";
+import { t as mermaidInlineExtension } from "./mermaid-C7xCVBrw.js";
+import oracle_default from "./extensions/oracle.js";
+import read_session_default from "./extensions/read-session.js";
+import read_web_page_default from "./extensions/read-web-page.js";
+import search_sessions_default from "./extensions/search-sessions.js";
+import session_name_default from "./extensions/session-name.js";
+import skill_default from "./extensions/skill.js";
+import system_prompt_default from "./extensions/system-prompt.js";
+import task_default from "./extensions/task.js";
+import tool_harness_default from "./extensions/tool-harness.js";
+import undo_edit_default from "./extensions/undo-edit.js";
+import web_search_default from "./extensions/web-search.js";
+export { bash_default as bash, code_review_default as codeReview, commandPaletteExtension as commandPalette, create_file_default as createFile, edit_file_default as editFile, editor_default as editor, finder_default as finder, format_file_default as formatFile, github_default as github, glob_default as glob, grep_default as grep, handoff_default as handoff, librarian_default as librarian, look_at_default as lookAt, ls_default as ls, mermaidInlineExtension as mermaid, oracle_default as oracle, read_default as read, read_session_default as readSession, read_web_page_default as readWebPage, search_sessions_default as searchSessions, session_name_default as sessionName, skill_default as skill, system_prompt_default as systemPrompt, task_default as task, tool_harness_default as toolHarness, undo_edit_default as undoEdit, web_search_default as webSearch };