intellif-aihub 0.1.6__py3-none-any.whl → 0.1.8__py3-none-any.whl

aihub/__init__.py

@@ -1 +1 @@
-__version__ = "0.1.6"
+__version__ = "0.1.8"

aihub/models/model_training_platform.py

@@ -93,9 +93,6 @@ class CreateTrainingRequest(BaseModel):
     category_id: int = Field(alias="category_id", description="分类ID")
     project_id: int = Field(alias="project_id", description="项目ID")
     estimate_run_time: Optional[int] = Field(None, alias="estimate_run_time", description="预计运行时长(s)")
-    is_vip: Optional[bool] = Field(False, alias="is_vip", description="是否vip任务")
-    preempt_policy: Optional[PreemptPolicy] = Field(None, alias="preempt_policy", description="抢占策略")
-    vip_node_names: Optional[List[str]] = Field(default_factory=list, alias="vip_node_names", description="VIP节点名字")
     is_quota_schedule: Optional[bool] = Field(False, alias="is_quota_schedule", description="是否配额调度")
 
     model_config = {"use_enum_values": True}
@@ -150,10 +147,8 @@ class Training(BaseModel):
     finished_at: int = Field(alias="finished_at", description="结束时间")
     started_at: int = Field(alias="started_at", description="开始时间")
     estimate_run_time: int = Field(alias="estimate_run_time", description="预计运行时长")
-    is_vip: bool = Field(alias="is_vip", description="是否VIP")
     cluster_partition: str = Field(alias="cluster_partition", description="集群分区")
     preempt_policy: PreemptPolicy = Field(alias="preempt_policy", description="抢占策略")
-    vip_node_names: List[str] = Field(alias="vip_node_names", description="VIP节点")
     stop_op_user: Optional[User] = Field(None, alias="stop_op_user", description="停止操作人")
     use_new_log: bool = Field(alias="use_new_log", description="是否使用新版日志")
     is_quota_schedule: bool = Field(alias="is_quota_schedule", description="配额调度")
@@ -266,3 +261,7 @@ class ListTrainingContainersResponse(BaseModel):
     page_size: int = Field(alias="page_size", description="每页数量")
     page_num: int = Field(alias="page_num", description="当前页码")
     data: Optional[List[TrainingContainer]] = Field(default_factory=list, description="训练容器列表")
+
+
+class ListStoragesResponse(BaseModel):
+    data: List[Storage] = Field(default_factory=list, description="存储列表")

aihub/models/quota_schedule_management.py

@@ -101,9 +101,6 @@ class CreateTaskRequest(BaseModel):
     category_id: int = Field(alias="category_id", description="分类ID")
     project_id: int = Field(alias="project_id", description="项目ID")
     estimate_run_time: Optional[int] = Field(None, alias="estimate_run_time", description="预计运行时长(s)")
-    is_vip: Optional[bool] = Field(False, alias="is_vip", description="是否VIP")
-    preempt_policy: Optional[PreemptPolicy] = Field(None, alias="preempt_policy", description="抢占策略")
-    vip_node_names: Optional[List[str]] = Field(default_factory=list, alias="vip_node_names", description="VIP节点名字")
     enable_reschedule: Optional[bool] = Field(False, alias="enable_reschedule", description="启用失败重调度")
 
     model_config = {"use_enum_values": True}
@@ -145,10 +142,8 @@ class Task(BaseModel):
     finished_at: int = Field(alias="finished_at", description="结束时间")
     started_at: int = Field(alias="started_at", description="开始时间")
     estimate_run_time: int = Field(alias="estimate_run_time", description="预计运行时长")
-    is_vip: bool = Field(alias="is_vip", description="是否vip")
     cluster_partition: str = Field(alias="cluster_partition", description="集群分区")
     preempt_policy: PreemptPolicy = Field(alias="preempt_policy", description="抢占策略")
-    vip_node_names: Optional[List[str]] = Field(None, alias="vip_node_names", description="VIP节点")
     stop_op_user: Optional[User] = Field(None, alias="stop_op_user", description="停止操作人")
     use_new_log: bool = Field(alias="use_new_log", description="新版日志")
     is_quota_schedule: bool = Field(alias="is_quota_schedule", description="是否配额调度")

aihub/models/task_center.py

@@ -50,8 +50,8 @@ class ProjectInfo(BaseModel):
 class TaskDetailOtherInfo(BaseModel):
     """任务详情附加信息"""
     label_project_type: LabelProjectTypeEnum = Field(LabelProjectTypeEnum.IMAGE_CLASSIFICATION, description="标注枚举")
-    dataset_id: int = Field(alias="dataset_id", description="数据集ID")
-    dataset_version_id: int = Field(alias="dataset_version_id", description="数据集版本ID")
+    dataset_id: Optional[int] = Field(default=None, alias="dataset_id", description="数据集ID")
+    dataset_version_id: Optional[int] = Field(default=None, alias="dataset_version_id", description="数据集版本ID")
     doc_id: int = Field(alias="doc_id", description="文档中心文档ID")
     doc_type: str = Field(alias="doc_type", default="doc_center", description="文档类型")
     label_projects: Optional[List[ProjectInfo]] = Field(alias="label_projects", default=None,
@@ -83,23 +83,25 @@ class CreateTaskResp(BaseModel):
 class LabelTaskDetail(BaseModel):
     """任务详情"""
     name: str = Field(description="任务名称")
-    description: Optional[str] = Field(alias="description", description="任务描述")
-    task_priority: Optional[str] = Field(alias="task_priority", description="优先级")
-    type: Optional[str] = Field(alias="type", description="任务类型")
-    receiver_id: Optional[int] = Field(alias="receiver_id", description="接收人ID")
-    project_id: Optional[int] = Field(None, description="项目ID")
-    other_info: TaskDetailOtherInfo = Field(alias="other_info", description="附加信息")
-    estimated_delivery_at: Optional[int] = Field(None, description="预计交付时间")
+    description: Optional[str] = Field(default=None, alias="description", description="任务描述")
+    task_priority: Optional[str] = Field(default=None, alias="task_priority", description="优先级")
+    type: Optional[str] = Field(default=None, alias="type", description="任务类型")
+    receiver_id: Optional[int] = Field(default=None, alias="receiver_id", description="接收人ID")
+    project_id: Optional[int] = Field(default=None, description="项目ID")
+    other_info: Optional[TaskDetailOtherInfo] = Field(default=None, alias="other_info", description="附加信息")
+    estimated_delivery_at: Optional[int] = Field(default=None, description="预计交付时间")
 
     @field_serializer("other_info")
-    def serialize_other_info(self, value: TaskDetailOtherInfo) -> str:
+    def serialize_other_info(self, value: Optional[TaskDetailOtherInfo]) -> Optional[str]:
         """将 other_info 序列化为 JSON 字符串"""
-        return value.model_dump_json()
+        return value.model_dump_json() if value else None
 
     @field_validator("other_info", mode="before")
     @classmethod
     def parse_other_info(cls, value):
         """将字符串解析为 TaskDetailOtherInfo 对象"""
+        if value is None:
+            return None
         if isinstance(value, str):
             try:
                 # 解析 JSON 字符串为字典

aihub/services/model_training_platform.py

@@ -33,6 +33,7 @@ from ..models.model_training_platform import (
     ListTrainingUsersResponse,
     ListTrainingContainersResponse,
     Pod,
+    ListStoragesResponse,
 )
 
 _BASE = "/model-training-platform/api/v1"
@@ -43,6 +44,7 @@ class ModelTrainingPlatformService:
 
     def __init__(self, http: httpx.Client):
         self._training = _Training(http)
+        self._storage = _Storage(http)
 
     def create_training(self, payload: CreateTrainingRequest) -> int:
         """创建训练任务
@@ -167,10 +169,22 @@ class ModelTrainingPlatformService:
         """
         return self._training.list_training_containers(payload)
 
+    def list_storages(self) -> ListStoragesResponse:
+        """获取存储列表
+
+        Returns:
+            ListStoragesResponse: 存储列表
+        """
+        return self._storage.list_storages()
+
     @property
     def training(self) -> _Training:
         return self._training
 
+    @property
+    def storage(self) -> _Storage:
+        return self._storage
+
 
 class _Training:
 
@@ -178,20 +192,14 @@ class _Training:
         self._http = http
 
     def create(self, payload: CreateTrainingRequest) -> int:
-        resp = self._http.post(
-            f"{_BASE}/trainings",
-            json=payload.model_dump(by_alias=True, exclude_none=True),
-        )
+        resp = self._http.post(f"{_BASE}/trainings", json=payload.model_dump(by_alias=True, exclude_none=True))
         wrapper = APIWrapper[CreateTrainingResponse].model_validate(resp.json())
         if wrapper.code != 0:
             raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
         return wrapper.data.id
 
     def list(self, payload: ListTrainingsRequest) -> ListTrainingsResponse:
-        resp = self._http.get(
-            f"{_BASE}/trainings",
-            params=payload.model_dump(by_alias=True, exclude_none=True),
-        )
+        resp = self._http.get(f"{_BASE}/trainings", params=payload.model_dump(by_alias=True, exclude_none=True))
         wrapper = APIWrapper[ListTrainingsResponse].model_validate(resp.json())
         if wrapper.code != 0:
             raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
@@ -210,13 +218,9 @@ class _Training:
         if wrapper.code != 0:
             raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
 
-    def list_training_pods(
-            self, training_id: int, payload: ListTrainingPodsRequest
-    ) -> ListTrainingPodsResponse:
-        resp = self._http.get(
-            f"{_BASE}/trainings/{training_id}/pods",
-            params=payload.model_dump(by_alias=True, exclude_none=True),
-        )
+    def list_training_pods(self, training_id: int, payload: ListTrainingPodsRequest) -> ListTrainingPodsResponse:
+        resp = self._http.get(f"{_BASE}/trainings/{training_id}/pods",
+                              params=payload.model_dump(by_alias=True, exclude_none=True))
         wrapper = APIWrapper[ListTrainingPodsResponse].model_validate(resp.json())
         if wrapper.code != 0:
             raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
@@ -229,51 +233,50 @@ class _Training:
             raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
         return wrapper.data
 
-    def get_training_logs_new(
-            self, training_id: int, pod_id: int
-    ) -> GetTrainingPodLogsNewResponse:
+    def get_training_logs_new(self, training_id: int, pod_id: int) -> GetTrainingPodLogsNewResponse:
         resp = self._http.get(f"{_BASE}/trainings/{training_id}/pods/{pod_id}/logs/new")
         wrapper = APIWrapper[GetTrainingPodLogsNewResponse].model_validate(resp.json())
         if wrapper.code != 0:
             raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
         return wrapper.data
 
-    def get_training_spec(
-            self, training_id: int, pod_id: int
-    ) -> GetTrainingPodSpecResponse:
+    def get_training_spec(self, training_id: int, pod_id: int) -> GetTrainingPodSpecResponse:
         resp = self._http.get(f"{_BASE}/trainings/{training_id}/pods/{pod_id}/spec")
         wrapper = APIWrapper[GetTrainingPodSpecResponse].model_validate(resp.json())
         if wrapper.code != 0:
             raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
         return wrapper.data
 
-    def get_training_events(
-            self, training_id: int, pod_id: int
-    ) -> GetTrainingPodEventsResponse:
+    def get_training_events(self, training_id: int, pod_id: int) -> GetTrainingPodEventsResponse:
         resp = self._http.get(f"{_BASE}/trainings/{training_id}/pods/{pod_id}/events")
         wrapper = APIWrapper[GetTrainingPodEventsResponse].model_validate(resp.json())
         if wrapper.code != 0:
             raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
         return wrapper.data
 
-    def list_training_users(
-            self, payload: ListTrainingUsersRequest
-    ) -> ListTrainingUsersResponse:
-        resp = self._http.get(
-            f"{_BASE}/training-users", params=payload.model_dump(by_alias=True)
-        )
+    def list_training_users(self, payload: ListTrainingUsersRequest) -> ListTrainingUsersResponse:
+        resp = self._http.get(f"{_BASE}/training-users", params=payload.model_dump(by_alias=True))
         wrapper = APIWrapper[ListTrainingUsersResponse].model_validate(resp.json())
         if wrapper.code != 0:
             raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
         return wrapper.data
 
-    def list_training_containers(
-            self, payload: ListTrainingContainersRequest
-    ) -> ListTrainingContainersResponse:
-        resp = self._http.get(
-            f"{_BASE}/training-containers", params=payload.model_dump(by_alias=True)
-        )
+    def list_training_containers(self, payload: ListTrainingContainersRequest) -> ListTrainingContainersResponse:
+        resp = self._http.get(f"{_BASE}/training-containers", params=payload.model_dump(by_alias=True))
         wrapper = APIWrapper[ListTrainingContainersResponse].model_validate(resp.json())
         if wrapper.code != 0:
             raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
         return wrapper.data
+
+
+class _Storage:
+
+    def __init__(self, http: httpx.Client):
+        self._http = http
+
+    def list_storages(self) -> ListStoragesResponse:
+        resp = self._http.get(f"{_BASE}/storages")
+        wrapper = APIWrapper[ListStoragesResponse].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data

{intellif_aihub-0.1.6.dist-info → intellif_aihub-0.1.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intellif-aihub
-Version: 0.1.6
+Version: 0.1.8
 Summary: Intellif AI-hub SDK.
 Author-email: Platform Team <aihub@example.com>
 License-Expression: Apache-2.0
@@ -44,7 +44,7 @@ aihub_sdk/
 ```bash
 # PyPI 安装
 pip install intellif-aihub
-# 运行环境：Python ≥ 3.8
+# 运行环境：Python ≥ 3.9
 ```
 
 ---

{intellif_aihub-0.1.6.dist-info → intellif_aihub-0.1.8.dist-info}/RECORD

@@ -1,4 +1,4 @@
-aihub/__init__.py,sha256=n3oM6B_EMz93NsTI18NNZd-jKFcUPzUkbIKj5VFK5ok,22
+aihub/__init__.py,sha256=C69ADlbQREQlR15trneyA2sk8x0-oH4rDAX5fsv19_U,22
 aihub/client.py,sha256=bIYL-NoXarSFnPwLzEcWqUjdiTZ7sau5DmGz8YX4-RQ,5144
 aihub/exceptions.py,sha256=l2cMAvipTqQOio3o11fXsCCSCevbuK4PTsxofkobFjk,500
 aihub/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -10,10 +10,10 @@ aihub/models/document_center.py,sha256=JVA-6hYSY-efmhDmb6nNJEFk-MK5TZjXpwjwxWPfs
 aihub/models/eval.py,sha256=4Gon4Sg4dOkyCx3KH2mO5ip3AhrBwrPC0UZA447HeoQ,910
 aihub/models/labelfree.py,sha256=nljprYO6ECuctTVbHqriQ73N5EEyYURhBrnU28Ngfvc,1589
 aihub/models/model_center.py,sha256=K94u6tsIP86Ws3P93O8uoO1gBmnamVnoxlDzvjiPLSo,5719
-aihub/models/model_training_platform.py,sha256=ewkbYifUlGgdxsReXeqXxfSmrgwfACDo5Rtlz4hNmpM,11864
-aihub/models/quota_schedule_management.py,sha256=RDRoQypS2tNaNmiU_9Aj7yciG8knTzg6r4kNmgE8leQ,12743
+aihub/models/model_training_platform.py,sha256=UrSbKDQq6evu4Gsw1RYO_INxkLbFHnf1-FgbDIpK4VI,11510
+aihub/models/quota_schedule_management.py,sha256=1nZAX9FbFwruT46U7kKEsECCbvfY-c4-xyYVTki5NVg,12256
 aihub/models/tag_resource_management.py,sha256=-FgiKyDIG7bZagzVRf-8rXWuqH9GyciDadxz5W2f3I8,2195
-aihub/models/task_center.py,sha256=Koz1Bet6PCJRLmoUMqz0eCQCRr0y4yEK1KmIgLXfOHM,5269
+aihub/models/task_center.py,sha256=7PlfAsOGBfv_KqfWmPYaWyle4WCKJdc8tTl9aRid_Ws,5502
 aihub/models/user_system.py,sha256=0L_pBkWL9v1tv_mclOyRgCyWibuuj_XU3mPoe2v48vQ,12216
 aihub/models/workflow_center.py,sha256=5li_nZ_UuiAfVG5waSwP3mYViQZ6Z7dlwxv67HNUc_I,19879
 aihub/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -24,7 +24,7 @@ aihub/services/document_center.py,sha256=dG67Ji-DOnzL2t-4x4gVfMt9fbSj_IjVHCLw5R-
 aihub/services/eval.py,sha256=V1nBISIyYWg9JJO24xzy4-kit9NsaCYp1EWIX_fgJkQ,2128
 aihub/services/labelfree.py,sha256=xua62UWhVXTxJjHRyy86waaormnJjmpQwepcARBy_h0,1450
 aihub/services/model_center.py,sha256=QZXlldPZrbC6YkVG3eGw5_53qvFAim2QlXg3DflByTA,6215
-aihub/services/model_training_platform.py,sha256=UR8JGUbZKHauusxCq_hMPtyRT1QziqD_6mMFjK80uBY,9648
+aihub/services/model_training_platform.py,sha256=38o6HJnyi3htFzpX7qj6UhzdqTchcXLRTYU0nM7ffJg,10176
 aihub/services/quota_schedule_management.py,sha256=UYOMwjXxJTgkpN6Rv5GzlcejtpZfu23PXlSKr0WihTY,9586
 aihub/services/reporter.py,sha256=ot93SmhxgwDJOzlHSCwlxDOuSydTWUEUQ-Ctp97wJBQ,669
 aihub/services/tag_resource_management.py,sha256=Bm_inSIzZbTc-e4LU9kvwtsPpM_yLwm8xzdrALjb6uY,2666
@@ -35,8 +35,8 @@ aihub/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aihub/utils/download.py,sha256=yC3SoL5uE68pMB-IsNz233wj-gFrHB7D7ALzQA5JkFM,2155
 aihub/utils/http.py,sha256=rSNh4uNP7E3YGm3H1indRHctxC5Wu5xNBPvDrb9UHt4,421
 aihub/utils/s3.py,sha256=ISIBP-XdBPkURpXnN56ZnIWokOOg2SRUh_qvxJk-G1Q,2187
-intellif_aihub-0.1.6.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-intellif_aihub-0.1.6.dist-info/METADATA,sha256=BVzQ1sfR3t5GCyC1NWhTJrgfIJNP3_FSr8ddqHiXIOY,2916
-intellif_aihub-0.1.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-intellif_aihub-0.1.6.dist-info/top_level.txt,sha256=vIvTtSIN73xv46BpYM-ctVGnyOiUQ9EWP_6ngvdIlvw,6
-intellif_aihub-0.1.6.dist-info/RECORD,,
+intellif_aihub-0.1.8.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+intellif_aihub-0.1.8.dist-info/METADATA,sha256=KvnHK0E9xn_TyvLVtR7XPX6LWQ2sY4iVG7lYDcAO7s0,2916
+intellif_aihub-0.1.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+intellif_aihub-0.1.8.dist-info/top_level.txt,sha256=vIvTtSIN73xv46BpYM-ctVGnyOiUQ9EWP_6ngvdIlvw,6
+intellif_aihub-0.1.8.dist-info/RECORD,,