瀏覽代碼

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 查看文件

text_embeddings[i] = embedding.get_embedding() text_embeddings[i] = embedding.get_embedding()
else: else:
embedding_queue_indices.append(i) embedding_queue_indices.append(i)

# release database connection, because embedding may take a long time
db.session.close()

if embedding_queue_indices: if embedding_queue_indices:
embedding_queue_texts = [texts[i] for i in embedding_queue_indices] embedding_queue_texts = [texts[i] for i in embedding_queue_indices]
embedding_queue_embeddings = [] embedding_queue_embeddings = []

Loading…
取消
儲存