From 433ae5daf5cf8b16aed9c0264b6cdcf3f106448f Mon Sep 17 00:00:00 2001 From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com> Date: Mon, 2 Dec 2024 15:54:49 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E9=80=89=E6=8B=A9?= =?UTF-8?q?=E4=B8=8D=E5=8F=AF=E7=94=A8=E7=9A=84=E6=9C=AC=E5=9C=B0=E6=A8=A1?= =?UTF-8?q?=E5=9E=8B=E8=BF=9B=E8=A1=8C=E5=90=91=E9=87=8F=E5=8C=96,?= =?UTF-8?q?=E7=8A=B6=E6=80=81=E4=B8=80=E7=9B=B4=E6=98=AF=E6=8E=92=E9=98=9F?= =?UTF-8?q?=E4=B8=AD=20(#1736)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/embedding/task/embedding.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/apps/embedding/task/embedding.py b/apps/embedding/task/embedding.py index 3e63c26b2..7087636c2 100644 --- a/apps/embedding/task/embedding.py +++ b/apps/embedding/task/embedding.py @@ -17,7 +17,7 @@ from django.db.models import QuerySet from common.config.embedding_config import ModelManage from common.event import ListenerManagement, UpdateProblemArgs, UpdateEmbeddingDatasetIdArgs, \ UpdateEmbeddingDocumentIdArgs -from dataset.models import Document, Status +from dataset.models import Document, Status, TaskType, State from ops import celery_app from setting.models import Model from setting.models_provider import get_model @@ -66,8 +66,8 @@ def embedding_by_document(document_id, model_id): """ def exception_handler(e): - QuerySet(Document).filter(id=document_id).update( - **{'status': Status.error, 'update_time': datetime.datetime.now()}) + ListenerManagement.update_status(QuerySet(Document).filter(id=document_id), TaskType.EMBEDDING, + State.FAILURE) max_kb_error.error( f'获取向量模型失败：{str(e)}{traceback.format_exc()}') @@ -102,7 +102,6 @@ def embedding_by_dataset(dataset_id, model_id): max_kb.info(f"数据集文档:{[d.name for d in document_list]}") for document in document_list: try: - print(document.id, model_id) embedding_by_document.delay(document.id, model_id) except Exception as e: pass