Переглянути джерело

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

tags/1.9.1
AkisAya 1 місяць тому
джерело
коміт
66196459d5
Аккаунт користувача з таким Email не знайдено
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 = []

Завантаження…
Відмінити
Зберегти