phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-12 20:12:33 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: Update embedding model key format in GeminiInferenceAdapter

Browse source 
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.
This commit is contained in:
jperezde 2025-10-15 17:16:14 +02:00
parent ef4bc70bbe
commit b155f92355
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/remote/inference/gemini/gemini.py
View file

@ -14,7 +14,7 @@ class GeminiInferenceAdapter(OpenAIMixin):
provider_data_api_key_field: str = "gemini_api_key" provider_data_api_key_field: str = "gemini_api_key"
embedding_model_metadata: dict[str, dict[str, int]] = { 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): def get_base_url(self):