hte-cli 0.2.16__tar.gz → 0.2.17__tar.gz

{hte_cli-0.2.16 → hte_cli-0.2.17}/PKG-INFO

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

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

{hte_cli-0.2.16 → hte_cli-0.2.17}/src/hte_cli/cli.py

@@ -408,6 +408,21 @@ def session_join(ctx, session_id: str, force_setup: bool):
             answer=result.answer,
         )
 
+        # Extract agent_id from task files for CyberGym post-hoc verification
+        agent_id = None
+        if files_zip:
+            try:
+                from io import BytesIO
+                from zipfile import ZipFile
+                with ZipFile(BytesIO(files_zip)) as zf:
+                    if "difficulty_levels.json" in zf.namelist():
+                        with zf.open("difficulty_levels.json") as f:
+                            import json
+                            difficulty_info = json.load(f)
+                            agent_id = difficulty_info.get("agent_id")
+            except Exception:
+                pass  # Not a CyberGym task or malformed zip
+
         console.print()
         console.print("[green]Task completed![/green]")
         console.print(f"Answer: {result.answer}")
@@ -428,7 +443,7 @@ def session_join(ctx, session_id: str, force_setup: bool):
                     eval_log_bytes=eval_log_bytes,
                     score=result.score,
                     score_binarized=result.score_binarized,
-                    agent_id=None,  # TODO: extract from task files if needed
+                    agent_id=agent_id,
                 )
             except APIError as e:
                 console.print(f"[red]Failed to upload result: {e}[/red]")