|
@@ -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()
|
|
|
|
+
|
|
|
|
+ if np.isnan(normalized_embedding).any():
|
|
|
|
+
|
|
|
|
+ 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()
|