applied-cli 0.6.4__tar.gz → 0.6.5__tar.gz

{applied_cli-0.6.4 → applied_cli-0.6.5}/PKG-INFO

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

{applied_cli-0.6.4 → applied_cli-0.6.5}/applied_cli/cli.py

@@ -1928,6 +1928,15 @@ def scenario_bulk_run(
     anonymous: bool = typer.Option(
         False, "--anonymous", help="Run with an anonymous contact"
     ),
+    wait: bool = typer.Option(
+        False, "--wait", help="Poll until all runs finish, then print final status"
+    ),
+    wait_timeout: float = typer.Option(
+        300.0, "--wait-timeout", help="Max seconds to wait with --wait (default 300)"
+    ),
+    poll_interval: float = typer.Option(
+        3.0, "--poll-interval", help="Seconds between status polls with --wait"
+    ),
     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"
@@ -1944,6 +1953,9 @@ def scenario_bulk_run(
             contact_email=contact_email,
             contact_id=contact_id,
             anonymous=anonymous,
+            wait=wait,
+            wait_timeout=wait_timeout,
+            poll_interval=poll_interval,
             output_format=format,
         )
     )

{applied_cli-0.6.4 → applied_cli-0.6.5}/applied_cli/tools.py

@@ -8,6 +8,7 @@ import asyncio
 import difflib
 import json
 import re
+import time
 from contextlib import suppress
 from html.parser import HTMLParser
 from typing import Any
@@ -6082,6 +6083,41 @@ async def scenario_run_delete(
     return f"Scenario run {run_id} deleted successfully."
 
 
+def _bulk_status_counts(counts: dict | None) -> dict[str, int]:
+    """Normalize bulk-status counts to lowercase keys with int values."""
+    lowered: dict[str, int] = {}
+    for key, value in (counts or {}).items():
+        lowered[str(key).lower()] = int(value or 0)
+    return lowered
+
+
+def _bulk_pending_count(counts: dict | None) -> int:
+    """Count runs still queued or running (case-insensitive)."""
+    normalized = _bulk_status_counts(counts)
+    return normalized.get("queued", 0) + normalized.get("running", 0)
+
+
+async def _await_bulk_run(
+    client: AppliedClient,
+    job_id: str,
+    *,
+    timeout: float,
+    poll_interval: float,
+) -> tuple[dict, bool]:
+    """Poll a bulk run until no runs are queued/running or the timeout elapses.
+
+    Returns (latest_status_payload, timed_out).
+    """
+    start = time.monotonic()
+    status = await client.get_scenario_bulk_run_status(job_id)
+    while _bulk_pending_count(status.get("counts")) > 0:
+        if time.monotonic() - start >= timeout:
+            return status, True
+        await asyncio.sleep(poll_interval)
+        status = await client.get_scenario_bulk_run_status(job_id)
+    return status, False
+
+
 async def _resolve_contact_override(
     client: AppliedClient,
     *,
@@ -6123,6 +6159,9 @@ async def scenario_bulk_run(
     contact_id: str | None = None,
     contact_email: str | None = None,
     anonymous: bool = False,
+    wait: bool = False,
+    wait_timeout: float = 300.0,
+    poll_interval: float = 3.0,
     output_format: str = "text",
 ) -> str:
     """
@@ -6134,6 +6173,10 @@ async def scenario_bulk_run(
     to run the scenarios as a contact that has an email, so the test conversation
     carries it.
 
+    With wait=True, this polls until every run finishes (or the timeout elapses)
+    and returns the final status, so you can run a benchmark and read results in
+    one call instead of polling scenario_bulk_status yourself.
+
     Args:
         client: Authenticated AppliedClient
         scenario_ids: List of scenario UUIDs to run
@@ -6144,9 +6187,12 @@ async def scenario_bulk_run(
         contact_id: Run scenarios as this existing contact (gives test convos its email)
         contact_email: Resolve/create a contact with this email and run as them
         anonymous: Run with an anonymous contact (mode='anonymous')
+        wait: Poll until all runs finish (or wait_timeout elapses)
+        wait_timeout: Max seconds to wait when wait=True (default 300)
+        poll_interval: Seconds between status polls when wait=True (default 3)
 
     Returns:
-        Summary of runs created
+        Summary of runs created (plus the final status when wait=True)
     """
     resolved_scenario_ids = list(scenario_ids or [])
     if not resolved_scenario_ids:
@@ -6196,6 +6242,26 @@ async def scenario_bulk_run(
         "contact_override": result.get("contact_override"),
     }
 
+    job_id = payload.get("job_id")
+    final_status: dict | None = None
+    timed_out = False
+    if wait and job_id:
+        try:
+            final_status, timed_out = await _await_bulk_run(
+                client,
+                str(job_id),
+                timeout=wait_timeout,
+                poll_interval=poll_interval,
+            )
+        except AppliedAPIError as e:
+            return _format_error(e)
+        counts = _bulk_status_counts(final_status.get("counts"))
+        payload["final_counts"] = counts
+        payload["timed_out"] = timed_out
+        payload["duration_seconds"] = final_status.get("duration_seconds")
+        payload["completed_at"] = final_status.get("completed_at")
+        payload["failed"] = final_status.get("failed") or []
+
     if output_format == "json":
         return to_json(payload)
 
@@ -6212,6 +6278,23 @@ async def scenario_bulk_run(
         output += f"scenario_run_ids: {preview_ids}\n"
         if len(run_ids) > 10:
             output += f"more_runs: {len(run_ids) - 10}\n"
+
+    if final_status is not None:
+        counts = payload["final_counts"]
+        output += "\n# Final Status\n"
+        output += "timed_out: " + ("true (still pending)" if timed_out else "false") + "\n"
+        output += f"completed: {counts.get('completed', 0)}\n"
+        output += f"failed: {counts.get('failed', 0)}\n"
+        pending = counts.get("queued", 0) + counts.get("running", 0)
+        output += f"still_pending: {pending}\n"
+        if payload.get("duration_seconds") is not None:
+            output += f"duration_seconds: {payload['duration_seconds']}\n"
+        failed_runs = payload.get("failed") or []
+        if failed_runs:
+            output += f"\n# Failed Runs ({len(failed_runs)})\n"
+            output += to_json(failed_runs)
+        return output
+
     output += "\nTip: use scenario_bulk_status(job_id, include_runs=True) or scenario_run_list(bulk_job_id=job_id) to get per-run details with scenario mappings."
     return output
 
@@ -6245,14 +6328,14 @@ async def scenario_bulk_status(
             payload.pop("runs", None)
         return to_json(payload)
 
-    counts = result.get("counts") or {}
+    counts = _bulk_status_counts(result.get("counts"))
     output = "# Bulk Run Status\n"
     output += f"job_id: {result.get('job_id')}\n"
     output += f"total: {result.get('total')}\n"
-    output += f"queued: {counts.get('QUEUED', 0)}\n"
-    output += f"running: {counts.get('RUNNING', 0)}\n"
-    output += f"completed: {counts.get('COMPLETED', 0)}\n"
-    output += f"failed: {counts.get('FAILED', 0)}\n"
+    output += f"queued: {counts.get('queued', 0)}\n"
+    output += f"running: {counts.get('running', 0)}\n"
+    output += f"completed: {counts.get('completed', 0)}\n"
+    output += f"failed: {counts.get('failed', 0)}\n"
     output += f"created_at: {result.get('created_at')}\n"
     output += f"updated_at: {result.get('updated_at')}\n"
     if result.get("completed_at"):

{applied_cli-0.6.4 → applied_cli-0.6.5}/applied_cli/v2/scenarios.py

@@ -30,6 +30,9 @@ class ScenariosBulkRunInput(StrictInput):
     contact_id: str | None = None
     contact_email: str | None = None
     anonymous: bool = False
+    wait: bool = False
+    wait_timeout: float = 300.0
+    poll_interval: float = 3.0
 
 
 class ScenariosBulkCancelInput(StrictInput):
@@ -828,6 +831,48 @@ async def scenarios_bulk_run_handler(
         "duplicated_scenarios": result.get("duplicated_scenarios"),
         "contact_override": result.get("contact_override"),
     }
+
+    job_id = payload.get("job_id")
+    if params.wait and job_id:
+        from applied_cli.tools import _await_bulk_run, _bulk_status_counts
+
+        try:
+            final_status, timed_out = await _await_bulk_run(
+                client,
+                str(job_id),
+                timeout=params.wait_timeout,
+                poll_interval=params.poll_interval,
+            )
+        except AppliedAPIError as exc:
+            return _api_error_result(exc)
+        counts = _bulk_status_counts(final_status.get("counts"))
+        payload["final_counts"] = counts
+        payload["timed_out"] = timed_out
+        payload["duration_seconds"] = final_status.get("duration_seconds")
+        payload["failed"] = final_status.get("failed") or []
+        pending = counts.get("queued", 0) + counts.get("running", 0)
+        summary = (
+            f"Bulk job {job_id} "
+            + ("timed out with " if timed_out else "finished: ")
+            + f"{counts.get('completed', 0)} completed, "
+            + f"{counts.get('failed', 0)} failed"
+            + (f", {pending} still pending" if pending else "")
+            + "."
+        )
+        warnings = []
+        if counts.get("failed"):
+            warnings.append(f"{counts['failed']} run(s) failed.")
+        if timed_out:
+            warnings.append("Timed out before all runs finished.")
+        return ToolResult(
+            data=payload,
+            summary=summary,
+            warnings=warnings,
+            next_actions=[
+                "Use scenarios_bulk_status with include_runs=true to inspect runs.",
+            ],
+        )
+
     queued = payload.get("queued") or 0
     return ToolResult(
         data=payload,
@@ -1064,7 +1109,8 @@ def scenario_specs() -> list[ToolSpec]:
                 "Run selected scenarios or every scenario in a benchmark and "
                 "return the queued job metadata. Pass contact_email or contact_id "
                 "to run as a contact with an email (fixes 'Email is not present' "
-                "failures on test conversations)."
+                "failures on test conversations). Pass wait=true to block until "
+                "all runs finish and return the final status in one call."
             ),
             input_model=ScenariosBulkRunInput,
             output_model=None,

{applied_cli-0.6.4 → applied_cli-0.6.5}/applied_cli.egg-info/PKG-INFO

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

{applied_cli-0.6.4 → applied_cli-0.6.5}/applied_cli.egg-info/SOURCES.txt

@@ -51,6 +51,7 @@ tests/test_knowledge_content_tools.py
 tests/test_recovery.py
 tests/test_scenario_bulk_cancel.py
 tests/test_scenario_bulk_run_contact.py
+tests/test_scenario_bulk_run_wait.py
 tests/test_toolkit_contract.py
 tests/test_v2_agents.py
 tests/test_v2_articles.py

{applied_cli-0.6.4 → applied_cli-0.6.5}/pyproject.toml

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

applied_cli-0.6.5/tests/test_scenario_bulk_run_wait.py

@@ -0,0 +1,107 @@
+import json
+
+import pytest
+
+from applied_cli import tools
+
+
+class FakeWaitClient:
+    """Bulk client whose status transitions to done after N polls."""
+
+    def __init__(self, status_sequence):
+        self._status_sequence = list(status_sequence)
+        self._poll = 0
+        self.status_calls = 0
+
+    async def list_scenarios(self, benchmark_id=None, limit=500, **kwargs):
+        return [{"id": "s1"}, {"id": "s2"}]
+
+    async def bulk_run_scenarios(
+        self, scenario_ids=None, target_agent_id=None, contact_override=None
+    ):
+        return {
+            "job_id": "job-1",
+            "total": len(scenario_ids or []),
+            "queued": len(scenario_ids or []),
+            "scenario_run_ids": ["r1", "r2"],
+        }
+
+    async def get_scenario_bulk_run_status(self, job_id):
+        self.status_calls += 1
+        idx = min(self._poll, len(self._status_sequence) - 1)
+        self._poll += 1
+        return self._status_sequence[idx]
+
+
+@pytest.mark.asyncio
+async def test_wait_polls_until_no_pending(monkeypatch):
+    # Avoid real sleeping between polls.
+    async def _no_sleep(_seconds):
+        return None
+
+    monkeypatch.setattr(tools.asyncio, "sleep", _no_sleep)
+
+    client = FakeWaitClient(
+        status_sequence=[
+            {"counts": {"queued": 2, "running": 0, "completed": 0, "failed": 0}},
+            {"counts": {"queued": 0, "running": 1, "completed": 1, "failed": 0}},
+            {
+                "counts": {"queued": 0, "running": 0, "completed": 2, "failed": 0},
+                "duration_seconds": 12.5,
+                "completed_at": "2026-06-05T10:00:00Z",
+                "failed": [],
+            },
+        ]
+    )
+    result = await tools.scenario_bulk_run(
+        client, benchmark_id="bench-1", wait=True, output_format="json"
+    )
+    data = json.loads(result)
+    assert data["timed_out"] is False
+    assert data["final_counts"]["completed"] == 2
+    assert data["duration_seconds"] == 12.5
+    assert client.status_calls == 3
+
+
+@pytest.mark.asyncio
+async def test_wait_times_out_when_runs_stay_pending(monkeypatch):
+    async def _no_sleep(_seconds):
+        return None
+
+    monkeypatch.setattr(tools.asyncio, "sleep", _no_sleep)
+
+    # Always pending → must hit the timeout path.
+    client = FakeWaitClient(
+        status_sequence=[
+            {"counts": {"queued": 2, "running": 0, "completed": 0, "failed": 0}}
+        ]
+    )
+    result = await tools.scenario_bulk_run(
+        client,
+        benchmark_id="bench-1",
+        wait=True,
+        wait_timeout=0.0,  # immediate timeout after first poll
+        output_format="json",
+    )
+    data = json.loads(result)
+    assert data["timed_out"] is True
+    assert data["final_counts"]["queued"] == 2
+
+
+@pytest.mark.asyncio
+async def test_no_wait_returns_started_summary():
+    client = FakeWaitClient(status_sequence=[{"counts": {}}])
+    result = await tools.scenario_bulk_run(
+        client, benchmark_id="bench-1", output_format="json"
+    )
+    data = json.loads(result)
+    assert "final_counts" not in data
+    assert client.status_calls == 0  # no polling when wait is False
+
+
+def test_bulk_status_counts_normalizes_case_and_types():
+    assert tools._bulk_status_counts({"QUEUED": 2, "Running": "1"}) == {
+        "queued": 2,
+        "running": 1,
+    }
+    assert tools._bulk_pending_count({"QUEUED": 3, "RUNNING": 4, "COMPLETED": 9}) == 7