ソースを参照

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

tags/1.9.1
AkisAya 1ヶ月前
コミット
66196459d5
コミッターのメールアドレスに関連付けられたアカウントが存在しません
1個のファイルの変更4行の追加0行の削除
  1. 4
    0
      api/core/rag/embedding/cached_embedding.py

+ 4
- 0
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 = []

読み込み中…
キャンセル
保存