ap-client 0.6.0.dev1__tar.gz → 0.6.0.dev2__tar.gz

{ap_client-0.6.0.dev1 → ap_client-0.6.0.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: ap-client
-Version: 0.6.0.dev1
+Version: 0.6.0.dev2
 Summary: Agent Platform API Client & CLI
 Requires-Python: >=3.10
 Requires-Dist: irepo-artifact==0.2.1

{ap_client-0.6.0.dev1 → ap_client-0.6.0.dev2}/ap_client/api.py

@@ -1125,8 +1125,15 @@ class APIClient:
         benchmark: Optional[str] = None,
         experiment: Optional[str] = None,
         ak_ids: Optional[list[str]] = None,
+        priority: Optional[str] = None,
+        priority_weight: Optional[int] = None,
     ) -> dict:
-        """Create a group."""
+        """Create a group with optional scheduling defaults for its Jobs.
+
+        Omitted priority fields use the server defaults: medium and weight 0.
+        """
+        if priority is not None and priority not in {"high", "medium"}:
+            raise ValueError("priority must be one of high/medium")
         if ak_ids is not None and not isinstance(ak_ids, list):
             raise ValueError("ak_ids must be a list")
         references = list(ak_ids or [])
@@ -1164,6 +1171,10 @@ class APIClient:
             body["queue_id"] = effective_queue_id
         if max_concurrency is not None:
             body["max_concurrency"] = max_concurrency
+        if priority is not None:
+            body["priority"] = priority
+        if priority_weight is not None:
+            body["priority_weight"] = priority_weight
         if eval_config is not None:
             body["eval_config"] = eval_config
         if model_base_url_collection is not None:
@@ -1881,9 +1892,15 @@ class APIClient:
         meta_job_type: Optional[str] = None,
         meta_teacher_direction: Optional[str] = None,
         meta_train_stage: Optional[str] = None,
+        benchmark: Optional[str] = None,
+        priority: Optional[str] = None,
+        priority_weight: Optional[int] = None,
+        priority_update_mode: 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_* 等派生字段。
@@ -1891,7 +1908,19 @@ class APIClient:
         ``meta_*`` 独立属性为部分更新（传了才改，不支持清除）；新服务端
         优先读属性，老服务端忽略属性——因此过渡期内同时也会将属性合并进
         tags 轨（双写）。
+
+        Priority changes update the Group defaults. The default existing mode
+        also updates all current Jobs, including PostProcess; incremental mode
+        changes only the defaults for future Jobs. Updating priority alone uses
+        weight 0; updating only weight preserves the current priority tier.
         """
+        if priority is not None and priority not in {"high", "medium"}:
+            raise ValueError("priority must be one of high/medium")
+        if priority_update_mode is not None:
+            if priority_update_mode not in {"existing", "incremental"}:
+                raise ValueError("priority_update_mode must be one of existing/incremental")
+            if priority is None and priority_weight is None:
+                raise ValueError("priority_update_mode requires priority or priority_weight")
         body: dict = {}
 
         if max_concurrency is not None:
@@ -1906,6 +1935,14 @@ class APIClient:
             body["model"] = model
         if scaffold is not None:
             body["scaffold"] = scaffold
+        if benchmark is not None:
+            body["benchmark"] = benchmark
+        if priority is not None:
+            body["priority"] = priority
+        if priority_weight is not None:
+            body["priority_weight"] = priority_weight
+        if priority_update_mode is not None:
+            body["priority_update_mode"] = priority_update_mode
         meta_attr_tags = _merge_meta_tags(
             None,
             meta_model=meta_model,

{ap_client-0.6.0.dev1 → ap_client-0.6.0.dev2}/ap_client/cli.py

@@ -642,7 +642,18 @@ def _print_formatted(data: dict, output_format: str) -> None:
             import yaml
         except ImportError as exc:
             raise typer.BadParameter("--format yaml requires PyYAML to be installed") from exc
-        typer.echo(yaml.safe_dump(data, allow_unicode=True, sort_keys=False))
+
+        class LosslessDumper(yaml.SafeDumper):
+            pass
+
+        def represent_string(dumper, value):
+            # YAML normalizes literal NEL/line separators in unquoted scalars.
+            # Double quotes make the emitter escape them while keeping Chinese readable.
+            style = '"' if any(character in value for character in "\x85\u2028\u2029") else None
+            return dumper.represent_scalar("tag:yaml.org,2002:str", value, style=style)
+
+        LosslessDumper.add_representer(str, represent_string)
+        typer.echo(yaml.dump(data, Dumper=LosslessDumper, allow_unicode=True, sort_keys=False))
         return
     _print_json(data)
 
@@ -3219,12 +3230,12 @@ def job_create(
     priority: Optional[str] = typer.Option(
         None,
         "--priority",
-        help="Scheduling priority tier: high or medium (default: medium)",
+        help="Scheduling tier: high or medium. Omit both priority flags to inherit Group defaults; otherwise an omitted tier is medium",
     ),
     priority_weight: Optional[int] = typer.Option(
         None,
         "--priority-weight",
-        help="Fine-grained priority weight within tier, 0-10 (default: 0)",
+        help="Priority weight, 0-10. Omit both priority flags to inherit Group defaults; otherwise an omitted weight is 0",
     ),
     resource_mode: Optional[str] = typer.Option(
         None,
@@ -5457,6 +5468,18 @@ def group_create(
         min=1,
         help="Maximum concurrent jobs. Server default: 50.",
     ),
+    priority: Optional[str] = typer.Option(
+        None,
+        "--priority",
+        help="Default priority inherited by Jobs: high or medium (server default: medium)",
+    ),
+    priority_weight: Optional[int] = typer.Option(
+        None,
+        "--priority-weight",
+        min=0,
+        max=10,
+        help="Default priority weight inherited by Jobs, 0-10 (server default: 0)",
+    ),
     eval_config: Optional[str] = typer.Option(
         None, "--eval-config", help="Evaluation config (JSON)"
     ),
@@ -5569,6 +5592,8 @@ def group_create(
 ):
     """Create a Group."""
     output_format = _normalize_output_format(output_format)
+    if priority is not None and priority not in {"high", "medium"}:
+        raise typer.BadParameter("--priority must be one of high/medium")
     try:
         for reference in ak_id or []:
             validate_ak_id(reference)
@@ -5650,6 +5675,8 @@ def group_create(
             checkpoint=checkpoint_dict,
             benchmark=benchmark,
             experiment=experiment,
+            **({"priority": priority} if priority is not None else {}),
+            **({"priority_weight": priority_weight} if priority_weight is not None else {}),
         )
     except ManagedAKProtocolError as exc:
         _emit_error(str(exc))
@@ -5662,6 +5689,10 @@ def group_create(
             print(f"  ak_ids: {response_ak_ids(result)}")
         if result.get("queue_id") is not None:
             print(f"  queue_id: {result.get('queue_id')}")
+        if result.get("priority") is not None:
+            print(f"  priority: {result.get('priority')}")
+        if result.get("priority_weight") is not None:
+            print(f"  priority_weight: {result.get('priority_weight')}")
         if result.get("template") is not None:
             print(f"  template: {result.get('template')}")
         if result.get("agenthub_revision") is not None:
@@ -6066,6 +6097,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",
@@ -6085,14 +6119,40 @@ def group_update(
         "--meta-train-stage",
         help="Set meta_train_stage tag (options: ap meta-train-stage list)",
     ),
+    priority: Optional[str] = typer.Option(
+        None,
+        "--priority",
+        help="Group priority default: high or medium; --priority alone sets weight to 0",
+    ),
+    priority_weight: Optional[int] = typer.Option(
+        None,
+        "--priority-weight",
+        min=0,
+        max=10,
+        help="Group priority weight default, 0-10; weight alone preserves the current tier",
+    ),
+    priority_update_mode: Optional[str] = typer.Option(
+        None,
+        "--priority-update-mode",
+        help="existing updates defaults and all current Jobs (including PostProcess); incremental updates only defaults (server default: existing)",
+    ),
     output_format: str = typer.Option(
         None,
         "--format",
         help="Output format: plain/table/json/yaml (default: AP_FORMAT or command default)",
     ),
 ):
-    """Update Group properties."""
+    """Update Group properties and scheduling defaults inherited by Jobs."""
     output_format = _normalize_output_format(output_format)
+    if priority is not None and priority not in {"high", "medium"}:
+        raise typer.BadParameter("--priority must be one of high/medium")
+    if priority_update_mode is not None:
+        if priority_update_mode not in {"existing", "incremental"}:
+            raise typer.BadParameter("--priority-update-mode must be one of existing/incremental")
+        if priority is None and priority_weight is None:
+            raise typer.BadParameter(
+                "--priority-update-mode requires --priority or --priority-weight"
+            )
     eval_config_dict = _parse_json_object(eval_config, "--eval-config")
     model_base_url_collection_list = _parse_model_base_url_collection_option(
         model_base_url_collection, model_base_url_file
@@ -6127,7 +6187,10 @@ def group_update(
             dataset,
             model_name,
             scaffold,
+            benchmark,
             tags_list,
+            priority,
+            priority_weight,
         )
     ):
         raise typer.BadParameter("At least one field must be provided to update")
@@ -6141,7 +6204,15 @@ 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 {}),
+        **({"priority": priority} if priority is not None else {}),
+        **({"priority_weight": priority_weight} if priority_weight is not None else {}),
+        **(
+            {"priority_update_mode": priority_update_mode}
+            if priority_update_mode is not None
+            else {}
+        ),
     )
     if output_format == "plain":
         _print_group_action_plain(
@@ -6155,6 +6226,9 @@ def group_update(
                 "dataset",
                 "model",
                 "scaffold",
+                "benchmark",
+                "priority",
+                "priority_weight",
                 "updated_at",
             ],
         )

{ap_client-0.6.0.dev1 → ap_client-0.6.0.dev2}/ap_client/managed_ak.py

@@ -138,20 +138,31 @@ def _safe_identifier(value: Any) -> str | None:
 
 
 class ManagedAKProtocolError(RuntimeError):
-    """A response cannot confirm that the requested AK reference was honored."""
+    """An unconfirmed AK reference, with only safe receipt identities retained."""
 
     def __init__(self, ak_id: str, response: Any):
         self.ak_id = ak_id
-        self.response = response
+        self.response: dict[str, Any] = {}
         self.job_ids: list[str] = []
         self.group_id = None
         if isinstance(response, dict):
             self.group_id = _safe_identifier(response.get("group_id"))
-            candidates = [response.get("job_id")]
+            if self.group_id:
+                self.response["group_id"] = self.group_id
+            job_id = _safe_identifier(response.get("job_id"))
+            if job_id:
+                self.response["job_id"] = job_id
+                self.job_ids.append(job_id)
             jobs = response.get("jobs")
             if isinstance(jobs, list):
-                candidates.extend(job.get("job_id") for job in jobs if isinstance(job, dict))
-            self.job_ids = [value for value in map(_safe_identifier, candidates) if value]
+                job_ids = [
+                    value
+                    for job in jobs
+                    if isinstance(job, dict)
+                    if (value := _safe_identifier(job.get("job_id")))
+                ]
+                self.job_ids.extend(job_ids)
+                self.response["jobs"] = [{"job_id": value} for value in job_ids]
         super().__init__(
             f"Managed AK protocol mismatch: server did not acknowledge ak_id={ak_id}. "
             f"Returned Job IDs: {self.job_ids}. Verify these jobs before resubmitting; "

{ap_client-0.6.0.dev1 → ap_client-0.6.0.dev2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "ap-client"
-version = "0.6.0.dev1"
+version = "0.6.0.dev2"
 description = "Agent Platform API Client & CLI"
 readme = { text = "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.", content-type = "text/markdown" }
 requires-python = ">=3.10"