The gemini function configures Google’s Gemini as your model provider.

import { createAgent, gemini } from "@inngest/agent-kit";

const agent = createAgent({
  name: "Code writer",
  system: "You are an expert TypeScript programmer.",
  model: gemini({ model: "gemini-pro" }),
});

Configuration

The gemini function accepts a model name string or a configuration object:

const agent = createAgent({
  model: gemini({
    model: "gemini-pro",
    apiKey: process.env.GOOGLE_API_KEY,
    baseUrl: "https://generativelanguage.googleapis.com/v1/",
    defaultParameters: {
      generationConfig: {
        temperature: 1.5,
      },
    },
  }),
});

Options

model
string
required

ID of the model to use. See the model endpoint compatibility table for details on which models work with the Gemini API.

apiKey
string

The Google API key to use for authenticating your request. By default we’ll search for and use the GOOGLE_API_KEY environment variable.

baseUrl
string
default:"https://generativelanguage.googleapis.com/v1/"

The base URL for the Gemini API.

defaultParameters
object

The default parameters to use for the model.

See Gemini’s models.generateContent reference.

Available Models

Gemini
"gemini-1.5-flash"
"gemini-1.5-flash-8b"
"gemini-1.5-pro"
"gemini-1.0-pro"
"text-embedding-004"
"aqa"

For the latest list of available models, see Google’s Gemini model overview.

Limitations

Gemini models do not currently support function without parameters.