15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

refactor: replace Status with State for consistent status handling

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
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(

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

@ -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