Gemini Embedding

About

Google Gemini provides state-of-the-art embedding models that convert text into high-dimensional vectors.

Authentication

Toolbox uses your Application Default Credentials (ADC) to authorize with the Gemini API client.

Optionally, you can use an API key obtain an API Key from the Google AI Studio.

We recommend using an API key for testing and using application default credentials for production.

Behavior

Automatic Vectorization

When a tool parameter is configured with embeddedBy: <your-gemini-model-name>, the Toolbox intercepts the raw text input from the client and sends it to the Gemini API. The resulting numerical array is then formatted before being passed to your database source.

Dimension Matching

The dimension field must match the expected size of your database column (e.g., a vector(768) column in PostgreSQL). This setting is supported by newer models since 2024 only. You cannot set this value if using the earlier model (models/embedding-001). Check out available Gemini models for more information.

Example

embeddingModels:
  gemini-model:
    kind: gemini
    model: gemini-embedding-001
    apiKey: ${GOOGLE_API_KEY}
    dimension: 768

Reference