From b155f92355dddfe6a08302370f99e9cfad10c940 Mon Sep 17 00:00:00 2001 From: jperezde Date: Wed, 15 Oct 2025 17:16:14 +0200 Subject: [PATCH] fix: Update embedding model key format in GeminiInferenceAdapter Changed the embedding model key from "text-embedding-004" to "models/text-embedding-004" in the GeminiInferenceAdapter class to align with the new model structure. --- llama_stack/providers/remote/inference/gemini/gemini.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/providers/remote/inference/gemini/gemini.py b/llama_stack/providers/remote/inference/gemini/gemini.py index bb66b94d5..cb2a7ffc5 100644 --- a/llama_stack/providers/remote/inference/gemini/gemini.py +++ b/llama_stack/providers/remote/inference/gemini/gemini.py @@ -14,7 +14,7 @@ class GeminiInferenceAdapter(OpenAIMixin): provider_data_api_key_field: str = "gemini_api_key" embedding_model_metadata: dict[str, dict[str, int]] = { - "text-embedding-004": {"embedding_dimension": 768, "context_length": 2048}, + "models/text-embedding-004": {"embedding_dimension": 768, "context_length": 2048}, } def get_base_url(self):