hte-cli 0.2.13__tar.gz → 0.2.15__tar.gz

{hte_cli-0.2.13 → hte_cli-0.2.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hte-cli
-Version: 0.2.13
+Version: 0.2.15
 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.13 → hte_cli-0.2.15}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "hte-cli"
-version = "0.2.13"
+version = "0.2.15"
 description = "Human Time-to-Completion Evaluation CLI"
 readme = "README.md"
 requires-python = ">=3.11"

{hte_cli-0.2.13 → hte_cli-0.2.15}/src/hte_cli/cli.py

@@ -354,6 +354,15 @@ def session_join(ctx, session_id: str, force_setup: bool):
         console.print("[dim]Skipping setup (use --force-setup to re-run)[/dim]")
         console.print()
 
+    # Check if session was cancelled during setup
+    try:
+        updated_session = api.join_session(session_id)
+        if updated_session.get("status") == "cancelled":
+            console.print("[yellow]Session was cancelled. Exiting.[/yellow]")
+            sys.exit(0)
+    except APIError:
+        pass  # Continue if we can't check - server might be temporarily unavailable
+
     # Step 4: Show instructions
     if session_info.get("instructions"):
         console.print(Panel(session_info["instructions"], title="Task Instructions"))
@@ -362,18 +371,19 @@ def session_join(ctx, session_id: str, force_setup: bool):
     # Step 3: Run the task using TaskRunner
     step_num = "3" if (not is_reconnect or force_setup) and images else "2" if (not is_reconnect or force_setup) else "1"
     console.print(f"[bold]Step {step_num}:[/bold] Starting task environment...")
+    console.print("[dim]Launching Docker containers...[/dim]")
+    console.print()
 
     events.docker_started()
 
     runner = TaskRunner()
     eval_log_bytes = None
     try:
-        with console.status("[dim]Launching Docker containers (this may take a minute)...[/dim]"):
-            result = runner.run_from_assignment(
-                assignment=assignment,
-                compose_yaml=compose_yaml,
-                files_zip=files_zip,
-            )
+        result = runner.run_from_assignment(
+            assignment=assignment,
+            compose_yaml=compose_yaml,
+            files_zip=files_zip,
+        )
         # Read eval log before cleanup
         if result.eval_log_path and result.eval_log_path.exists():
             eval_log_bytes = result.eval_log_path.read_bytes()