Google AI Embedding Models

To use a hosted Google AI embedding model, specify the google path in the from field of your configuration.

Include the model ID in the from field; a model ID is required. For example, google:gemini-embedding-2 selects Google's latest generally available embedding model.

The following parameters are specific to Google AI embedding models:

Parameter	Description	Default
google_api_key	The API key for accessing Google AI.	-
dimensions	The output dimensionality of the embeddings. Some embedding models support dynamic output sizes.	-

Below is an example configuration in spicepod.yaml:

embeddings:
  - from: google:gemini-embedding-2
    name: gemini_embeddings
    params:
      google_api_key: ${ secrets:GEMINI_API_KEY }
      dimensions: 768 # optional parameter

  - from: google:gemini-embedding-001
    name: legacy_embeddings
    params:
      google_api_key: ${ secrets:GEMINI_API_KEY }

See Google AI Embedding Models for a list of supported embedding models.

For detailed instructions and examples on running vector searches, refer to the Vector-Based Search documentation.