intellif-aihub 0.1.18__tar.gz → 0.1.19__tar.gz

{intellif_aihub-0.1.18 → intellif_aihub-0.1.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intellif-aihub
-Version: 0.1.18
+Version: 0.1.19
 Summary: Intellif AI-hub SDK.
 Author-email: Platform Team <aihub@example.com>
 License-Expression: Apache-2.0

{intellif_aihub-0.1.18 → intellif_aihub-0.1.19}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name            = "intellif-aihub"
-version = "0.1.18"
+version = "0.1.19"
 description     = "Intellif AI-hub SDK."
 readme          = {file = "README.md", content-type = "text/markdown"}
 requires-python = ">=3.9"

intellif_aihub-0.1.19/src/aihub/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.19"

{intellif_aihub-0.1.18 → intellif_aihub-0.1.19}/src/aihub/models/labelfree.py

@@ -1,12 +1,29 @@
 from __future__ import annotations
 
+from enum import Enum
 from typing import Optional
 
 from pydantic import BaseModel, Field
 
 
+class LabelProjectStatus(Enum):
+    """标注状态"""
+
+    Pending = "pending"
+    """未开始"""
+    Loading = "loading"
+    """数据读取中"""
+    Error = "failed"
+    """数据读取异常"""
+    In_Progress = "ready"
+    """进行中"""
+    Finished = "finished"
+    """标注完成"""
+
+
 class Stats(BaseModel):
     """标注统计信息"""
+
     total_annotations: int = Field(alias="total_annotations", description="总数据量")
     labeled_annotations: int = Field(alias="labeled_annotations", description="已标注数据量")
     total_labels: int = Field(alias="total_labels", description="总标签量")
@@ -19,9 +36,12 @@ class Stats(BaseModel):
 
 class GetGlobalStatsResponse(BaseModel):
     """标注统计概况"""
+
     global_stats: Stats = Field(alias="global_stats")
     valid_ten_percent: bool = Field(alias="valid_ten_percent", description="是否完成验收10%")
     valid_fifty_percent: bool = Field(alias="valid_fifty_percent", description="是否完成验收50%")
     valid_hundred_percent: bool = Field(alias="valid_hundred_percent", description="是否完成验收100%")
     data_exported_count: int = Field(alias="data_exported_count", description="已导出数据次数")
     exported_dataset_name: str = Field(alias="exported_dataset_name", description="最新数据集名称")
+    status: LabelProjectStatus = Field(description="状态")
+    model_config = {"use_enum_values": True}

{intellif_aihub-0.1.18 → intellif_aihub-0.1.19}/src/aihub/models/task_center.py

@@ -138,8 +138,8 @@ class LabelValidateStage(Enum):
     """10%阶段"""
     FIFTY_PERCENT = "标注阶段（50%）"
     """50%阶段"""
-    HUNDRED_PERCENT = "标注阶段（100%）"
-    """100%阶段（已交付）"""
+    LABEL_FINISHED = "标注阶段（100%）"
+    """标注完成"""
 
 
 class LabelValidateReq(BaseModel):

{intellif_aihub-0.1.18 → intellif_aihub-0.1.19}/src/intellif_aihub.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intellif-aihub
-Version: 0.1.18
+Version: 0.1.19
 Summary: Intellif AI-hub SDK.
 Author-email: Platform Team <aihub@example.com>
 License-Expression: Apache-2.0

{intellif_aihub-0.1.18 → intellif_aihub-0.1.19}/tests/test_task_center.py

@@ -12,14 +12,16 @@ TOKEN = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOjQ5MDY2ODUwODAsImlhdCI6MT
 
 class TestTaskCenter(unittest.TestCase):
     def test_create_label_task(self):
-        from src.aihub.models.task_center import LabelProjectTypeEnum
+        from src.aihub.models.task_center import LabelProjectTypeEnum, LabelValidateStage
         from src.aihub.client import Client
         from src.aihub.models.task_center import TaskCenterPriorityEnum
+        from src.aihub.models.labelfree import LabelProjectStatus
+        import time
 
         # 创建任务
         client = Client(base_url=BASE_URL, token=TOKEN)
         task_id = client.task_center.create_label_task(
-            name="test_task",
+            name="test_tas2k232",
             dataset_version_name="re/V1",
             feishu_doc_name="人脸质量人脸照片分类",
             task_receiver_name="hyc",
@@ -34,18 +36,24 @@ class TestTaskCenter(unittest.TestCase):
             # 获取任务信息
             task_item = client.task_center.get(task_id)
             # 使用sdkl
+            if not task_item.other_info.label_projects:
+                print("任务未完成，请稍后...")
+                time.sleep(5)
+                continue
             p = task_item.other_info.label_projects[0]
             label_stats = client.labelfree.get_project_global_stats(p.label_project_name)
             # 等待标注完成、完成100%验收、存在数据导出
-            if (
-                    (label_stats.global_stats.total_annotations == label_stats.global_stats.labeled_annotations)
-                and (label_stats.valid_hundred_percent)
-                and label_stats.data_exported_count != 0
-            ):
+            if label_stats.status == LabelProjectStatus.Finished and label_stats.data_exported_count != 0:
                 exported_dataset_name = label_stats.exported_dataset_name
+                client.task_center.validate_label_project(
+                    task_id, p.label_project_name, LabelValidateStage.LABEL_FINISHED, True
+                )
 
-                client.task_center.validate_label_project(task_id, {})
                 break
+            else:
+                print("任务未完成，请稍后...")
+                time.sleep(5)
+                continue
 
         # 下载数据
         client.dataset_management.run_download(exported_dataset_name, local_dir="./output")
@@ -56,4 +64,10 @@ class TestTaskCenter(unittest.TestCase):
         from src.aihub.models.task_center import LabelValidateStage
 
         client = Client(base_url=BASE_URL, token=TOKEN)
-        client.task_center.validate_label_project(1923, "project_893437", LabelValidateStage.HUNDRED_PERCENT, True)
+        client.task_center.validate_label_project(1923, "project_893437", LabelValidateStage.LABEL_FINISHED, True)
+
+    def test_get_label_project(self):
+        from src.aihub.client import Client
+
+        client = Client(base_url=BASE_URL, token=TOKEN)
+        client.labelfree.get_project_global_stats("project_889552")

intellif_aihub-0.1.18/src/aihub/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.1.18"