Просмотр исходного кода

fix full-doc mode document doesn't reindex after enable or un_archive (#16737)

tags/1.2.0
Jyong 7 месяцев назад
Родитель
Сommit
58c62f0a34
Аккаунт пользователя с таким Email не найден

+ 1
- 1
api/core/indexing_runner.py Просмотреть файл

}, },
) )
if dataset_document.doc_form == IndexType.PARENT_CHILD_INDEX: if dataset_document.doc_form == IndexType.PARENT_CHILD_INDEX:
child_chunks = document_segment.child_chunks
child_chunks = document_segment.get_child_chunks()
if child_chunks: if child_chunks:
child_documents = [] child_documents = []
for child_chunk in child_chunks: for child_chunk in child_chunks:

+ 17
- 0
api/models/dataset.py Просмотреть файл

else: else:
return [] return []


def get_child_chunks(self):
process_rule = self.document.dataset_process_rule
if process_rule.mode == "hierarchical":
rules = Rule(**process_rule.rules_dict)
if rules.parent_mode:
child_chunks = (
db.session.query(ChildChunk)
.filter(ChildChunk.segment_id == self.id)
.order_by(ChildChunk.position.asc())
.all()
)
return child_chunks or []
else:
return []
else:
return []

@property @property
def sign_content(self): def sign_content(self):
return self.get_sign_content() return self.get_sign_content()

+ 1
- 1
api/tasks/add_document_to_index_task.py Просмотреть файл

}, },
) )
if dataset_document.doc_form == IndexType.PARENT_CHILD_INDEX: if dataset_document.doc_form == IndexType.PARENT_CHILD_INDEX:
child_chunks = segment.child_chunks
child_chunks = segment.get_child_chunks()
if child_chunks: if child_chunks:
child_documents = [] child_documents = []
for child_chunk in child_chunks: for child_chunk in child_chunks:

+ 1
- 1
api/tasks/deal_dataset_vector_index_task.py Просмотреть файл

}, },
) )
if dataset_document.doc_form == IndexType.PARENT_CHILD_INDEX: if dataset_document.doc_form == IndexType.PARENT_CHILD_INDEX:
child_chunks = segment.child_chunks
child_chunks = segment.get_child_chunks()
if child_chunks: if child_chunks:
child_documents = [] child_documents = []
for child_chunk in child_chunks: for child_chunk in child_chunks:

+ 1
- 1
api/tasks/enable_segment_to_index_task.py Просмотреть файл



index_processor = IndexProcessorFactory(dataset_document.doc_form).init_index_processor() index_processor = IndexProcessorFactory(dataset_document.doc_form).init_index_processor()
if dataset_document.doc_form == IndexType.PARENT_CHILD_INDEX: if dataset_document.doc_form == IndexType.PARENT_CHILD_INDEX:
child_chunks = segment.child_chunks
child_chunks = segment.get_child_chunks()
if child_chunks: if child_chunks:
child_documents = [] child_documents = []
for child_chunk in child_chunks: for child_chunk in child_chunks:

+ 1
- 1
api/tasks/enable_segments_to_index_task.py Просмотреть файл

) )


if dataset_document.doc_form == IndexType.PARENT_CHILD_INDEX: if dataset_document.doc_form == IndexType.PARENT_CHILD_INDEX:
child_chunks = segment.child_chunks
child_chunks = segment.get_child_chunks()
if child_chunks: if child_chunks:
child_documents = [] child_documents = []
for child_chunk in child_chunks: for child_chunk in child_chunks:

Загрузка…
Отмена
Сохранить