From 2ba36f87e37c35104c48f9cebf12f17398d0ee1a Mon Sep 17 00:00:00 2001 From: yangdx Date: Sat, 1 Feb 2025 09:55:05 +0800 Subject: [PATCH] Add support for list input in quantize_embedding function - Convert list to numpy array if needed - Maintain existing functionality --- lightrag/utils.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/lightrag/utils.py b/lightrag/utils.py index b9308b66..29665c3e 100644 --- a/lightrag/utils.py +++ b/lightrag/utils.py @@ -463,8 +463,12 @@ def cosine_similarity(v1, v2): return dot_product / (norm1 * norm2) -def quantize_embedding(embedding: np.ndarray, bits=8) -> tuple: +def quantize_embedding(embedding: Union[np.ndarray, list], bits=8) -> tuple: """Quantize embedding to specified bits""" + # Convert list to numpy array if needed + if isinstance(embedding, list): + embedding = np.array(embedding) + # Calculate min/max values for reconstruction min_val = embedding.min() max_val = embedding.max()