Browse Source

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

tags/1.9.1
AkisAya 1 month ago
parent
commit
66196459d5
No account linked to committer's email address
1 changed files with 4 additions and 0 deletions
  1. 4
    0
      api/core/rag/embedding/cached_embedding.py

+ 4
- 0
api/core/rag/embedding/cached_embedding.py 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 = []

Loading…
Cancel
Save