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

/** Tool to retrieve knowledge from the File Search Stores. */
const fileSearchArgsBaseSchema = 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 = z.infer<typeof fileSearchArgsBaseSchema>;

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