diff --git a/apps/common/event/listener_manage.py b/apps/common/event/listener_manage.py index 040fad65a..f27d2e627 100644 --- a/apps/common/event/listener_manage.py +++ b/apps/common/event/listener_manage.py @@ -92,7 +92,7 @@ class ListenerManagement: @staticmethod def embedding_by_paragraph_data_list(data_list, paragraph_id_list, embedding_model: Embeddings): maxkb_logger.info(_('Start--->Embedding paragraph: {paragraph_id_list}').format(paragraph_id_list=paragraph_id_list)) - status = Status.success + status = State.SUCCESS try: # 删除段落 VectorStore.get_embedding_vector().delete_by_paragraph_ids(paragraph_id_list) @@ -105,7 +105,7 @@ class ListenerManagement: except Exception as e: maxkb_logger.error(_('Vectorized paragraph: {paragraph_id_list} error {error} {traceback}').format( paragraph_id_list=paragraph_id_list, error=str(e), traceback=traceback.format_exc())) - status = Status.error + status = State.FAILURE finally: QuerySet(Paragraph).filter(id__in=paragraph_id_list).update(**{'status': status}) maxkb_logger.info( diff --git a/apps/knowledge/task/handler.py b/apps/knowledge/task/handler.py index 9734f0202..1fa6fb503 100644 --- a/apps/knowledge/task/handler.py +++ b/apps/knowledge/task/handler.py @@ -1,7 +1,6 @@ # coding=utf-8 -import logging import re import traceback @@ -11,7 +10,8 @@ from django.utils.translation import gettext_lazy as _ from common.utils.fork import ChildLink, Fork from common.utils.logger import maxkb_logger from common.utils.split_model import get_split_model -from knowledge.models.knowledge import KnowledgeType, Document, Knowledge, Status +from knowledge.models import State +from knowledge.models.knowledge import KnowledgeType, Document, Knowledge def get_save_handler(knowledge_id, selector): @@ -89,7 +89,7 @@ def get_sync_web_document_handler(knowledge_id): meta={'source_url': source_url, 'selector': selector}, type=KnowledgeType.WEB, char_length=0, - status=Status.error).save() + status=State.FAILURE).save() return handler