@springbrand/agent-runtime 0.1.0

package/src/pi/tool/web-search/index.ts

@@ -0,0 +1,2 @@
+export * from "./api";
+export * from "./web-search";

package/src/pi/tool/web-search/web-search.ts

@@ -0,0 +1,127 @@
+/**
+ * Adapted from ttttmr/pi-web-search at commit
+ * 118b3eee3f5900cef3141745109a300315630592.
+ * Copyright (c) ttttmr. Licensed under the MIT license.
+ * https://github.com/ttttmr/pi-web-search
+ */
+
+import type { AgentTool } from "@earendil-works/pi-agent-core";
+import { Type } from "@earendil-works/pi-ai";
+import type { PiToolCandidate } from "../compiler";
+import type { WebSearch, WebSearchResult } from "./api";
+
+export const webSearchParameters = Type.Object({
+  query: Type.String({
+    description: "The search query or question to answer",
+  }),
+  urls: Type.Optional(Type.Array(Type.String(), {
+    description: "Additional URLs to analyze along with search (up to 20)",
+    maxItems: 20,
+  })),
+});
+
+export interface WebSearchDetails extends Omit<WebSearchResult, "text"> {
+  resultCount: number;
+}
+
+function progress(
+  onUpdate: Parameters<
+    AgentTool<typeof webSearchParameters, WebSearchDetails>["execute"]
+  >[3],
+  text: string,
+): void {
+  onUpdate?.({
+    content: [{ type: "text", text }],
+    details: {} as WebSearchDetails,
+  });
+}
+
+export function webSearchPiToolCandidate(
+  search: WebSearch,
+): PiToolCandidate {
+  const tool: AgentTool<typeof webSearchParameters, WebSearchDetails> = {
+    name: "web_search",
+    label: "Web search",
+    description:
+      "Primary tool for web discovery, current facts, cited research, and public URL analysis using the current model provider. " +
+      "Make the first query answer-oriented: preserve the user's distinguishing terms and the evidence being sought. " +
+      "When a relevant URL is already known, include it in urls instead of fetching it separately unless raw HTTP access is required. " +
+      "Optionally include up to 20 URLs.",
+    parameters: webSearchParameters,
+    async execute(_toolCallId, input, signal, onUpdate) {
+      const hasUrls = Boolean(input.urls?.length);
+      progress(
+        onUpdate,
+        hasUrls
+          ? `Searching and analyzing ${input.urls!.length} URL(s)...`
+          : `Searching for "${input.query}"...`,
+      );
+      const result = await search(input, {
+        signal,
+        onProgress: (text) => progress(onUpdate, text),
+      });
+      const additionalResults = (result.searchResults ?? []).filter(
+        (item) =>
+          item.url &&
+          !result.sources.some((source) => source.url === item.url),
+      );
+      let summary = result.text;
+
+      if (result.failed?.length) {
+        summary +=
+          `\n\n## URL Status\n✅ Retrieved: ${result.retrieved?.length ?? 0}` +
+          `\n❌ Failed: ${result.failed.length}`;
+        for (const failure of result.failed) {
+          summary += `\n- ${failure.url ?? "Unknown URL"}: ${failure.status ?? "failed"}`;
+        }
+      }
+      if (result.sources.length) {
+        summary += `\n\n## Sources\n${result.sources.map(
+          (source, index) =>
+            `${index + 1}. [${source.title}](${source.url})`,
+        ).join("\n")}`;
+      }
+      if (additionalResults.length) {
+        const visible = additionalResults.slice(0, 8);
+        summary += `\n\n## Additional Search Results\n${visible.map(
+          (item, index) => {
+            const label = item.title || item.url || `Result ${index + 1}`;
+            const url = item.url ? ` - ${item.url}` : "";
+            const metadata = [
+              item.source,
+              item.type,
+              item.status,
+              item.query ? `query=${item.query}` : undefined,
+            ].filter(Boolean).join(", ");
+            return `${index + 1}. ${label}${url}${
+              metadata ? ` (${metadata})` : ""
+            }`;
+          },
+        ).join("\n")}`;
+        if (additionalResults.length > visible.length) {
+          summary += `\n... and ${
+            additionalResults.length - visible.length
+          } more results in tool details.`;
+        }
+      }
+
+      const { text: _text, ...standardized } = result;
+      const details: WebSearchDetails = {
+        ...standardized,
+        resultCount:
+          result.searchResults?.length || result.sources.length,
+      };
+      return {
+        content: [{ type: "text", text: summary }],
+        details,
+      };
+    },
+  };
+  return {
+    owner: "runtime-base",
+    authorized: true,
+    requiredExecutionLevel: "safe",
+    source: "action",
+    tool,
+  };
+}