refactor: replace Status with State for consistent status handling

Author: liuruibin

apps/common/event/listener_manage.py

@@ -92,7 +92,7 @@ def embedding_by_paragraph_list(paragraph_id_list, embedding_model: Embeddings):
     @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 @@ def is_save_function():
         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(

apps/knowledge/task/handler.py

@@ -1,7 +1,6 @@
 # coding=utf-8
 
 
-import logging
 import re
 import traceback
 
@@ -11,7 +10,8 @@
 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 handler(source_url: str, selector, response: Fork.Response):
                      meta={'source_url': source_url, 'selector': selector},
                      type=KnowledgeType.WEB,
                      char_length=0,
-                     status=Status.error).save()
+                     status=State.FAILURE).save()
 
     return handler