intellif-aihub 0.1.10__tar.gz → 0.1.12__tar.gz

{intellif_aihub-0.1.10/src/intellif_aihub.egg-info → intellif_aihub-0.1.12}/PKG-INFO

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

{intellif_aihub-0.1.10 → intellif_aihub-0.1.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name            = "intellif-aihub"
-version         = "0.1.10"
+version         = "0.1.12"
 description     = "Intellif AI-hub SDK."
 readme          = {file = "README.md", content-type = "text/markdown"}
 requires-python = ">=3.9"
@@ -46,3 +46,22 @@ dev = [
     "mkdocs-material>=9.6.15",
     "mkdocstrings-python>=1.11.1",
 ]
+
+
+[tool.black]
+line-length = 120
+target-version = ["py39", "py310"]
+include = '\.pyi?$'
+exclude = '''
+/(
+    \.git
+  | \.hg
+  | \.mypy_cache
+  | \.tox
+  | \.venv
+  | _build
+  | buck-out
+  | build
+  | dist
+)/
+'''

intellif_aihub-0.1.12/src/aihub/__init__.py

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

{intellif_aihub-0.1.10 → intellif_aihub-0.1.12}/src/aihub/client.py

@@ -1,8 +1,10 @@
 from __future__ import annotations
 
 import os
+import sys
 
 import httpx
+from loguru import logger
 
 from .exceptions import APIError
 from .services import artifact
@@ -65,7 +67,14 @@ class Client:
     user_system: UserSystemService = None
     workflow_center: WorkflowCenterService = None
 
-    def __init__(self, *, base_url: str, token: str | None = None, timeout: float = 60.0):
+    def __init__(
+        self,
+        *,
+        base_url: str,
+        token: str | None = None,
+        timeout: float = 60.0,
+        log_level: str = "INFO",
+    ):
         """AI-HUB python SDK 客户端
 
         Args:
@@ -77,6 +86,15 @@ class Client:
             >>> client = Client(base_url="xxx", token="xxxx")
 
         """
+        logger.remove()
+        logger.add(
+            sys.stdout,
+            colorize=True,
+            format="<green>{time}</green> <level>{message}</level>",
+            level=log_level,
+        )
+        logger.info(f"AI-HUB Python SDK initialized with log level: {log_level}")
+
         if not base_url:
             raise ValueError("base_url必须填写")
 

{intellif_aihub-0.1.10 → intellif_aihub-0.1.12}/src/aihub/models/model_center.py

@@ -3,29 +3,33 @@ from __future__ import annotations
 from enum import IntEnum
 from typing import List, Optional
 
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field, ConfigDict
 
 
 class User(BaseModel):
     """用户"""
+
     id: int = Field(description="用户ID")
     name: str = Field(description="用户名")
 
 
 class ModelType(BaseModel):
     """模型类型"""
+
     id: int = Field(description="类型ID")
     name: str = Field(description="类型名称")
 
 
 class DeployPlatform(BaseModel):
     """部署平台"""
+
     id: int = Field(description="部署平台ID")
     name: str = Field(description="部署平台名称")
 
 
 class QuantLevel(BaseModel):
     """量化等级"""
+
     id: int = Field(description="量化等级ID")
     name: str = Field(description="量化等级名称")
 
@@ -40,6 +44,7 @@ class ModelStatus(IntEnum):
 
 class Model(BaseModel):
     """模型详情"""
+
     id: int = Field(description="模型ID")
     name: str = Field(description="模型名称")
     description: str = Field(description="描述")
@@ -53,17 +58,19 @@ class Model(BaseModel):
     created_at: int = Field(alias="created_at", description="创建时间戳 (ms)")
     updated_at: int = Field(alias="updated_at", description="更新时间戳 (ms)")
 
-    model_config = {"use_enum_values": True}
+    model_config = ConfigDict(protected_namespaces=(), use_enum_values=True)
 
 
 class ListModelTypesRequest(BaseModel):
     """查询模型类型列表请求"""
+
     page_size: int = Field(999, alias="page_size", description="每页数量")
     page_num: int = Field(1, alias="page_num", description="当前页码")
 
 
 class ListModelTypesResponse(BaseModel):
     """查询模型类型列表返回"""
+
     total: int = Field(description="总条数")
     page_size: int = Field(alias="page_size", description="每页数量")
     page_num: int = Field(alias="page_num", description="当前页码")
@@ -72,12 +79,14 @@ class ListModelTypesResponse(BaseModel):
 
 class ListDeployPlatformsRequest(BaseModel):
     """查询部署平台列表请求"""
+
     page_size: int = Field(999, alias="page_size", description="每页数量")
     page_num: int = Field(1, alias="page_num", description="当前页码")
 
 
 class ListDeployPlatformsResponse(BaseModel):
     """查询部署平台列表返回"""
+
     total: int = Field(description="总条数")
     page_size: int = Field(alias="page_size", description="每页数量")
     page_num: int = Field(alias="page_num", description="当前页码")
@@ -86,12 +95,14 @@ class ListDeployPlatformsResponse(BaseModel):
 
 class ListQuantLevelsRequest(BaseModel):
     """查询量化等级列表请求"""
+
     page_size: int = Field(999, alias="page_size", description="每页数量")
     page_num: int = Field(1, alias="page_num", description="当前页码")
 
 
 class ListQuantLevelsResponse(BaseModel):
     """查询量化等级列表返回"""
+
     total: int = Field(description="总条数")
     page_size: int = Field(alias="page_size", description="每页数量")
     page_num: int = Field(alias="page_num", description="当前页码")
@@ -100,6 +111,7 @@ class ListQuantLevelsResponse(BaseModel):
 
 class ListModelsRequest(BaseModel):
     """查询模型列表请求"""
+
     page_size: int = Field(20, alias="page_size", description="每页数量")
     page_num: int = Field(1, alias="page_num", description="当前页码")
     name: Optional[str] = Field(None, description="名称过滤")
@@ -107,6 +119,7 @@ class ListModelsRequest(BaseModel):
 
 class ListModelsResponse(BaseModel):
     """查询模型列表返回"""
+
     total: int = Field(description="总条数")
     page_size: int = Field(alias="page_size", description="每页数量")
     page_num: int = Field(alias="page_num", description="当前页码")
@@ -115,6 +128,7 @@ class ListModelsResponse(BaseModel):
 
 class CreateModelRequest(BaseModel):
     """创建模型请求"""
+
     name: str = Field(description="模型名称")
     description: Optional[str] = Field(None, description="描述")
     model_type_id: int = Field(alias="model_type_id", description="模型类型ID")
@@ -122,15 +136,18 @@ class CreateModelRequest(BaseModel):
     deploy_platform_id: Optional[int] = Field(None, alias="deploy_platform_id", description="部署平台ID")
     param_cnt: Optional[str] = Field(None, alias="param_cnt", description="参数量")
     quant_level_id: Optional[int] = Field(None, alias="quant_level_id", description="量化等级ID")
+    model_config = ConfigDict(protected_namespaces=())
 
 
 class CreateModelResponse(BaseModel):
     """创建模型返回"""
+
     id: int = Field(description="模型ID")
 
 
 class EditModelRequest(BaseModel):
     """编辑模型请求"""
+
     id: int = Field(description="模型ID")
     name: str = Field(description="模型名称")
     description: Optional[str] = Field(None, description="描述")
@@ -139,3 +156,4 @@ class EditModelRequest(BaseModel):
     deploy_platform_id: Optional[int] = Field(None, alias="deploy_platform_id", description="部署平台ID")
     param_cnt: Optional[str] = Field(None, alias="param_cnt", description="参数量")
     quant_level_id: Optional[int] = Field(None, alias="quant_level_id", description="量化等级ID")
+    model_config = ConfigDict(protected_namespaces=())

{intellif_aihub-0.1.10 → intellif_aihub-0.1.12}/src/aihub/models/model_training_platform.py

@@ -3,17 +3,19 @@ from __future__ import annotations
 from enum import IntEnum
 from typing import List, Optional
 
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field, ConfigDict
 
 
 class Env(BaseModel):
     """环境变量"""
+
     key: str = Field(description="变量名")
     value: str = Field(description="变量值")
 
 
 class Sku(BaseModel):
     """sku"""
+
     cpu: int = Field(description="CPU数")
     gpu: int = Field(description="GPU数")
     memory: int = Field(description="内存GiB")
@@ -21,6 +23,7 @@ class Sku(BaseModel):
 
 class VirtualCluster(BaseModel):
     """虚拟集群"""
+
     id: int = Field(description="ID")
     name: str = Field(description="名称")
     gpu_type: str = Field(alias="gpu_type", description="GPU类型，A800/4090/3090/2080")
@@ -30,6 +33,7 @@ class VirtualCluster(BaseModel):
 
 class Storage(BaseModel):
     """存储"""
+
     id: int = Field(description="存储ID")
     name: str = Field(description="存储名称")
     path: str = Field(description="挂载路径")
@@ -42,12 +46,14 @@ class Storage(BaseModel):
 
 class Category(BaseModel):
     """分类"""
+
     id: int = Field(description="分类ID")
     name: str = Field(description="分类名称")
 
 
 class Project(BaseModel):
     """项目"""
+
     id: int = Field(description="项目ID")
     name: str = Field(description="项目名称")
     description: str = Field(description="项目描述")
@@ -55,6 +61,7 @@ class Project(BaseModel):
 
 class User(BaseModel):
     """用户"""
+
     id: int = Field(description="用户ID")
     name: str = Field(description="用户名")
 
@@ -69,13 +76,15 @@ class TrainingStatus(IntEnum):
 
 
 class PreemptPolicy(IntEnum):
-    """抢占策略：1-等待任务完成；2-停止任务"""
+    """抢占策略：0-非VIP任务；1-等待任务完成；2-停止任务"""
+    NONE = 0  # 非VIP任务
     WAIT = 1  # 等待任务完成
     STOP = 2  # 停止任务
 
 
 class CreateTrainingRequest(BaseModel):
     """创建训练任务请求"""
+
     framework: str = Field(description="训练框架，如PyTorchJob/MpiJobMpiRun/MpiJobDeepspeed")
     name: str = Field(description="任务名称")
     description: Optional[str] = Field(None, description="描述")
@@ -100,11 +109,13 @@ class CreateTrainingRequest(BaseModel):
 
 class CreateTrainingResponse(BaseModel):
     """创建训练任务返回"""
+
     id: int = Field(description="训练任务ID")
 
 
 class ListTrainingsRequest(BaseModel):
     """查询训练任务列表请求"""
+
     page_size: int = Field(20, alias="page_size", description="每页数量")
     page_num: int = Field(1, alias="page_num", description="当前页码")
     user_id: Optional[int] = Field(None, alias="user_id", description="用户过滤")
@@ -120,6 +131,7 @@ class ListTrainingsRequest(BaseModel):
 
 class Training(BaseModel):
     """训练任务详情"""
+
     id: int = Field(description="任务ID")
     framework: str = Field(description="训练框架")
     name: str = Field(description="名称")
@@ -153,11 +165,12 @@ class Training(BaseModel):
     use_new_log: bool = Field(alias="use_new_log", description="是否使用新版日志")
     is_quota_schedule: bool = Field(alias="is_quota_schedule", description="配额调度")
 
-    model_config = {"use_enum_values": True}
+    model_config = ConfigDict(protected_namespaces=(), use_enum_values=True)
 
 
 class ListTrainingsResponse(BaseModel):
     """查询训练任务列表返回"""
+
     total: int = Field(description="总条数")
     page_size: int = Field(alias="page_size", description="每页数量")
     page_num: int = Field(alias="page_num", description="当前页码")
@@ -166,6 +179,7 @@ class ListTrainingsResponse(BaseModel):
 
 class Pod(BaseModel):
     """训练任务Pod"""
+
     id: int = Field(description="ID")
     namespace: str = Field(description="Namespace")
     name: str = Field(description="名称")
@@ -182,12 +196,14 @@ class Pod(BaseModel):
 
 class ListTrainingPodsRequest(BaseModel):
     """查询训练任务Pod列表请求"""
+
     page_size: int = Field(20, alias="page_size", description="每页数量")
     page_num: int = Field(1, alias="page_num", description="当前页码")
 
 
 class ListTrainingPodsResponse(BaseModel):
     """查询训练任务Pod列表返回"""
+
     total: int = Field(description="总条数")
     page_size: int = Field(alias="page_size", description="每页数量")
     page_num: int = Field(alias="page_num", description="当前页码")
@@ -196,39 +212,46 @@ class ListTrainingPodsResponse(BaseModel):
 
 class PodLogInfo(BaseModel):
     """pod日志信息"""
+
     name: str = Field(description="日志名称")
     url: str = Field(description="URL")
 
 
 class GetTrainingPodLogsNewResponse(BaseModel):
     """查询训练任务Pod日志返回"""
+
     logs: List[PodLogInfo] = Field(description="日志列表")
 
 
 class GetTrainingPodSpecResponse(BaseModel):
     """查询训练任务Pod Spec返回"""
+
     spec: str = Field(description="Pod YAML Spec")
 
 
 class GetTrainingPodEventsResponse(BaseModel):
     """查询训练任务Pod Event返回"""
+
     events: str = Field(description="事件内容")
 
 
 class TrainingUser(BaseModel):
     """训练任务用户"""
+
     user_id: int = Field(alias="user_id", description="用户ID")
     username: str = Field(description="用户名")
 
 
 class ListTrainingUsersRequest(BaseModel):
     """查询训练任务用户列表请求"""
+
     page_size: int = Field(20, alias="page_size", description="每页数量")
     page_num: int = Field(1, alias="page_num", description="当前页码")
 
 
 class ListTrainingUsersResponse(BaseModel):
     """查询训练任务用户列表返回"""
+
     total: int = Field(description="总条数")
     page_size: int = Field(alias="page_size", description="每页数量")
     page_num: int = Field(alias="page_num", description="当前页码")
@@ -237,6 +260,7 @@ class ListTrainingUsersResponse(BaseModel):
 
 class Container(BaseModel):
     """容器"""
+
     namespace: str = Field(description="Namespace")
     pod_name: str = Field(alias="pod_name", description="Pod名")
     container_name: str = Field(alias="container_name", description="容器名")
@@ -244,6 +268,7 @@ class Container(BaseModel):
 
 class TrainingContainer(BaseModel):
     """训练容器"""
+
     training_id: int = Field(alias="training_id", description="训练ID")
     training_name: str = Field(alias="training_name", description="训练名")
     containers: List[Container] = Field(description="容器列表")
@@ -251,12 +276,14 @@ class TrainingContainer(BaseModel):
 
 class ListTrainingContainersRequest(BaseModel):
     """查询训练任务容器列表请求"""
+
     page_size: int = Field(20, alias="page_size", description="每页数量")
     page_num: int = Field(1, alias="page_num", description="当前页码")
 
 
 class ListTrainingContainersResponse(BaseModel):
     """查询训练任务容器列表返回"""
+
     total: int = Field(description="总条数")
     page_size: int = Field(alias="page_size", description="每页数量")
     page_num: int = Field(alias="page_num", description="当前页码")

{intellif_aihub-0.1.10 → intellif_aihub-0.1.12}/src/aihub/models/task_center.py

@@ -57,6 +57,8 @@ class TaskDetailOtherInfo(BaseModel):
     label_projects: Optional[List[ProjectInfo]] = Field(alias="label_projects", default=None,
                                                         description="关联标注项目列表")
 
+    model_config = {"use_enum_values": True}
+
 
 class CreateTaskReq(BaseModel):
     """创建标注任务请求"""

{intellif_aihub-0.1.10 → intellif_aihub-0.1.12}/src/aihub/models/workflow_center.py

@@ -329,6 +329,8 @@ class TaskNode(BaseModel):
     virtual_cluster: VirtualCluster = Field(alias="virtual_cluster", description="虚拟集群")
     avg_gpu_util: float = Field(alias="avg_gpu_util", description="平均GPU利用率")
 
+    model_config = {"use_enum_values": True}
+
 
 class Param(BaseModel):
     """参数"""
@@ -375,6 +377,8 @@ class Run(BaseModel):
     avg_gpu_util: float = Field(alias="avg_gpu_util", description="平均GPU利用率")
     total_gpu_time: float = Field(alias="total_gpu_time", description="总的gpu用时")
 
+    model_config = {"use_enum_values": True}
+
 
 class ListRunsRequest(BaseModel):
     """查询运行实例列表请求"""
@@ -384,6 +388,8 @@ class ListRunsRequest(BaseModel):
     user_id: Optional[int] = Field(None, alias="user_id", description="用户过滤")
     status: Optional[RunStatus] = Field(None, description="状态过滤")
 
+    model_config = {"use_enum_values": True}
+
 
 class ListRunsResponse(BaseModel):
     """查询运行实例列表返回"""

{intellif_aihub-0.1.10 → intellif_aihub-0.1.12/src/intellif_aihub.egg-info}/PKG-INFO

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

intellif_aihub-0.1.10/src/aihub/__init__.py

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