intellif-aihub 0.1.6__tar.gz → 0.1.8__tar.gz

{intellif_aihub-0.1.6 → intellif_aihub-0.1.8}/PKG-INFO

@@ -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 → intellif_aihub-0.1.8}/README.md

@@ -23,7 +23,7 @@ aihub_sdk/
 ```bash
 # PyPI 安装
 pip install intellif-aihub
-# 运行环境：Python ≥ 3.8
+# 运行环境：Python ≥ 3.9
 ```
 
 ---

{intellif_aihub-0.1.6 → intellif_aihub-0.1.8}/pyproject.toml

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

intellif_aihub-0.1.8/src/aihub/__init__.py

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

{intellif_aihub-0.1.6 → intellif_aihub-0.1.8}/src/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="存储列表")

{intellif_aihub-0.1.6 → intellif_aihub-0.1.8}/src/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="是否配额调度")

{intellif_aihub-0.1.6 → intellif_aihub-0.1.8}/src/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 字符串为字典

{intellif_aihub-0.1.6 → intellif_aihub-0.1.8}/src/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 → intellif_aihub-0.1.8}/src/intellif_aihub.egg-info/PKG-INFO

@@ -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 → intellif_aihub-0.1.8}/tests/test_model_training_platform.py

@@ -74,3 +74,6 @@ class TestModelTrainingPlatform(unittest.TestCase):
         containers = client.model_training_platform.list_training_containers(ListTrainingContainersRequest())
         data = containers.data or []
         self.assertIsInstance(data, list)
+
+        storages = client.model_training_platform.list_storages()
+        self.assertIsInstance(storages.data, list)

intellif_aihub-0.1.6/src/aihub/__init__.py

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