New PR to add Google Search grounding to

const llmResponse = await generate({
  model: gemini15Pro,
  config: {
    version: input.modelVersion,
    vertexRetrieval: {
      datastore: {
        dataStoreId: 'abc',
        projectId: 'extensions-testing',
        location: 'global',
      },
    },
    googleSearchRetrieval: {},
  },
});

const llmResponse = await generate({
  model: gemini15Pro,
  config: {
    version: input.modelVersion,
    vertexRetrieval: {
      datastore: {
        dataStoreId: 'abc',
        projectId: 'extensions-testing',
        location: 'global',
      },
    },
    googleSearchRetrieval: {},
  },
});