浏览代码

chore: skip document segments fetching with non-existed dataset of DatasetDocument in add_document_to_index_task task (#17784)

tags/1.3.0
Bowen Liang 6 个月前
父节点
当前提交
636a0ba37f
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 4 次插入5 次删除
  1. 4
    5
      api/tasks/add_document_to_index_task.py

+ 4
- 5
api/tasks/add_document_to_index_task.py 查看文件

@@ -37,6 +37,10 @@ def add_document_to_index_task(dataset_document_id: str):
indexing_cache_key = "document_{}_indexing".format(dataset_document.id)

try:
dataset = dataset_document.dataset
if not dataset:
raise Exception(f"Document {dataset_document.id} dataset {dataset_document.dataset_id} doesn't exist.")

segments = (
db.session.query(DocumentSegment)
.filter(
@@ -77,11 +81,6 @@ def add_document_to_index_task(dataset_document_id: str):
document.children = child_documents
documents.append(document)

dataset = dataset_document.dataset

if not dataset:
raise Exception("Document has no dataset")

index_type = dataset.doc_form
index_processor = IndexProcessorFactory(index_type).init_index_processor()
index_processor.load(dataset, documents)

正在加载...
取消
保存