You can get an API key from the Google AI Studio.

​
Chat model

We recommend configuring Gemini 2.0 Flash as your chat model.
config.yaml
models:
  - name: Gemini 2.0 Flash
    provider: gemini
    model: gemini-2.0-flash
    apiKey: <YOUR_GEMINI_API_KEY>

​
Autocomplete model

Gemini currently does not offer any autocomplete models. Click here to see a list of autocomplete model providers.

​
Embeddings model

We recommend configuring text-embedding-004 as your embeddings model.
config.yaml
models:
  - name: Gemini Embeddings
    provider: gemini
    model: models/text-embedding-004 
    apiKey: <YOUR_GEMINI_API_KEY>
    roles:
      - embed

​
Reranking model

Gemini currently does not offer any reranking models. Click here to see a list of reranking model providers.