Преглед изворни кода

Ensure consistent float type for cached embedding return values (#10185)

kazuya-awano пре 4 месеци
родитељ
комит
2d6865d421
1 измењених фајлова са 2 додато и 1 уклоњено
  1. 2 1
      api/core/rag/embedding/cached_embedding.py

+ 2 - 1
api/core/rag/embedding/cached_embedding.py

@@ -102,7 +102,8 @@ class CacheEmbedding(Embeddings):
         embedding = redis_client.get(embedding_cache_key)
         if embedding:
             redis_client.expire(embedding_cache_key, 600)
-            return list(np.frombuffer(base64.b64decode(embedding), dtype="float"))
+            decoded_embedding = np.frombuffer(base64.b64decode(embedding), dtype="float")
+            return [float(x) for x in decoded_embedding]
         try:
             embedding_result = self._model_instance.invoke_text_embedding(
                 texts=[text], user=self._user, input_type=EmbeddingInputType.QUERY