@ai-sdk/google

Vercel AI SDK - Google Generative AI Provider

The Google provider contains language model support for the Google Generative AI APIs. It creates language model objects that can be used with the generateText, streamText, generateObject, and streamObject AI functions.

Setup

The Google provider is available in the @ai-sdk/google module. You can install it with

npm i @ai-sdk/google

Provider Instance

You can import createGoogleGenerativeAI from @ai-sdk/google and create a provider instance with various settings:

import { createGoogleGenerativeAI } from '@ai-sdk/google';

const google = createGoogleGenerativeAI({
  // optional base URL for proxies etc.:
  baseURL: '',

  // optional API key, default to env property GOOGLE_GENERATIVE_AI_API_KEY:
  apiKey: '',

  // optional custom headers:
  headers: {
    'custom-header': 'value',
  },
});

The AI SDK also provides a shorthand google import with a Google provider instance that uses defaults:

import { google } from '@ai-sdk/google';

Models

You can create models that call the Google Generative AI API using the provider instance. The first argument is the model id, e.g. models/gemini-pro. The models support tool calls and some have multi-modal capabilities.

const model = google('models/gemini-pro');

Google Generative AI models support also some model specific settings that are not part of the standard call settings. You can pass them as an options argument:

const model = google('models/gemini-pro', {
  topK: 0.2,
});