Vercel AI SDK

import { z } from "zod";
import { tool } from "ai";
import { Valyu, SearchType as ValyuSearchSDKType } from "valyu-js"; // Ensure ValyuSearchSDKType is the correct export for search types

export const valyuDeepSearchTool = tool({
  description:
    "Search Valyu for real-time academic papers, web content, market data, etc. Use for specific, up-to-date information across various domains.",
  parameters: z.object({
    query: z
      .string()
      .describe(
        'Detailed search query (e.g., "latest advancements in AI for healthcare" or "current price of Bitcoin").'
      ),
    searchType: z
      .enum(["all", "web", "market", "academic", "proprietary"])
      .describe(
        'Search domain: "academic", "web", "market", "all", or "proprietary" for specific Valyu datasets.'
      ),
  }),
  execute: async ({ query, searchType }) => {
    const VALYU_API_KEY = process.env.VALYU_API_KEY;
    if (!VALYU_API_KEY) {
      console.error("VALYU_API_KEY is not set.");
      return JSON.stringify({
        success: false,
        error: "Valyu API key not configured.",
        results: [],
      });
    }

    const valyu = new Valyu(VALYU_API_KEY);

    const searchTypeMap: { [key: string]: ValyuSearchSDKType } = {
      all: "all",
      web: "web",
      market: "all",
      academic: "proprietary",
      proprietary: "proprietary",
    };
    const mappedSearchType: ValyuSearchSDKType =
      searchTypeMap[searchType] || "all";

    try {
      console.log(
        `[ValyuDeepSearchTool] Query: "${query}", LLM Type: ${searchType}, Valyu Type: ${mappedSearchType}`
      );
      const response = await valyu.search(
        query,
        {
          searchType: mappedSearchType,
          maxNumResults: 5,
          maxPrice: 50.0,
          relevanceThreshold: 0.5,
          ...(searchType === "academic"
            ? { includedSources: ["valyu/valyu-arxiv"] }
            : {}),
        }
      );

      if (!response.success) {
        console.error("[ValyuDeepSearchTool] API Error:", response.error);
        return JSON.stringify({
          success: false,
          error: response.error || "Valyu API request failed.",
          query,
          results: [],
        });
      }

      console.log(
        `[ValyuDeepSearchTool] Success. Results: ${response.results?.length}, TX_ID: ${response.tx_id}`
      );
      return JSON.stringify({
        success: true,
        query,
        results: response.results || [],
        tx_id: response.tx_id,
      });
    } catch (error) {
      const errorMessage =
        error instanceof Error ? error.message : "Unknown error.";
      console.error("[ValyuDeepSearchTool] Exception:", errorMessage);
      return JSON.stringify({
        success: false,
        error: errorMessage,
        query,
        results: [],
      });
    }
  },
});

import { z as zodWeb } from "zod"; // Use a different alias for Zod if imported again in the same scope
import { tool as toolWeb } from "ai";
import {
  Valyu as ValyuWeb,
  SearchType as ValyuWebSearchSDKType,
} from "valyu-js";

export const valyuWebSearchTool = toolWeb({
  description:
    "Perform a web search using Valyu for up-to-date information from the internet.",
  parameters: zodWeb.object({
    query: zodWeb.string().describe("The search query for the web."),
  }),
  execute: async ({ query }) => {
    const VALYU_API_KEY = process.env.VALYU_API_KEY;
    if (!VALYU_API_KEY) {
      console.error("VALYU_API_KEY is not set for web search.");
      return JSON.stringify({
        success: false,
        error: "Valyu API key not configured.",
        results: [],
      });
    }
    const valyu = new ValyuWeb(VALYU_API_KEY);
    try {
      console.log(`[ValyuWebSearchTool] Web Query: "${query}"`);
      const response = await valyu.search(
        query,
        {
          searchType: "web" as ValyuWebSearchSDKType, // Fixed to web search
          maxNumResults: 5,
          maxPrice: 30.0, // Adjust maxPrice for web search if needed
          relevanceThreshold: 0.5,
        }
      );
      if (!response.success) {
        console.error("[ValyuWebSearchTool] API Error:", response.error);
        return JSON.stringify({
          success: false,
          error: response.error || "Valyu Web API request failed.",
          query,
          results: [],
        });
      }
      console.log(
        `[ValyuWebSearchTool] Success. Results: ${response.results?.length}, TX_ID: ${response.tx_id}`
      );
      return JSON.stringify({
        success: true,
        query,
        results: response.results || [],
        tx_id: response.tx_id,
      });
    } catch (error) {
      const errorMessage =
        error instanceof Error ? error.message : "Unknown error.";
      console.error("[ValyuWebSearchTool] Exception:", errorMessage);
      return JSON.stringify({
        success: false,
        error: errorMessage,
        query,
        results: [],
      });
    }
  },
});

import { streamText, CoreMessage } from "ai";
import { openai } from "@ai-sdk/openai";
// Import your tools (ensure paths are correct)
// import { valyuDeepSearchTool } from './path-to-your-tools-file';
// import { valyuWebSearchTool } from './path-to-your-tools-file';

async function POST(req: Request) {
  const aiMessages: CoreMessage[] = []; // Placeholder for your actual messages

  const result = await streamText({
    model: openai("gpt-4o"),
    messages: aiMessages,
    system: `You are a helpful AI. Use valyuDeepSearchTool for varied searches or valyuWebSearchTool for web info. Cite sources: [Title](URL).`,
    tools: {
      valyuDeepSearch: valyuDeepSearchTool, // For comprehensive search
      valyuWebSearch: valyuWebSearchTool, // For dedicated web search
      // Add other tools as needed
    },
    toolChoice: "auto",
  });

  return result.toAIStreamResponse();
}

System Prompt:

You are an AI assistant.
- For general web searches, use "valyuWebSearchTool" with a "query".
- For specific research (academic, market data), use "valyuDeepSearchTool" with a "query" and relevant "searchType" ('academic', 'market', 'all', 'proprietary').
- **Always cite information from search results using Markdown links: [Source Title](URL_from_result).**