intellif-aihub 0.1.10__py3-none-any.whl → 0.1.12__py3-none-any.whl

aihub/__init__.py

@@ -1 +1 @@
-__version__ = "0.1.10"
+__version__ = "0.1.12"

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必须填写")
 

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=())

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="当前页码")

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):
     """创建标注任务请求"""

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.dist-info → intellif_aihub-0.1.12.dist-info}/METADATA

@@ -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.dist-info → intellif_aihub-0.1.12.dist-info}/RECORD

@@ -1,5 +1,5 @@
-aihub/__init__.py,sha256=z0zCHFTcKSR0tJ6h5qrpNmRVP21QIPP8N0p7quCnnm0,23
-aihub/client.py,sha256=bIYL-NoXarSFnPwLzEcWqUjdiTZ7sau5DmGz8YX4-RQ,5144
+aihub/__init__.py,sha256=LcIlFjHZFfiF9Rd4UHoakmombOFkxIYk00I181frGBM,23
+aihub/client.py,sha256=nVELjkyVOG6DKJjurYn59fCoT5JsSayUweiH7bvKcAo,5547
 aihub/exceptions.py,sha256=l2cMAvipTqQOio3o11fXsCCSCevbuK4PTsxofkobFjk,500
 aihub/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aihub/models/artifact.py,sha256=4xIWV3xfZuZWcCdmZGEZ5k_rvV4oc5C_3gapw5O-2vQ,4586
@@ -9,13 +9,13 @@ aihub/models/dataset_management.py,sha256=HmmOW0bA2byQNVavrDG0K5L6pAeGrTFv-ap9pv
 aihub/models/document_center.py,sha256=od9bzx6krAS6ktIA-ChxeqGcch0v2wsS1flY2vuHXBc,1340
 aihub/models/eval.py,sha256=4Gon4Sg4dOkyCx3KH2mO5ip3AhrBwrPC0UZA447HeoQ,910
 aihub/models/labelfree.py,sha256=nljprYO6ECuctTVbHqriQ73N5EEyYURhBrnU28Ngfvc,1589
-aihub/models/model_center.py,sha256=LkvRom4woQ8hYvAwd-LZ15tDRPrWSxvxqzZr_DEoRj0,5765
-aihub/models/model_training_platform.py,sha256=JVX1vsRQmUcJP2SN1S3FEnYvvX1xomlM7BidPOEM_z4,11609
+aihub/models/model_center.py,sha256=DNMchrN0pYDcTMHApWNNVMrARF_i9Ng5xlAwHX5isYw,5935
+aihub/models/model_training_platform.py,sha256=2zir5i-XvuxKKVYr4wuNYUC7nwMzetdtCRoysZ1W_Tc,11725
 aihub/models/quota_schedule_management.py,sha256=LdXwKkpJd0jUFSHtTHUlFLlH-NUSmgywWtxwFg57CNk,12368
 aihub/models/tag_resource_management.py,sha256=-FgiKyDIG7bZagzVRf-8rXWuqH9GyciDadxz5W2f3I8,2195
-aihub/models/task_center.py,sha256=7PlfAsOGBfv_KqfWmPYaWyle4WCKJdc8tTl9aRid_Ws,5502
+aihub/models/task_center.py,sha256=HE21Q4Uj-vt9LHezHnqBYgdinhrh4iJPOq8VXbSMllU,5548
 aihub/models/user_system.py,sha256=0L_pBkWL9v1tv_mclOyRgCyWibuuj_XU3mPoe2v48vQ,12216
-aihub/models/workflow_center.py,sha256=k9VnvNarM9si491LusJCikKMKHNPw8zqMldFYeDgz60,19983
+aihub/models/workflow_center.py,sha256=4xtI1WZ38ceXJ8gwDBj-QNjOiRlLO_8kGiQybdudJPY,20121
 aihub/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aihub/services/artifact.py,sha256=PtCGhYFpFK_hppQr7A0bdXvceXzmYSwEpRj-PE2rIcQ,11473
 aihub/services/data_warehouse.py,sha256=awvlJdggo8ph6sXweXXVp4GLRuUSD46LoD0QQksXRts,2964
@@ -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.10.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-intellif_aihub-0.1.10.dist-info/METADATA,sha256=yTFE22-ezVoAmrLnQ7SDDpNeIIVgKVFmIyATWPzMLoc,2917
-intellif_aihub-0.1.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-intellif_aihub-0.1.10.dist-info/top_level.txt,sha256=vIvTtSIN73xv46BpYM-ctVGnyOiUQ9EWP_6ngvdIlvw,6
-intellif_aihub-0.1.10.dist-info/RECORD,,
+intellif_aihub-0.1.12.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+intellif_aihub-0.1.12.dist-info/METADATA,sha256=LPpYwvlgxhqRiSk6-PD7BEY0F8pVjw-kOZH_oU4Vwug,2917
+intellif_aihub-0.1.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+intellif_aihub-0.1.12.dist-info/top_level.txt,sha256=vIvTtSIN73xv46BpYM-ctVGnyOiUQ9EWP_6ngvdIlvw,6
+intellif_aihub-0.1.12.dist-info/RECORD,,