AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 11:59:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
kazuya-awano 2024-11-29 10:18:41 +09:00 committed by GitHub
parent 0f1133729f
commit 2d6865d421
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/core/rag/embedding/cached_embedding.py
View File

@ -102,7 +102,8 @@ class CacheEmbedding(Embeddings):
embedding = redis_client.get(embedding_cache_key) embedding = redis_client.get(embedding_cache_key)
if embedding: if embedding:
redis_client.expire(embedding_cache_key, 600) 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: try:
embedding_result = self._model_instance.invoke_text_embedding( embedding_result = self._model_instance.invoke_text_embedding(
texts=[text], user=self._user, input_type=EmbeddingInputType.QUERY texts=[text], user=self._user, input_type=EmbeddingInputType.QUERY