diff --git a/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py b/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py index 12de57617..93e3dd26a 100644 --- a/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py +++ b/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py @@ -117,6 +117,7 @@ class LiteLLMEmbeddingEngine(EmbeddingEngine): api_key=self.api_key, api_base=self.endpoint, api_version=self.api_version, + dimensions=self.dimensions, ) return [data["embedding"] for data in response.data]