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

aihub/services/labelfree.py

@@ -1,44 +1,44 @@
-from __future__ import annotations
-
-import httpx
-
-from ..exceptions import APIError
-from ..models.common import APIWrapper
-from ..models.labelfree import *
-
-_BASE = "/labelfree/api/v2"
-
-
-class LabelfreeService:
-    def __init__(self, http: httpx.Client):
-        self._project = _Project(http)
-
-    def get_project_global_stats(self, project_name: str) -> GetGlobalStatsResponse:
-        """获取标注项目的进展信息
-
-        Args:
-            project_name: 标注项目名称
-
-        Returns:
-            GetGlobalStatsResponse
-
-        """
-        return self._project.get_global_stats(project_name)
-
-    @property
-    def project(self) -> _Project:
-        return self._project
-
-
-class _Project:
-    def __init__(self, http: httpx.Client):
-        self._http = http
-
-    def get_global_stats(self, project_name: str) -> GetGlobalStatsResponse:
-        resp = self._http.get(
-            f"{_BASE}/projects/global_stats", params={"project_name": project_name}
-        )
-        wrapper = APIWrapper[GetGlobalStatsResponse].model_validate(resp.json())
-        if wrapper.code != 0:
-            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
-        return wrapper.data
+from __future__ import annotations
+
+import httpx
+
+from ..exceptions import APIError
+from ..models.common import APIWrapper
+from ..models.labelfree import GetGlobalStatsResponse
+
+_BASE = "/labelfree/api/v2"
+
+
+class LabelfreeService:
+    def __init__(self, http: httpx.Client):
+        self._project = _Project(http)
+
+    def get_project_global_stats(self, project_name: str) -> GetGlobalStatsResponse:
+        """获取标注项目的进展信息
+
+        Args:
+            project_name: 标注项目名称
+
+        Returns:
+            GetGlobalStatsResponse
+
+        """
+        return self._project.get_global_stats(project_name)
+
+    @property
+    def project(self) -> _Project:
+        return self._project
+
+
+class _Project:
+    def __init__(self, http: httpx.Client):
+        self._http = http
+
+    def get_global_stats(self, project_name: str) -> GetGlobalStatsResponse:
+        resp = self._http.get(
+            f"{_BASE}/projects/global_stats", params={"project_name": project_name}
+        )
+        wrapper = APIWrapper[GetGlobalStatsResponse].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data

aihub/services/model_training_platform.py

@@ -1,135 +1,209 @@
-from __future__ import annotations
-
-import httpx
-
-from ..exceptions import APIError
-from ..models.common import APIWrapper
-from ..models.model_training_platform import *
-
-_BASE = "/model-training-platform/api/v1"
-
-
-class ModelTrainingPlatformService:
-
-    def __init__(self, http: httpx.Client):
-        self._training = _Training(http)
-
-    def create_training(self, payload: CreateTrainingRequest) -> int:
-        return self._training.create(payload)
-
-    def list_trainings(self, payload: ListTrainingsRequest) -> ListTrainingsResponse:
-        return self._training.list(payload)
-
-    def get_training(self, training_id: int) -> Training:
-        return self._training.get(training_id)
-
-    def stop_training(self, training_id: int) -> None:
-        self._training.stop(training_id)
-
-    def list_training_pods(self, training_id: int, payload: ListTrainingPodsRequest) -> ListTrainingPodsResponse:
-        return self._training.list_training_pods(training_id, payload)
-
-    def get_training_pod(self, training_id: int, pod_id: int) -> Pod:
-        return self._training.get_training_pod(training_id, pod_id)
-
-    def get_pod_logs_new(self, training_id: int, pod_id: int) -> GetTrainingPodLogsNewResponse:
-        return self._training.get_training_logs_new(training_id, pod_id)
-
-    def get_pod_spec(self, training_id: int, pod_id: int) -> GetTrainingPodSpecResponse:
-        return self._training.get_training_spec(training_id, pod_id)
-
-    def get_pod_events(self, training_id: int, pod_id: int) -> GetTrainingPodEventsResponse:
-        return self._training.get_training_events(training_id, pod_id)
-
-    def list_training_users(self, payload: ListTrainingUsersRequest) -> ListTrainingUsersResponse:
-        return self._training.list_training_users(payload)
-
-    def list_training_containers(self, payload: ListTrainingContainersRequest) -> ListTrainingContainersResponse:
-        return self._training.list_training_containers(payload)
-
-    @property
-    def training(self) -> _Training:
-        return self._training
-
-
-class _Training:
-
-    def __init__(self, http: httpx.Client):
-        self._http = http
-
-    def create(self, payload: CreateTrainingRequest) -> int:
-        resp = self._http.post(f"{_BASE}/trainings", json=payload.model_dump(by_alias=True, exclude_none=True))
-        wrapper = APIWrapper[CreateTrainingResponse].model_validate(resp.json())
-        if wrapper.code != 0:
-            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
-        return wrapper.data.id
-
-    def list(self, payload: ListTrainingsRequest) -> ListTrainingsResponse:
-        resp = self._http.get(f"{_BASE}/trainings", params=payload.model_dump(by_alias=True, exclude_none=True))
-        wrapper = APIWrapper[ListTrainingsResponse].model_validate(resp.json())
-        if wrapper.code != 0:
-            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
-        return wrapper.data
-
-    def get(self, training_id: int) -> Training:
-        resp = self._http.get(f"{_BASE}/trainings/{training_id}")
-        wrapper = APIWrapper[Training].model_validate(resp.json())
-        if wrapper.code != 0:
-            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
-        return wrapper.data
-
-    def stop(self, training_id: int) -> None:
-        resp = self._http.post(f"{_BASE}/trainings/{training_id}/stop")
-        wrapper = APIWrapper[dict].model_validate(resp.json())
-        if wrapper.code != 0:
-            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
-
-    def list_training_pods(self, training_id: int, payload: ListTrainingPodsRequest) -> ListTrainingPodsResponse:
-        resp = self._http.get(f"{_BASE}/trainings/{training_id}/pods",
-                              params=payload.model_dump(by_alias=True, exclude_none=True))
-        wrapper = APIWrapper[ListTrainingPodsResponse].model_validate(resp.json())
-        if wrapper.code != 0:
-            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
-        return wrapper.data
-
-    def get_training_pod(self, training_id: int, pod_id: int) -> Pod:
-        resp = self._http.get(f"{_BASE}/trainings/{training_id}/pods/{pod_id}")
-        wrapper = APIWrapper[Pod].model_validate(resp.json())
-        if wrapper.code != 0:
-            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
-        return wrapper.data
-
-    def get_training_logs_new(self, training_id: int, pod_id: int) -> GetTrainingPodLogsNewResponse:
-        resp = self._http.get(f"{_BASE}/trainings/{training_id}/pods/{pod_id}/logs/new")
-        wrapper = APIWrapper[GetTrainingPodLogsNewResponse].model_validate(resp.json())
-        if wrapper.code != 0:
-            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
-        return wrapper.data
-
-    def get_training_spec(self, training_id: int, pod_id: int) -> GetTrainingPodSpecResponse:
-        resp = self._http.get(f"{_BASE}/trainings/{training_id}/pods/{pod_id}/spec")
-        wrapper = APIWrapper[GetTrainingPodSpecResponse].model_validate(resp.json())
-        if wrapper.code != 0:
-            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
-        return wrapper.data
-
-    def get_training_events(self, training_id: int, pod_id: int) -> GetTrainingPodEventsResponse:
-        resp = self._http.get(f"{_BASE}/trainings/{training_id}/pods/{pod_id}/events")
-        wrapper = APIWrapper[GetTrainingPodEventsResponse].model_validate(resp.json())
-        if wrapper.code != 0:
-            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
-        return wrapper.data
-
-    def list_training_users(self, payload: ListTrainingUsersRequest) -> ListTrainingUsersResponse:
-        resp = self._http.get(f"{_BASE}/training-users", params=payload.model_dump(by_alias=True))
-        wrapper = APIWrapper[ListTrainingUsersResponse].model_validate(resp.json())
-        if wrapper.code != 0:
-            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
-        return wrapper.data
-
-    def list_training_containers(self, payload: ListTrainingContainersRequest) -> ListTrainingContainersResponse:
-        resp = self._http.get(f"{_BASE}/training-containers", params=payload.model_dump(by_alias=True))
-        wrapper = APIWrapper[ListTrainingContainersResponse].model_validate(resp.json())
-        if wrapper.code != 0:
-            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
-        return wrapper.data
+from __future__ import annotations
+
+import httpx
+
+from ..exceptions import APIError
+from ..models.common import APIWrapper
+from ..models.model_training_platform import (
+    CreateTrainingRequest,
+    ListTrainingsRequest,
+    ListTrainingsResponse,
+    ListTrainingPodsRequest,
+    GetTrainingPodLogsNewResponse,
+    GetTrainingPodEventsResponse,
+    Training,
+    CreateTrainingResponse,
+    ListTrainingUsersRequest,
+    ListTrainingPodsResponse,
+    GetTrainingPodSpecResponse,
+    ListTrainingContainersRequest,
+    ListTrainingUsersResponse,
+    ListTrainingContainersResponse,
+    Pod,
+)
+
+_BASE = "/model-training-platform/api/v1"
+
+
+class ModelTrainingPlatformService:
+
+    def __init__(self, http: httpx.Client):
+        self._training = _Training(http)
+
+    def create_training(self, payload: CreateTrainingRequest) -> int:
+        """创建训练任务
+
+        Args:
+            payload (CreateTrainingRequest): 创建训练任务参数
+
+        Returns:
+            int: 训练任务ID
+        """
+        return self._training.create(payload)
+
+    def list_trainings(self, payload: ListTrainingsRequest) -> ListTrainingsResponse:
+        return self._training.list(payload)
+
+    def get_training(self, training_id: int) -> Training:
+        """获取训练任务详情
+
+        Args:
+            training_id (int): 训练任务ID
+        Returns:
+            Training
+
+        """
+        return self._training.get(training_id)
+
+    def stop_training(self, training_id: int) -> None:
+        """停止训练瑞文
+
+        Args:
+            training_id (int): 训练任务ID
+
+        Returns:
+
+        """
+        self._training.stop(training_id)
+
+    def list_training_pods(
+        self, training_id: int, payload: ListTrainingPodsRequest
+    ) -> ListTrainingPodsResponse:
+        return self._training.list_training_pods(training_id, payload)
+
+    def get_training_pod(self, training_id: int, pod_id: int) -> Pod:
+        return self._training.get_training_pod(training_id, pod_id)
+
+    def get_pod_logs_new(
+        self, training_id: int, pod_id: int
+    ) -> GetTrainingPodLogsNewResponse:
+        return self._training.get_training_logs_new(training_id, pod_id)
+
+    def get_pod_spec(self, training_id: int, pod_id: int) -> GetTrainingPodSpecResponse:
+        return self._training.get_training_spec(training_id, pod_id)
+
+    def get_pod_events(
+        self, training_id: int, pod_id: int
+    ) -> GetTrainingPodEventsResponse:
+        return self._training.get_training_events(training_id, pod_id)
+
+    def list_training_users(
+        self, payload: ListTrainingUsersRequest
+    ) -> ListTrainingUsersResponse:
+        return self._training.list_training_users(payload)
+
+    def list_training_containers(
+        self, payload: ListTrainingContainersRequest
+    ) -> ListTrainingContainersResponse:
+        return self._training.list_training_containers(payload)
+
+    @property
+    def training(self) -> _Training:
+        return self._training
+
+
+class _Training:
+
+    def __init__(self, http: httpx.Client):
+        self._http = http
+
+    def create(self, payload: CreateTrainingRequest) -> int:
+        resp = self._http.post(
+            f"{_BASE}/trainings",
+            json=payload.model_dump(by_alias=True, exclude_none=True),
+        )
+        wrapper = APIWrapper[CreateTrainingResponse].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data.id
+
+    def list(self, payload: ListTrainingsRequest) -> ListTrainingsResponse:
+        resp = self._http.get(
+            f"{_BASE}/trainings",
+            params=payload.model_dump(by_alias=True, exclude_none=True),
+        )
+        wrapper = APIWrapper[ListTrainingsResponse].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data
+
+    def get(self, training_id: int) -> Training:
+        resp = self._http.get(f"{_BASE}/trainings/{training_id}")
+        wrapper = APIWrapper[Training].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data
+
+    def stop(self, training_id: int) -> None:
+        resp = self._http.post(f"{_BASE}/trainings/{training_id}/stop")
+        wrapper = APIWrapper[dict].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+
+    def list_training_pods(
+        self, training_id: int, payload: ListTrainingPodsRequest
+    ) -> ListTrainingPodsResponse:
+        resp = self._http.get(
+            f"{_BASE}/trainings/{training_id}/pods",
+            params=payload.model_dump(by_alias=True, exclude_none=True),
+        )
+        wrapper = APIWrapper[ListTrainingPodsResponse].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data
+
+    def get_training_pod(self, training_id: int, pod_id: int) -> Pod:
+        resp = self._http.get(f"{_BASE}/trainings/{training_id}/pods/{pod_id}")
+        wrapper = APIWrapper[Pod].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data
+
+    def get_training_logs_new(
+        self, training_id: int, pod_id: int
+    ) -> GetTrainingPodLogsNewResponse:
+        resp = self._http.get(f"{_BASE}/trainings/{training_id}/pods/{pod_id}/logs/new")
+        wrapper = APIWrapper[GetTrainingPodLogsNewResponse].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data
+
+    def get_training_spec(
+        self, training_id: int, pod_id: int
+    ) -> GetTrainingPodSpecResponse:
+        resp = self._http.get(f"{_BASE}/trainings/{training_id}/pods/{pod_id}/spec")
+        wrapper = APIWrapper[GetTrainingPodSpecResponse].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data
+
+    def get_training_events(
+        self, training_id: int, pod_id: int
+    ) -> GetTrainingPodEventsResponse:
+        resp = self._http.get(f"{_BASE}/trainings/{training_id}/pods/{pod_id}/events")
+        wrapper = APIWrapper[GetTrainingPodEventsResponse].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data
+
+    def list_training_users(
+        self, payload: ListTrainingUsersRequest
+    ) -> ListTrainingUsersResponse:
+        resp = self._http.get(
+            f"{_BASE}/training-users", params=payload.model_dump(by_alias=True)
+        )
+        wrapper = APIWrapper[ListTrainingUsersResponse].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data
+
+    def list_training_containers(
+        self, payload: ListTrainingContainersRequest
+    ) -> ListTrainingContainersResponse:
+        resp = self._http.get(
+            f"{_BASE}/training-containers", params=payload.model_dump(by_alias=True)
+        )
+        wrapper = APIWrapper[ListTrainingContainersResponse].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data

aihub/services/quota_schedule_management.py

@@ -1,18 +1,182 @@
-from __future__ import annotations
-
-import os
-from pathlib import Path
-
-_ENV_KEY = "PRE_STOP_SENTINEL_FILE"
-_DEFAULT_SENTINEL = "/tmp/pre_stop_sentinel_file"
-_SENTINEL_PATH = Path(os.getenv(_ENV_KEY, _DEFAULT_SENTINEL))
-
-
-def is_pre_stopped() -> bool:
-    return _SENTINEL_PATH.exists()
-
-
-class PreStopService:
-    @staticmethod
-    def is_pre_stopped() -> bool:
-        return is_pre_stopped()
+from __future__ import annotations
+
+import os
+from pathlib import Path
+
+import httpx
+
+from ..exceptions import APIError
+from ..models.common import APIWrapper
+from ..models.quota_schedule_management import *
+
+_ENV_KEY = "PRE_STOP_SENTINEL_FILE"
+_DEFAULT_SENTINEL = "/tmp/pre_stop_sentinel_file"
+_SENTINEL_PATH = Path(os.getenv(_ENV_KEY, _DEFAULT_SENTINEL))
+_BASE = "/quota-schedule-management/api/v1"
+
+
+def is_pre_stopped() -> bool:
+    return _SENTINEL_PATH.exists()
+
+
+class PreStopService:
+    @staticmethod
+    def is_pre_stopped() -> bool:
+        return is_pre_stopped()
+
+
+class QuotaScheduleManagementService:
+
+    def __init__(self, http: httpx.Client):
+        self._task = _Task(http)
+
+    def create_task(self, payload: CreateTaskRequest) -> int:
+        """创建调度任务
+
+        Args:
+            payload: 创建任务的各项参数
+
+        Returns:
+            int: 后端生成的任务 ID
+        """
+        return self._task.create(payload)
+
+    def list_tasks(self, payload: ListTasksRequest) -> ListTasksResponse:
+        """分页查询任务列表
+
+        Args:
+            payload: 查询条件（分页 / 过滤）
+
+        Returns:
+            ListTasksResponse: 列表及分页信息
+        """
+        return self._task.list(payload)
+
+    def get_task(self, task_id: int) -> Task:
+        """获取任务详情
+
+        Args:
+            task_id: 任务 ID
+
+        Returns:
+            Task: 任务完整信息
+        """
+        return self._task.get(task_id)
+
+    def stop_task(self, task_id: int) -> None:
+        """停止任务
+
+        Args:
+            task_id: 任务 ID
+
+        Returns:
+            None
+        """
+        self._task.stop(task_id)
+
+    def list_task_pods(self, task_id: int, payload: ListTaskPodsRequest) -> ListTaskPodsResponse:
+        return self._task.list_pods(task_id, payload)
+
+    def get_task_pod(self, task_id: int, pod_id: int) -> Pod:
+        return self._task.get_pod(task_id, pod_id)
+
+    def get_pod_logs_new(self, task_id: int, pod_id: int) -> List[PodLogInfo]:
+        return self._task.get_logs_new(task_id, pod_id).logs
+
+    def get_pod_spec(self, task_id: int, pod_id: int) -> str:
+        return self._task.get_spec(task_id, pod_id).spec
+
+    def get_pod_events(self, task_id: int, pod_id: int) -> str:
+        return self._task.get_events(task_id, pod_id).events
+
+    def list_task_users(self, payload: ListTaskUsersRequest) -> ListTaskUsersResponse:
+        return self._task.list_users(payload)
+
+    def get_metrics_overview(self, payload: GetMetricsOverviewRequest) -> GetMetricsOverviewResponse:
+        return self._task.get_metrics_overview(payload)
+
+    @property
+    def task(self) -> _Task:
+        return self._task
+
+
+class _Task:
+
+    def __init__(self, http: httpx.Client):
+        self._http = http
+
+    def create(self, payload: CreateTaskRequest) -> int:
+        resp = self._http.post(f"{_BASE}/tasks", json=payload.model_dump(by_alias=True, exclude_none=True))
+        wrapper = APIWrapper[CreateTaskResponse].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data.id
+
+    def list(self, payload: ListTasksRequest) -> ListTasksResponse:
+        resp = self._http.get(f"{_BASE}/tasks", params=payload.model_dump(by_alias=True, exclude_none=True))
+        wrapper = APIWrapper[ListTasksResponse].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data
+
+    def get(self, task_id: int) -> Task:
+        resp = self._http.get(f"{_BASE}/tasks/{task_id}")
+        wrapper = APIWrapper[Task].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data
+
+    def stop(self, task_id: int) -> None:
+        resp = self._http.post(f"{_BASE}/tasks/{task_id}/stop")
+        wrapper = APIWrapper[dict].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+
+    def list_pods(self, task_id: int, payload: ListTaskPodsRequest) -> ListTaskPodsResponse:
+        resp = self._http.get(f"{_BASE}/tasks/{task_id}/pods", params=payload.model_dump(by_alias=True))
+        wrapper = APIWrapper[ListTaskPodsResponse].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data
+
+    def get_pod(self, task_id: int, pod_id: int) -> Pod:
+        resp = self._http.get(f"{_BASE}/tasks/{task_id}/pods/{pod_id}")
+        wrapper = APIWrapper[Pod].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data
+
+    def get_logs_new(self, task_id: int, pod_id: int) -> GetTaskPodLogsNewResponse:
+        resp = self._http.get(f"{_BASE}/tasks/{task_id}/pods/{pod_id}/logs/new")
+        wrapper = APIWrapper[GetTaskPodLogsNewResponse].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data
+
+    def get_spec(self, task_id: int, pod_id: int) -> GetTaskPodSpecResponse:
+        resp = self._http.get(f"{_BASE}/tasks/{task_id}/pods/{pod_id}/spec")
+        wrapper = APIWrapper[GetTaskPodSpecResponse].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data
+
+    def get_events(self, task_id: int, pod_id: int) -> GetTaskPodEventsResponse:
+        resp = self._http.get(f"{_BASE}/tasks/{task_id}/pods/{pod_id}/events")
+        wrapper = APIWrapper[GetTaskPodEventsResponse].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data
+
+    def list_users(self, payload: ListTaskUsersRequest) -> ListTaskUsersResponse:
+        resp = self._http.get(f"{_BASE}/task-users", params=payload.model_dump(by_alias=True))
+        wrapper = APIWrapper[ListTaskUsersResponse].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data
+
+    def get_metrics_overview(self, payload: GetMetricsOverviewRequest) -> GetMetricsOverviewResponse:
+        resp = self._http.get(f"{_BASE}/metrics/overview", params=payload.model_dump(by_alias=True))
+        wrapper = APIWrapper[GetMetricsOverviewResponse].model_validate(resp.json())
+        if wrapper.code != 0:
+            raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
+        return wrapper.data