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

aihub/__init__.py

@@ -1 +1 @@
-__version__ = "0.1.3"
+__version__ = "0.1.4"

aihub/client.py

@@ -1,91 +1,95 @@
-from __future__ import annotations
-
-import os
-
-import httpx
-
-from .exceptions import APIError
-from .services import artifact
-from .services import dataset_management
-from .services import document_center
-from .services import labelfree
-from .services import model_training_platform
-from .services import tag_resource_management
-from .services import task_center
-from .services import user_system
-from .services.artifact import ArtifactService
-from .services.dataset_management import DatasetManagementService
-from .services.document_center import DocumentCenterService
-from .services.labelfree import LabelfreeService
-from .services.model_training_platform import ModelTrainingPlatformService
-from .services.tag_resource_management import TagResourceManagementService
-from .services.task_center import TaskCenterService
-from .services.user_system import UserSystemService
-
-
-class Client:
-    """AI-HUB python SDK 客户端
-
-    Attributes:
-        dataset_management (DatasetManagementService): 数据集管理服务
-        labelfree (LabelfreeService): 标注服务
-        task_center (TaskCenterService): 任务中心
-        artifact (ArtifactService): 制品管理
-
-    """
-
-    labelfree: LabelfreeService = None
-    tag_resource_management: TagResourceManagementService = None
-    document_center: DocumentCenterService = None
-    task_center: TaskCenterService = None
-    dataset_management: DatasetManagementService = None
-    artifact: ArtifactService = None
-    user_system: UserSystemService = None
-    model_training_platform: ModelTrainingPlatformService = None
-
-    def __init__(self, *, base_url: str, token: str | None = None, timeout: float = 60.0):
-        """AI-HUB python SDK 客户端
-
-        Args:
-            base_url (str): 服务地址
-            token (str): 密钥，显式传入，或在环境变量AI_HUB_TOKEN中设置
-
-        Examples:
-            >>> from aihub.client import Client
-            >>> client = Client(base_url="xxx", token="xxxx")
-
-        """
-        if not base_url:
-            raise ValueError("base_url必须填写")
-
-        token = os.getenv("AI_HUB_TOKEN") or token
-        if not token:
-            raise ValueError("缺少token：请显式传入，或在环境变量AI_HUB_TOKEN中设置")
-
-        self._http = httpx.Client(
-            base_url=base_url.rstrip("/"),
-            timeout=timeout,
-            headers={"Authorization": f"Bearer {token}"},
-            # event_hooks={"response": [self._raise_for_status]},
-        )
-        self.dataset_management = dataset_management.DatasetManagementService(self._http)
-        self.labelfree = labelfree.LabelfreeService(self._http)
-        self.tag_resource_management = tag_resource_management.TagResourceManagementService(self._http)
-        self.document_center = document_center.DocumentCenterService(self._http)
-        self.task_center = task_center.TaskCenterService(self._http)
-        self.artifact = artifact.ArtifactService(self._http)
-        self.user_system = user_system.UserSystemService(self._http)
-        self.model_training_platform = model_training_platform.ModelTrainingPlatformService(self._http)
-
-    @staticmethod
-    def _raise_for_status(r: httpx.Response):
-        try:
-            r.raise_for_status()
-        except httpx.HTTPStatusError as e:
-            raise APIError(f"{e.response.status_code}: {e.response.text}") from e
-
-    def __enter__(self):
-        return self
-
-    def __exit__(self, *a):
-        self._http.close()
+from __future__ import annotations
+
+import os
+
+import httpx
+
+from .exceptions import APIError
+from .services import artifact
+from .services import dataset_management
+from .services import document_center
+from .services import labelfree
+from .services import model_training_platform
+from .services import quota_schedule_management
+from .services import tag_resource_management
+from .services import task_center
+from .services import user_system
+from .services.artifact import ArtifactService
+from .services.dataset_management import DatasetManagementService
+from .services.document_center import DocumentCenterService
+from .services.labelfree import LabelfreeService
+from .services.model_training_platform import ModelTrainingPlatformService
+from .services.quota_schedule_management import QuotaScheduleManagementService
+from .services.tag_resource_management import TagResourceManagementService
+from .services.task_center import TaskCenterService
+from .services.user_system import UserSystemService
+
+
+class Client:
+    """AI-HUB python SDK 客户端
+
+    Attributes:
+        dataset_management (DatasetManagementService): 数据集管理服务
+        labelfree (LabelfreeService): 标注服务
+        task_center (TaskCenterService): 任务中心
+        artifact (ArtifactService): 制品管理
+
+    """
+
+    labelfree: LabelfreeService = None
+    tag_resource_management: TagResourceManagementService = None
+    document_center: DocumentCenterService = None
+    task_center: TaskCenterService = None
+    dataset_management: DatasetManagementService = None
+    artifact: ArtifactService = None
+    user_system: UserSystemService = None
+    model_training_platform: ModelTrainingPlatformService = None
+    quota_schedule_management: QuotaScheduleManagementService = None
+
+    def __init__(self, *, base_url: str, token: str | None = None, timeout: float = 60.0):
+        """AI-HUB python SDK 客户端
+
+        Args:
+            base_url (str): 服务地址
+            token (str): 密钥，显式传入，或在环境变量AI_HUB_TOKEN中设置
+
+        Examples:
+            >>> from aihub.client import Client
+            >>> client = Client(base_url="xxx", token="xxxx")
+
+        """
+        if not base_url:
+            raise ValueError("base_url必须填写")
+
+        token = os.getenv("AI_HUB_TOKEN") or token
+        if not token:
+            raise ValueError("缺少token：请显式传入，或在环境变量AI_HUB_TOKEN中设置")
+
+        self._http = httpx.Client(
+            base_url=base_url.rstrip("/"),
+            timeout=timeout,
+            headers={"Authorization": f"Bearer {token}"},
+            # event_hooks={"response": [self._raise_for_status]},
+        )
+        self.dataset_management = dataset_management.DatasetManagementService(self._http)
+        self.labelfree = labelfree.LabelfreeService(self._http)
+        self.tag_resource_management = tag_resource_management.TagResourceManagementService(self._http)
+        self.document_center = document_center.DocumentCenterService(self._http)
+        self.task_center = task_center.TaskCenterService(self._http)
+        self.artifact = artifact.ArtifactService(self._http)
+        self.user_system = user_system.UserSystemService(self._http)
+        self.model_training_platform = model_training_platform.ModelTrainingPlatformService(self._http)
+        self.quota_schedule_management = quota_schedule_management.QuotaScheduleManagementService(self._http)
+
+    @staticmethod
+    def _raise_for_status(r: httpx.Response):
+        try:
+            r.raise_for_status()
+        except httpx.HTTPStatusError as e:
+            raise APIError(f"{e.response.status_code}: {e.response.text}") from e
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, *a):
+        self._http.close()

aihub/exceptions.py

@@ -1,18 +1,18 @@
-from __future__ import annotations
-
-from typing import Any, Optional
-
-
-class SDKError(Exception):
-    pass
-
-
-class APIError(SDKError):
-    def __init__(self, message: str, *, status: Optional[int] = None, detail: Any = None) -> None:
-        super().__init__(message)
-        self.message: str = message
-        self.status: Optional[int] = status
-        self.detail: Any = detail
-
-    def __str__(self) -> str:
-        return f"[HTTP {self.status}] {self.message}" if self.status else self.message
+from __future__ import annotations
+
+from typing import Any, Optional
+
+
+class SDKError(Exception):
+    pass
+
+
+class APIError(SDKError):
+    def __init__(self, message: str, *, status: Optional[int] = None, detail: Any = None) -> None:
+        super().__init__(message)
+        self.message: str = message
+        self.status: Optional[int] = status
+        self.detail: Any = detail
+
+    def __str__(self) -> str:
+        return f"[HTTP {self.status}] {self.message}" if self.status else self.message

aihub/models/artifact.py

@@ -1,137 +1,137 @@
-# !/usr/bin/env python
-# -*-coding:utf-8 -*-
-"""制品管理模型模块
-
-该模块定义了制品管理相关的数据模型，包括制品类型、创建制品请求、制品响应等。
-"""
-from __future__ import annotations
-
-from enum import Enum
-from typing import List, Optional
-
-from pydantic import BaseModel
-
-
-class ArtifactType(str, Enum):
-    """制品类型枚举
-
-    定义了系统支持的各种制品类型。
-    """
-
-    dataset = "dataset"  # 数据集类型
-    model = "model"  # 模型类型
-    metrics = "metrics"  # 指标类型
-    log = "log"  # 日志类型
-    checkpoint = "checkpoint"  # 检查点类型
-    image = "image"  # 图像类型
-    prediction = "prediction"  # 预测结果类型
-    other = "other"  # 其他类型
-
-
-class CreateArtifactsReq(BaseModel):
-    """创建制品请求模型
-
-    用于向服务器发送创建制品的请求。
-    """
-
-    entity_id: str
-    """实体ID，通常是运行ID，用于关联制品与特定运行"""
-
-    entity_type: ArtifactType = ArtifactType.other
-    """制品类型，指定制品的类型，默认为other"""
-
-    src_path: str
-    """源路径，制品在系统中的路径标识"""
-
-    is_dir: bool = False
-    """是否为目录，True表示制品是一个目录，False表示是单个文件"""
-
-
-class CreateArtifactsResponseData(BaseModel):
-    """创建制品响应数据模型
-
-    服务器创建制品后返回的数据。
-    """
-
-    id: int  # 制品ID
-    s3_path: str  # S3存储路径
-
-
-class CreateArtifactsResponseModel(BaseModel):
-    """创建制品响应模型
-
-    服务器对创建制品请求的完整响应。
-    """
-
-    code: int  # 响应码，0表示成功
-    msg: str = ""  # 响应消息
-    data: CreateArtifactsResponseData | None = None  # 响应数据
-
-
-class CreatEvalReq(BaseModel):
-    """创建评估请求模型
-
-    用于创建模型评估的请求。
-    """
-
-    dataset_id: int  # 数据集ID
-    dataset_version_id: int  # 数据集版本ID
-    prediction_artifact_path: str  # 预测结果制品路径
-    evaled_artifact_path: str  # 评估结果制品路径
-    run_id: str  # 运行ID
-    user_id: int  # 用户ID
-    report: dict = {}  # 评估报告
-
-
-class ArtifactResp(BaseModel):
-    """制品响应模型
-
-    表示一个制品的详细信息。
-    """
-
-    id: int  # 制品ID
-    entity_type: str  # 实体类型
-    entity_id: str  # 实体ID
-    src_path: str  # 源路径
-    s3_path: str  # S3存储路径
-    is_dir: bool  # 是否为目录
-
-
-class ArtifactRespData(BaseModel):
-    """制品响应数据模型
-
-    包含分页信息和制品列表的响应数据。
-    """
-
-    total: int  # 总记录数
-    page_size: int  # 每页大小
-    page_num: int  # 页码
-    data: List[ArtifactResp]  # 制品列表
-
-
-class ArtifactRespModel(BaseModel):
-    """制品响应模型
-
-    服务器对获取制品请求的完整响应。
-    """
-
-    code: int  # 响应码，0表示成功
-    msg: str = ""  # 响应消息
-    data: ArtifactRespData  # 响应数据
-
-
-# 无限大的页面大小，用于一次性获取所有制品
-InfinityPageSize = 10000 * 100
-
-
-class StsResp(BaseModel):
-    """STS临时凭证响应模型
-
-    包含访问S3存储所需的临时凭证信息。
-    """
-
-    access_key_id: Optional[str] = None  # 访问密钥ID
-    secret_access_key: Optional[str] = None  # 秘密访问密钥
-    session_token: Optional[str] = None  # 会话令牌
-    expiration: Optional[int] = None  # 过期时间
-    endpoint: Optional[str] = None  # 端点URL
+# !/usr/bin/env python
+# -*-coding:utf-8 -*-
+"""制品管理模型模块
+
+该模块定义了制品管理相关的数据模型，包括制品类型、创建制品请求、制品响应等。
+"""
+from __future__ import annotations
+
+from enum import Enum
+from typing import List, Optional
+
+from pydantic import BaseModel
+
+
+class ArtifactType(str, Enum):
+    """制品类型枚举
+
+    定义了系统支持的各种制品类型。
+    """
+
+    dataset = "dataset"  # 数据集类型
+    model = "model"  # 模型类型
+    metrics = "metrics"  # 指标类型
+    log = "log"  # 日志类型
+    checkpoint = "checkpoint"  # 检查点类型
+    image = "image"  # 图像类型
+    prediction = "prediction"  # 预测结果类型
+    other = "other"  # 其他类型
+
+
+class CreateArtifactsReq(BaseModel):
+    """创建制品请求模型
+
+    用于向服务器发送创建制品的请求。
+    """
+
+    entity_id: str
+    """实体ID，通常是运行ID，用于关联制品与特定运行"""
+
+    entity_type: ArtifactType = ArtifactType.other
+    """制品类型，指定制品的类型，默认为other"""
+
+    src_path: str
+    """源路径，制品在系统中的路径标识"""
+
+    is_dir: bool = False
+    """是否为目录，True表示制品是一个目录，False表示是单个文件"""
+
+
+class CreateArtifactsResponseData(BaseModel):
+    """创建制品响应数据模型
+
+    服务器创建制品后返回的数据。
+    """
+
+    id: int  # 制品ID
+    s3_path: str  # S3存储路径
+
+
+class CreateArtifactsResponseModel(BaseModel):
+    """创建制品响应模型
+
+    服务器对创建制品请求的完整响应。
+    """
+
+    code: int  # 响应码，0表示成功
+    msg: str = ""  # 响应消息
+    data: CreateArtifactsResponseData | None = None  # 响应数据
+
+
+class CreatEvalReq(BaseModel):
+    """创建评估请求模型
+
+    用于创建模型评估的请求。
+    """
+
+    dataset_id: int  # 数据集ID
+    dataset_version_id: int  # 数据集版本ID
+    prediction_artifact_path: str  # 预测结果制品路径
+    evaled_artifact_path: str  # 评估结果制品路径
+    run_id: str  # 运行ID
+    user_id: int  # 用户ID
+    report: dict = {}  # 评估报告
+
+
+class ArtifactResp(BaseModel):
+    """制品响应模型
+
+    表示一个制品的详细信息。
+    """
+
+    id: int  # 制品ID
+    entity_type: str  # 实体类型
+    entity_id: str  # 实体ID
+    src_path: str  # 源路径
+    s3_path: str  # S3存储路径
+    is_dir: bool  # 是否为目录
+
+
+class ArtifactRespData(BaseModel):
+    """制品响应数据模型
+
+    包含分页信息和制品列表的响应数据。
+    """
+
+    total: int  # 总记录数
+    page_size: int  # 每页大小
+    page_num: int  # 页码
+    data: List[ArtifactResp]  # 制品列表
+
+
+class ArtifactRespModel(BaseModel):
+    """制品响应模型
+
+    服务器对获取制品请求的完整响应。
+    """
+
+    code: int  # 响应码，0表示成功
+    msg: str = ""  # 响应消息
+    data: ArtifactRespData  # 响应数据
+
+
+# 无限大的页面大小，用于一次性获取所有制品
+InfinityPageSize = 10000 * 100
+
+
+class StsResp(BaseModel):
+    """STS临时凭证响应模型
+
+    包含访问S3存储所需的临时凭证信息。
+    """
+
+    access_key_id: Optional[str] = None  # 访问密钥ID
+    secret_access_key: Optional[str] = None  # 秘密访问密钥
+    session_token: Optional[str] = None  # 会话令牌
+    expiration: Optional[int] = None  # 过期时间
+    endpoint: Optional[str] = None  # 端点URL

aihub/models/common.py

@@ -1,13 +1,13 @@
-from __future__ import annotations
-
-from typing import Generic, Optional, TypeVar
-
-from pydantic import BaseModel
-
-T = TypeVar("T")
-
-
-class APIWrapper(BaseModel, Generic[T]):
-    code: int
-    msg: Optional[str] = None
-    data: Optional[T]
+from __future__ import annotations
+
+from typing import Generic, Optional, TypeVar
+
+from pydantic import BaseModel
+
+T = TypeVar("T")
+
+
+class APIWrapper(BaseModel, Generic[T]):
+    code: int
+    msg: Optional[str] = None
+    data: Optional[T]