@xynogen/pix-9router 0.4.4 → 0.6.0

package/src/fetch.ts

@@ -1,260 +0,0 @@
-/**
- * fetch.ts — fetch tool via 9Router API, with curl fallback.
- *
- * Environment:
- *   ROUTER_API_BASE  — router API base URL (default: https://9router.example.com/v1)
- *   ROUTER_API_KEY   — bearer token for the router
- */
-
-import { StringEnum } from "@earendil-works/pi-ai";
-import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
-import { Type } from "typebox";
-import { apiPost, curl, isCancelled } from "./http.js";
-import { makeRenderCall, makeRenderResult } from "./render.js";
-
-type RouterOutcome = "running" | "fallback" | "success" | "cancelled" | "error";
-type FetchFormat = "markdown" | "text" | "html";
-
-export interface FetchResultDetails {
-	_type: "fetchResult";
-	outcome: RouterOutcome;
-	url: string;
-	format: FetchFormat;
-	source?: "api" | "curl-fallback" | "failed";
-	chars?: number;
-	truncated?: boolean;
-}
-
-export interface FetchParams {
-	url: string;
-	format?: FetchFormat;
-	max_characters?: number;
-}
-
-interface FetchResult {
-	content: { type: "text"; text: string }[];
-	details: FetchResultDetails;
-	isError?: boolean;
-}
-
-interface FetchDependencies {
-	apiPost: typeof apiPost;
-	curl: typeof curl;
-}
-
-/**
- * Extract readable page content from the exa fetch JSON envelope.
- * Shape: { provider, url, title, content: { format, text }, ... }
- * Falls back to the raw string when it isn't the expected JSON shape.
- */
-export function formatFetchResult(raw: string): string {
-	let parsed: unknown;
-	try {
-		parsed = JSON.parse(raw);
-	} catch {
-		return raw;
-	}
-	if (typeof parsed !== "object" || parsed === null) return raw;
-
-	const obj = parsed as {
-		title?: string | null;
-		url?: string | null;
-		content?: { text?: string | null } | string | null;
-	};
-
-	const text = typeof obj.content === "string" ? obj.content : (obj.content?.text ?? undefined);
-	if (typeof text !== "string") return raw;
-
-	const header: string[] = [];
-	if (obj.title) header.push(`# ${obj.title}`);
-	if (obj.url) header.push(`URL: ${obj.url}`);
-
-	return header.length > 0 ? `${header.join("\n")}\n\n${text}` : text;
-}
-
-export async function executeFetch(
-	params: FetchParams,
-	signal?: AbortSignal,
-	onUpdate?: (result: FetchResult) => void,
-	dependencies: FetchDependencies = { apiPost, curl },
-): Promise<FetchResult> {
-	const maxChars = params.max_characters ?? 1000;
-	const fmt = params.format ?? "markdown";
-	let apiMsg = "";
-
-	try {
-		onUpdate?.({
-			content: [{ type: "text", text: `Fetching (${fmt}): ${params.url}...` }],
-			details: {
-				_type: "fetchResult",
-				outcome: "running",
-				url: params.url,
-				format: fmt,
-			},
-		});
-
-		const raw = await dependencies.apiPost(
-			"/web/fetch",
-			{
-				model: "exa",
-				url: params.url,
-				format: fmt,
-				max_characters: maxChars,
-			},
-			signal,
-		);
-
-		const formatted = formatFetchResult(raw);
-		const truncated =
-			maxChars > 0 && formatted.length > maxChars
-				? `${formatted.slice(0, maxChars)}\n\n[truncated]`
-				: formatted;
-
-		return {
-			content: [{ type: "text", text: truncated.slice(0, 20_000) }],
-			details: {
-				_type: "fetchResult",
-				outcome: "success",
-				url: params.url,
-				format: fmt,
-				source: "api",
-				chars: formatted.length,
-				truncated: maxChars > 0 && formatted.length > maxChars,
-			},
-		};
-	} catch (apiErr: unknown) {
-		apiMsg = apiErr instanceof Error ? apiErr.message : String(apiErr);
-		if (isCancelled(apiErr, signal)) {
-			return {
-				content: [{ type: "text", text: `Fetch cancelled.\nAPI: ${apiMsg}` }],
-				details: {
-					_type: "fetchResult",
-					outcome: "cancelled",
-					url: params.url,
-					format: fmt,
-				},
-			};
-		}
-		onUpdate?.({
-			content: [
-				{
-					type: "text",
-					text: `API failed: ${apiMsg}\nFalling back to curl...`,
-				},
-			],
-			details: {
-				_type: "fetchResult",
-				outcome: "running",
-				url: params.url,
-				format: fmt,
-			},
-		});
-	}
-
-	try {
-		let html = await dependencies.curl(["-L", params.url]);
-		const chars = html.length;
-		const truncated = maxChars > 0 && html.length > maxChars;
-		if (truncated) html = `${html.slice(0, maxChars)}\n\n[truncated]`;
-
-		const banner =
-			"[FALLBACK — raw curl] API unavailable, content may differ from exa formatted output.";
-		const body =
-			fmt === "html"
-				? `${banner}\n\n${html.slice(0, 19_500)}`
-				: `${banner}\n\nRaw HTML (format='${fmt}' not applied):\n\n${html.slice(0, 19_000)}`;
-
-		return {
-			content: [{ type: "text", text: body.slice(0, 20_000) }],
-			details: {
-				_type: "fetchResult",
-				outcome: "fallback",
-				url: params.url,
-				format: fmt,
-				source: "curl-fallback",
-				chars,
-				truncated,
-			},
-		};
-	} catch (curlErr: unknown) {
-		const curlMsg = curlErr instanceof Error ? curlErr.message : String(curlErr);
-		return {
-			content: [
-				{
-					type: "text",
-					text: `Fetch failed (both API and curl).\nAPI: ${apiMsg}\nCurl: ${curlMsg}`,
-				},
-			],
-			details: {
-				_type: "fetchResult",
-				outcome: "error",
-				url: params.url,
-				format: fmt,
-				source: "failed",
-			},
-			isError: true,
-		};
-	}
-}
-
-export default function registerFetch(pi: ExtensionAPI): void {
-	pi.registerTool({
-		name: "fetch",
-		label: "Fetch",
-		renderShell: "self",
-		description: "Fetch a web page as markdown, text, or raw HTML via exa. Returns page content.",
-		promptSnippet:
-			"fetch(url, format, max_characters?) — format: 'markdown', 'text', or 'html'. Read page content via exa.",
-		promptGuidelines: [
-			"fetch: format='markdown' for readable content, 'text' for plain extraction, 'html' for raw source. max_characters caps size (default 1000, 0 = unlimited; 5000-10000 for typical pages).",
-		],
-		renderCall: makeRenderCall<unknown>("fetch", (args) =>
-			String((args as Partial<FetchParams>).url ?? ""),
-		),
-		renderResult: (
-			(renderTerminal) => (result, options, theme, context) =>
-				renderTerminal(result, options, theme, {
-					...context,
-					isError: context.isError && options.expanded,
-				})
-		)(
-			makeRenderResult<FetchResultDetails>({
-				tool: "fetch",
-				target: (details) => details.url,
-				meta: (details) => {
-					if (details.outcome === "error") return "failed";
-					if (details.outcome === "cancelled") return "cancelled";
-					const chars = `${details.chars ?? 0} chars`;
-					return details.outcome === "fallback"
-						? `${chars} · curl fallback`
-						: `${chars} · ${details.format}`;
-				},
-				status: (details) =>
-					details.outcome === "error"
-						? "error"
-						: details.outcome === "fallback" || details.outcome === "cancelled"
-							? "warning"
-							: "success",
-			}),
-		),
-		parameters: Type.Object({
-			url: Type.String({ description: "URL to fetch" }),
-			format: Type.Optional(
-				StringEnum(["markdown", "text", "html"] as const, {
-					description:
-						'Optional choice. Enter exactly "markdown" for readable Markdown, "text" for plain text, or "html" for raw HTML. Default is "markdown".',
-				}),
-			),
-			max_characters: Type.Optional(
-				Type.Number({
-					description: "Max characters (default 1000, 0 = unlimited)",
-					default: 1000,
-				}),
-			),
-		}),
-
-		async execute(_toolCallId, params, signal, onUpdate) {
-			return executeFetch(params as FetchParams, signal, onUpdate);
-		},
-	});
-}

package/src/search.ts

@@ -1,282 +0,0 @@
-/**
- * search.ts — search tool via 9Router API, with curl fallback.
- *
- * Environment:
- *   ROUTER_API_BASE  — router API base URL (default: https://9router.example.com/v1)
- *   ROUTER_API_KEY   — bearer token for the router
- */
-
-import { StringEnum } from "@earendil-works/pi-ai";
-import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
-import { Type } from "typebox";
-import { routerBaseUrl } from "./data.js";
-import { apiPost, auth, curl, isCancelled } from "./http.js";
-import { makeRenderCall, makeRenderResult } from "./render.js";
-
-type RouterOutcome = "running" | "fallback" | "success" | "cancelled" | "error";
-type SearchType = "web" | "news";
-
-export interface SearchResultDetails {
-	_type: "searchResult";
-	outcome: RouterOutcome;
-	query: string;
-	searchType: SearchType;
-	source?: "api" | "curl-fallback" | "failed";
-	resultCount?: number;
-}
-
-export interface SearchParams {
-	query: string;
-	search_type?: SearchType;
-	max_results?: number;
-}
-
-interface SearchResult {
-	content: { type: "text"; text: string }[];
-	details: SearchResultDetails;
-	isError?: boolean;
-}
-
-interface SearchDependencies {
-	apiPost: typeof apiPost;
-	curl: typeof curl;
-}
-
-interface SearchResultItem {
-	title?: string | null;
-	url?: string | null;
-	snippet?: string | null;
-	published_at?: string | null;
-	metadata?: { author?: string | null } | null;
-	author?: string | null;
-}
-
-/** Parse and format a raw exa search response with its authoritative result count. */
-export function parseSearchResults(raw: string): { text: string; count: number } {
-	let parsed: unknown;
-	try {
-		parsed = JSON.parse(raw);
-	} catch {
-		return { text: raw.slice(0, 20_000), count: 0 };
-	}
-
-	const obj = parsed as Record<string, unknown>;
-	const results = obj?.results;
-	if (!Array.isArray(results)) {
-		return { text: JSON.stringify(parsed, null, 2).slice(0, 20_000), count: 0 };
-	}
-
-	if (results.length === 0) return { text: "No results.", count: 0 };
-
-	const lines: string[] = [];
-	results.forEach((r: SearchResultItem, i: number) => {
-		const title = (r.title ?? "").trim() || r.url || "(untitled)";
-		lines.push(`${i + 1}. ${title}`);
-		if (r.url) lines.push(`   ${r.url}`);
-		const meta: string[] = [];
-		const author = r.author ?? r.metadata?.author;
-		if (author) meta.push(author);
-		if (r.published_at) meta.push(r.published_at.slice(0, 10));
-		if (meta.length > 0) lines.push(`   (${meta.join(" — ")})`);
-		const snippet = (r.snippet ?? "").trim();
-		if (snippet) lines.push(`   ${snippet.replace(/\s+/g, " ").slice(0, 300)}`);
-		lines.push("");
-	});
-
-	return {
-		text: lines.join("\n").trim().slice(0, 20_000),
-		count: results.length,
-	};
-}
-
-/** Format a raw exa search response as a compact markdown list. */
-export function formatSearchResults(raw: string): string {
-	return parseSearchResults(raw).text;
-}
-
-export async function executeSearch(
-	params: SearchParams,
-	signal?: AbortSignal,
-	onUpdate?: (result: SearchResult) => void,
-	dependencies: SearchDependencies = { apiPost, curl },
-): Promise<SearchResult> {
-	const max = Math.min(params.max_results ?? 5, 10);
-	const searchType = params.search_type ?? "web";
-	let apiMsg = "";
-
-	try {
-		onUpdate?.({
-			content: [{ type: "text", text: `Searching (${searchType}): ${params.query}...` }],
-			details: {
-				_type: "searchResult",
-				outcome: "running",
-				query: params.query,
-				searchType,
-			},
-		});
-
-		const raw = await dependencies.apiPost(
-			"/search",
-			{
-				model: "exa",
-				query: params.query,
-				search_type: searchType,
-				max_results: max,
-			},
-			signal,
-		);
-		const parsed = parseSearchResults(raw);
-
-		return {
-			content: [{ type: "text", text: parsed.text }],
-			details: {
-				_type: "searchResult",
-				outcome: "success",
-				query: params.query,
-				searchType,
-				source: "api",
-				resultCount: parsed.count,
-			},
-		};
-	} catch (apiErr: unknown) {
-		apiMsg = apiErr instanceof Error ? apiErr.message : String(apiErr);
-		if (isCancelled(apiErr, signal)) {
-			return {
-				content: [{ type: "text", text: `Search cancelled.\nAPI: ${apiMsg}` }],
-				details: {
-					_type: "searchResult",
-					outcome: "cancelled",
-					query: params.query,
-					searchType,
-				},
-			};
-		}
-		onUpdate?.({
-			content: [
-				{
-					type: "text",
-					text: `API failed: ${apiMsg}\nFalling back to curl...`,
-				},
-			],
-			details: {
-				_type: "searchResult",
-				outcome: "running",
-				query: params.query,
-				searchType,
-			},
-		});
-	}
-
-	try {
-		const body = JSON.stringify({
-			model: "exa",
-			query: params.query,
-			search_type: searchType,
-			max_results: max,
-		});
-		const raw = await dependencies.curl([
-			"-X",
-			"POST",
-			"-H",
-			"Content-Type: application/json",
-			...(auth() ? ["-H", `Authorization: Bearer ${auth()}`] : []),
-			"-d",
-			body,
-			`${routerBaseUrl()}/search`,
-		]);
-		const parsed = parseSearchResults(raw);
-
-		return {
-			content: [
-				{
-					type: "text",
-					text: `[FALLBACK — curl] API called via curl instead of fetch.\n\n${parsed.text.slice(0, 19_500)}`,
-				},
-			],
-			details: {
-				_type: "searchResult",
-				outcome: "fallback",
-				query: params.query,
-				searchType,
-				source: "curl-fallback",
-				resultCount: parsed.count,
-			},
-		};
-	} catch (curlErr: unknown) {
-		const curlMsg = curlErr instanceof Error ? curlErr.message : String(curlErr);
-		return {
-			content: [
-				{
-					type: "text",
-					text: `Search failed (both API and curl).\nAPI: ${apiMsg}\nCurl: ${curlMsg}`,
-				},
-			],
-			details: {
-				_type: "searchResult",
-				outcome: "error",
-				query: params.query,
-				searchType,
-				source: "failed",
-			},
-			isError: true,
-		};
-	}
-}
-
-export default function registerSearch(pi: ExtensionAPI): void {
-	pi.registerTool({
-		name: "search",
-		label: "Search",
-		renderShell: "self",
-		description: "Web or news search via exa. Returns title, url, and snippet for each result.",
-		promptSnippet:
-			"search(query, search_type, max_results?) — search_type: 'web' or 'news'. Defaults to 5 results, max 10.",
-		promptGuidelines: [
-			"search: search_type='web' for general web results, 'news' for recent news articles.",
-		],
-		renderCall: makeRenderCall<unknown>("search", (args) =>
-			String((args as Partial<SearchParams>).query ?? ""),
-		),
-		renderResult: (
-			(renderTerminal) => (result, options, theme, context) =>
-				renderTerminal(result, options, theme, {
-					...context,
-					isError: context.isError && options.expanded,
-				})
-		)(
-			makeRenderResult<SearchResultDetails>({
-				tool: "search",
-				target: (details) => `“${details.query}”`,
-				meta: (details) => {
-					if (details.outcome === "error") return "failed";
-					if (details.outcome === "cancelled") return "cancelled";
-					const count = details.resultCount ?? 0;
-					return `${count} ${count === 1 ? "result" : "results"} · ${details.searchType}`;
-				},
-				status: (details) =>
-					details.outcome === "error"
-						? "error"
-						: details.outcome === "fallback" || details.outcome === "cancelled"
-							? "warning"
-							: "success",
-			}),
-		),
-		parameters: Type.Object({
-			query: Type.String({ description: "Search query" }),
-			search_type: StringEnum(["web", "news"] as const, {
-				description:
-					'Required choice. Enter exactly "web" for general web results or "news" for recent news articles.',
-			}),
-			max_results: Type.Optional(
-				Type.Number({
-					description: "Max results (default 5, max 10)",
-					default: 5,
-				}),
-			),
-		}),
-
-		async execute(_toolCallId, params, signal, onUpdate) {
-			return executeSearch(params as SearchParams, signal, onUpdate);
-		},
-	});
-}