diff --git a/llama-index-core/llama_index/core/base/embeddings/base.py b/llama-index-core/llama_index/core/base/embeddings/base.py index 04414093b0b086f2ba30c105ccdd6104fc895469..5fb65b2fdea5b3dd62e7f4fa25be2f034331ac3b 100644 --- a/llama-index-core/llama_index/core/base/embeddings/base.py +++ b/llama-index-core/llama_index/core/base/embeddings/base.py @@ -29,7 +29,7 @@ class SimilarityMode(str, Enum): def mean_agg(embeddings: List[Embedding]) -> Embedding: """Mean aggregation for embeddings.""" - return list(np.array(embeddings).mean(axis=0)) + return np.array(embeddings).mean(axis=0).tolist() def similarity(