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

2026-01-08 07:14:14 +00:00 · 2025-09-28 13:44:51 +08:00
parent a5387b304e
commit 66196459d5
1 changed files with 4 additions and 0 deletions
--- a/api/core/rag/embedding/cached_embedding.py
+++ b/api/core/rag/embedding/cached_embedding.py
@@ -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 = []