@mammothb/pi-websearch 0.1.0

package/LICENSE

@@ -0,0 +1,18 @@
+MIT License
+
+Copyright (c) 2026 mammothb
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and
+associated documentation files (the "Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the
+following conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or substantial
+portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT
+LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO
+EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
+USE OR OTHER DEALINGS IN THE SOFTWARE.

package/README.md

@@ -0,0 +1,14 @@
+# pi-websearch
+
+A [pi](https://pi.dev) extension that adds a `websearch` tool for searching the web.
+
+## Usage
+
+Once installed, the LLM can call the `websearch` tool to search the web for current information.
+
+## Development
+
+```bash
+# Test locally
+pi -e ./index.ts
+```

package/index.ts

@@ -0,0 +1,6 @@
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import { createWebsearchTool } from "./src/websearch.js";
+
+export default function (pi: ExtensionAPI) {
+  pi.registerTool(createWebsearchTool());
+}

package/package.json

@@ -0,0 +1,33 @@
+{
+  "name": "@mammothb/pi-websearch",
+  "version": "0.1.0",
+  "description": "A pi extension that adds a websearch tool for searching the web",
+  "keywords": [
+    "pi-package"
+  ],
+  "license": "MIT",
+  "files": [
+    "index.ts",
+    "src"
+  ],
+  "pi": {
+    "extensions": [
+      "./index.ts"
+    ]
+  },
+  "scripts": {
+    "test": "vitest run",
+    "test:coverage": "vitest run --coverage"
+  },
+  "devDependencies": {
+    "@vitest/coverage-v8": "4.1.7",
+    "typescript": "6.0.3",
+    "vitest": "4.1.7"
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-ai": "*",
+    "@earendil-works/pi-coding-agent": "*",
+    "@earendil-works/pi-tui": "*",
+    "typebox": "*"
+  }
+}

package/src/websearch.ts

@@ -0,0 +1,50 @@
+import type { ToolDefinition } from "@earendil-works/pi-coding-agent";
+import Type from "typebox";
+
+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
+> {
+  return {
+    name: "websearch",
+    label: "Web Search",
+    description: `- Search the web using the session's web search provider - performs real-time web searches and can scrape content from specific URLs.
+- Provides up-to-date information for current events and recent data.
+- Supports configurable result counts and returns the content from the most relevant websites.
+- 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)
+  - 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 },
+      };
+    },
+  };
+}