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

aihub/__init__.py

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

aihub/models/eval.py

@@ -1,23 +1,54 @@
 # !/usr/bin/env python
 # -*-coding:utf-8 -*-
+from enum import Enum
 from typing import Dict, List, Optional
 
 from pydantic import BaseModel, Field
 
 
-class CreateEvalReq(BaseModel):
-    """创建评测任务"""
+class BaseEvalReq(BaseModel):
+    """评测任务基础请求模型"""
+
+    run_id: str = Field(description="运行ID")
+    type: str = Field(description="评测类型，支持 'llm' 和 'cv'")
+    prediction_artifact_path: str = Field(description="推理产物的路径")
+    user_id: int = Field(0, description="用户ID，默认0")
+
+
+class ClientType(Enum):
+    """客户端类型枚举"""
+
+    Workflow = "workflow"
+    sdk = "sdk"
+
+
+class CreateLLMEvalReq(BaseEvalReq):
+    """创建LLM类型评测任务请求"""
+
+    type: str = Field(default="llm", description="评测类型，固定为 'llm'")
     dataset_id: int = Field(description="数据集ID")
     dataset_version_id: int = Field(description="数据集版本ID")
-    prediction_artifact_path: str = Field(description="推理产物的路径")
     evaled_artifact_path: str = Field(description="评测结果产物的路径")
-    run_id: str = Field(description="运行ID")
-    user_id: int = Field(0, description="用户ID")
-    report: Dict = Field(default_factory=dict, description="评测报告")
+    report: Dict = Field(description="评测报告")
+    is_public: bool = Field(default=False, description="是否公开")
+    client_type: ClientType = Field(default=ClientType.Workflow, description="客户端类型")
+    model_config = {"use_enum_values": True}
+
+
+class CreateCVEvalReq(BaseEvalReq):
+    """创建CV类型评测任务请求"""
+
+    type: str = Field(default="cv", description="评测类型，固定为 'cv'")
+    metrics_artifact_path: str = Field(description="指标产物的路径")
+    ground_truth_artifact_path: str = Field(description="真实标签产物的路径")
+    is_public: bool = Field(default=False, description="是否公开")
+    client_type: ClientType = Field(default=ClientType.Workflow, description="客户端类型")
+    model_config = {"use_enum_values": True}
 
 
 class EvalRun(BaseModel):
     """评测任务的运行实体"""
+
     id: int = Field(description="评测的运行ID")
     name: str = Field(description="评测名称")
     description: str = Field(description="评测描述")
@@ -37,15 +68,20 @@ class EvalRun(BaseModel):
     eval_config: Optional[Dict] = Field(default=None, description="评测配置")
     created_at: int = Field(description="创建时间")
     updated_at: int = Field(description="更新时间")
+    is_public: bool = Field(default=False, description="是否公开")
+    client_type: ClientType = Field(default=ClientType.Workflow, description="客户端类型")
+    model_config = {"use_enum_values": True, "protected_namespaces": ()}
 
 
 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="状态过滤")
@@ -58,11 +94,19 @@ class ListEvalReq(BaseModel):
     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列表过滤")
+    model_config = {"use_enum_values": True, "protected_namespaces": ()}
 
 
 class ListEvalResp(BaseModel):
     """列出评测任务响应"""
+
     total: int = Field(description="总数")
     page_size: int = Field(description="页面大小")
     page_num: int = Field(description="页码")
     data: List[EvalRun] = Field(description="评测运行列表")
+
+
+class GrantPermissionReq(BaseModel):
+    """授权权限请求"""
+
+    user_ids: list[int] = Field(description="用户ID数组")

aihub/models/task_center.py

@@ -42,6 +42,7 @@ class CreateTaskOtherInfo(BaseModel):
     doc_type: str = Field(alias="doc_type", default="doc_center", description="文档类型")
 
     model_config = {"use_enum_values": True}
+    auto_valid_interval: int = Field(3, description="自动验收间隔")
 
 
 class ProjectInfo(BaseModel):

aihub/models/user_system.py

@@ -328,7 +328,7 @@ class SearchUsersResponse(BaseModel):
     total: int = Field(description="总数")
     page_size: int = Field(alias="page_size", description="单页条数")
     page_num: int = Field(alias="page_num", description="当前页")
-    data: List[User] = Field(description="用户列表")
+    data: Optional[List[User]] = Field(description="用户列表")
 
 
 # 此行放在文件末尾，否则序列化报错

aihub/services/eval.py

@@ -7,12 +7,21 @@
 - **创建评测任务 / 评测报告**
 - **获取评测任务列表**
 """
+from typing import List
 
 import httpx
 
 from ..exceptions import APIError
 from ..models.common import APIWrapper
-from ..models.eval import CreateEvalReq, CreateEvalResp, ListEvalReq, ListEvalResp
+from ..models.eval import (
+    CreateLLMEvalReq,
+    CreateCVEvalReq,
+    CreateEvalResp,
+    ListEvalReq,
+    ListEvalResp,
+    GrantPermissionReq,
+    ClientType,
+)
 
 _BASE = "/eval-platform/api/v1"
 
@@ -31,16 +40,21 @@ class EvalService:
         evaled_artifact_path: str,
         report_json: dict,
         run_id,
+        user_id: int = 0,
+        is_public: bool = False,
+        access_user_ids: List[int] = None,
     ) -> int:
         """创建评测报告
 
         Args:
+            is_public (bool): 是否公开
             run_id (str): RUN ID
             report_json (dict): 报告内容
             evaled_artifact_path:   评测结果制品路径
             prediction_artifact_path: 推理结果制品路径
             dataset_version_name (str): 数据集名称
-
+            user_id (int, optional): 用户ID，默认为0
+            access_user_ids (list): 授权访问的用户id
 
         Returns:
             id (int): 评测报告id
@@ -49,19 +63,65 @@ class EvalService:
         from .dataset_management import DatasetManagementService
 
         dataset_service = DatasetManagementService(self._http)
-        dataset_version = dataset_service.get_dataset_version_by_name(
-            dataset_version_name
-        )
-        payload = CreateEvalReq(
+        dataset_version = dataset_service.get_dataset_version_by_name(dataset_version_name)
+        payload = CreateLLMEvalReq(
             dataset_id=dataset_version.dataset_id,
             dataset_version_id=dataset_version.id,
             evaled_artifact_path=evaled_artifact_path,
             prediction_artifact_path=prediction_artifact_path,
             report=report_json,
             run_id=run_id,
+            user_id=user_id,
+            is_public=is_public,
+            type="llm",
+            client_type=ClientType.Workflow,
         )
+        resp = self._eval.create(payload)
+        if is_public is False and access_user_ids:
+            self.grant_permission(user_ids=access_user_ids, run_id=resp)
+
+        return resp
 
-        return self._eval.create(payload)
+    def create_cv_run(
+        self,
+        run_id: str,
+        prediction_artifact_path: str,
+        metrics_artifact_path: str,
+        ground_truth_artifact_path: str,
+        user_id: int = 0,
+        is_public: bool = False,
+        access_user_ids: List[int] = None,
+    ) -> int:
+        """创建 CV 类型评测运行
+
+        Args:
+            access_user_ids: 授权访问的用户ID
+            is_public (bool): 是否公开
+            run_id (str): 运行ID
+            prediction_artifact_path (str): 推理产物的路径
+            metrics_artifact_path (str): 指标产物的路径
+            ground_truth_artifact_path (str): 真实标签产物的路径
+            user_id (int, optional): 用户ID，默认为0
+            access_user_ids (list): 授权访问的用户id
+
+        Returns:
+            id (int): 评测运行id
+        """
+        payload = CreateCVEvalReq(
+            run_id=run_id,
+            prediction_artifact_path=prediction_artifact_path,
+            metrics_artifact_path=metrics_artifact_path,
+            ground_truth_artifact_path=ground_truth_artifact_path,
+            user_id=user_id,
+            is_public=is_public,
+            type="cv",
+            client_type=ClientType.Workflow,
+        )
+        resp = self._eval.create(payload)
+        if is_public is False and access_user_ids:
+            self.grant_permission(user_ids=access_user_ids, run_id=resp)
+
+        return resp
 
     def list(
         self,
@@ -114,18 +174,40 @@ class EvalService:
 
         return self._eval.list(payload)
 
+    def grant_permission(self, user_ids: List[int], run_id: int):
+        """授权访问
+
+        Args:
+            user_ids (list): 授权信息
+            run_id (int): 任务ID
+
+        Returns:
+            dict: 授权信息
+        """
+        req = GrantPermissionReq(
+            user_ids=user_ids,
+        )
+        return self._eval.grant_permission(req, run_id)
+
 
 class _Eval:
     def __init__(self, http: httpx.Client):
         self._http = http
 
-    def create(self, payload: CreateEvalReq) -> int:
+    def create(self, payload) -> int:
         resp = self._http.post(f"{_BASE}/run/", json=payload.model_dump())
         wrapper = APIWrapper[CreateEvalResp].model_validate(resp.json())
         if wrapper.code != 0:
             raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
         return wrapper.data.eval_run.id
 
+    def grant_permission(self, payload, task_id):
+        resp = self._http.post(f"{_BASE}/run/{task_id}/permissions", json=payload.model_dump())
+        wrapper = APIWrapper[CreateEvalResp].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data
+
     def list(self, payload: ListEvalReq) -> ListEvalResp:
         # Build query parameters, excluding None values
         params = {}

aihub/services/task_center.py

@@ -84,6 +84,7 @@ class TaskCenterService:
         label_type: LabelProjectTypeEnum = LabelProjectTypeEnum.IMAGE_CLASSIFICATION,
         description: str = "",
         task_priority: TaskCenterPriorityEnum = TaskCenterPriorityEnum.low,
+        auto_valid_interval: int = 3,
     ) -> int:
         """创建标注任务
 
@@ -107,6 +108,7 @@ task_priority="low", estimated_delivery_at= "2025-08-01")
             label_type (LabelProjectTypeEnum): 标注项目类型，默认为图像分类
             description (str): 任务描述，默认为空
             task_priority (TaskCenterPriorityEnum): 任务优先级，默认为低优先级
+            auto_valid_interval(str): 标注自动验收时间（默认三天）
         Returns:
             任务ID
         """
@@ -153,6 +155,7 @@ task_priority="low", estimated_delivery_at= "2025-08-01")
             dataset_version_id=dataset_version_id,
             doc_id=doc_id,
             doc_type="doc_center",
+            auto_valid_interval=auto_valid_interval,
         )
         estimated_delivery_at_timestamp = date_str_to_timestamp(estimated_delivery_at)
         task_req = CreateTaskReq(

aihub/services/user_system.py

@@ -290,9 +290,9 @@ class UserSystemService:
     def delete_user(self, user_id: int) -> None:
         """删除用户
 
-         Args:
-             user_id: 目标用户 ID
-         """
+        Args:
+            user_id: 目标用户 ID
+        """
         self._user.delete(user_id)
 
     def set_user_roles(self, user_id: int, payload: SetUserRolesRequest) -> None:
@@ -304,7 +304,12 @@ class UserSystemService:
         """
         self._user.set_roles(user_id, payload)
 
-    def get_user_menus(self, user_id: int, parent_id: int | None = None, auth: str | None = None, ) -> List[TreeMenu]:
+    def get_user_menus(
+        self,
+        user_id: int,
+        parent_id: int | None = None,
+        auth: str | None = None,
+    ) -> List[TreeMenu]:
         """查询用户可见菜单
 
         Args:
@@ -317,7 +322,7 @@ class UserSystemService:
         """
         return self._user.get_menus(user_id, parent_id=parent_id, auth=auth)
 
-    def search_users(self, payload: SearchUsersRequest) -> SearchUsersResponse:
+    def search_users(self, payload: SearchUsersRequest) -> List[User]:
         """条件搜索用户
 
         Args:
@@ -326,13 +331,15 @@ class UserSystemService:
         Returns:
             SearchUsersResponse: 搜索结果
         """
-        return self._user.search(payload)
+        resp = self._user.search(payload)
+        return resp.data
 
-    def search_one(self, payload: SearchUsersRequest) -> int:
+    def search_one(self, nickname: str) -> int:
         """搜索单个用户并返回其 ID
 
         Args:
-            payload: 搜索条件
+            nickname: 用户昵称
+
 
         Returns:
             int: 命中的用户 ID
@@ -340,7 +347,10 @@ class UserSystemService:
         Raises:
             APIError: 未找到用户时抛出
         """
-        return self._user.search_one(payload)
+        req = SearchUsersRequest(
+            nickname=nickname,
+        )
+        return self._user.search_one(req)
 
     @property
     def auth(self) -> _Auth:
@@ -568,6 +578,13 @@ class _User:
 
     def search_one(self, req: SearchUsersRequest) -> int:
         resp = self.search(req)
+
+        if resp.data is None:
+            raise APIError(f"no user found")
+
+        if len(resp.data) > 1:
+            raise APIError("more than one user found")
+
         for user in resp.data:
             if user.nickname == req.nickname:
                 return user.id

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

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

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

@@ -1,4 +1,4 @@
-aihub/__init__.py,sha256=8XalsVoLEfXslFvdtUEmkNOuYShzOzYOcFbgmOz1oSk,23
+aihub/__init__.py,sha256=zmP2TRnzKPjZJ1eiBcT-cRInsji6FW-OVD3FafQFCc4,23
 aihub/client.py,sha256=Fu3jlEy21T4nJDV5EXTDujy1_B3Pf6CSTyPwkj3PPuE,5574
 aihub/exceptions.py,sha256=l2cMAvipTqQOio3o11fXsCCSCevbuK4PTsxofkobFjk,500
 aihub/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -7,37 +7,37 @@ 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=YFgaRQv8v_mR7i8t6KJ4w8l2TGugsL1BK4VRFr6iqd8,3304
+aihub/models/eval.py,sha256=eebUv31GJ_gf7-vCYCLknN6utsPZw9G3t6ikZb9gK0I,4865
 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=rGCvrkDjHLLBQ46DtDA7UAJ4eBQ0jwxpQmSB1z3UHmw,6135
-aihub/models/user_system.py,sha256=58f4w5g_SHUhX4vvJK6RXChCr6IDjX4i957suWYaxg8,12249
+aihub/models/task_center.py,sha256=D4cc3fd4zOhQH5X8VdQmdHckrnYjBGhjg55dBaJkous,6209
+aihub/models/user_system.py,sha256=__JLHIUFCi2Ke3Wg7v0G5hqE2WoNJNTG4Gsj5E6vv3o,12259
 aihub/models/workflow_center.py,sha256=4xtI1WZ38ceXJ8gwDBj-QNjOiRlLO_8kGiQybdudJPY,20121
 aihub/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aihub/services/artifact.py,sha256=lfOrgOT2AlH1w-75NLcQGOhVWdhmJcWD1gESPpUzqUw,11257
 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=ryXjbC2eCTTU4ot3vjJ4_HhGldjCuZ0LS4OYVwWaCoc,4505
+aihub/services/eval.py,sha256=n1ytGzPJz-bwRyHoXaAugh7-u7Dak4aMYWklX3X_Pso,7287
 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
 aihub/services/quota_schedule_management.py,sha256=UYOMwjXxJTgkpN6Rv5GzlcejtpZfu23PXlSKr0WihTY,9586
 aihub/services/reporter.py,sha256=ot93SmhxgwDJOzlHSCwlxDOuSydTWUEUQ-Ctp97wJBQ,669
 aihub/services/tag_resource_management.py,sha256=Bm_inSIzZbTc-e4LU9kvwtsPpM_yLwm8xzdrALjb6uY,2666
-aihub/services/task_center.py,sha256=ljy4lhyqVe2lnytp1h72mliqXUbSvEELT9KmxxPKdKc,8003
-aihub/services/user_system.py,sha256=IqWL4bnsKyyzuGT5l6adnw0qNXlH9PSo1-C_pFyOSzA,18868
+aihub/services/task_center.py,sha256=UJXSo-8wjGvxAey2QPer84NrZ9t4xJK6FRQdJkm4o7w,8175
+aihub/services/user_system.py,sha256=1OKqQDTwuaLILeNk8yA2g1CG6BMX9MsJZFZw_-PIlAg,19141
 aihub/services/workflow_center.py,sha256=caKxOlba0J1s1RUK6RUm1ndJSwAcZXEakRanu3sGKPU,17468
 aihub/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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.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,,
+intellif_aihub-0.1.22.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+intellif_aihub-0.1.22.dist-info/METADATA,sha256=DuDrCBkMMCiEHFQtkQwE1AdvA2JfVJmwWwvUMEMp6HM,2949
+intellif_aihub-0.1.22.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+intellif_aihub-0.1.22.dist-info/top_level.txt,sha256=vIvTtSIN73xv46BpYM-ctVGnyOiUQ9EWP_6ngvdIlvw,6
+intellif_aihub-0.1.22.dist-info/RECORD,,