@ai-sdk/google/src/tool/google-search.ts

The **[Google Generative AI provider](https://ai-sdk.dev/providers/ai-sdk-providers/google-generative-ai)** for the [AI SDK](https://ai-sdk.dev/docs) contains language model support for the [Google Generative AI](https://ai.google/discover/generativeai/)

import {
  createProviderToolFactory,
  lazySchema,
  zodSchema,
} from '@ai-sdk/provider-utils';
import { z } from 'zod/v4';

// https://ai.google.dev/gemini-api/docs/google-search
// https://ai.google.dev/api/generate-content#GroundingSupport
// https://cloud.google.com/vertex-ai/generative-ai/docs/grounding/grounding-with-google-search

export const googleSearchToolArgsBaseSchema = z
  .object({
    searchTypes: z
      .object({
        webSearch: z.object({}).optional(),
        imageSearch: z.object({}).optional(),
      })
      .optional(),

    timeRangeFilter: z
      .object({
        startTime: z.string(),
        endTime: z.string(),
      })
      .optional(),
  })
  .passthrough();

export type GoogleSearchToolArgs = z.infer<
  typeof googleSearchToolArgsBaseSchema
>;

const googleSearchToolArgsSchema = lazySchema(() =>
  zodSchema(googleSearchToolArgsBaseSchema),
);

export const googleSearch = createProviderToolFactory<{}, GoogleSearchToolArgs>(
  {
    id: 'google.google_search',
    inputSchema: googleSearchToolArgsSchema,
  },
);