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

fix: float values are not json for nan value close #11827 (#11840)

Browse Source 
Signed-off-by: yihong0618 <zouzou0208@gmail.com>
This commit is contained in:
yihong 2024-12-19 20:50:20 +08:00 committed by GitHub
parent 12d45e9114
commit 5a8a901560
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -65,6 +65,11 @@ class CacheEmbedding(Embeddings):
for vector in embedding_result.embeddings: for vector in embedding_result.embeddings:
try: try:
normalized_embedding = (vector / np.linalg.norm(vector)).tolist() normalized_embedding = (vector / np.linalg.norm(vector)).tolist()
# stackoverflow best way: https://stackoverflow.com/questions/20319813/how-to-check-list-containing-nan
if np.isnan(normalized_embedding).any():
# for issue #11827 float values are not json compliant
logger.warning(f"Normalized embedding is nan: {normalized_embedding}")
continue
embedding_queue_embeddings.append(normalized_embedding) embedding_queue_embeddings.append(normalized_embedding)
except IntegrityError: except IntegrityError:
db.session.rollback() db.session.rollback()