fix db connection error in embed_documents() (#26196)

This commit is contained in:
AkisAya
2025-09-28 13:44:51 +08:00
committed by GitHub
parent a5387b304e
commit 66196459d5

View File

@@ -42,6 +42,10 @@ class CacheEmbedding(Embeddings):
text_embeddings[i] = embedding.get_embedding()
else:
embedding_queue_indices.append(i)
# release database connection, because embedding may take a long time
db.session.close()
if embedding_queue_indices:
embedding_queue_texts = [texts[i] for i in embedding_queue_indices]
embedding_queue_embeddings = []