From 8db35c45c1861c71ecd475cb28c5d3ec74812878 Mon Sep 17 00:00:00 2001 From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com> Date: Tue, 7 Jan 2025 18:21:13 +0800 Subject: [PATCH] fix: Document upload prompt: missing status list field (#1991) --- apps/dataset/serializers/document_serializers.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/apps/dataset/serializers/document_serializers.py b/apps/dataset/serializers/document_serializers.py index 3d24c01e6..33b0838ba 100644 --- a/apps/dataset/serializers/document_serializers.py +++ b/apps/dataset/serializers/document_serializers.py @@ -700,7 +700,11 @@ class DocumentSerializers(ApiMixin, serializers.Serializer): _document.save() return self.one() - def refresh(self, state_list, with_valid=True): + def refresh(self, state_list=None, with_valid=True): + if state_list is None: + state_list = [State.PENDING.value, State.STARTED.value, State.SUCCESS.value, State.FAILURE.value, + State.REVOKE.value, + State.REVOKED.value, State.IGNORED.value] if with_valid: self.is_valid(raise_exception=True) document_id = self.data.get("document_id")