applied-cli 0.6.2__tar.gz → 0.6.3__tar.gz

{applied_cli-0.6.2 → applied_cli-0.6.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: applied-cli
-Version: 0.6.2
+Version: 0.6.3
 Summary: CLI and shared client library for Applied Labs AI support agents
 Author: Applied Labs
 License-Expression: MIT

{applied_cli-0.6.2 → applied_cli-0.6.3}/applied_cli/cli.py

@@ -1956,6 +1956,27 @@ def scenario_bulk_status(
     typer.echo(result)
 
 
+@app.command("scenario-bulk-cancel")
+def scenario_bulk_cancel(
+    job_id: str = typer.Argument(..., help="Bulk run job ID"),
+    apply: bool = typer.Option(
+        False, "--apply", help="Cancel the pending runs (default is a dry-run plan)"
+    ),
+    shop_id: str = typer.Option(None, "--shop-id", help="Override shop ID"),
+    format: str = typer.Option(
+        "text", "--format", "-f", help="Output format: text or json"
+    ),
+) -> None:
+    """Cancel a stuck bulk run by deleting its queued/running scenario runs."""
+    client = get_client(shop_id=shop_id)
+    result = asyncio.run(
+        tools.scenario_bulk_cancel(
+            client, job_id=job_id, apply=apply, output_format=format
+        )
+    )
+    typer.echo(result)
+
+
 @app.command("audit-metric-fields")
 def audit_metric_fields(
     is_active: bool = typer.Option(

{applied_cli-0.6.2 → applied_cli-0.6.3}/applied_cli/tools.py

@@ -6230,6 +6230,95 @@ async def scenario_bulk_status(
     return output
 
 
+_PENDING_RUN_STATUSES = ("queued", "running")
+
+
+async def scenario_bulk_cancel(
+    client: AppliedClient,
+    job_id: str,
+    *,
+    apply: bool = False,
+    output_format: str = "text",
+) -> str:
+    """
+    Cancel a stuck bulk scenario run job.
+
+    A bulk run can get stuck with runs left in 'queued' or 'running' forever, so
+    the job never reports complete. This cancels the job by deleting those pending
+    runs; completed and failed runs (and their result conversations) are preserved.
+    Once no pending runs remain, the job reports as finished.
+
+    This replaces the manual workaround of deleting the 'agent-test-bulk-job:'
+    key from browser localStorage.
+
+    Args:
+        client: Authenticated AppliedClient
+        job_id: The bulk run job UUID
+        apply: If True, delete the pending runs; otherwise just report the plan
+        output_format: 'text' (default) or 'json'
+
+    Returns:
+        Summary of pending runs cancelled (or that would be cancelled).
+    """
+    try:
+        runs = await client.list_scenario_runs(
+            bulk_job_id=job_id, latest=False, limit=1000
+        )
+    except AppliedAPIError as e:
+        return _format_error(e)
+
+    pending = [r for r in runs if str(r.get("status") or "") in _PENDING_RUN_STATUSES]
+    result: dict[str, Any] = {
+        "job_id": job_id,
+        "apply": apply,
+        "total_runs": len(runs),
+        "pending_runs": len(pending),
+        "terminal_runs": len(runs) - len(pending),
+        "cancelled": 0,
+        "errors": [],
+    }
+
+    if not runs:
+        result["message"] = (
+            f"No runs found for job {job_id} — it may already be cleared or the "
+            f"job id is invalid. Nothing to cancel."
+        )
+        return to_json(result) if output_format == "json" else result["message"]
+
+    if not pending:
+        result["message"] = (
+            f"Job {job_id} has no pending (queued/running) runs — nothing to "
+            f"cancel. {result['terminal_runs']} run(s) already finished."
+        )
+        return to_json(result) if output_format == "json" else result["message"]
+
+    if not apply:
+        result["message"] = (
+            f"Would cancel {len(pending)} pending run(s) (queued/running) for job "
+            f"{job_id}; {result['terminal_runs']} finished run(s) preserved. "
+            f"Re-run with --apply to cancel."
+        )
+        return to_json(result) if output_format == "json" else result["message"]
+
+    cancelled = 0
+    for run in pending:
+        run_id = run.get("id")
+        if not run_id:
+            continue
+        try:
+            await client.delete_scenario_run(str(run_id))
+            cancelled += 1
+        except AppliedAPIError as e:
+            result["errors"].append({"run_id": str(run_id), "error": str(e)})
+    result["cancelled"] = cancelled
+    result["message"] = (
+        f"Cancelled {cancelled} pending run(s) for job {job_id}; "
+        f"{result['terminal_runs']} finished run(s) preserved."
+        + (f" {len(result['errors'])} error(s)." if result["errors"] else "")
+    )
+    return to_json(result) if output_format == "json" else result["message"]
+
+
 async def _load_audit_target_summaries(
     client: AppliedClient,
     ratings: list[dict[str, Any]],

{applied_cli-0.6.2 → applied_cli-0.6.3}/applied_cli/v2/domains.py

@@ -119,6 +119,7 @@ DOMAIN_TOOL_RENAMES: dict[str, dict[str, str]] = {
         "scenario_run_delete": "scenarios_runs_delete",
         "scenario_bulk_run": "scenarios_bulk_run",
         "scenario_bulk_status": "scenarios_bulk_status",
+        "scenario_bulk_cancel": "scenarios_bulk_cancel",
     },
     "taxonomy": {
         "taxonomy_list": "taxonomy_items_list",

{applied_cli-0.6.2 → applied_cli-0.6.3}/applied_cli/v2/scenarios.py

@@ -29,6 +29,11 @@ class ScenariosBulkRunInput(StrictInput):
     contact_override: dict[str, Any] | None = None
 
 
+class ScenariosBulkCancelInput(StrictInput):
+    job_id: str
+    apply: bool = False
+
+
 class BenchmarksListInput(StrictInput):
     agent_id: str | None = None
     limit: int = 50
@@ -850,6 +855,38 @@ async def scenarios_bulk_status_handler(
     )
 
 
+async def scenarios_bulk_cancel_handler(
+    client: AppliedClient,
+    params: ScenariosBulkCancelInput,
+) -> ToolResult[Any]:
+    from applied_cli import tools as legacy_tools
+
+    raw = await legacy_tools.scenario_bulk_cancel(
+        client,
+        job_id=params.job_id,
+        apply=params.apply,
+        output_format="json",
+    )
+    try:
+        data = json.loads(raw)
+    except (json.JSONDecodeError, TypeError):
+        return ToolResult(data={"message": raw}, summary=str(raw))
+
+    next_actions = []
+    if not params.apply and data.get("pending_runs"):
+        next_actions.append("Re-run with apply=true to cancel the pending runs.")
+    elif data.get("cancelled"):
+        next_actions.append(
+            f"Poll scenarios_bulk_status with job_id='{params.job_id}' to confirm "
+            f"the job now reports finished."
+        )
+    return ToolResult(
+        data=data,
+        summary=data.get("message", "Bulk cancel processed."),
+        next_actions=next_actions,
+    )
+
+
 def scenario_specs() -> list[ToolSpec]:
     return [
         ToolSpec(
@@ -1034,4 +1071,18 @@ def scenario_specs() -> list[ToolSpec]:
             read_write_mode="read",
             tags=["scenario_bulk_status", "native"],
         ),
+        ToolSpec(
+            name="scenarios_bulk_cancel",
+            namespace="scenarios",
+            description=(
+                "Cancel a stuck bulk scenario run job by deleting its queued/"
+                "running runs (completed/failed runs are preserved). Dry-run by "
+                "default; set apply=true to cancel."
+            ),
+            input_model=ScenariosBulkCancelInput,
+            output_model=None,
+            handler=scenarios_bulk_cancel_handler,
+            read_write_mode="write",
+            tags=["scenario_bulk_cancel", "native"],
+        ),
     ]

{applied_cli-0.6.2 → applied_cli-0.6.3}/applied_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: applied-cli
-Version: 0.6.2
+Version: 0.6.3
 Summary: CLI and shared client library for Applied Labs AI support agents
 Author: Applied Labs
 License-Expression: MIT

{applied_cli-0.6.2 → applied_cli-0.6.3}/applied_cli.egg-info/SOURCES.txt

@@ -49,6 +49,7 @@ tests/test_conversation_tools.py
 tests/test_flow_tools.py
 tests/test_knowledge_content_tools.py
 tests/test_recovery.py
+tests/test_scenario_bulk_cancel.py
 tests/test_toolkit_contract.py
 tests/test_v2_agents.py
 tests/test_v2_articles.py

{applied_cli-0.6.2 → applied_cli-0.6.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "applied-cli"
-version = "0.6.2"
+version = "0.6.3"
 description = "CLI and shared client library for Applied Labs AI support agents"
 readme = "README.md"
 requires-python = ">=3.11"

applied_cli-0.6.3/tests/test_scenario_bulk_cancel.py

@@ -0,0 +1,87 @@
+import json
+
+import pytest
+
+from applied_cli import tools
+
+
+class FakeBulkClient:
+    def __init__(self, runs):
+        self._runs = runs
+        self.deleted = []
+
+    async def list_scenario_runs(
+        self, scenario_id=None, benchmark_id=None, bulk_job_id=None, latest=False,
+        limit=50,
+    ):
+        return list(self._runs)
+
+    async def delete_scenario_run(self, run_id):
+        self.deleted.append(run_id)
+
+
+STUCK_RUNS = [
+    {"id": "r1", "status": "completed"},
+    {"id": "r2", "status": "queued"},
+    {"id": "r3", "status": "running"},
+    {"id": "r4", "status": "failed"},
+]
+
+
+@pytest.mark.asyncio
+async def test_dry_run_reports_pending_without_deleting():
+    client = FakeBulkClient(STUCK_RUNS)
+    result = await tools.scenario_bulk_cancel(client, "job-1", output_format="json")
+    data = json.loads(result)
+    assert data["pending_runs"] == 2
+    assert data["terminal_runs"] == 2
+    assert data["cancelled"] == 0
+    assert client.deleted == []  # dry run deletes nothing
+
+
+@pytest.mark.asyncio
+async def test_apply_deletes_only_pending_runs():
+    client = FakeBulkClient(STUCK_RUNS)
+    result = await tools.scenario_bulk_cancel(
+        client, "job-1", apply=True, output_format="json"
+    )
+    data = json.loads(result)
+    assert data["cancelled"] == 2
+    # Only the queued/running runs are deleted; completed/failed preserved.
+    assert set(client.deleted) == {"r2", "r3"}
+
+
+@pytest.mark.asyncio
+async def test_no_pending_runs_is_a_noop():
+    client = FakeBulkClient(
+        [{"id": "r1", "status": "completed"}, {"id": "r2", "status": "failed"}]
+    )
+    result = await tools.scenario_bulk_cancel(
+        client, "job-1", apply=True, output_format="text"
+    )
+    assert "nothing to" in result.lower()
+    assert client.deleted == []
+
+
+@pytest.mark.asyncio
+async def test_unknown_job_reports_no_runs():
+    client = FakeBulkClient([])
+    result = await tools.scenario_bulk_cancel(client, "missing", output_format="text")
+    assert "No runs found" in result
+    assert client.deleted == []
+
+
+@pytest.mark.asyncio
+async def test_v2_scenarios_bulk_cancel_handler_structured():
+    from applied_cli.v2.scenarios import (
+        ScenariosBulkCancelInput,
+        scenarios_bulk_cancel_handler,
+    )
+
+    client = FakeBulkClient(STUCK_RUNS)
+    result = await scenarios_bulk_cancel_handler(
+        client, ScenariosBulkCancelInput(job_id="job-1", apply=False)
+    )
+    assert result.data["pending_runs"] == 2
+    assert "apply=true" in " ".join(result.next_actions)
+    assert client.deleted == []