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.