intellif-aihub 0.1.18__py3-none-any.whl → 0.1.20__py3-none-any.whl

aihub/__init__.py

@@ -1 +1 @@
-__version__ = "0.1.18"
+__version__ = "0.1.20"

aihub/models/eval.py

@@ -1,6 +1,6 @@
 # !/usr/bin/env python
 # -*-coding:utf-8 -*-
-from typing import Dict
+from typing import Dict, List, Optional
 
 from pydantic import BaseModel, Field
 
@@ -19,8 +19,50 @@ class CreateEvalReq(BaseModel):
 class EvalRun(BaseModel):
     """评测任务的运行实体"""
     id: int = Field(description="评测的运行ID")
+    name: str = Field(description="评测名称")
+    description: str = Field(description="评测描述")
+    user_id: int = Field(description="用户ID")
+    model_id: int = Field(description="模型ID")
+    model_name: str = Field(description="模型名称")
+    dataset_id: int = Field(description="数据集ID")
+    dataset_version_id: int = Field(description="数据集版本ID")
+    dataset_name: str = Field(description="数据集名称")
+    status: str = Field(description="状态")
+    prediction_artifact_path: str = Field(description="推理产物路径")
+    evaled_artifact_path: str = Field(description="评测结果产物路径")
+    run_id: str = Field(description="运行ID")
+    dataset_summary: Dict = Field(default_factory=dict, description="数据集摘要")
+    metrics_summary: Dict = Field(default_factory=dict, description="指标摘要")
+    viz_summary: Dict = Field(default_factory=dict, description="可视化摘要")
+    eval_config: Optional[Dict] = Field(default=None, description="评测配置")
+    created_at: int = Field(description="创建时间")
+    updated_at: int = Field(description="更新时间")
 
 
 class CreateEvalResp(BaseModel):
     """创建评测任务的返回结果"""
     eval_run: EvalRun = Field(alias="eval_run", description="评测运行信息")
+
+
+class ListEvalReq(BaseModel):
+    """列出评测任务请求"""
+    page_size: int = Field(20, description="页面大小")
+    page_num: int = Field(1, description="页码")
+    status: Optional[str] = Field(None, description="状态过滤")
+    name: Optional[str] = Field(None, description="名称过滤")
+    model_id: Optional[int] = Field(None, description="模型ID过滤")
+    dataset_id: Optional[int] = Field(None, description="数据集ID过滤")
+    dataset_version_id: Optional[int] = Field(None, description="数据集版本ID过滤")
+    run_id: Optional[str] = Field(None, description="运行ID过滤")
+    user_id: Optional[int] = Field(None, description="用户ID过滤")
+    model_ids: Optional[str] = Field(None, description="模型ID列表过滤")
+    dataset_ids: Optional[str] = Field(None, description="数据集ID列表过滤")
+    dataset_version_ids: Optional[str] = Field(None, description="数据集版本ID列表过滤")
+
+
+class ListEvalResp(BaseModel):
+    """列出评测任务响应"""
+    total: int = Field(description="总数")
+    page_size: int = Field(description="页面大小")
+    page_num: int = Field(description="页码")
+    data: List[EvalRun] = Field(description="评测运行列表")

aihub/models/labelfree.py

@@ -1,12 +1,29 @@
 from __future__ import annotations
 
+from enum import Enum
 from typing import Optional
 
 from pydantic import BaseModel, Field
 
 
+class LabelProjectStatus(Enum):
+    """标注状态"""
+
+    Pending = "pending"
+    """未开始"""
+    Loading = "loading"
+    """数据读取中"""
+    Error = "failed"
+    """数据读取异常"""
+    In_Progress = "ready"
+    """进行中"""
+    Finished = "finished"
+    """标注完成"""
+
+
 class Stats(BaseModel):
     """标注统计信息"""
+
     total_annotations: int = Field(alias="total_annotations", description="总数据量")
     labeled_annotations: int = Field(alias="labeled_annotations", description="已标注数据量")
     total_labels: int = Field(alias="total_labels", description="总标签量")
@@ -19,9 +36,12 @@ class Stats(BaseModel):
 
 class GetGlobalStatsResponse(BaseModel):
     """标注统计概况"""
+
     global_stats: Stats = Field(alias="global_stats")
     valid_ten_percent: bool = Field(alias="valid_ten_percent", description="是否完成验收10%")
     valid_fifty_percent: bool = Field(alias="valid_fifty_percent", description="是否完成验收50%")
     valid_hundred_percent: bool = Field(alias="valid_hundred_percent", description="是否完成验收100%")
     data_exported_count: int = Field(alias="data_exported_count", description="已导出数据次数")
     exported_dataset_name: str = Field(alias="exported_dataset_name", description="最新数据集名称")
+    status: LabelProjectStatus = Field(description="状态")
+    model_config = {"use_enum_values": True}

aihub/models/task_center.py

@@ -138,8 +138,8 @@ class LabelValidateStage(Enum):
     """10%阶段"""
     FIFTY_PERCENT = "标注阶段（50%）"
     """50%阶段"""
-    HUNDRED_PERCENT = "标注阶段（100%）"
-    """100%阶段（已交付）"""
+    LABEL_FINISHED = "标注阶段（100%）"
+    """标注完成"""
 
 
 class LabelValidateReq(BaseModel):

aihub/services/eval.py

@@ -5,13 +5,14 @@
 本模块围绕 **“模型评测（Run → Report）”** 提供能力：
 
 - **创建评测任务 / 评测报告**
+- **获取评测任务列表**
 """
 
 import httpx
 
 from ..exceptions import APIError
 from ..models.common import APIWrapper
-from ..models.eval import CreateEvalReq, CreateEvalResp
+from ..models.eval import CreateEvalReq, CreateEvalResp, ListEvalReq, ListEvalResp
 
 _BASE = "/eval-platform/api/v1"
 
@@ -62,6 +63,57 @@ class EvalService:
 
         return self._eval.create(payload)
 
+    def list(
+        self,
+        page_size: int = 20,
+        page_num: int = 1,
+        status: str = None,
+        name: str = None,
+        model_id: int = None,
+        dataset_id: int = None,
+        dataset_version_id: int = None,
+        run_id: str = None,
+        user_id: int = None,
+        model_ids: str = None,
+        dataset_ids: str = None,
+        dataset_version_ids: str = None,
+    ) -> ListEvalResp:
+        """列出评测结果
+
+        Args:
+            page_size (int): 页面大小，默认为20
+            page_num (int): 页码，默认为1
+            status (str, optional): 状态过滤
+            name (str, optional): 名称过滤
+            model_id (int, optional): 模型ID过滤
+            dataset_id (int, optional): 数据集ID过滤
+            dataset_version_id (int, optional): 数据集版本ID过滤
+            run_id (str, optional): 运行ID过滤
+            user_id (int, optional): 用户ID过滤
+            model_ids (str, optional): 模型ID列表过滤（逗号分隔）
+            dataset_ids (str, optional): 数据集ID列表过滤（逗号分隔）
+            dataset_version_ids (str, optional): 数据集版本ID列表过滤（逗号分隔）
+
+        Returns:
+            ListEvalResp: 评测结果列表响应
+        """
+        payload = ListEvalReq(
+            page_size=page_size,
+            page_num=page_num,
+            status=status,
+            name=name,
+            model_id=model_id,
+            dataset_id=dataset_id,
+            dataset_version_id=dataset_version_id,
+            run_id=run_id,
+            user_id=user_id,
+            model_ids=model_ids,
+            dataset_ids=dataset_ids,
+            dataset_version_ids=dataset_version_ids,
+        )
+
+        return self._eval.list(payload)
+
 
 class _Eval:
     def __init__(self, http: httpx.Client):
@@ -73,3 +125,16 @@ class _Eval:
         if wrapper.code != 0:
             raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
         return wrapper.data.eval_run.id
+
+    def list(self, payload: ListEvalReq) -> ListEvalResp:
+        # Build query parameters, excluding None values
+        params = {}
+        for field, value in payload.model_dump().items():
+            if value is not None:
+                params[field] = value
+
+        resp = self._http.get(f"{_BASE}/run/", params=params)
+        wrapper = APIWrapper[ListEvalResp].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data

{intellif_aihub-0.1.18.dist-info → intellif_aihub-0.1.20.dist-info}/METADATA

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

{intellif_aihub-0.1.18.dist-info → intellif_aihub-0.1.20.dist-info}/RECORD

@@ -1,4 +1,4 @@
-aihub/__init__.py,sha256=6BiuMUkhwQp6bzUZSF8np8F1NwCltEtK0sPBF__tepU,23
+aihub/__init__.py,sha256=8XalsVoLEfXslFvdtUEmkNOuYShzOzYOcFbgmOz1oSk,23
 aihub/client.py,sha256=Fu3jlEy21T4nJDV5EXTDujy1_B3Pf6CSTyPwkj3PPuE,5574
 aihub/exceptions.py,sha256=l2cMAvipTqQOio3o11fXsCCSCevbuK4PTsxofkobFjk,500
 aihub/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -7,13 +7,13 @@ aihub/models/common.py,sha256=qmabc2LkAdQJXIcpT1P35zxd0Lc8yDYdD4ame1iF4Bs,241
 aihub/models/data_warehouse.py,sha256=zXvWwg7ySoFJMdqQ_1UMTNEKDMhu1hDHlWdBAXdizBk,3905
 aihub/models/dataset_management.py,sha256=4DuQ0zM7jv73SJiqvieHLtn2Y-T6FIFV9r7bgzyCtDo,10790
 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/eval.py,sha256=YFgaRQv8v_mR7i8t6KJ4w8l2TGugsL1BK4VRFr6iqd8,3304
+aihub/models/labelfree.py,sha256=YUnUv0tjYSFAFzYtmbnLOha8rnDe32sb50HkPOclAzU,2016
 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=uxGZ5t2KM4aAu7Z93Zard-gbiDPwJCiezMRgIGd9Rtk,6149
+aihub/models/task_center.py,sha256=rGCvrkDjHLLBQ46DtDA7UAJ4eBQ0jwxpQmSB1z3UHmw,6135
 aihub/models/user_system.py,sha256=58f4w5g_SHUhX4vvJK6RXChCr6IDjX4i957suWYaxg8,12249
 aihub/models/workflow_center.py,sha256=4xtI1WZ38ceXJ8gwDBj-QNjOiRlLO_8kGiQybdudJPY,20121
 aihub/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -21,7 +21,7 @@ aihub/services/artifact.py,sha256=lfOrgOT2AlH1w-75NLcQGOhVWdhmJcWD1gESPpUzqUw,11
 aihub/services/data_warehouse.py,sha256=awvlJdggo8ph6sXweXXVp4GLRuUSD46LoD0QQksXRts,2964
 aihub/services/dataset_management.py,sha256=R7mFsJ1dNOI_p5yNj_rQdLolRC0UKEN4WejE7uOjVlE,21379
 aihub/services/document_center.py,sha256=dG67Ji-DOnzL2t-4x4gVfMt9fbSj_IjVHCLw5R-VTkQ,1813
-aihub/services/eval.py,sha256=V1nBISIyYWg9JJO24xzy4-kit9NsaCYp1EWIX_fgJkQ,2128
+aihub/services/eval.py,sha256=ryXjbC2eCTTU4ot3vjJ4_HhGldjCuZ0LS4OYVwWaCoc,4505
 aihub/services/labelfree.py,sha256=xua62UWhVXTxJjHRyy86waaormnJjmpQwepcARBy_h0,1450
 aihub/services/model_center.py,sha256=QZXlldPZrbC6YkVG3eGw5_53qvFAim2QlXg3DflByTA,6215
 aihub/services/model_training_platform.py,sha256=38o6HJnyi3htFzpX7qj6UhzdqTchcXLRTYU0nM7ffJg,10176
@@ -36,8 +36,8 @@ aihub/utils/di.py,sha256=vFUzno5WbRKu6-pj8Hnz9IqT7xb9UDZQ4qpOFH1YAtM,11812
 aihub/utils/download.py,sha256=ZZVbcC-PnN3PumV7ZiJ_-srkt4HPPovu2F6Faa2RrPE,1830
 aihub/utils/http.py,sha256=AmfHHNjptuuSFx2T1twWCnerR_hLN_gd0lUs8z36ERA,547
 aihub/utils/s3.py,sha256=ISIBP-XdBPkURpXnN56ZnIWokOOg2SRUh_qvxJk-G1Q,2187
-intellif_aihub-0.1.18.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-intellif_aihub-0.1.18.dist-info/METADATA,sha256=XQqYqji6SA8pj_BondfV-zZLv5JjqgMqAJjc1l5FOuU,2949
-intellif_aihub-0.1.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-intellif_aihub-0.1.18.dist-info/top_level.txt,sha256=vIvTtSIN73xv46BpYM-ctVGnyOiUQ9EWP_6ngvdIlvw,6
-intellif_aihub-0.1.18.dist-info/RECORD,,
+intellif_aihub-0.1.20.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+intellif_aihub-0.1.20.dist-info/METADATA,sha256=Jr4_my80mlDP_JR7KyR6Ayu5dzBI1WZ_quXnBfbfDAA,2949
+intellif_aihub-0.1.20.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+intellif_aihub-0.1.20.dist-info/top_level.txt,sha256=vIvTtSIN73xv46BpYM-ctVGnyOiUQ9EWP_6ngvdIlvw,6
+intellif_aihub-0.1.20.dist-info/RECORD,,