hte-cli 0.2.24__py3-none-any.whl → 0.2.25__py3-none-any.whl

hte_cli/cli.py

@@ -172,6 +172,17 @@ def session_join(ctx, session_id: str, force_setup: bool):
         console.print("[red]Not logged in. Run: hte-cli auth login[/red]")
         sys.exit(1)
 
+    # Check Docker is running before we start (with retry prompt)
+    while True:
+        docker_ok, docker_error = _check_docker()
+        if docker_ok:
+            break
+        console.print(f"[red]{docker_error}[/red]")
+        console.print()
+        if not click.confirm("Start Docker and retry?", default=True):
+            sys.exit(1)
+        console.print("[dim]Checking Docker again...[/dim]")
+
     api = APIClient(config)
 
     # Step 1: Join session
@@ -201,8 +212,11 @@ def session_join(ctx, session_id: str, force_setup: bool):
     # Check if reconnecting (session already in_progress)
     is_reconnect = session_info.get("status") == "in_progress"
 
-    if is_reconnect and not force_setup:
-        console.print("[yellow]Reconnecting to existing session...[/yellow]")
+    # Always run setup on reconnect - previous attempt may have failed
+    # (e.g., image pull failed, Docker wasn't running, etc.)
+    if is_reconnect:
+        force_setup = True
+        console.print("[yellow]Reconnecting to existing session (re-running setup)...[/yellow]")
         console.print()
 
     console.print(
@@ -219,7 +233,11 @@ def session_join(ctx, session_id: str, force_setup: bool):
     import time
     from hte_cli.events import EventStreamer
     from hte_cli.runner import TaskRunner
-    from hte_cli.image_utils import extract_images_from_compose, pull_image_with_progress
+    from hte_cli.image_utils import (
+        extract_images_from_compose,
+        extract_image_platforms_from_compose,
+        pull_image_with_progress,
+    )
 
     # Create event streamer
     events = EventStreamer(api, session_id)
@@ -285,9 +303,11 @@ def session_join(ctx, session_id: str, force_setup: bool):
     failed_images = []
 
     if not is_reconnect or force_setup:
-        # Extract images from compose
+        # Extract images and their platforms from compose
+        image_platforms = {}
         if compose_yaml:
             images = extract_images_from_compose(compose_yaml)
+            image_platforms = extract_image_platforms_from_compose(compose_yaml)
 
         # Send setup_started event (includes CLI version for debugging)
         events.setup_started(images=images, cli_version=__version__)
@@ -298,9 +318,11 @@ def session_join(ctx, session_id: str, force_setup: bool):
 
             console.print(f"[bold]Step 2:[/bold] Pulling {len(images)} Docker image(s)...")
             pull_start = time.monotonic()
+            pull_errors = {}
 
             for img in images:
                 short_name = img.split("/")[-1][:40]
+                platform = image_platforms.get(img)
 
                 # Check if already cached
                 if check_image_exists_locally(img):
@@ -310,6 +332,7 @@ def session_join(ctx, session_id: str, force_setup: bool):
 
                 # Need to pull - show progress
                 last_status = ["connecting..."]
+                last_error = [""]
                 with console.status(
                     f"[yellow]↓[/yellow] {short_name} [dim]connecting...[/dim]"
                 ) as status:
@@ -328,14 +351,23 @@ def session_join(ctx, session_id: str, force_setup: bool):
                                     status.update(
                                         f"[yellow]↓[/yellow] {short_name} [dim]{display}[/dim]"
                                     )
+                        # Capture error messages
+                        if "error" in line.lower() or "denied" in line.lower():
+                            last_error[0] = line
 
-                    success = pull_image_with_progress(img, on_progress=show_progress)
+                    success = pull_image_with_progress(
+                        img, platform=platform, on_progress=show_progress
+                    )
 
                 if success:
                     console.print(f"  [green]✓[/green] {short_name} [dim](downloaded)[/dim]")
                     pulled_images.append(img)
                 else:
-                    console.print(f"  [red]✗[/red] {short_name} [dim](failed)[/dim]")
+                    platform_note = f" (platform: {platform})" if platform else ""
+                    console.print(f"  [red]✗[/red] {short_name}{platform_note} [dim](failed)[/dim]")
+                    if last_error[0]:
+                        console.print(f"      [dim]{last_error[0][:60]}[/dim]")
+                        pull_errors[img] = last_error[0]
                     failed_images.append(img)
 
             pull_duration = time.monotonic() - pull_start
@@ -347,6 +379,20 @@ def session_join(ctx, session_id: str, force_setup: bool):
             )
             console.print()
 
+            # Fail fast if any required image couldn't be pulled
+            if failed_images:
+                console.print(
+                    f"[red]Error: Failed to pull {len(failed_images)} required Docker image(s).[/red]"
+                )
+                console.print()
+                console.print("[yellow]Troubleshooting:[/yellow]")
+                console.print("  1. Check Docker is running: docker info")
+                console.print("  2. Try manual pull: docker pull python:3.12-slim --platform linux/amd64")
+                console.print("  3. Check network connectivity")
+                console.print()
+                console.print("Session remains active - you can retry with: hte-cli session join " + session_id)
+                sys.exit(1)
+
         # Send setup_completed - THIS STARTS THE TIMER ON SERVER
         total_setup = time.monotonic() - setup_start_time
         events.setup_completed(total_seconds=total_setup)
@@ -644,7 +690,7 @@ def _check_docker() -> tuple[bool, str | None]:
             timeout=10,
         )
         if result.returncode != 0:
-            return False, "Docker is not running. Start Docker Desktop or the Docker daemon."
+            return False, "Docker is not running. Start Docker (Docker Desktop, colima, or dockerd)."
     except FileNotFoundError:
         return False, "Docker is not installed. Install from https://docs.docker.com/get-docker/"
     except Exception as e:

hte_cli/image_utils.py

@@ -38,6 +38,33 @@ def extract_images_from_compose(compose_yaml: str) -> list[str]:
         return []
 
 
+def extract_image_platforms_from_compose(compose_yaml: str) -> dict[str, str | None]:
+    """
+    Extract Docker image names and their platforms from a compose.yaml string.
+
+    Args:
+        compose_yaml: Docker Compose YAML content
+
+    Returns:
+        Dict mapping image names to their platform (or None if no platform specified)
+    """
+    try:
+        compose_data = yaml.safe_load(compose_yaml)
+        if not compose_data or "services" not in compose_data:
+            return {}
+
+        image_platforms = {}
+        for service_name, service_config in compose_data.get("services", {}).items():
+            if isinstance(service_config, dict) and "image" in service_config:
+                image = service_config["image"]
+                platform = service_config.get("platform")
+                image_platforms[image] = platform
+        return image_platforms
+    except yaml.YAMLError as e:
+        logger.warning(f"Failed to parse compose.yaml: {e}")
+        return {}
+
+
 def check_image_exists_locally(image: str) -> bool:
     """
     Check if a Docker image exists locally.
@@ -61,16 +88,20 @@ def check_image_exists_locally(image: str) -> bool:
 
 def pull_image_with_progress(
     image: str,
+    platform: str | None = None,
     on_progress: Callable[[str, str], None] | None = None,
     on_complete: Callable[[str, bool], None] | None = None,
+    on_error: Callable[[str, str], None] | None = None,
 ) -> bool:
     """
     Pull a Docker image with progress callbacks using PTY for real progress output.
 
     Args:
         image: Image name to pull
+        platform: Optional platform to pull (e.g., "linux/amd64")
         on_progress: Callback(image, status_line) called for each progress update
         on_complete: Callback(image, success) called when pull completes
+        on_error: Callback(image, error_message) called when pull fails
 
     Returns:
         True if pull succeeded, False otherwise
@@ -79,8 +110,12 @@ def pull_image_with_progress(
         # Use PTY to get real progress output from docker
         master_fd, slave_fd = pty.openpty()
 
+        cmd = ["docker", "pull", image]
+        if platform:
+            cmd.extend(["--platform", platform])
+
         process = subprocess.Popen(
-            ["docker", "pull", image],
+            cmd,
             stdout=slave_fd,
             stderr=slave_fd,
             stdin=slave_fd,

{hte_cli-0.2.24.dist-info → hte_cli-0.2.25.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hte-cli
-Version: 0.2.24
+Version: 0.2.25
 Summary: Human Time-to-Completion Evaluation CLI
 Project-URL: Homepage, https://github.com/sean-peters-au/lyptus-mono
 Author: Lyptus Research

{hte_cli-0.2.24.dist-info → hte_cli-0.2.25.dist-info}/RECORD

@@ -1,15 +1,15 @@
 hte_cli/__init__.py,sha256=fDGXp-r8bIoLtlQnn5xJ_CpwMhonvk9bGjZQsjA2mDI,914
 hte_cli/__main__.py,sha256=63n0gNGfskidWDU0aAIF2N8lylVCLYKVIkrN9QiORoo,107
 hte_cli/api_client.py,sha256=m42kfFZS72Nu_VuDwxRsLNy4ziCcvgk7KNWBh9gwqy0,9257
-hte_cli/cli.py,sha256=ZZSEbvFKJBeWPwlfABdXMvAcz5kfsywH033leFFkO7M,24184
+hte_cli/cli.py,sha256=5aKf-k7qw3e1tmwpy34KDivIJ5l__2W-OEkGynCbQbU,26354
 hte_cli/config.py,sha256=42Xv__YMSeRLs2zhGukJkIXFKtnBtYCHnONfViGyt2g,3387
 hte_cli/errors.py,sha256=1J5PpxcUKBu6XjigMMCPOq4Zc12tnv8LhAsiaVFWLQM,2762
 hte_cli/events.py,sha256=oDKCS-a0IZ7bz7xkwQj5eM4DoDCYvnclAGohrMTWf8s,5644
-hte_cli/image_utils.py,sha256=TLwJdswUQrSD2bQcAXW03R8j8WG2pbHzd12TWcE7zy4,6418
+hte_cli/image_utils.py,sha256=nVHhUY-QZ4uPpGSx3ByOiVGOnm9T11p_cVlb39FQb_Y,7717
 hte_cli/runner.py,sha256=SWl9FF4X3e9eBbZyL0ujhmmSL5OK8J6st-Ty0jD5AWM,14550
 hte_cli/scorers.py,sha256=B0ZjQ3Fh-VDkc_8CDc86yW7vpdimbV3RSqs7l-VeUIg,6629
 hte_cli/version_check.py,sha256=WVZyGy2XfAghQYdd2N9-0Qfg-7pgp9gt4761-PnmacI,1708
-hte_cli-0.2.24.dist-info/METADATA,sha256=tnFgGGfZ15wjb6fz_Bgzuo9ApfgjRpwv6HSxVIVu7Os,3820
-hte_cli-0.2.24.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-hte_cli-0.2.24.dist-info/entry_points.txt,sha256=XbyEEi1H14DFAt0Kdl22e_IRVEGzimSzYSh5HlhKlFA,41
-hte_cli-0.2.24.dist-info/RECORD,,
+hte_cli-0.2.25.dist-info/METADATA,sha256=Sqc87sNbJMTRSJaR71y4Y6DpXjSyJ7-UDAix0p-bRpw,3820
+hte_cli-0.2.25.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+hte_cli-0.2.25.dist-info/entry_points.txt,sha256=XbyEEi1H14DFAt0Kdl22e_IRVEGzimSzYSh5HlhKlFA,41
+hte_cli-0.2.25.dist-info/RECORD,,