Gemini

Parameters for GeminiEmbedder, which embeds documents with Gemini embedding models via the Gemini API or Vertex AI.

Gemini Embedder embeds documents using Google's Gemini embedding models through the Gemini API or Vertex AI.

Parameters

ParameterTypeDescriptionDefault
idstrThe model ID to use for embeddings"gemini-embedding-001"
task_typestrType of task for embedding generation"RETRIEVAL_QUERY"
titleOptional[str]Optional title for the content being embeddedNone
dimensionsOptional[int]Output dimensions of the embedding1536
api_keyOptional[str]Google API keyEnvironment variable GOOGLE_API_KEY
request_paramsOptional[Dict[str, Any]]Additional parameters for embedding requestsNone
client_paramsOptional[Dict[str, Any]]Additional parameters for client initializationNone
gemini_clientOptional[Client]Pre-configured google.genai clientNone
vertexaiboolUse the Vertex AI API. Also enabled by setting GOOGLE_GENAI_USE_VERTEXAI=trueFalse
project_idOptional[str]Google Cloud project ID for Vertex AI. Falls back to GOOGLE_CLOUD_PROJECTNone
locationOptional[str]Google Cloud region for Vertex AI. Falls back to GOOGLE_CLOUD_LOCATIONNone
enable_batchboolEmbed multiple texts per API call when adding content to vector databasesFalse
batch_sizeintNumber of texts per batch when enable_batch is True100