Skip to main content

Gemini

info

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

Chat model

We recommend configuring Gemini 1.5 Pro as your chat model.

config.yaml
models:
- name: Gemini 1.5 Pro
provider: gemini
model: gemini-1.5-pro-latest
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.