From a009484942387540e73aa5a0658f092de074be99 Mon Sep 17 00:00:00 2001
From: Botong Zhu <945263008@qq.com>
Date: Thu, 29 Feb 2024 01:18:10 +0800
Subject: [PATCH] Fix: mean_agg returning none-serializable numpy float64
(#11458)
---
llama-index-core/llama_index/core/base/embeddings/base.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
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 04414093b0..5fb65b2fde 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(
--
GitLab