@mammothb/pi-websearch 0.1.0 → 0.3.0

package/src/websearch.ts

@@ -1,22 +1,88 @@
-import type { ToolDefinition } from "@earendil-works/pi-coding-agent";
-import Type from "typebox";
+import type { ImageContent, TextContent } from "@earendil-works/pi-ai";
+import {
+  keyText,
+  type Theme,
+  type ToolDefinition,
+} from "@earendil-works/pi-coding-agent";
+import { Container, Spacer, Text } from "@earendil-works/pi-tui";
+import type { WebsearchConfig } from "./config";
+import { createProvider } from "./lib/providers";
+import type { SearchArgs } from "./lib/types";
+import { WebsearchParameters } from "./lib/types";
+
+const COLLAPSED_PREVIEW_LINES = 7;
 
 interface WebsearchDetails {
   query: string;
-  numResults: number;
 }
 
-const Parameters = Type.Object({
-  query: Type.String({ description: "Search query" }),
-  numResults: Type.Optional(
-    Type.Number({ description: "Number of results (default: 8)" }),
-  ),
-});
-
-export function createWebsearchTool(): ToolDefinition<
-  typeof Parameters,
-  WebsearchDetails
-> {
+function isTextContent(c: TextContent | ImageContent): c is TextContent {
+  return c.type === "text";
+}
+
+function renderExpandableResult(
+  details: WebsearchDetails,
+  textContent: string,
+  expanded: boolean,
+  theme: Theme,
+): Container {
+  const container = new Container();
+
+  const title = details.query;
+  container.addChild(
+    new Text(
+      theme.fg("syntaxKeyword", "query: ") + theme.fg("syntaxString", title),
+    ),
+  );
+
+  if (!textContent) {
+    return container;
+  }
+
+  container.addChild(new Spacer(1));
+
+  if (expanded) {
+    container.addChild(new Text(textContent));
+  } else {
+    const lines = textContent
+      .split("\n")
+      .filter(
+        (line, index, arr) =>
+          line.length > 0 || index === 0 || index < arr.length - 1,
+      );
+    const previewLines = lines.slice(0, COLLAPSED_PREVIEW_LINES);
+    const remaining = Math.max(0, lines.length - previewLines.length);
+
+    const preview = previewLines.join("\n");
+    container.addChild(new Text(preview));
+
+    if (remaining > 0) {
+      container.addChild(new Spacer(1));
+      const expandKey = keyText("app.tools.expand") || "Ctrl+O";
+      container.addChild(
+        new Text(
+          theme.fg("muted", `... (${remaining} more lines, `) +
+            theme.fg("dim", expandKey) +
+            theme.fg("muted", " to expand)"),
+        ),
+      );
+    }
+  }
+  return container;
+}
+
+export function createWebsearchTool(
+  config: WebsearchConfig,
+): ToolDefinition<typeof WebsearchParameters, WebsearchDetails> {
+  const year = new Date().getFullYear();
+  const { defaults } = config;
+  const provider = createProvider(config);
+
+  const usageNotes =
+    config.provider === "searxng"
+      ? "\n  - Results are fetched from a self-hosted SearXNG metasearch instance"
+      : "\n  - Supports live crawling modes when available: 'fallback' (backup if cached unavailable) or 'preferred' (prioritize live crawling)\n  - Search types when available: 'auto' (balanced), 'fast' (quick results), 'deep' (comprehensive search)";
+
   return {
     name: "websearch",
     label: "Web Search",
@@ -26,25 +92,65 @@ export function createWebsearchTool(): ToolDefinition<
 - Use this tool for accessing information beyond knowledge cutoff.
 - Searches are performed automatically within a single API call.
 
-Usage notes:
-  - Supports live crawling modes when available: 'fallback' (backup if cached unavailable) or 'preferred' (prioritize live crawling)
-  - Search types when available: 'auto' (balanced), 'fast' (quick results), 'deep' (comprehensive search)
+Usage notes:${usageNotes}
   - Configurable context length for optimal LLM integration`,
     promptSnippet: "Search the web",
-    parameters: Parameters,
-    async execute(_toolCallId, params, _signal, _onUpdate, _ctx) {
-      const query = params.query;
-      const numResults = params.numResults ?? 8;
-
-      return {
-        content: [
-          {
-            type: "text",
-            text: `Search results for: ${query} (${numResults} results requested)`,
-          },
-        ],
-        details: { query, numResults },
+    promptGuidelines: [
+      "Use websearch to find current information, documentation, or answers that require up-to-date web data. Always cite sources from search results.",
+      `The current year is ${year}. You MUST use this year when searching for recent information or current events.\n- Example: If the current year is ${year} and the user asks for "latest AI news", search for "AI news ${year}", NOT "AI news ${year - 1}"`,
+    ],
+    parameters: WebsearchParameters,
+    async execute(_toolCallId, params, signal, _onUpdate, _ctx) {
+      const args: SearchArgs = {
+        query: params.query,
+        type: params.type ?? defaults.type,
+        numResults: params.numResults ?? defaults.numResults,
+        livecrawl: params.livecrawl ?? defaults.livecrawl,
+        contextMaxCharacters:
+          params.contextMaxCharacters ?? defaults.contextMaxCharacters,
       };
+
+      try {
+        const result = await provider.search(args, signal);
+
+        return {
+          content: [
+            {
+              type: "text",
+              text:
+                result ??
+                "No search results found. Please try a different query.",
+            },
+          ],
+          details: { query: params.query },
+        };
+      } catch (err) {
+        const message = err instanceof Error ? err.message : String(err);
+        throw new Error(`Web search failed: ${message}`);
+      }
+    },
+    renderCall(args, theme, _context) {
+      return new Text(
+        theme.fg("toolTitle", theme.bold("websearch ")) +
+          theme.fg("muted", `"${args.query}"`),
+      );
+    },
+    renderResult(result, options, theme, _context) {
+      if (options.isPartial) {
+        return new Text(theme.fg("warning", "Searching..."));
+      }
+
+      const textContent = result.content
+        .filter(isTextContent)
+        .map((c) => c.text)
+        .join("\n");
+
+      return renderExpandableResult(
+        result.details,
+        textContent,
+        options.expanded,
+        theme,
+      );
     },
   };
 }