hte-cli 0.2.1__py3-none-any.whl → 0.2.3__py3-none-any.whl

hte_cli/cli.py

@@ -281,26 +281,63 @@ def session_join(ctx, session_id: str, force_setup: bool):
 
     # Step 3: Run setup (skip if reconnecting without force)
     setup_start_time = time.monotonic()
-    if not is_reconnect or force_setup:
-        # Send setup_started event
-        events.setup_started({"cli_version": __version__, "task_id": assignment["task_id"]})
+    images = []
+    pulled_images = []
+    cached_images = []
+    failed_images = []
 
-        # Pull images if we have compose
+    if not is_reconnect or force_setup:
+        # Extract images from compose
         if compose_yaml:
             images = extract_images_from_compose(compose_yaml)
-            if images:
-                console.print(f"[bold]Pulling {len(images)} Docker image(s)...[/bold]")
-                events.image_pull_started({})
-                for img in images:
-                    short_name = img.split("/")[-1][:40]
-                    with console.status(f"[yellow]↓[/yellow] {short_name}") as status:
-                        success = pull_image_with_progress(img)
-                    if success:
-                        console.print(f"  [green]✓[/green] {short_name}")
-                    else:
-                        console.print(f"  [red]✗[/red] {short_name} (failed)")
-                events.image_pull_completed({})
-                console.print()
+
+        # Send setup_started event
+        events.setup_started(images=images)
+
+        # Pull images if we have any
+        if images:
+            from hte_cli.image_utils import check_image_exists_locally
+
+            console.print(f"[bold]Pulling {len(images)} Docker image(s)...[/bold]")
+            pull_start = time.monotonic()
+
+            for img in images:
+                short_name = img.split("/")[-1][:40]
+
+                # Check if already cached
+                if check_image_exists_locally(img):
+                    console.print(f"  [green]✓[/green] {short_name} [dim](cached)[/dim]")
+                    cached_images.append(img)
+                    continue
+
+                # Need to pull - show progress
+                with console.status(f"[yellow]↓[/yellow] {short_name} [dim]connecting...[/dim]") as status:
+                    def show_progress(image: str, line: str):
+                        # Parse docker pull output for layer progress
+                        # Lines look like: "abc123: Downloading [====>   ] 10MB/50MB"
+                        if ": " in line:
+                            parts = line.split(": ", 1)
+                            if len(parts) == 2:
+                                layer_status = parts[1][:50]  # Truncate
+                                status.update(f"[yellow]↓[/yellow] {short_name} [dim]{layer_status}[/dim]")
+
+                    success = pull_image_with_progress(img, 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]")
+                    failed_images.append(img)
+
+            pull_duration = time.monotonic() - pull_start
+            events.image_pull_completed(
+                duration_seconds=pull_duration,
+                pulled=pulled_images,
+                cached=cached_images,
+                failed=failed_images,
+            )
+            console.print()
 
         # Send setup_completed - THIS STARTS THE TIMER ON SERVER
         total_setup = time.monotonic() - setup_start_time

{hte_cli-0.2.1.dist-info → hte_cli-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hte-cli
-Version: 0.2.1
+Version: 0.2.3
 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.1.dist-info → hte_cli-0.2.3.dist-info}/RECORD

@@ -1,7 +1,7 @@
 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=ppVNUPxQVen3R90UVkA33Cy35N7Un5sh42ml0dPz9nw,39368
+hte_cli/cli.py,sha256=xXlsXfX-mXUHUszR55-u7bCftSgPszF-prm1-8Yfinw,40730
 hte_cli/config.py,sha256=42Xv__YMSeRLs2zhGukJkIXFKtnBtYCHnONfViGyt2g,3387
 hte_cli/errors.py,sha256=1J5PpxcUKBu6XjigMMCPOq4Zc12tnv8LhAsiaVFWLQM,2762
 hte_cli/events.py,sha256=Zn-mroqaLHNzdT4DFf8st1Qclglshihdc09dBfCN070,5522
@@ -9,7 +9,7 @@ hte_cli/image_utils.py,sha256=454yoZEI1duNYrZC8UjhfZzDRP4Nxdrf2TvnZ_54G1k,4439
 hte_cli/runner.py,sha256=DhC8FMjHwfLR193iP4thLDRZrNssYA9KH1WYKU2JKeg,13535
 hte_cli/scorers.py,sha256=sFoPJePRt-K191-Ga4cVmrldruJclYXTOLkU_C9nCDI,6025
 hte_cli/version_check.py,sha256=WVZyGy2XfAghQYdd2N9-0Qfg-7pgp9gt4761-PnmacI,1708
-hte_cli-0.2.1.dist-info/METADATA,sha256=r2fEYVYX0wHf-S2pD7ysYt-Ln8Js9wfulcl4RgTWob4,3767
-hte_cli-0.2.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-hte_cli-0.2.1.dist-info/entry_points.txt,sha256=XbyEEi1H14DFAt0Kdl22e_IRVEGzimSzYSh5HlhKlFA,41
-hte_cli-0.2.1.dist-info/RECORD,,
+hte_cli-0.2.3.dist-info/METADATA,sha256=WGhP-PeFCMDUsn5mj_1vXEbvALy1gx_hA5MvMWrwK-w,3767
+hte_cli-0.2.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+hte_cli-0.2.3.dist-info/entry_points.txt,sha256=XbyEEi1H14DFAt0Kdl22e_IRVEGzimSzYSh5HlhKlFA,41
+hte_cli-0.2.3.dist-info/RECORD,,