From a0c0e5933aa408bdcc93dca9b46d8e1877641a0c Mon Sep 17 00:00:00 2001 From: jperezde Date: Wed, 15 Oct 2025 17:16:59 +0200 Subject: [PATCH] feat: Add new embedding model "models/gemini-embedding-001" to GeminiInferenceAdapter Introduced a new embedding model with a dimension of 3072 and a context length of 2048 to the GeminiInferenceAdapter class. --- llama_stack/providers/remote/inference/gemini/gemini.py | 1 + 1 file changed, 1 insertion(+) diff --git a/llama_stack/providers/remote/inference/gemini/gemini.py b/llama_stack/providers/remote/inference/gemini/gemini.py index cb2a7ffc5..56767a50e 100644 --- a/llama_stack/providers/remote/inference/gemini/gemini.py +++ b/llama_stack/providers/remote/inference/gemini/gemini.py @@ -15,6 +15,7 @@ class GeminiInferenceAdapter(OpenAIMixin): provider_data_api_key_field: str = "gemini_api_key" embedding_model_metadata: dict[str, dict[str, int]] = { "models/text-embedding-004": {"embedding_dimension": 768, "context_length": 2048}, + "models/gemini-embedding-001": {"embedding_dimension": 3072, "context_length": 2048}, } def get_base_url(self):