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

/** Tool to retrieve knowledge from the File Search Stores. */
const fileSearchArgsBaseSchema = lazySchema(() =>
  zodSchema(
    z.looseObject({
      /** The names of the file_search_stores to retrieve from.
       *  Example: `fileSearchStores/my-file-search-store-123`
       */
      fileSearchStoreNames: z
        .array(z.string())
        .describe(
          'The names of the file_search_stores to retrieve from. Example: `fileSearchStores/my-file-search-store-123`',
        ),
      /** The number of file search retrieval chunks to retrieve. */
      topK: z
        .number()
        .int()
        .positive()
        .describe('The number of file search retrieval chunks to retrieve.')
        .optional(),

      /** Metadata filter to apply to the file search retrieval documents.
       *  See https://google.aip.dev/160 for the syntax of the filter expression.
       */
      metadataFilter: z
        .string()
        .describe(
          'Metadata filter to apply to the file search retrieval documents. See https://google.aip.dev/160 for the syntax of the filter expression.',
        )
        .optional(),
    }),
  ),
);

export type GoogleFileSearchToolArgs = InferSchema<
  typeof fileSearchArgsBaseSchema
>;

export const fileSearch = createProviderExecutedToolFactory<
  {},
  {},
  GoogleFileSearchToolArgs
>({
  id: 'google.file_search',
  inputSchema: lazySchema(() => zodSchema(z.object({}))),
  outputSchema: lazySchema(() => zodSchema(z.object({}))),
});