json-render/examples/chat/lib/tools/search.ts

import { tool, generateText } from "ai";
import { gateway } from "@ai-sdk/gateway";
import { z } from "zod";

/**
 * Web search tool using Perplexity Sonar via AI Gateway.
 *
 * Perplexity Sonar models have built-in internet access and return
 * synthesized answers with citations. This is wrapped as a regular tool
 * (with an `execute` function) so that ToolLoopAgent can loop: it calls
 * the model, gets results, and feeds them back for the next step.
 */
export const webSearch = tool({
  description:
    "Search the web for current information on any topic. Use this when the user asks about something not covered by the specialized tools (weather, crypto, GitHub, Hacker News). Returns a synthesized answer based on real-time web data.",
  inputSchema: z.object({
    query: z
      .string()
      .describe(
        "The search query — be specific and include relevant context for better results",
      ),
  }),
  execute: async ({ query }) => {
    try {
      const { text } = await generateText({
        model: gateway("perplexity/sonar"),
        prompt: query,
      });
      return { content: text };
    } catch (error) {
      return {
        error: `Search failed: ${error instanceof Error ? error.message : "Unknown error"}`,
      };
    }
  },
});