mammoth-cli 2.2.0__tar.gz → 2.2.2__tar.gz

{mammoth_cli-2.2.0 → mammoth_cli-2.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mammoth-cli
-Version: 2.2.0
+Version: 2.2.2
 Summary: Command-line interface for the Mammoth Analytics platform
 Project-URL: homepage, https://mammoth.io
 Project-URL: repository, https://github.com/EdgeMetric/mammothsdk
@@ -17,7 +17,7 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: <3.15,>=3.12
 Requires-Dist: jsonschema>=4.20
 Requires-Dist: keyring<26,>=25.7
-Requires-Dist: mammoth-io<0.9,>=0.8.3
+Requires-Dist: mammoth-io<0.9,>=0.8.4
 Requires-Dist: platformdirs<5,>=4.10
 Requires-Dist: pydantic<3,>=2.12.5
 Requires-Dist: pyyaml<7,>=6

{mammoth_cli-2.2.0 → mammoth_cli-2.2.2}/mammoth_cli/__init__.py

@@ -6,7 +6,7 @@ not implement a second HTTP client and does not call private SDK members.
 
 from __future__ import annotations
 
-__version__ = "2.2.0"
+__version__ = "2.2.2"
 
 # The versioned machine output envelope contract.
 SCHEMA_VERSION = 1

{mammoth_cli-2.2.0 → mammoth_cli-2.2.2}/mammoth_cli/bundled_skill/mammoth-cli/references/command-index.md

@@ -1,12 +1,12 @@
 # Published CLI command catalog
 
-This generated catalog is a lookup table for every command in the published CLI manifest, currently 588; each entry carries its release status line. It is a command-contract index, not an assertion that every backend/OpenAPI operation has a CLI binding. For an exact local CLI contract, run `mammoth schema get COMMAND_ID`; `mammoth capability list` is an API-binding inventory and can omit typed/local CLI routes. For focused workflows, read the [recipes index](recipes/index.md). Sensitive structured input must come from a private file or pipe; never put secrets in literal argv.
+This generated catalog is a lookup table for every command in the published CLI manifest, currently 589; each entry carries its release status line. It is a command-contract index, not an assertion that every backend/OpenAPI operation has a CLI binding. For an exact local CLI contract, run `mammoth schema get COMMAND_ID`; `mammoth capability list` is an API-binding inventory and can omit typed/local CLI routes. For focused workflows, read the [recipes index](recipes/index.md). Sensitive structured input must come from a private file or pipe; never put secrets in literal argv.
 
 Load only the applicable domain file:
 
 - [activity](commands/activity.md) — 2 published commands
 - [addon](commands/addon.md) — 7 published commands
-- [agent](commands/agent.md) — 13 published commands
+- [agent](commands/agent.md) — 14 published commands
 - [ai](commands/ai.md) — 5 published commands
 - [annotation](commands/annotation.md) — 5 published commands
 - [auth](commands/auth.md) — 3 published commands

{mammoth_cli-2.2.0 → mammoth_cli-2.2.2}/mammoth_cli/bundled_skill/mammoth-cli/references/commands/agent.md

@@ -92,6 +92,16 @@ Result: `AgentRunStopResult`; mutation `benign_mutation`, confirmation `none`, w
 
 Status on release: untried; no live run recorded.
 
+### `agent.run.units.set`
+
+Run: `mammoth agent run units set`. Exact input fields: `mammoth schema get agent.run.units.set`.
+
+Example: `mammoth agent run units set resource-123 --input '{"step": 1, "kind": "sample", "units": [{"sample_key": "Status"}]}'`. Placeholders are illustrative; resolve IDs and input from observed reads.
+
+Result: `AgentRunUnitsSetResult`; mutation `benign_mutation`, confirmation `none`, wait policy `not_async`.
+
+Status on release: untried; no live run recorded.
+
 ### `agent.session.delete`
 
 Run: `mammoth agent session delete`. Exact input fields: `mammoth schema get agent.session.delete`.

{mammoth_cli-2.2.0 → mammoth_cli-2.2.2}/mammoth_cli/commands/agent.py

@@ -263,3 +263,11 @@ def _run_control(invocation: Invocation) -> HandlerResult:
 
 
 agent_run_pause = agent_run_resume = agent_run_stop = agent_run_extend = _run_control
+
+
+def agent_run_units_set(invocation: Invocation) -> HandlerResult:
+    """Report the objects one plan step of a run will work on, as queued units."""
+    run_id = _require_string_positional(invocation, "run id")
+    document = _bound_document(invocation)
+    kwargs = {field: _require_field(document, field) for field in ("step", "kind", "units")}
+    return _session_call(invocation, run_id=run_id, **kwargs)

{mammoth_cli-2.2.0 → mammoth_cli-2.2.2}/mammoth_cli/commands/dataset.py

@@ -681,12 +681,23 @@ def dataset_delete(invocation: Invocation) -> HandlerResult:
         if hint is not None:
             action = f"{action}. {hint}"
         enforce_confirmation(invocation, policy=POLICY_PROMPT_OR_YES, action=action)
-        data = service.call(_symbol(invocation), dataset_id=dataset_id, project_id=project_id)
+        ack = service.call(_symbol(invocation), dataset_id=dataset_id, project_id=project_id)
+        # The server acknowledges a delete with a job handle. Settle it here so
+        # the handle's id is kept as ``job.id`` (the shape ``view delete`` has).
+        data = _with_job(ack, service.wait_if_job(ack))
     if hint is not None and isinstance(data, dict):
         data = {**data, "hint": hint}
     return data, _meta(invocation, auth.workspace_id, project_id)
 
 
+def _with_job(handle: Any, settled: Any) -> Any:
+    """Return the settled result with the acknowledged job's id as ``job.id``."""
+    job_id = handle.get("job_id") if isinstance(handle, dict) else None
+    if job_id is None or not isinstance(settled, dict):
+        return settled
+    return {**settled, "job": {"id": job_id}}
+
+
 def dataset_bulk_delete(invocation: Invocation) -> HandlerResult:
     """Permanently delete the listed datasets. ``dataset_ids`` from ``--input``; ``--yes`` required.
 

{mammoth_cli-2.2.0 → mammoth_cli-2.2.2}/mammoth_cli/commands/registry.py

@@ -544,6 +544,7 @@ HANDLERS: dict[str, Handler] = {
     "agent.run.resume": agent_cmd.agent_run_resume,
     "agent.run.status": agent_cmd.agent_run_status,
     "agent.run.stop": agent_cmd.agent_run_stop,
+    "agent.run.units.set": agent_cmd.agent_run_units_set,
     "agent.session.delete": agent_cmd.agent_session_delete,
     "agent.session.list": agent_cmd.agent_session_list,
     "agent.session.messages": agent_cmd.agent_session_messages,

{mammoth_cli-2.2.0 → mammoth_cli-2.2.2}/mammoth_cli/services/input_fields.py

@@ -19,6 +19,7 @@ _HANDLER_OWNED_FIELDS: dict[str, frozenset[str]] = {
             "run.resume",
             "run.stop",
             "run.extend",
+            "run.units.set",
         )
     },
     # Skill handlers deliberately derive filesystem roots from the running

{mammoth_cli-2.2.0 → mammoth_cli-2.2.2}/mammoth_cli/services/positionals.py

@@ -140,6 +140,14 @@ POSITIONAL_OVERRIDES: dict[str, tuple[PositionalSpec, ...]] = {
             help="ID of the recorded action (from agent action list).",
         ),
     ),
+    "agent.run.units.set": (
+        PositionalSpec(
+            name="run_id",
+            type=str,
+            required=True,
+            help="ID of the run (from agent run status or agent run list).",
+        ),
+    ),
     **{
         f"agent.run.{verb}": (
             PositionalSpec(

{mammoth_cli-2.2.0 → mammoth_cli-2.2.2}/pyproject.toml

@@ -1,13 +1,13 @@
 [project]
 name = "mammoth-cli"
-version = "2.2.0"
+version = "2.2.2"
 description = "Command-line interface for the Mammoth Analytics platform"
 authors = [{name = "Mammoth Analytics", email = "support@mammoth.io"}]
 readme = "README.md"
 license = "LicenseRef-Proprietary"
 requires-python = ">=3.12,<3.15"
 dependencies = [
-    "mammoth-io (>=0.8.3,<0.9)",
+    "mammoth-io (>=0.8.4,<0.9)",
     "typer (>=0.27,<0.28)",
     "rich (>=15,<16)",
     "platformdirs (>=4.10,<5)",

{mammoth_cli-2.2.0 → mammoth_cli-2.2.2}/spec/manifests/commands/agent.yaml

@@ -368,6 +368,52 @@ commands:
     command: agent.run.status
     ids: {}
     kind: object
+- command_id: agent.run.units.set
+  command_path: agent run units set
+  disposition: command
+  alias_of: null
+  operation_ids: []
+  positionals:
+  - name: run_id
+    type: str
+    required: true
+    metavar: RUN_ID
+    falls_back_to_field: null
+    help: ID of the run (from agent run status or agent run list).
+  options: []
+  sdk_symbol: mammoth.api.agents.AgentsAPI.run_units_set
+  sdk_conversion: Call mammoth.api.agents.AgentsAPI.run_units_set with the session id from --session, the
+    positional run id and validated request fields.
+  request_model: AgentRunUnitsSetRequest
+  result_model: AgentRunUnitsSetResult
+  mutation_class: benign_mutation
+  edits_target: false
+  confirmation: none
+  wait_policy: not_async
+  pagination_policy: none
+  acceptance_evidence: live_disposable_project
+  live_test: null
+  live_exemption_reason: Live coverage runs on the test server with a chat session that has a run.
+  contract_fixture: null
+  required_fixture_guard: null
+  secret_fields: []
+  human_example: mammoth agent run units set RUN_ID --help
+  agent_example: 'mammoth agent run units set resource-123 --input ''{"step": 1, "kind": "sample", "units": [{"sample_key": "Status"}]}'''
+  unit_tests:
+  - UT-AGENTRUNUNITSSET
+  contract_tests:
+  - CT-AGENTRUNUNITSSET
+  draft_test: null
+  undo_test: null
+  known_restrictions: 'Writes. Reports once the objects a plan step will work on (kind is dataset, view and so on;
+    1 to 1000 units, each id with an optional name and project_id) so the run panel can list them and
+    measure progress and the time left. A unit already listed is left as it is. Needs --session SESSION_ID;
+    an embedded chat supplies it. The server, not the model, counts the units done and the time left.'
+  reviewed_by: primary
+  readback:
+    command: agent.run.status
+    ids: {}
+    kind: object
 - command_id: agent.session.delete
   command_path: agent session delete
   disposition: command

{mammoth_cli-2.2.0 → mammoth_cli-2.2.2}/spec/manifests/sdk-catalog.source.yaml

@@ -383,6 +383,13 @@ sdk_methods:
   wait_policy: not_async
   pagination_policy: none
   openapi_operation_ids: []
+- sdk_symbol: mammoth.api.agents.AgentsAPI.run_units_set
+  canonical_command: agent.run.units.set
+  alias_of: null
+  mutation_class: benign_mutation
+  wait_policy: not_async
+  pagination_policy: none
+  openapi_operation_ids: []
 - sdk_symbol: mammoth.api.agents.AgentsAPI.session_delete
   canonical_command: agent.session.delete
   alias_of: null

{mammoth_cli-2.2.0 → mammoth_cli-2.2.2}/spec/manifests/sdk-methods.yaml

@@ -631,6 +631,18 @@ methods:
   pagination_policy: none
   secret_fields: []
   reviewed_by: primary
+- sdk_symbol: mammoth.api.agents.AgentsAPI.run_units_set
+  implementation_origin: mammoth.api.agents.AgentsAPI.run_units_set
+  signature: 'run_units_set(self, session_id: ''str'', run_id: ''str'', step: ''int'', kind: ''str'', units: ''list[dict[str, Any]]'') -> ''dict[str, Any]'''
+  openapi_operation_ids: []
+  canonical_command: agent.run.units.set
+  alias_of: null
+  no_command_reason: null
+  mutation_class: benign_mutation
+  wait_policy: not_async
+  pagination_policy: none
+  secret_fields: []
+  reviewed_by: primary
 - sdk_symbol: mammoth.api.agents.AgentsAPI.session_delete
   implementation_origin: mammoth.api.agents.AgentsAPI.session_delete
   signature: 'session_delete(self, session_id: ''str'') -> ''dict[str, Any]'''