ap-client 0.6.0.dev1__tar.gz → 0.6.1.dev0__tar.gz

{ap_client-0.6.0.dev1 → ap_client-0.6.1.dev0}/PKG-INFO

@@ -1,18 +1,20 @@
 Metadata-Version: 2.5
 Name: ap-client
-Version: 0.6.0.dev1
+Version: 0.6.1.dev0
 Summary: Agent Platform API Client & CLI
 Requires-Python: >=3.10
+Requires-Dist: click>=8.0.0
 Requires-Dist: irepo-artifact==0.2.1
 Requires-Dist: pyyaml>=6.0
 Requires-Dist: requests>=2.28.0
 Requires-Dist: rich>=13.0.0
 Requires-Dist: typer>=0.9.0
+Requires-Dist: urllib3>=1.26
 Requires-Dist: websockets>=13.0
 Provides-Extra: all
-Requires-Dist: instance-repo[oss]<2.0.0,>=1.3.2; extra == 'all'
+Requires-Dist: instance-repo[oss]<2.0.0,>=1.4.0.dev3; extra == 'all'
 Provides-Extra: dataset
-Requires-Dist: instance-repo[oss]<2.0.0,>=1.3.2; extra == 'dataset'
+Requires-Dist: instance-repo[oss]<2.0.0,>=1.4.0.dev3; extra == 'dataset'
 Description-Content-Type: text/markdown
 
 A lightweight Python SDK and command line interface for Agent Platform. It provides helpers for configuring API access and managing templates, datasets, jobs, and groups.

{ap_client-0.6.0.dev1 → ap_client-0.6.1.dev0}/ap_client/api.py

@@ -1881,9 +1881,12 @@ class APIClient:
         meta_job_type: Optional[str] = None,
         meta_teacher_direction: Optional[str] = None,
         meta_train_stage: Optional[str] = None,
+        benchmark: Optional[str] = None,
     ) -> dict:
         """Update group properties.
 
+        ``benchmark`` is omitted when None; pass an empty string to clear it.
+
         ``tags``（`key:value` 数组）在客户端先与 Group 现有 tags 合并，再以
         全量替换语义提交（服务端 PATCH 是整体替换）；同名 key 覆盖，未提及
         的 key 保留。服务端会从 tags 同步 meta_* 等派生字段。
@@ -1906,6 +1909,8 @@ class APIClient:
             body["model"] = model
         if scaffold is not None:
             body["scaffold"] = scaffold
+        if benchmark is not None:
+            body["benchmark"] = benchmark
         meta_attr_tags = _merge_meta_tags(
             None,
             meta_model=meta_model,

{ap_client-0.6.0.dev1 → ap_client-0.6.1.dev0}/ap_client/cli.py

@@ -25,6 +25,7 @@ from ap_client.api import (
 )
 from ap_client.config import ENV_VAR_SPECS, ConfigurationError, _parse_bool, normalize_output_format
 from ap_client.dataset_commands import register as _register_dataset_commands
+from ap_client.dfs_commands import register as _register_dfs_commands
 from ap_client.exporter import export_group, export_job, export_jobs
 from ap_client.fs_commands import register as _register_fs_commands
 from ap_client.instance_commands import register as _register_instance_commands
@@ -167,6 +168,7 @@ app.add_typer(checkpoint_app, name="checkpoint")
 # ap dataset version/split/access 嵌套在 dataset 下;instance/fs 是一等资源。
 _register_dataset_commands(dataset_app)
 _register_instance_commands(app)
+_register_dfs_commands(app)
 _register_fs_commands(app)
 _register_akpool_commands(app)
 
@@ -6066,6 +6068,9 @@ def group_update(
     scaffold: Optional[str] = typer.Option(
         None, "--scaffold", "-s", help="Scaffold/agent name (benchmark perspective)"
     ),
+    benchmark: Optional[str] = typer.Option(
+        None, "--benchmark", help="Benchmark name; pass an empty string to clear"
+    ),
     tags: Optional[str] = typer.Option(
         None,
         "--tags",
@@ -6127,6 +6132,7 @@ def group_update(
             dataset,
             model_name,
             scaffold,
+            benchmark,
             tags_list,
         )
     ):
@@ -6141,6 +6147,7 @@ def group_update(
         dataset=dataset,
         model=model_name,
         scaffold=scaffold,
+        **({"benchmark": benchmark} if benchmark is not None else {}),
         **({"tags": tags_list} if tags_list is not None else {}),
     )
     if output_format == "plain":
@@ -6155,6 +6162,7 @@ def group_update(
                 "dataset",
                 "model",
                 "scaffold",
+                "benchmark",
                 "updated_at",
             ],
         )

{ap_client-0.6.0.dev1 → ap_client-0.6.1.dev0}/ap_client/dataset_commands.py

@@ -123,7 +123,10 @@ _NO_BENCHMARK = "none"
 _FORMAT_HELP = "Output format: plain/table/json/yaml (default: AP_FORMAT or command default)"
 
 _METADATA_MODELS = ("version_first", "split_first")
-_VERSION_SPLIT_METADATA_MODEL = "version_first"
+# IR SDK >= 1.4.0.dev3 的业务入口固定 split_first：business_metadata_model 拒绝
+# version_first（仅保留给低层 legacy 读）。带 version 的 split 查询在两种模型下
+# 结果一致（实测），显式 split_first 与 SDK 归一值一致且新旧 SDK 均接受。
+_VERSION_SPLIT_METADATA_MODEL = "split_first"
 
 #: CLI **不替 SDK 决定元数据模型**:不下发 ``metadata_model``,由 SDK 按 profile 归一
 #: (当前为 ``split_first``)。
@@ -1186,9 +1189,26 @@ def dataset_get(
             "dataset_name": dataset,
             "registered": False,
         }
-        parts = dataset.split("/")
-        if len(parts) == 2 and all(parts):
-            rendered.update(dataset_l1=parts[0], dataset_l2=parts[1])
+        from .dataset_name import (
+            DatasetNameError,
+            DatasetNameUpgradeRequired,
+            parse_dataset_name,
+        )
+
+        try:
+            name = parse_dataset_name(dataset)
+        except DatasetNameUpgradeRequired as exc:
+            raise _fail(
+                f"Failed to query dataset {dataset}: {exc}",
+                "pip install 'ap-client[dataset]' --upgrade",
+                exit_code=2,
+            ) from exc
+        except DatasetNameError:
+            # 非规范名:不凭空发明 level 字段(与历史行为一致)。
+            pass
+        else:
+            rendered["dataset_l1"] = name.l1
+            rendered["dataset_l2"] = name.l2
     else:
         rendered = _registered_dict(detail)
     _emit_detail(rendered, output_format, order=_DATASET_DETAIL_ORDER, always=("status",))
@@ -1197,7 +1217,9 @@ def dataset_get(
 
 
 def dataset_create(
-    dataset: str = typer.Argument(..., help="Dataset name (e.g. alibaba/my-bench)"),
+    dataset: str = typer.Argument(
+        ..., help="Dataset name: L1 or L1/L2 (e.g. my-bench or alibaba/my-bench)"
+    ),
     benchmark: str = typer.Option(..., "--benchmark", help="Benchmark name or source_id"),
     description: Optional[str] = typer.Option(None, "--description", help="Dataset description"),
     visibility: str = typer.Option(
@@ -1216,20 +1238,32 @@ def dataset_create(
 
     from instance_repo.models import DatasetSeries
 
+    from .dataset_name import (
+        DatasetNameError,
+        DatasetNameUpgradeRequired,
+        parse_dataset_name,
+    )
     from .irepo_sdk import _reason
 
-    parts = dataset.split("/")
-    if len(parts) != 2 or not all(parts):
+    try:
+        name = parse_dataset_name(dataset)
+    except DatasetNameUpgradeRequired as exc:
+        raise _fail(
+            f"Failed to create dataset {dataset}: {exc}",
+            "pip install 'ap-client[dataset]' --upgrade",
+            exit_code=2,
+        ) from exc
+    except DatasetNameError:
         raise _fail(
-            f"Failed to create dataset {dataset}: the name must be two segments <team>/<name>",
-            "e.g. your-team/my-bench",
+            f"Failed to create dataset {dataset}: the name must be one segment (L1) "
+            "or two segments (L1/L2)",
+            "e.g. my-bench or your-team/my-bench",
             exit_code=2,
         )
-    l1, l2 = parts
     series = DatasetSeries(
         dataset_name=dataset,
-        dataset_l1=l1,
-        dataset_l2=l2,
+        dataset_l1=name.l1,
+        dataset_l2=name.l2,
         benchmark_id=benchmark_id,
         visibility=resolved_visibility,
         description=description or None,

ap_client-0.6.1.dev0/ap_client/dataset_name.py

@@ -0,0 +1,53 @@
+"""DatasetName 解析 Adapter:唯一入口委托给已安装的 InstanceRepo SDK。
+
+AP CLI 不维护第三套 dataset 名解析:``instance_repo.paths.DatasetName`` 是客户端
+路径契约的唯一事实源(L1 单段 ``A`` / 两段 ``A/B``,与 apiserver 的 namespace
+guard 及 conformance 契约对齐)。SDK 缺失或过旧(没有 ``DatasetName.parse``
+能力)时,给出明确的升级提示,不静默降级、不本地重写解析规则。
+"""
+
+from __future__ import annotations
+
+from typing import Any
+
+__all__ = [
+    "DatasetNameError",
+    "DatasetNameUpgradeRequired",
+    "parse_dataset_name",
+]
+
+#: SDK 缺失/无 DatasetName 能力时的升级提示(与 irepo_sdk 的 SdkMissing 同范式)。
+_UPGRADE_HINT = (
+    "The installed instance-repo SDK does not provide DatasetName parsing "
+    "(single-segment L1 dataset names need it).\n"
+    "Install it with: pip install 'ap-client[dataset]' --upgrade"
+)
+
+
+class DatasetNameUpgradeRequired(RuntimeError):
+    """已安装的 instance-repo SDK 缺少 DatasetName 能力,必须升级。"""
+
+
+class DatasetNameError(ValueError):
+    """Dataset 名不符合 IR DatasetName 契约(单段或两段、无非法段)。"""
+
+
+def parse_dataset_name(value: str) -> Any:
+    """Parse ``value`` through the installed IR ``DatasetName`` capability.
+
+    Returns the SDK ``DatasetName`` (``canonical`` / ``l1`` / ``l2`` /
+    ``segments`` / ``leaf``)。Raises :class:`DatasetNameUpgradeRequired` when the
+    installed SDK lacks the capability; raises :class:`DatasetNameError` when
+    the value is not a valid L1 or L1/L2 name.
+    """
+    try:
+        from instance_repo.paths import DatasetName
+    except ImportError as exc:
+        raise DatasetNameUpgradeRequired(_UPGRADE_HINT) from exc
+    parse = getattr(DatasetName, "parse", None)
+    if not callable(parse):
+        raise DatasetNameUpgradeRequired(_UPGRADE_HINT)
+    try:
+        return parse(value)
+    except Exception as exc:  # noqa: BLE001 - SDK 校验失败统一转成 CLI 契约
+        raise DatasetNameError(str(exc)) from exc

{ap_client-0.6.0.dev1 → ap_client-0.6.1.dev0}/ap_client/dataset_register.py

@@ -66,11 +66,25 @@ def _claim_without_workspace(
 
 
 def _create_owned_dataset(repo: Any, dataset: str, benchmark_id: str) -> dict:
-    parts = dataset.split("/")
-    if len(parts) != 2 or not all(parts):
+    from .dataset_name import (
+        DatasetNameError,
+        DatasetNameUpgradeRequired,
+        parse_dataset_name,
+    )
+
+    try:
+        name = parse_dataset_name(dataset)
+    except DatasetNameUpgradeRequired as exc:
+        raise datasets._fail(
+            f"Failed to create dataset {dataset}: {exc}",
+            "pip install 'ap-client[dataset]' --upgrade",
+            exit_code=2,
+        ) from exc
+    except DatasetNameError:
         raise datasets._fail(
-            f"Failed to create dataset {dataset}: the name must be two segments <team>/<name>",
-            "Use a dataset name such as your-team/my-bench",
+            f"Failed to create dataset {dataset}: the name must be one segment (L1) "
+            "or two segments (L1/L2)",
+            "Use a dataset name such as my-bench or your-team/my-bench",
             exit_code=2,
         )
 
@@ -78,8 +92,8 @@ def _create_owned_dataset(repo: Any, dataset: str, benchmark_id: str) -> dict:
 
     fields: dict[str, Any] = {
         "dataset_name": dataset,
-        "dataset_l1": parts[0],
-        "dataset_l2": parts[1],
+        "dataset_l1": name.l1,
+        "dataset_l2": name.l2,
         "visibility": "private",
     }
     if benchmark_id: