From 91494073700dea93133f82e6ef95f964f335608e Mon Sep 17 00:00:00 2001 From: minecraft1024a Date: Tue, 19 Aug 2025 20:19:51 +0800 Subject: [PATCH] =?UTF-8?q?fix(cache):=20=E5=85=BC=E5=AE=B9=E5=85=83?= =?UTF-8?q?=E7=BB=84=E5=BD=A2=E5=BC=8F=E7=9A=84=20embedding=5Fresult?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit get_embedding 方法的返回值可能是一个包含向量和模型名称的元组 --- src/common/cache_manager.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/common/cache_manager.py b/src/common/cache_manager.py index a8306a106..be11aaff3 100644 --- a/src/common/cache_manager.py +++ b/src/common/cache_manager.py @@ -134,7 +134,9 @@ class CacheManager: if semantic_query and self.embedding_model: embedding_result = await self.embedding_model.get_embedding(semantic_query) if embedding_result: - validated_embedding = self._validate_embedding(embedding_result) + # embedding_result是一个元组(embedding_vector, model_name)，取第一个元素 + embedding_vector = embedding_result[0] if isinstance(embedding_result, tuple) else embedding_result + validated_embedding = self._validate_embedding(embedding_vector) if validated_embedding is not None: query_embedding = np.array([validated_embedding], dtype='float32') @@ -263,7 +265,9 @@ class CacheManager: try: embedding_result = await self.embedding_model.get_embedding(semantic_query) if embedding_result: - validated_embedding = self._validate_embedding(embedding_result) + # embedding_result是一个元组(embedding_vector, model_name)，取第一个元素 + embedding_vector = embedding_result[0] if isinstance(embedding_result, tuple) else embedding_result + validated_embedding = self._validate_embedding(embedding_vector) if validated_embedding is not None: embedding = np.array([validated_embedding], dtype='float32') # 写入 L1 Vector