intellif-aihub 0.1.7__tar.gz → 0.1.9__tar.gz

{intellif_aihub-0.1.7 → intellif_aihub-0.1.9}/PKG-INFO

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

{intellif_aihub-0.1.7 → intellif_aihub-0.1.9}/pyproject.toml

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

intellif_aihub-0.1.9/src/aihub/__init__.py

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

intellif_aihub-0.1.9/src/aihub/models/artifact.py

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

{intellif_aihub-0.1.7 → intellif_aihub-0.1.9}/src/aihub/models/data_warehouse.py

@@ -7,14 +7,14 @@ from pydantic import BaseModel, Field
 
 
 class SearchType(IntEnum):
-    """搜索类型"""
+    """搜索类型：1-SQL；2-图片；3-文本"""
     SQL = 1  # SQL
     IMAGE = 2  # 图片
     TEXT = 3  # 文本
 
 
 class SearchStatus(IntEnum):
-    """搜索任务状态"""
+    """搜索任务状态：1-等待中；2-运行中；3-成功；4-失败"""
     WAITING = 1  # 等待中
     RUNNING = 2  # 运行中
     SUCCESS = 3  # 成功

{intellif_aihub-0.1.7 → intellif_aihub-0.1.9}/src/aihub/models/dataset_management.py

@@ -7,7 +7,7 @@ from pydantic import BaseModel, Field
 
 
 class DatasetVersionStatus(IntEnum):
-    """数据集版本状态"""
+    """数据集版本状态：1-等待中；2-运行中；3-成功；4-失败；5-加载meta；6-构建index"""
     Waiting = 1  # 等待中
     Running = 2  # 运行中
     Success = 3  # 成功
@@ -17,7 +17,7 @@ class DatasetVersionStatus(IntEnum):
 
 
 class UploadType(IntEnum):
-    """上传类型"""
+    """上传类型：1-本地上传；3-服务器路径上传；4-Labelfree；5-数据接入"""
     LOCAL = 1  # 本地上传
     SERVER_PATH = 3  # 服务器路径上传
     LABELFREE = 4  # Labelfree
@@ -28,7 +28,7 @@ class CreateDatasetRequest(BaseModel):
     """创建数据集请求"""
     name: str = Field(description="数据集名称")
     description: str = Field(description="数据集描述")
-    tags: List[int] = Field(description="标签ID列表")
+    tags: List[int] = Field(description="标签ID列表，通过标签管理系统查询")
     cover_img: Optional[str] = Field(None, alias="cover_img", description="封面图片URL")
     create_by: Optional[int] = Field(None, alias="create_by", description="创建人")
     is_private: Optional[bool] = Field(None, alias="is_private", description="是否私有")

{intellif_aihub-0.1.7 → intellif_aihub-0.1.9}/src/aihub/models/document_center.py

@@ -7,7 +7,7 @@ from pydantic import BaseModel, Field
 
 
 class DocumentType(IntEnum):
-    """文档类型"""
+    """文档类型：0-Markdown；1-飞书文档生成的pdf"""
     MARKDOWN = 0  # Markdown
     PDF = 1  # 飞书文档生成的pdf
 

{intellif_aihub-0.1.7 → intellif_aihub-0.1.9}/src/aihub/models/model_center.py

@@ -31,7 +31,7 @@ class QuantLevel(BaseModel):
 
 
 class ModelStatus(IntEnum):
-    """模型状态"""
+    """模型状态：1-Waiting；2-Creating；3-Success；4-Fail"""
     Waiting = 1
     Creating = 2
     Success = 3

{intellif_aihub-0.1.7 → intellif_aihub-0.1.9}/src/aihub/models/model_training_platform.py

@@ -60,7 +60,7 @@ class User(BaseModel):
 
 
 class TrainingStatus(IntEnum):
-    """训练任务状态"""
+    """训练任务状态：1-Waiting；2-Running；3-Success；4-Failed；5-Stopped"""
     Waiting = 1
     Running = 2
     Success = 3
@@ -69,7 +69,7 @@ class TrainingStatus(IntEnum):
 
 
 class PreemptPolicy(IntEnum):
-    """抢占策略"""
+    """抢占策略：1-等待任务完成；2-停止任务"""
     WAIT = 1  # 等待任务完成
     STOP = 2  # 停止任务
 

{intellif_aihub-0.1.7 → intellif_aihub-0.1.9}/src/aihub/models/quota_schedule_management.py

@@ -66,7 +66,7 @@ class SourceTask(BaseModel):
 
 
 class TaskStatus(IntEnum):
-    """任务状态"""
+    """任务状态：1-Waiting；2-Running；3-Success；4-Failed；5-Stopped；6-Stopping"""
     Waiting = 1
     Running = 2
     Success = 3
@@ -76,7 +76,7 @@ class TaskStatus(IntEnum):
 
 
 class PreemptPolicy(IntEnum):
-    """抢占策略"""
+    """抢占策略：1-等待任务完成；2-停止任务"""
     WAIT = 1  # 等待任务完成
     STOP = 2  # 停止任务
 

{intellif_aihub-0.1.7 → intellif_aihub-0.1.9}/src/aihub/models/task_center.py

@@ -83,13 +83,13 @@ 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")
+    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(None, description="预计交付时间")
+    estimated_delivery_at: Optional[int] = Field(default=None, description="预计交付时间")
 
     @field_serializer("other_info")
     def serialize_other_info(self, value: Optional[TaskDetailOtherInfo]) -> Optional[str]:

{intellif_aihub-0.1.7 → intellif_aihub-0.1.9}/src/aihub/models/workflow_center.py

@@ -304,7 +304,7 @@ class IOParam(BaseModel):
 
 
 class TaskNodeStatus(IntEnum):
-    """任务节点状态"""
+    """任务节点状态：1-Waiting；2-Running；3-Success；4-Fail"""
     Waiting = 1
     Running = 2
     Success = 3
@@ -343,7 +343,7 @@ class NodeVirtualCluster(BaseModel):
 
 
 class RunStatus(IntEnum):
-    """运行状态"""
+    """运行状态：1-Waiting；2-Running；3-Success；4-Failed；5-Stopped"""
     Waiting = 1
     Running = 2
     Success = 3

{intellif_aihub-0.1.7 → intellif_aihub-0.1.9}/src/aihub/services/task_center.py

@@ -13,7 +13,6 @@ from __future__ import annotations
 import datetime
 
 import httpx
-from loguru import logger
 
 from .tag_resource_management import TagResourceManagementService
 from .user_system import UserSystemService
@@ -176,12 +175,10 @@ class _TaskCenter:
         self._http = http
 
     def create(self, payload: CreateTaskReq) -> int:
-        logger.debug(f"create task: {payload}")
         resp = self._http.post(
             f"{_BASE}/tasks",
             json=payload.model_dump(),
         )
-        logger.debug(f"create task response: {resp.text}")
         if resp.status_code != 200:
             raise APIError(
                 message="API Error", status=resp.status_code, detail=resp.json()
@@ -191,11 +188,9 @@ class _TaskCenter:
         return wrapper.data.id
 
     def get(self, task_id: int) -> LabelTaskDetail:
-        logger.debug(f"get task: {task_id}")
         resp = self._http.get(
             f"{_BASE}/tasks/{task_id}",
         )
-        logger.debug(f"get task response: {resp.text}")
         if resp.status_code != 200:
             raise APIError(
                 message="API Error", status=resp.status_code, detail=resp.json()

{intellif_aihub-0.1.7 → intellif_aihub-0.1.9}/src/intellif_aihub.egg-info/PKG-INFO

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

intellif_aihub-0.1.7/src/aihub/__init__.py

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

intellif_aihub-0.1.7/src/aihub/models/artifact.py

@@ -1,137 +0,0 @@
-# !/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: Optional[CreateArtifactsResponseData] = 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