zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: replace Status with State for consistent status handling

Browse Source
This commit is contained in:
CaptainB 2025-07-17 16:42:24 +08:00
parent 19a1b5a788
commit 7eeb42c9fd
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/common/event/listener_manage.py
View File

@ -92,7 +92,7 @@ class ListenerManagement:
@staticmethod @staticmethod
def embedding_by_paragraph_data_list(data_list, paragraph_id_list, embedding_model: Embeddings): 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)) maxkb_logger.info(_('Start--->Embedding paragraph: {paragraph_id_list}').format(paragraph_id_list=paragraph_id_list))
status = Status.success status = State.SUCCESS
try: try:
# 删除段落 # 删除段落
VectorStore.get_embedding_vector().delete_by_paragraph_ids(paragraph_id_list) VectorStore.get_embedding_vector().delete_by_paragraph_ids(paragraph_id_list)
@ -105,7 +105,7 @@ class ListenerManagement:
except Exception as e: except Exception as e:
maxkb_logger.error(_('Vectorized paragraph: {paragraph_id_list} error {error} {traceback}').format( 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())) paragraph_id_list=paragraph_id_list, error=str(e), traceback=traceback.format_exc()))
status = Status.error status = State.FAILURE
finally: finally:
QuerySet(Paragraph).filter(id__in=paragraph_id_list).update(**{'status': status}) QuerySet(Paragraph).filter(id__in=paragraph_id_list).update(**{'status': status})
maxkb_logger.info( maxkb_logger.info(

6
apps/knowledge/task/handler.py
View File

@ -1,7 +1,6 @@
# coding=utf-8 # coding=utf-8
import logging
import re import re
import traceback import traceback
@ -11,7 +10,8 @@ from django.utils.translation import gettext_lazy as _
from common.utils.fork import ChildLink, Fork from common.utils.fork import ChildLink, Fork
from common.utils.logger import maxkb_logger from common.utils.logger import maxkb_logger
from common.utils.split_model import get_split_model 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): 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}, meta={'source_url': source_url, 'selector': selector},
type=KnowledgeType.WEB, type=KnowledgeType.WEB,
char_length=0, char_length=0,
status=Status.error).save() status=State.FAILURE).save()
return handler return handler