@jarcelao/pi-exa-api 0.3.0 → 0.3.1

package/README.md

@@ -11,7 +11,7 @@ pi install npm:@jarcelao/pi-exa-api
 ```
 
 > [!NOTE]
-> This extension is tested up to `pi-coding-agent` v0.70.6
+> This extension is tested up to `pi-coding-agent` v0.74.0
 
 ## Configuration
 

package/extensions/formatters.ts

@@ -2,8 +2,8 @@
  * Result formatting utilities
  */
 
-import { keyHint } from "@mariozechner/pi-coding-agent";
-import type { Theme } from "@mariozechner/pi-coding-agent";
+import { keyHint } from "@earendil-works/pi-coding-agent";
+import type { Theme } from "@earendil-works/pi-coding-agent";
 
 import type {
   ExaSearchResult,

package/extensions/index.ts

@@ -9,34 +9,14 @@
  * Also registers the /exa-status command to check API key configuration.
  */
 
-import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
 
 import { getApiKey } from "./api-key.ts";
-import { createMissingApiKeyError } from "./errors.ts";
-import { mapSearchContentType, mapFetchContentType } from "./content-types.ts";
-import {
-  formatSearchResults,
-  formatFetchResult,
-  formatCodeContextResult,
-  parseCostDollars,
-} from "./formatters.ts";
 
 import { createExaSearchTool } from "./tools/search.ts";
 import { createExaFetchTool } from "./tools/fetch.ts";
 import { createExaCodeContextTool } from "./tools/code-context.ts";
 
-// Re-export all utilities for testing
-export {
-  getApiKey,
-  createMissingApiKeyError,
-  mapSearchContentType,
-  mapFetchContentType,
-  formatSearchResults,
-  formatFetchResult,
-  formatCodeContextResult,
-  parseCostDollars,
-};
-
 export default function exaSearchExtension(pi: ExtensionAPI): void {
   pi.on("session_start", async (_event: unknown, ctx: ExtensionContext) => {
     const hasKey = !!getApiKey();

package/extensions/tools/code-context.ts

@@ -2,28 +2,20 @@
  * Exa Code Context tool definition
  */
 
-import { mkdtemp, writeFile } from "node:fs/promises";
-import { tmpdir } from "node:os";
-import { join } from "node:path";
-import type { Theme } from "@mariozechner/pi-coding-agent";
-import {
-  DEFAULT_MAX_BYTES,
-  DEFAULT_MAX_LINES,
-  defineTool,
-  formatSize,
-  truncateHead,
-} from "@mariozechner/pi-coding-agent";
-import { Text } from "@mariozechner/pi-tui";
+import type { Theme } from "@earendil-works/pi-coding-agent";
+import { defineTool } from "@earendil-works/pi-coding-agent";
 import { Type, type Static } from "typebox";
 
-import { getApiKey } from "../api-key.ts";
-import { createMissingApiKeyError } from "../errors.ts";
-import {
-  formatCodeContextResult,
-  formatToolOutputPreview,
-  parseCostDollars,
-} from "../formatters.ts";
+import { formatCodeContextResult, parseCostDollars } from "../formatters.ts";
 import type { CodeContextDetails, CodeContextResponse } from "../types.ts";
+import {
+  requireApiKey,
+  truncateAndSave,
+  renderToolCall,
+  renderToolResult,
+  formatCost,
+  EXA_CONTEXT_API_URL,
+} from "./shared.ts";
 
 // Tool parameter schema
 export const ExaCodeContextParams = Type.Object({
@@ -62,15 +54,9 @@ export function createExaCodeContextTool() {
       _onUpdate: unknown,
       _ctx: unknown,
     ) {
-      const apiKey = getApiKey();
-      if (!apiKey) {
-        throw createMissingApiKeyError();
-      }
+      const apiKey = requireApiKey();
 
       // Ensure tokensNum is the correct type: number or "dynamic"
-      // The schema accepts both string and number, but the Exa API requires:
-      // - A number (e.g., 5000)
-      // - The literal string "dynamic"
       let tokensNum: string | number = params.tokensNum ?? "dynamic";
       if (typeof tokensNum === "string" && tokensNum !== "dynamic") {
         tokensNum = Number(tokensNum);
@@ -78,7 +64,7 @@ export function createExaCodeContextTool() {
 
       let response: CodeContextResponse;
       try {
-        const httpResponse = await fetch("https://api.exa.ai/context", {
+        const httpResponse = await fetch(EXA_CONTEXT_API_URL, {
           method: "POST",
           headers: {
             "Content-Type": "application/json",
@@ -101,24 +87,8 @@ export function createExaCodeContextTool() {
         throw new Error(`Exa Context API error: ${message}`);
       }
 
-      let output = formatCodeContextResult(response);
-
-      const truncation = truncateHead(output, {
-        maxLines: DEFAULT_MAX_LINES,
-        maxBytes: DEFAULT_MAX_BYTES,
-      });
-
-      let result = truncation.content;
-
-      if (truncation.truncated) {
-        const tempDir = await mkdtemp(join(tmpdir(), "pi-exa-"));
-        const tempFile = join(tempDir, "output.txt");
-        await writeFile(tempFile, output, "utf8");
-        result += `\n\n[Output truncated: ${truncation.outputLines} of ${truncation.totalLines} lines`;
-        result += ` (${formatSize(truncation.outputBytes)} of ${formatSize(truncation.totalBytes)}).`;
-        result += ` Full output saved to: ${tempFile}]`;
-      }
-
+      const output = formatCodeContextResult(response);
+      const result = await truncateAndSave(output);
       const cost = parseCostDollars(response.costDollars);
 
       return {
@@ -133,31 +103,20 @@ export function createExaCodeContextTool() {
     },
 
     renderCall(args: CodeContextParams, theme: Theme) {
-      const preview = args.query.length > 50 ? args.query.slice(0, 50) + "..." : args.query;
       const desc = `${args.tokensNum ?? "dynamic"} tokens`;
-      const text =
-        theme.fg("toolTitle", theme.bold("exa_code_context ")) +
-        theme.fg("muted", preview) +
-        theme.fg("dim", ` ${desc}`);
-      return new Text(text, 0, 0);
+      return renderToolCall("exa_code_context", args.query, desc, theme);
     },
 
     renderResult(result, options, theme, context) {
       const details = result.details as CodeContextDetails | undefined;
-      const text = (context.lastComponent as Text | undefined) ?? new Text("", 0, 0);
-
-      let header = "";
-      if (details) {
-        const cost = details.cost ? ` • $${details.cost.total.toFixed(6)}` : "";
-        header = theme.fg(
-          "success",
-          `✓ ${details.resultsCount} sources • ${details.outputTokens} tokens${cost}`,
-        );
-      }
-
-      const preview = formatToolOutputPreview(result, options, theme);
-      text.setText(preview ? `${header}\n${preview}` : header);
-      return text;
+      const cost = details ? formatCost(details.cost) : "";
+      const header = details
+        ? theme.fg(
+            "success",
+            `✓ ${details.resultsCount} sources • ${details.outputTokens} tokens${cost}`,
+          )
+        : "";
+      return renderToolResult(header, result, options, theme, context);
     },
   });
 }

package/extensions/tools/fetch.ts

@@ -2,26 +2,21 @@
  * Exa Fetch tool definition
  */
 
-import { mkdtemp, writeFile } from "node:fs/promises";
-import { tmpdir } from "node:os";
-import { join } from "node:path";
-import type { ExtensionContext, Theme } from "@mariozechner/pi-coding-agent";
-import {
-  DEFAULT_MAX_BYTES,
-  DEFAULT_MAX_LINES,
-  defineTool,
-  formatSize,
-  truncateHead,
-} from "@mariozechner/pi-coding-agent";
-import { Text } from "@mariozechner/pi-tui";
+import type { ExtensionContext, Theme } from "@earendil-works/pi-coding-agent";
+import { defineTool } from "@earendil-works/pi-coding-agent";
 import { Type, type Static } from "typebox";
 import Exa from "exa-js";
 
-import { getApiKey } from "../api-key.ts";
-import { createMissingApiKeyError } from "../errors.ts";
 import { mapFetchContentType } from "../content-types.ts";
-import { formatFetchResult, formatToolOutputPreview } from "../formatters.ts";
+import { formatFetchResult } from "../formatters.ts";
 import type { FetchContentType, FetchDetails, ExaSearchResult } from "../types.ts";
+import {
+  requireApiKey,
+  truncateAndSave,
+  renderToolCall,
+  renderToolResult,
+  formatCost,
+} from "./shared.ts";
 
 // Tool parameter schema
 export const ExaFetchParams = Type.Object({
@@ -58,11 +53,7 @@ export function createExaFetchTool() {
       _onUpdate: unknown,
       _ctx: ExtensionContext,
     ) {
-      const apiKey = getApiKey();
-      if (!apiKey) {
-        throw createMissingApiKeyError();
-      }
-
+      const apiKey = requireApiKey();
       const exa = new Exa(apiKey);
 
       const contentsOptions: {
@@ -96,23 +87,8 @@ export function createExaFetchTool() {
       }
 
       const result = response.results[0] as ExaSearchResult;
-      let output = formatFetchResult(result, (params.contentType ?? "text") as FetchContentType);
-
-      const truncation = truncateHead(output, {
-        maxLines: DEFAULT_MAX_LINES,
-        maxBytes: DEFAULT_MAX_BYTES,
-      });
-
-      let content = truncation.content;
-
-      if (truncation.truncated) {
-        const tempDir = await mkdtemp(join(tmpdir(), "pi-exa-"));
-        const tempFile = join(tempDir, "output.txt");
-        await writeFile(tempFile, output, "utf8");
-        content += `\n\n[Output truncated: ${truncation.outputLines} of ${truncation.totalLines} lines`;
-        content += ` (${formatSize(truncation.outputBytes)} of ${formatSize(truncation.totalBytes)}).`;
-        content += ` Full output saved to: ${tempFile}]`;
-      }
+      const output = formatFetchResult(result, (params.contentType ?? "text") as FetchContentType);
+      const content = await truncateAndSave(output);
 
       return {
         content: [{ type: "text", text: content }],
@@ -125,30 +101,19 @@ export function createExaFetchTool() {
     },
 
     renderCall(args: FetchParams, theme: Theme) {
-      const urlPreview = args.url.length > 40 ? args.url.slice(0, 40) + "..." : args.url;
       const desc = args.contentType ?? "text";
-      const text =
-        theme.fg("toolTitle", theme.bold("exa_fetch ")) +
-        theme.fg("muted", urlPreview) +
-        theme.fg("dim", ` ${desc}`);
-      return new Text(text, 0, 0);
+      return renderToolCall("exa_fetch", args.url, desc, theme);
     },
 
     renderResult(result, options, theme, context) {
       const details = result.details as FetchDetails | undefined;
-      const text = (context.lastComponent as Text | undefined) ?? new Text("", 0, 0);
-
-      let header = "";
-      if (details) {
-        const cost = details.cost ? ` • $${details.cost.total.toFixed(6)}` : "";
-        header = details.title
+      const cost = details ? formatCost(details.cost) : "";
+      const header = details
+        ? details.title
           ? theme.fg("success", `✓ ${details.title}${cost}`)
-          : theme.fg("success", `✓ Fetched${cost}`);
-      }
-
-      const preview = formatToolOutputPreview(result, options, theme);
-      text.setText(preview ? `${header}\n${preview}` : header);
-      return text;
+          : theme.fg("success", `✓ Fetched${cost}`)
+        : "";
+      return renderToolResult(header, result, options, theme, context);
     },
   });
 }

package/extensions/tools/search.ts

@@ -2,26 +2,21 @@
  * Exa Search tool definition
  */
 
-import { mkdtemp, writeFile } from "node:fs/promises";
-import { tmpdir } from "node:os";
-import { join } from "node:path";
-import type { ExtensionContext, Theme } from "@mariozechner/pi-coding-agent";
-import {
-  DEFAULT_MAX_BYTES,
-  DEFAULT_MAX_LINES,
-  defineTool,
-  formatSize,
-  truncateHead,
-} from "@mariozechner/pi-coding-agent";
-import { Text } from "@mariozechner/pi-tui";
+import type { ExtensionContext, Theme } from "@earendil-works/pi-coding-agent";
+import { defineTool } from "@earendil-works/pi-coding-agent";
 import { Type, type Static } from "typebox";
 import Exa from "exa-js";
 
-import { getApiKey } from "../api-key.ts";
-import { createMissingApiKeyError } from "../errors.ts";
 import { mapSearchContentType } from "../content-types.ts";
-import { formatSearchResults, formatToolOutputPreview } from "../formatters.ts";
+import { formatSearchResults } from "../formatters.ts";
 import type { SearchContentType, SearchDetails, ExaSearchResult } from "../types.ts";
+import {
+  requireApiKey,
+  truncateAndSave,
+  renderToolCall,
+  renderToolResult,
+  formatCost,
+} from "./shared.ts";
 
 // Tool parameter schema
 export const ExaSearchParams = Type.Object({
@@ -63,11 +58,7 @@ export function createExaSearchTool() {
       _onUpdate: unknown,
       _ctx: ExtensionContext,
     ) {
-      const apiKey = getApiKey();
-      if (!apiKey) {
-        throw createMissingApiKeyError();
-      }
-
+      const apiKey = requireApiKey();
       const numResults = Math.max(1, Math.min(100, params.numResults ?? 10));
       const exa = new Exa(apiKey);
 
@@ -89,26 +80,12 @@ export function createExaSearchTool() {
         throw new Error(`Exa API error: ${message}`);
       }
 
-      let output = formatSearchResults({
+      const output = formatSearchResults({
         results: response.results as ExaSearchResult[],
         costDollars: response.costDollars as { total: number } | undefined,
       });
 
-      const truncation = truncateHead(output, {
-        maxLines: DEFAULT_MAX_LINES,
-        maxBytes: DEFAULT_MAX_BYTES,
-      });
-
-      let result = truncation.content;
-
-      if (truncation.truncated) {
-        const tempDir = await mkdtemp(join(tmpdir(), "pi-exa-"));
-        const tempFile = join(tempDir, "output.txt");
-        await writeFile(tempFile, output, "utf8");
-        result += `\n\n[Output truncated: ${truncation.outputLines} of ${truncation.totalLines} lines`;
-        result += ` (${formatSize(truncation.outputBytes)} of ${formatSize(truncation.totalBytes)}).`;
-        result += ` Full output saved to: ${tempFile}]`;
-      }
+      const result = await truncateAndSave(output);
 
       return {
         content: [{ type: "text", text: result }],
@@ -121,28 +98,15 @@ export function createExaSearchTool() {
     },
 
     renderCall(args: SearchParams, theme: Theme) {
-      const preview = args.query.length > 50 ? args.query.slice(0, 50) + "..." : args.query;
       const desc = `${args.numResults ?? 10} results • ${args.contentType ?? "highlights"}`;
-      const text =
-        theme.fg("toolTitle", theme.bold("exa_search ")) +
-        theme.fg("muted", preview) +
-        theme.fg("dim", ` ${desc}`);
-      return new Text(text, 0, 0);
+      return renderToolCall("exa_search", args.query, desc, theme);
     },
 
     renderResult(result, options, theme, context) {
       const details = result.details as SearchDetails | undefined;
-      const text = (context.lastComponent as Text | undefined) ?? new Text("", 0, 0);
-
-      let header = "";
-      if (details) {
-        const cost = details.cost ? ` • $${details.cost.total.toFixed(6)}` : "";
-        header = theme.fg("success", `✓ ${details.numResults} results${cost}`);
-      }
-
-      const preview = formatToolOutputPreview(result, options, theme);
-      text.setText(preview ? `${header}\n${preview}` : header);
-      return text;
+      const cost = details ? formatCost(details.cost) : "";
+      const header = details ? theme.fg("success", `✓ ${details.numResults} results${cost}`) : "";
+      return renderToolResult(header, result, options, theme, context);
     },
   });
 }

package/extensions/tools/shared.ts

@@ -0,0 +1,106 @@
+/**
+ * Shared utilities for Exa tools
+ */
+
+import { mkdtemp, writeFile } from "node:fs/promises";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import {
+  DEFAULT_MAX_BYTES,
+  DEFAULT_MAX_LINES,
+  formatSize,
+  truncateHead,
+} from "@earendil-works/pi-coding-agent";
+import { Text } from "@earendil-works/pi-tui";
+import type { Theme } from "@earendil-works/pi-coding-agent";
+
+import { getApiKey } from "../api-key.ts";
+import { createMissingApiKeyError } from "../errors.ts";
+import { formatToolOutputPreview } from "../formatters.ts";
+
+/** Exa Context API base URL */
+export const EXA_CONTEXT_API_URL = "https://api.exa.ai/context";
+
+/**
+ * Require an API key to be configured, throwing if missing.
+ * @returns The API key
+ */
+export function requireApiKey(): string {
+  const apiKey = getApiKey();
+  if (!apiKey) {
+    throw createMissingApiKeyError();
+  }
+  return apiKey;
+}
+
+/**
+ * Truncate output and save to temp file if needed.
+ * @param output - The full output string
+ * @returns Object with the (possibly truncated) content string
+ */
+export async function truncateAndSave(output: string): Promise<string> {
+  const truncation = truncateHead(output, {
+    maxLines: DEFAULT_MAX_LINES,
+    maxBytes: DEFAULT_MAX_BYTES,
+  });
+
+  let content = truncation.content;
+
+  if (truncation.truncated) {
+    const tempDir = await mkdtemp(join(tmpdir(), "pi-exa-"));
+    const tempFile = join(tempDir, "output.txt");
+    await writeFile(tempFile, output, "utf8");
+    content += `\n\n[Output truncated: ${truncation.outputLines} of ${truncation.totalLines} lines`;
+    content += ` (${formatSize(truncation.outputBytes)} of ${formatSize(truncation.totalBytes)}).`;
+    content += ` Full output saved to: ${tempFile}]`;
+  }
+
+  return content;
+}
+
+/**
+ * Format the tool call preview for TUI display.
+ * @param label - The tool label (e.g., "exa_search")
+ * @param preview - The main preview text
+ * @param desc - The description text
+ * @param theme - The theme object
+ */
+export function renderToolCall(label: string, preview: string, desc: string, theme: Theme): Text {
+  const truncatedPreview = preview.length > 50 ? preview.slice(0, 50) + "..." : preview;
+  const text =
+    theme.fg("toolTitle", theme.bold(`${label} `)) +
+    theme.fg("muted", truncatedPreview) +
+    theme.fg("dim", ` ${desc}`);
+  return new Text(text, 0, 0);
+}
+
+/**
+ * Format the tool result for TUI display with optional header and preview.
+ * @param details - The details object with optional cost info
+ * @param headerText - The header text to display
+ * @param result - The full result object
+ * @param options - The display options
+ * @param theme - The theme object
+ * @param context - The context object
+ */
+export function renderToolResult(
+  headerText: string,
+  result: { content: Array<{ type: string; text?: string }> },
+  options: { expanded: boolean },
+  theme: Theme,
+  context: { lastComponent?: unknown },
+): Text {
+  const text = (context.lastComponent as Text | undefined) ?? new Text("", 0, 0);
+  const preview = formatToolOutputPreview(result, options, theme);
+  text.setText(preview ? `${headerText}\n${preview}` : headerText);
+  return text;
+}
+
+/**
+ * Format cost for display.
+ * @param cost - Cost object with total property
+ * @returns Formatted cost string or empty string
+ */
+export function formatCost(cost?: { total: number }): string {
+  return cost ? ` • $${cost.total.toFixed(6)}` : "";
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@jarcelao/pi-exa-api",
-  "version": "0.3.0",
+  "version": "0.3.1",
   "description": "Web search and content fetching for pi via the Exa API",
   "keywords": [
     "pi-package"
@@ -22,18 +22,18 @@
     "format:check": "oxfmt --check"
   },
   "dependencies": {
-    "exa-js": "^2.10.1"
+    "exa-js": "^2.12.1"
   },
   "devDependencies": {
     "@types/node": "^25.5.0",
-    "oxfmt": "^0.41.0",
+    "oxfmt": "^0.48.0",
     "oxlint": "^1.56.0",
-    "typescript": "^5.9.3",
+    "typescript": "^6.0.3",
     "vitest": "^4.1.0"
   },
   "peerDependencies": {
-    "@mariozechner/pi-coding-agent": "*",
-    "@mariozechner/pi-tui": "*",
+    "@earendil-works/pi-coding-agent": "*",
+    "@earendil-works/pi-tui": "*",
     "typebox": "*"
   },
   "pi": {

package/tests/api-key.test.ts

@@ -1,5 +1,6 @@
 import { describe, it, expect, afterEach } from "vitest";
-import { getApiKey, createMissingApiKeyError } from "../extensions/index.ts";
+import { getApiKey } from "../extensions/api-key.ts";
+import { createMissingApiKeyError } from "../extensions/errors.ts";
 
 describe("API Key Management", () => {
   afterEach(() => {

package/tests/content-type-mapping.test.ts

@@ -1,5 +1,5 @@
 import { describe, it, expect } from "vitest";
-import { mapSearchContentType, mapFetchContentType } from "../extensions/index.ts";
+import { mapSearchContentType, mapFetchContentType } from "../extensions/content-types.ts";
 
 describe("ContentType Mapping (Search)", () => {
   it('should map "text" to contents.text', () => {

package/tests/extension-registration.test.ts

@@ -23,7 +23,7 @@ function createMockExtensionAPI() {
 
 function setup() {
   const api = createMockExtensionAPI();
-  exaSearchExtension(api as unknown as import("@mariozechner/pi-coding-agent").ExtensionAPI);
+  exaSearchExtension(api as unknown as import("@earendil-works/pi-coding-agent").ExtensionAPI);
   return api;
 }
 

package/tests/formatting.test.ts

@@ -4,7 +4,7 @@ import {
   formatFetchResult,
   formatCodeContextResult,
   parseCostDollars,
-} from "../extensions/index.ts";
+} from "../extensions/formatters.ts";
 
 describe("formatSearchResults", () => {
   it("should format basic result fields", () => {

package/extensions/temp-file.ts

@@ -1,19 +0,0 @@
-/**
- * Temp file utilities
- */
-
-import { mkdtemp, writeFile } from "node:fs/promises";
-import { tmpdir } from "node:os";
-import { join } from "node:path";
-
-/**
- * Write content to a temporary file.
- * @param content - The content to write
- * @returns The path to the created temporary file
- */
-export async function writeTempFile(content: string): Promise<string> {
-  const tempDir = await mkdtemp(join(tmpdir(), "pi-exa-"));
-  const tempFile = join(tempDir, "output.txt");
-  await writeFile(tempFile, content, "utf8");
-  return tempFile;
-}