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):