title: 'Google AI Embedding Models' sidebar_label: 'Google AI' sidebar_position: 4

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:

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 }

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 (opens in a new tab) for a list of supported embedding models.

For detailed instructions and examples on running vector searches, refer to the Vector-Based Search documentation (opens in a new tab).