diff --git a/api/apps/sdk/doc.py b/api/apps/sdk/doc.py index 25c78d520..65271dba4 100644 --- a/api/apps/sdk/doc.py +++ b/api/apps/sdk/doc.py @@ -688,9 +688,9 @@ def parse(tenant_id, dataset_id): continue if not doc: return get_error_data_result(message=f"You don't own the document {id}.") - if doc[0].progress != 0.0: + if 0.0 < doc[0].progress < 1.0: return get_error_data_result( - "Can't stop parsing document with progress at 0 or 100" + "Can't parse document that is currently being processed" ) info = {"run": "1", "progress": 0, "progress_msg": "", "chunk_num": 0, "token_num": 0} DocumentService.update_by_id(id, info) diff --git a/sdk/python/test/test_http_api/test_file_management_within_dataset/test_parse_documents.py b/sdk/python/test/test_http_api/test_file_management_within_dataset/test_parse_documents.py index d4a5ff0a7..f96a06613 100644 --- a/sdk/python/test/test_http_api/test_file_management_within_dataset/test_parse_documents.py +++ b/sdk/python/test/test_http_api/test_file_management_within_dataset/test_parse_documents.py @@ -205,8 +205,7 @@ class TestDocumentsParse: res = parse_documnet( get_http_api_auth, dataset_id, {"document_ids": document_ids} ) - assert res["code"] == 102 - assert res["message"] == "Can't stop parsing document with progress at 0 or 100" + assert res["code"] == 0 @pytest.mark.skip(reason="issues/6234") def test_duplicate_parse(self, get_http_api_auth, tmp_path):