intellif-aihub 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

aihub/models/quota_schedule_management.py

@@ -0,0 +1,239 @@
+from __future__ import annotations
+
+from typing import List, Optional
+
+from pydantic import BaseModel, Field
+
+
+class Env(BaseModel):
+    key: str
+    value: str
+
+
+class Sku(BaseModel):
+    cpu: int
+    gpu: int
+    memory: int
+
+
+class VirtualCluster(BaseModel):
+    id: int
+    name: str
+    gpu_type: str = Field(alias="gpu_type")
+    label: str
+    sku: Sku
+
+
+class Storage(BaseModel):
+    id: int
+    name: str
+    path: str
+    server_path: str = Field(alias="server_path")
+    server_host: str = Field(alias="server_host")
+    server_type: str = Field(alias="server_type")
+    permission: str
+    description: str
+
+
+class Category(BaseModel):
+    id: int
+    name: str
+
+
+class Project(BaseModel):
+    id: int
+    name: str
+    description: str
+
+
+class User(BaseModel):
+    id: int
+    name: str
+
+
+class SourceTask(BaseModel):
+    id: int
+    name: str
+
+
+class CreateTaskRequest(BaseModel):
+    priority: int
+    framework: str
+    name: str
+    description: Optional[str] = None
+    command: Optional[str] = None
+    image: str
+    virtual_cluster_id: int = Field(alias="virtual_cluster_id")
+    sku_cnt: int = Field(alias="sku_cnt")
+    enable_ssh: Optional[bool] = Field(False, alias="enable_ssh")
+    envs: Optional[List[Env]] = Field(default_factory=list, alias="envs")
+    storage_ids: Optional[List[int]] = Field(default_factory=list, alias="storage_ids")
+    instances: int
+    use_ib_network: Optional[bool] = Field(False, alias="use_ib_network")
+    always_pull_image: Optional[bool] = Field(False, alias="always_pull_image")
+    shm: Optional[int] = None
+    category_id: int = Field(alias="category_id")
+    project_id: int = Field(alias="project_id")
+    estimate_run_time: Optional[int] = Field(None, alias="estimate_run_time")
+    is_vip: Optional[bool] = Field(False, alias="is_vip")
+    preempt_policy: Optional[int] = Field(None, alias="preempt_policy")
+    vip_node_names: Optional[List[str]] = Field(default_factory=list, alias="vip_node_names")
+    enable_reschedule: Optional[bool] = Field(False, alias="enable_reschedule")
+
+
+class CreateTaskResponse(BaseModel):
+    id: int
+
+
+class Task(BaseModel):
+    id: int
+    priority: int
+    mtp_id: int = Field(alias="mtp_id")
+    framework: str
+    name: str
+    description: str
+    command: str
+    image: str
+    virtual_cluster: VirtualCluster = Field(alias="virtual_cluster")
+    sku_cnt: int = Field(alias="sku_cnt")
+    enable_ssh: bool = Field(alias="enable_ssh")
+    envs: Optional[List[Env]] = Field(default_factory=list, alias="envs")
+    storages: Optional[List[Storage]] = Field(default_factory=list, alias="storages")
+    instances: int
+    created_at: int = Field(alias="created_at")
+    username: str
+    user_id: int = Field(alias="user_id")
+    namespace: str
+    res_name: str = Field(alias="res_name")
+    status: int
+    use_ib_network: bool = Field(alias="use_ib_network")
+    always_pull_image: bool = Field(alias="always_pull_image")
+    shm: int
+    category: Category
+    project: Project
+    avg_gpu_util: float = Field(alias="avg_gpu_util")
+    finished_at: int = Field(alias="finished_at")
+    started_at: int = Field(alias="started_at")
+    estimate_run_time: int = Field(alias="estimate_run_time")
+    is_vip: bool = Field(alias="is_vip")
+    cluster_partition: str = Field(alias="cluster_partition")
+    preempt_policy: int = Field(alias="preempt_policy")
+    vip_node_names: Optional[List[str]] = Field(None, alias="vip_node_names")
+    stop_op_user: Optional[User] = Field(None, alias="stop_op_user")
+    use_new_log: bool = Field(alias="use_new_log")
+    is_quota_schedule: bool = Field(alias="is_quota_schedule")
+    enable_reschedule: bool = Field(alias="enable_reschedule")
+    remain_schedule_cnt: int = Field(alias="remain_schedule_cnt")
+    source_task: Optional[SourceTask] = Field(None, alias="source_task")
+
+
+class ListTasksRequest(BaseModel):
+    page_size: int = Field(20, alias="page_size")
+    page_num: int = Field(1, alias="page_num")
+    user_id: Optional[int] = Field(None, alias="user_id")
+    name: Optional[str] = None
+    virtual_cluster_id: Optional[int] = Field(None, alias="virtual_cluster_id")
+    status: Optional[int] = None
+    category_id: Optional[int] = Field(None, alias="category_id")
+    project_id: Optional[int] = Field(None, alias="project_id")
+    priority: Optional[int] = None
+
+
+class ListTasksResponse(BaseModel):
+    total: int
+    page_size: int = Field(alias="page_size")
+    page_num: int = Field(alias="page_num")
+    data: List[Task] = Field(default_factory=list)
+
+
+class Pod(BaseModel):
+    id: int
+    namespace: str
+    name: str
+    status: str
+    created_at: int = Field(alias="created_at")
+    started_at: int = Field(alias="started_at")
+    finished_at: int = Field(alias="finished_at")
+    host_ip: str = Field(alias="host_ip")
+    node_name: str = Field(alias="node_name")
+    ssh_port: int = Field(alias="ssh_port")
+    ssh_info: str = Field(alias="ssh_info")
+    use_new_log: bool = Field(alias="use_new_log")
+
+
+class ListTaskPodsRequest(BaseModel):
+    page_size: int = Field(20, alias="page_size")
+    page_num: int = Field(1, alias="page_num")
+
+
+class ListTaskPodsResponse(BaseModel):
+    total: int
+    page_size: int = Field(alias="page_size")
+    page_num: int = Field(alias="page_num")
+    data: List[Pod] = Field(default_factory=list)
+
+
+class PodLogInfo(BaseModel):
+    name: str
+    url: str
+
+
+class GetTaskPodLogsNewResponse(BaseModel):
+    logs: List[PodLogInfo]
+
+
+class GetTaskPodSpecResponse(BaseModel):
+    spec: str
+
+
+class GetTaskPodEventsResponse(BaseModel):
+    events: str
+
+
+class MachineOverview(BaseModel):
+    high: int
+    low: int
+    free: int
+
+
+class HighPrioritySummary(BaseModel):
+    group_id: int = Field(alias="group_id")
+    group_name: str = Field(alias="group_name")
+    used: int
+    total: int
+
+
+class MetricsOverview(BaseModel):
+    vc_id: int = Field(alias="vc_id")
+    vc_name: str = Field(alias="vc_name")
+    machine: MachineOverview
+    high_priority: List[HighPrioritySummary] = Field(alias="high_priority")
+
+
+class GetMetricsOverviewRequest(BaseModel):
+    page_size: int = Field(20, alias="page_size")
+    page_num: int = Field(1, alias="page_num")
+
+
+class GetMetricsOverviewResponse(BaseModel):
+    total: int
+    page_size: int = Field(alias="page_size")
+    page_num: int = Field(alias="page_num")
+    data: List[MetricsOverview] = Field(default_factory=list)
+
+
+class TaskUser(BaseModel):
+    user_id: int = Field(alias="user_id")
+    username: str
+
+
+class ListTaskUsersRequest(BaseModel):
+    page_size: int = Field(20, alias="page_size")
+    page_num: int = Field(1, alias="page_num")
+
+
+class ListTaskUsersResponse(BaseModel):
+    total: int
+    page_size: int = Field(alias="page_size")
+    page_num: int = Field(alias="page_num")
+    data: List[TaskUser] = Field(default_factory=list)

aihub/models/tag_resource_management.py

@@ -1,50 +1,50 @@
-# !/usr/bin/env python
-# -*-coding:utf-8 -*-
-
-from __future__ import annotations
-
-from typing import List, Optional
-
-from pydantic import BaseModel, Field
-
-
-class Project(BaseModel):
-    id: int
-    name: str
-
-
-class ProjectListData(BaseModel):
-    data: List[Project]
-
-
-class SelectProjectsResponse(BaseModel):
-    data: List[Project]
-
-
-class SkuBrief(BaseModel):
-    id: int
-    description: str
-    cpu: int
-    memory: int
-    gpu_type: int = Field(alias="gpu_type")
-    gpu_memory: int = Field(alias="gpu_memory")
-    network: int
-    created_at: int = Field(alias="created_at")
-
-
-class VirtualClusterBrief(BaseModel):
-    id: int
-    name: str
-    uuid: str
-    sku: Optional[SkuBrief] = None
-    created_at: int = Field(alias="created_at")
-
-
-class SelectVirtualClustersRequest(BaseModel):
-    user_id: int = Field(alias="user_id")
-    module_type: Optional[int] = Field(None, alias="module_type")
-    new_module_type: Optional[str] = Field(None, alias="new_module_type")
-
-
-class SelectVirtualClustersResponse(BaseModel):
-    data: Optional[List[VirtualClusterBrief]] = Field(default_factory=list)
+# !/usr/bin/env python
+# -*-coding:utf-8 -*-
+
+from __future__ import annotations
+
+from typing import List, Optional
+
+from pydantic import BaseModel, Field
+
+
+class Project(BaseModel):
+    id: int
+    name: str
+
+
+class ProjectListData(BaseModel):
+    data: List[Project]
+
+
+class SelectProjectsResponse(BaseModel):
+    data: List[Project]
+
+
+class SkuBrief(BaseModel):
+    id: int
+    description: str
+    cpu: int
+    memory: int
+    gpu_type: int = Field(alias="gpu_type")
+    gpu_memory: int = Field(alias="gpu_memory")
+    network: int
+    created_at: int = Field(alias="created_at")
+
+
+class VirtualClusterBrief(BaseModel):
+    id: int
+    name: str
+    uuid: str
+    sku: Optional[SkuBrief] = None
+    created_at: int = Field(alias="created_at")
+
+
+class SelectVirtualClustersRequest(BaseModel):
+    user_id: int = Field(alias="user_id")
+    module_type: Optional[int] = Field(None, alias="module_type")
+    new_module_type: Optional[str] = Field(None, alias="new_module_type")
+
+
+class SelectVirtualClustersResponse(BaseModel):
+    data: Optional[List[VirtualClusterBrief]] = Field(default_factory=list)

aihub/models/task_center.py

@@ -1,117 +1,117 @@
-from __future__ import annotations
-
-import json
-from enum import Enum
-from typing import Optional, List
-
-from pydantic import BaseModel, Field, field_serializer, field_validator
-
-
-class TaskCenterPriorityEnum(Enum):
-    """
-    任务优先级枚举
-    """
-
-    low = "low"
-    medium = "medium"
-    high = "high"
-
-
-class LabelProjectTypeEnum(Enum):
-    """
-    任务类型枚举
-     1 - 目标检测 2 - 语义分割 3 - 图片分类 4 - 实例分割 5 - 视频标注 6 - 人类偏好文本标注 7- 敏感预料文本标注 8 - 文本标注 9 - 关键点标注
-    """
-
-    OBJECT_DETECTION = 1
-    SEGMENTATION = 2
-    IMAGE_CLASSIFICATION = 3
-    INSTANCE_SEGMENTATION = 4
-    VIDEO_LABELING = 5
-    HUMAN_PREFERENCE_TEXT_LABELING = 6
-    SENSITIVE_TEXT_LABELING = 7
-    TEXT_LABELING = 8
-    KEYPOINT_LABELING = 9
-
-
-class CreateTaskOtherInfo(BaseModel):
-    label_project_type: LabelProjectTypeEnum = LabelProjectTypeEnum.IMAGE_CLASSIFICATION
-    dataset_id: int = Field(alias="dataset_id")
-    dataset_version_id: int = Field(alias="dataset_version_id")
-    doc_id: int = Field(alias="doc_id")
-    doc_type: str = Field(alias="doc_type", default="doc_center")
-
-
-class ProjectInfo(BaseModel):
-    label_project_id: int = Field(alias="label_project_id")
-    label_project_name: str = Field(alias="label_project_name")
-
-
-class TaskDetailOtherInfo(BaseModel):
-    label_project_type: LabelProjectTypeEnum = LabelProjectTypeEnum.IMAGE_CLASSIFICATION
-    dataset_id: int = Field(alias="dataset_id")
-    dataset_version_id: int = Field(alias="dataset_version_id")
-    doc_id: int = Field(alias="doc_id")
-    doc_type: str = Field(alias="doc_type", default="doc_center")
-    label_projects: Optional[List[ProjectInfo]] = Field(alias="label_projects")
-
-
-class CreateTaskReq(BaseModel):
-    name: str
-    description: Optional[str] = None
-    task_priority: Optional[str] = None
-    type: Optional[str] = None
-    receiver_id: Optional[int] = None
-    project_id: Optional[int] = None
-    other_info: CreateTaskOtherInfo = Field(alias="other_info")
-    estimated_delivery_at: Optional[int] = None
-
-    @field_serializer("other_info")
-    def serialize_other_info(self, value: CreateTaskOtherInfo) -> str:
-        """将 other_info 序列化为 JSON 字符串"""
-        return value.model_dump_json()
-
-
-class CreateTaskResp(BaseModel):
-    id: int = Field(alias="id")
-
-
-class LabelTaskDetail(BaseModel):
-    """任务详情"""
-
-    name: str
-    description: Optional[str] = Field(alias="description")
-    task_priority: Optional[str] = Field(alias="task_priority")
-    type: Optional[str] = Field(alias="type")
-    receiver_id: Optional[int] = Field(alias="receiver_id")
-    project_id: Optional[int] = None
-    other_info: TaskDetailOtherInfo = Field(alias="other_info")
-    estimated_delivery_at: Optional[int] = None
-
-    @field_serializer("other_info")
-    def serialize_other_info(self, value: TaskDetailOtherInfo) -> str:
-        """将 other_info 序列化为 JSON 字符串"""
-        return value.model_dump_json()
-
-    @field_validator("other_info", mode="before")
-    @classmethod
-    def parse_other_info(cls, value):
-        """将字符串解析为 TaskDetailOtherInfo 对象"""
-        if isinstance(value, str):
-            try:
-                # 解析 JSON 字符串为字典
-                data = json.loads(value)
-                # 创建 TaskDetailOtherInfo 对象
-                return TaskDetailOtherInfo(**data)
-            except (json.JSONDecodeError, TypeError, ValueError) as e:
-                raise ValueError(f"无法解析 other_info 字符串: {e}")
-        elif isinstance(value, dict):
-            # 如果传入的是字典，直接创建对象
-            return TaskDetailOtherInfo(**value)
-        elif isinstance(value, TaskDetailOtherInfo):
-            # 如果已经是对象，直接返回
-            return value
-        else:
-            raise ValueError(
-                f"other_info 必须是字符串、字典或 TaskDetailOtherInfo 对象，得到: {type(value)}"
-            )
+from __future__ import annotations
+
+import json
+from enum import Enum
+from typing import Optional, List
+
+from pydantic import BaseModel, Field, field_serializer, field_validator
+
+
+class TaskCenterPriorityEnum(Enum):
+    """
+    任务优先级枚举
+    """
+
+    low = "low"
+    medium = "medium"
+    high = "high"
+
+
+class LabelProjectTypeEnum(Enum):
+    """
+    任务类型枚举
+     1 - 目标检测 2 - 语义分割 3 - 图片分类 4 - 实例分割 5 - 视频标注 6 - 人类偏好文本标注 7- 敏感预料文本标注 8 - 文本标注 9 - 关键点标注
+    """
+
+    OBJECT_DETECTION = 1
+    SEGMENTATION = 2
+    IMAGE_CLASSIFICATION = 3
+    INSTANCE_SEGMENTATION = 4
+    VIDEO_LABELING = 5
+    HUMAN_PREFERENCE_TEXT_LABELING = 6
+    SENSITIVE_TEXT_LABELING = 7
+    TEXT_LABELING = 8
+    KEYPOINT_LABELING = 9
+
+
+class CreateTaskOtherInfo(BaseModel):
+    label_project_type: LabelProjectTypeEnum = LabelProjectTypeEnum.IMAGE_CLASSIFICATION
+    dataset_id: int = Field(alias="dataset_id")
+    dataset_version_id: int = Field(alias="dataset_version_id")
+    doc_id: int = Field(alias="doc_id")
+    doc_type: str = Field(alias="doc_type", default="doc_center")
+
+
+class ProjectInfo(BaseModel):
+    label_project_id: int = Field(alias="label_project_id")
+    label_project_name: str = Field(alias="label_project_name")
+
+
+class TaskDetailOtherInfo(BaseModel):
+    label_project_type: LabelProjectTypeEnum = LabelProjectTypeEnum.IMAGE_CLASSIFICATION
+    dataset_id: int = Field(alias="dataset_id")
+    dataset_version_id: int = Field(alias="dataset_version_id")
+    doc_id: int = Field(alias="doc_id")
+    doc_type: str = Field(alias="doc_type", default="doc_center")
+    label_projects: Optional[List[ProjectInfo]] = Field(alias="label_projects")
+
+
+class CreateTaskReq(BaseModel):
+    name: str
+    description: Optional[str] = None
+    task_priority: Optional[str] = None
+    type: Optional[str] = None
+    receiver_id: Optional[int] = None
+    project_id: Optional[int] = None
+    other_info: CreateTaskOtherInfo = Field(alias="other_info")
+    estimated_delivery_at: Optional[int] = None
+
+    @field_serializer("other_info")
+    def serialize_other_info(self, value: CreateTaskOtherInfo) -> str:
+        """将 other_info 序列化为 JSON 字符串"""
+        return value.model_dump_json()
+
+
+class CreateTaskResp(BaseModel):
+    id: int = Field(alias="id")
+
+
+class LabelTaskDetail(BaseModel):
+    """任务详情"""
+
+    name: str
+    description: Optional[str] = Field(alias="description")
+    task_priority: Optional[str] = Field(alias="task_priority")
+    type: Optional[str] = Field(alias="type")
+    receiver_id: Optional[int] = Field(alias="receiver_id")
+    project_id: Optional[int] = None
+    other_info: TaskDetailOtherInfo = Field(alias="other_info")
+    estimated_delivery_at: Optional[int] = None
+
+    @field_serializer("other_info")
+    def serialize_other_info(self, value: TaskDetailOtherInfo) -> str:
+        """将 other_info 序列化为 JSON 字符串"""
+        return value.model_dump_json()
+
+    @field_validator("other_info", mode="before")
+    @classmethod
+    def parse_other_info(cls, value):
+        """将字符串解析为 TaskDetailOtherInfo 对象"""
+        if isinstance(value, str):
+            try:
+                # 解析 JSON 字符串为字典
+                data = json.loads(value)
+                # 创建 TaskDetailOtherInfo 对象
+                return TaskDetailOtherInfo(**data)
+            except (json.JSONDecodeError, TypeError, ValueError) as e:
+                raise ValueError(f"无法解析 other_info 字符串: {e}")
+        elif isinstance(value, dict):
+            # 如果传入的是字典，直接创建对象
+            return TaskDetailOtherInfo(**value)
+        elif isinstance(value, TaskDetailOtherInfo):
+            # 如果已经是对象，直接返回
+            return value
+        else:
+            raise ValueError(
+                f"other_info 必须是字符串、字典或 TaskDetailOtherInfo 对象，得到: {type(value)}"
+            )