From 0857a7f57efc0cd1d440382b39843cecbfbc90d5 Mon Sep 17 00:00:00 2001 From: Arnav Agrawal Date: Tue, 8 Apr 2025 20:06:18 -0700 Subject: [PATCH] slice azure embeddings as well --- core/embedding/litellm_embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/embedding/litellm_embedding.py b/core/embedding/litellm_embedding.py index 4453161..3ac221d 100644 --- a/core/embedding/litellm_embedding.py +++ b/core/embedding/litellm_embedding.py @@ -54,7 +54,7 @@ class LiteLLMEmbeddingModel(BaseEmbeddingModel): try: model_params = {"model": self.model_config["model_name"]} - if self.model_config["model_name"] == "text-embedding-3-large": + if self.model_config["model_name"] in ["text-embedding-3-large", "azure/text-embedding-3-large"]: model_params["dimensions"] = PGVECTOR_MAX_DIMENSIONS # Add all model-specific parameters from the config