hud-python 0.4.51__py3-none-any.whl → 0.4.52__py3-none-any.whl

hud/utils/task_tracking.py

@@ -0,0 +1,223 @@
+"""Task tracking for async telemetry operations.
+
+This module provides infrastructure to track async tasks created during
+telemetry operations (status updates, metric logging) to ensure they
+complete before process shutdown, preventing telemetry loss.
+
+The task tracker maintains strong references to tasks and explicitly cleans
+them up when they complete via callbacks. This ensures tasks are not garbage
+collected before they finish executing.
+
+Thread Safety:
+    Uses threading.Lock (not asyncio.Lock) because done callbacks run
+    synchronously and need to modify the task set safely.
+
+Race Condition Prevention:
+    The wait_all() method uses a multi-pass approach to catch tasks that
+    are created while waiting for existing tasks to complete.
+
+This is an internal module used by async context managers and cleanup
+routines. Users typically don't interact with it directly.
+"""
+
+import asyncio
+import contextlib
+import logging
+import threading
+from collections.abc import Coroutine
+from typing import Any
+
+logger = logging.getLogger(__name__)
+
+# Module exports
+__all__ = ["TaskTracker", "track_task", "wait_all_tasks"]
+
+# Global singleton task tracker
+_global_tracker: "TaskTracker | None" = None
+
+
+class TaskTracker:
+    """Tracks async tasks to ensure completion before shutdown.
+
+    Maintains a set of tasks with thread-safe access for both async code
+    and synchronous callbacks. Tasks are automatically removed when they
+    complete via done callbacks.
+    """
+
+    def __init__(self) -> None:
+        self._tasks: set[asyncio.Task] = set()
+        # Use threading.Lock for synchronous access from done callbacks
+        self._lock = threading.Lock()
+
+    def track_task(self, coro: Coroutine[Any, Any, Any], name: str = "task") -> asyncio.Task | None:
+        """Create and track an async task.
+
+        Args:
+            coro: The coroutine to run
+            name: Descriptive name for debugging and logging
+
+        Returns:
+            The created asyncio.Task, or None if no event loop is available
+        """
+        try:
+            task = asyncio.create_task(coro, name=name)
+
+            # Add task to tracking set (thread-safe)
+            with self._lock:
+                self._tasks.add(task)
+                task_count = len(self._tasks)
+
+            # Setup cleanup callback
+            def cleanup_callback(completed_task: asyncio.Task) -> None:
+                """Remove completed task from tracking set and log failures."""
+                with self._lock:
+                    self._tasks.discard(completed_task)
+
+                # Log exceptions outside lock to avoid blocking
+                with contextlib.suppress(Exception):
+                    if not completed_task.cancelled():
+                        with contextlib.suppress(Exception):
+                            exc = completed_task.exception()
+                            if exc:
+                                logger.warning("Task '%s' failed: %s", name, exc)
+
+            task.add_done_callback(cleanup_callback)
+            logger.debug("Tracking task '%s' (total active: %d)", name, task_count)
+            return task
+
+        except RuntimeError as e:
+            # No event loop - fall back to fire_and_forget
+            logger.warning("Cannot track task '%s': %s", name, e)
+            from hud.utils.async_utils import fire_and_forget
+
+            fire_and_forget(coro, name)
+            return None
+
+    async def wait_all(self, *, timeout_seconds: float = 30.0) -> int:
+        """Wait for all tracked tasks to complete.
+
+        Uses a multi-pass approach to handle race conditions where tasks are
+        added while waiting for existing tasks to complete. This ensures that
+        status updates created near the end of execution are still waited for.
+
+        Args:
+            timeout_seconds: Maximum time to wait in seconds
+
+        Returns:
+            Number of tasks that completed
+        """
+        total_completed = 0
+        time_remaining = timeout_seconds
+        max_passes = 10  # Prevent infinite loops if tasks keep spawning
+
+        for pass_num in range(max_passes):
+            # Get snapshot of pending tasks (thread-safe)
+            with self._lock:
+                pending = [t for t in self._tasks if not t.done()]
+
+            if not pending:
+                if pass_num == 0:
+                    logger.debug("No pending tasks to wait for")
+                else:
+                    logger.debug("All tasks completed after %d passes", pass_num)
+                break
+
+            # Log progress
+            if pass_num == 0:
+                logger.info("Waiting for %d pending tasks...", len(pending))
+            else:
+                logger.debug("Pass %d: Waiting for %d tasks", pass_num + 1, len(pending))
+
+            # Wait for this batch (max 5s per pass to check for new tasks)
+            batch_timeout = min(time_remaining, 5.0) if time_remaining > 0 else 5.0
+            start_time = asyncio.get_event_loop().time()
+
+            try:
+                done, still_pending = await asyncio.wait(
+                    pending, timeout=batch_timeout, return_when=asyncio.ALL_COMPLETED
+                )
+            except Exception as e:
+                logger.error("Error waiting for tasks: %s", e)
+                break
+
+            # Update timing
+            elapsed = asyncio.get_event_loop().time() - start_time
+            time_remaining -= elapsed
+            total_completed += len(done)
+
+            # Handle timeout
+            if still_pending:
+                if time_remaining <= 0:
+                    logger.warning(
+                        "%d tasks still pending after %ss timeout - cancelling",
+                        len(still_pending),
+                        timeout_seconds,
+                    )
+                    for task in still_pending:
+                        task.cancel()
+                    break
+                # Otherwise continue to next pass
+            else:
+                # All tasks from this batch completed, check for new ones
+                with self._lock:
+                    new_pending = [t for t in self._tasks if not t.done()]
+
+                if not new_pending:
+                    # No new tasks were added - we're done
+                    break
+                # Otherwise loop to wait for the new tasks
+
+        if total_completed > 0:
+            logger.info("Completed %d tasks", total_completed)
+
+        return total_completed
+
+    def get_pending_count(self) -> int:
+        """Get number of pending tasks (thread-safe)."""
+        with self._lock:
+            return sum(1 for t in self._tasks if not t.done())
+
+
+def get_global_tracker() -> TaskTracker:
+    """Get or create the global task tracker."""
+    global _global_tracker
+    if _global_tracker is None:
+        _global_tracker = TaskTracker()
+    return _global_tracker
+
+
+def track_task(coro: Coroutine[Any, Any, Any], name: str = "task") -> asyncio.Task | None:
+    """Create and track an async task for telemetry operations.
+
+    This is a convenience function that uses the global tracker to ensure
+    the task completes before shutdown. Used internally by async context
+    managers for status updates and metric logging.
+
+    Args:
+        coro: The coroutine to track
+        name: Descriptive name for debugging
+
+    Returns:
+        The created task, or None if no event loop is available
+    """
+    tracker = get_global_tracker()
+    return tracker.track_task(coro, name)
+
+
+async def wait_all_tasks(*, timeout_seconds: float = 30.0) -> int:
+    """Wait for all tracked telemetry tasks to complete.
+
+    Ensures that all async telemetry operations (status updates, logs)
+    complete before the calling function returns, preventing telemetry loss.
+
+    Uses a multi-pass approach to handle race conditions where status updates
+    are created while waiting for other tasks to complete.
+
+    Args:
+        timeout_seconds: Maximum time to wait for tasks in seconds
+
+    Returns:
+        Number of tasks that completed
+    """
+    tracker = get_global_tracker()
+    return await tracker.wait_all(timeout_seconds=timeout_seconds)

hud/utils/tests/test_version.py

@@ -5,4 +5,4 @@ def test_import():
     """Test that the package can be imported."""
     import hud
 
-    assert hud.__version__ == "0.4.51"
+    assert hud.__version__ == "0.4.52"

hud/version.py

@@ -4,4 +4,4 @@ Version information for the HUD SDK.
 
 from __future__ import annotations
 
-__version__ = "0.4.51"
+__version__ = "0.4.52"

{hud_python-0.4.51.dist-info → hud_python-0.4.52.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hud-python
-Version: 0.4.51
+Version: 0.4.52
 Summary: SDK for the HUD platform.
 Project-URL: Homepage, https://github.com/hud-evals/hud-python
 Project-URL: Bug Tracker, https://github.com/hud-evals/hud-python/issues
@@ -48,6 +48,7 @@ Requires-Dist: opentelemetry-api>=1.34.1
 Requires-Dist: opentelemetry-exporter-otlp-proto-http>=1.34.1
 Requires-Dist: opentelemetry-instrumentation-mcp==0.47.0
 Requires-Dist: opentelemetry-sdk>=1.34.1
+Requires-Dist: packaging>=21.0
 Requires-Dist: pathspec>=0.12.1
 Requires-Dist: pillow>=11.1.0
 Requires-Dist: prompt-toolkit==3.0.51

{hud_python-0.4.51.dist-info → hud_python-0.4.52.dist-info}/RECORD

@@ -1,33 +1,33 @@
-hud/__init__.py,sha256=JMDFUE1pP0J1Xl_miBdt7ERvoffZmTzSFe8yxz512A8,552
+hud/__init__.py,sha256=0LQ9PyuU6yZx7Fxu8YJXKC1i3TTHjg3UrInThh759QE,653
 hud/__main__.py,sha256=YR8Dq8OhINOsVfQ55PmRXXg4fEK84Rt_-rMtJ5rvhWo,145
 hud/settings.py,sha256=disObWa-DgXzoDcCDp3y1dTPaNsbR0IvoMJL9Eg4zyo,3947
 hud/types.py,sha256=KCVrglSG0VK-AUCFgG5CLgDcYqokvYgZG1r2Bsc-tBg,11155
-hud/version.py,sha256=C-hl1MocZrnKED-WbF7OhqI9B81dhTZJ7N3-aE7cw6k,105
+hud/version.py,sha256=6eRuCjDlZn4_gfm4B_A9cRBz531OtlF4KrULLYylQVo,105
 hud/agents/__init__.py,sha256=UoIkljWdbq4bM0LD-mSaw6w826EqdEjOk7r6glNYwYQ,286
-hud/agents/base.py,sha256=5G8Aio50KlDMGi3RjIlzK6fNHlY8YK9y5jhhHTUN5CE,31626
+hud/agents/base.py,sha256=bv2eeBacvluIQDrcrbP6Ydj41i_9AXBA51x1y8mAC_g,31897
 hud/agents/claude.py,sha256=0zje6de9_H-QlyDf05BnvtwoLPFnYgXhHgQGEAjwktU,16006
 hud/agents/grounded_openai.py,sha256=UC_Z0BP1fvThB95cCYqvMgaQLYXFYB_8zlawwIphruY,11247
 hud/agents/langchain.py,sha256=1EgCy8jfjunsWxlPC5XfvfLS6_XZVrIF1ZjtHcrvhYw,9584
-hud/agents/lite_llm.py,sha256=_3wbUiYCp7q8Vyu9rhaoJDvmb_bsyUsLYWP3iQJ2bHo,2239
+hud/agents/lite_llm.py,sha256=RW_VcMB-NIsxSfVhiQ2tvXSNQp3rTQg5h5Ax1b56bP0,2238
 hud/agents/openai.py,sha256=O1xV1h1l-W8lmnmXqTYr5CwnmnaniMqOxAZbl2CTTng,14576
 hud/agents/openai_chat_generic.py,sha256=_vAID9dZ_UxL0elYwafskRcsdrSsLsxJ4zPrP58oBiw,12151
 hud/agents/misc/__init__.py,sha256=LbVpHl2bDtheGPixbRRKsEjujwzmrXs7sCS8u1sYfAk,219
 hud/agents/misc/integration_test_agent.py,sha256=MYHsynQ_fooexxSGU_tN_zcQEUMrVCnlrdr6_KIj_Vw,2093
 hud/agents/misc/response_agent.py,sha256=uMuRDkz5QgaMQliNzBRepond5sb7KyqIiKm3LstjVnw,3753
 hud/agents/tests/__init__.py,sha256=W-O-_4i34d9TTyEHV-O_q1Ai1gLhzwDaaPo02_TWQIY,34
-hud/agents/tests/test_base.py,sha256=KDgaSlxiiHrNRq64zLjenq_6LcKb75Ln4NZCvci-I8M,28643
+hud/agents/tests/test_base.py,sha256=JBd1C9V8XOVWy2q53aLlwhoIVQKulrawl8SUWbP0VrM,28420
 hud/agents/tests/test_claude.py,sha256=0nZnfsbGoECvsLPdmaRnc9jVmrehVvc3kxeyiCQI2Cc,13807
 hud/agents/tests/test_client.py,sha256=uikgh6yhjPPX2RBU4XJQMz1mNox9uXjuwsP8t93id18,13337
 hud/agents/tests/test_grounded_openai_agent.py,sha256=VK8lUvHIjWicMX00VKPE-FZyjiJqTEhb80MuRRa9fVc,5437
 hud/agents/tests/test_openai.py,sha256=dnAFAoBKZf-5dtDpj6UC3q7oZv2tdMFcniPU0emfImw,8020
-hud/cli/__init__.py,sha256=D32zOZKmp-KDTtXj7H2uYPTh-fOmnwGmIitHAU6eVKs,41422
+hud/cli/__init__.py,sha256=ICDXVcw5QBGbSESBujXjrD2k0muk8HsUozY_SxHkruM,41086
 hud/cli/__main__.py,sha256=fDH7XITyuDITwSDIVwRso06aouADO0CzTHKqp5TOwJE,143
 hud/cli/analyze.py,sha256=4u5oYfJMquOjT9PzzRTYVcTZDxDi0ilNP_g532_hpOU,14716
 hud/cli/build.py,sha256=aFFWGtO1sQeTonvLsuMeDKG62IkakASbsP7FonGYuxc,22381
 hud/cli/clone.py,sha256=AwVDIuhr8mHb1oT2Af2HrD25SiTdwATpE6zd93vzLgA,6099
 hud/cli/debug.py,sha256=jtFW8J5F_3rhq1Hf1_SkJ7aLS3wjnyIs_LsC8k5cnzc,14200
 hud/cli/dev.py,sha256=mZW66fbtnCHplwGf1UWy72qfVYXnjPBUqVFAOD4v9Hg,23012
-hud/cli/eval.py,sha256=b91oYXBpODe_R2sRTWItjGceYSzSP5keKVLMkC7uc-Q,26926
+hud/cli/eval.py,sha256=D6BXRNXlEpCHyyryaJTjPKUdHhFgETC-h8kqwYz9vTg,25524
 hud/cli/get.py,sha256=sksKrdzBGZa7ZuSoQkc0haj-CvOGVSSikoVXeaUd3N4,6274
 hud/cli/init.py,sha256=Iz0fcE8ao50xChCKwbapTwjAPRY0ZDqV6XHLsvCpRC4,9952
 hud/cli/list_func.py,sha256=EVi2Vc3Lb3glBNJxFx4MPnZknZ4xmuJz1OFg_dc8a_E,7177
@@ -52,16 +52,16 @@ hud/cli/rl/wait_utils.py,sha256=FyIvqYWLOydANTetukoE5Rp2AOQi67qkiAlIJp4HpL8,2577
 hud/cli/tests/__init__.py,sha256=ZrGVkmH7DHXGqOvjOSNGZeMYaFIRB2K8c6hwr8FPJ-8,68
 hud/cli/tests/test_analyze.py,sha256=inbRvi7KJKoMYrcqXU6RSayoh7mAOGVrRknm6BLQFes,11055
 hud/cli/tests/test_analyze_metadata.py,sha256=TP8rVRDFcdkZrdmKFIR9yUxuxiaAkc-6UmjV4vrJaJM,10025
-hud/cli/tests/test_build.py,sha256=YbI8ICGgV7sYwkoE_sUpcf-ItTAB-i6vYAqziML68Mg,13552
+hud/cli/tests/test_build.py,sha256=OZhEHZAMjn1LVyWNzNlawWnyyn8z2gk2VxSNvDZ4f1A,13624
 hud/cli/tests/test_cli_init.py,sha256=_H0bAn5_skJ91Zj8P5P_wtZoPWvrN7jMhPZvmnnf0n8,11289
 hud/cli/tests/test_cli_main.py,sha256=0wMho9p9NcGjp0jLiUtCQh_FYdbMaCJtSY3sBbSgPwA,697
 hud/cli/tests/test_clone.py,sha256=oC2mf-41QQVc7ODJkjrWbVPNMB2fDW3nZ6jY6w93gvQ,4458
 hud/cli/tests/test_cursor.py,sha256=ZfxAFKJesJ3UV1JBoASSRlv6BXbpvVEk_pjxUg1jnf4,9821
 hud/cli/tests/test_debug.py,sha256=bQ76d_0HJfthHBSECmGNv499ZE57CIOKsanMlNfNHGk,18036
-hud/cli/tests/test_eval.py,sha256=URvNJBlwgbA8MTZhMbuTmGa2TETo4TPRm_G482CSQVM,22021
+hud/cli/tests/test_eval.py,sha256=9tsjHHA43IUiEy5AK3avdURZE3H3AFWTeeb87QW-R98,22329
 hud/cli/tests/test_list_func.py,sha256=pkG4TtJJBMi9Xk8KBNFBlGcam7kwz01IRsjfQBL2PxM,10700
 hud/cli/tests/test_main_module.py,sha256=6RhwCcdRSN2uQV6-adti40ZcLd3u-mPR1ai6wL64c6Y,1105
-hud/cli/tests/test_mcp_server.py,sha256=2RzAdhHrBOn-FLsMfZ4CJd2z7xJEA-W7Ytd1RTIP_os,890
+hud/cli/tests/test_mcp_server.py,sha256=2VYdxUVNhYq4BTXl1yJZWCZvYy7UZSUhYOcRhP1JccM,904
 hud/cli/tests/test_pull.py,sha256=ToSJrlfn13pYnrWWt3W_S7qFFjwvoZC2UisrZVrxujo,13155
 hud/cli/tests/test_push.py,sha256=V71KP5gEDo7Z9ccFpjidBjYliFg_KCfnZoZYbBXjguE,12875
 hud/cli/tests/test_registry.py,sha256=-o9MvQTcBElteqrg0XW8Bg59KrHCt88ZyPqeaAlyyTg,9539
@@ -82,12 +82,13 @@ hud/cli/utils/remote_runner.py,sha256=OOSJ6wU_gS_hJaURDfxZcyekjIIwPQKGN_Pq64tin3
 hud/cli/utils/runner.py,sha256=7HxVGa6OTflQnO0FSuRs273wnAmbm7cFRU9RTGL3-Wo,4390
 hud/cli/utils/server.py,sha256=EE5DJ0RAmXCEjMcZycpAsAxxCj6sOdIsXqPh38kK2ew,7416
 hud/cli/utils/source_hash.py,sha256=EDD3KC4pLGBVoSL5UTv1GvF2TAs2ThNHNOhP_5Sbub4,2979
-hud/cli/utils/tasks.py,sha256=bpH2mQkfgKUpgh1J0NtVxMxcM1jDZk2GAAPQMcqAUz4,826
+hud/cli/utils/tasks.py,sha256=lX9SeM5XekrTBx2HWKMA0BQ7R3Q8jqGgi4G5vIHsnJM,994
+hud/cli/utils/version_check.py,sha256=_e4H5Xxz2f9PVNSG8hAV_cJhS2zCX69RAodMRPTg-24,7279
 hud/clients/README.md,sha256=XNE3mch95ozDgVqfwCGcrhlHY9CwT1GKfNANNboowto,3826
 hud/clients/__init__.py,sha256=N5M_gZv4nP7dLRwpAiaqqaxyaLieGW6397FszeG7JGw,364
-hud/clients/base.py,sha256=yyQ4ctrXVyPps7Q_JGq2wilr1diNetWZ05wUJ4YCRng,14203
+hud/clients/base.py,sha256=Ywq3CzsWRx1cgwMDUCgSK_N1kxPfFTV7h-8Ah2cxDOw,14215
 hud/clients/fastmcp.py,sha256=1xaAg7DwMcwt_GRx2n3OsZaX-UMEQCZCaLDK4cr2HhQ,9178
-hud/clients/mcp_use.py,sha256=WE_99LxilPnfYo2yXsxQOl3Rt8eNyYuvvIycVzpUzk0,14627
+hud/clients/mcp_use.py,sha256=2Pb8bhAEmD9TJYKb-2JbxmyIsh4ttE_pP-kldrt6AH0,14750
 hud/clients/tests/__init__.py,sha256=sKOtJFFa4mDIXh1U6O8ZUHjigE8CiRMQ2PzJTIBZuVE,33
 hud/clients/tests/test_client_integration.py,sha256=kohU6jfCNfwSnAushHeB1_CmDlRfQc7VBL0GEdJYSeI,4198
 hud/clients/tests/test_fastmcp.py,sha256=4q3TzDjuieTZa89taiNJIrzbUncNkYOG4MaubypA21k,13030
@@ -98,8 +99,8 @@ hud/clients/utils/mcp_use_retry.py,sha256=knsgOTR3YFXshmPFfPQE6K6C5GpR1ZBJe2J7oz
 hud/clients/utils/retry.py,sha256=mMs2T_mAlb8AYhSqMR4AmCw7838gqCC4mdG3zjMAYM4,5744
 hud/clients/utils/retry_transport.py,sha256=Rsq25eiKKt_pM1bas78QEZvO0illK97X_3opmaS3A3w,6809
 hud/datasets/__init__.py,sha256=-g05iDy76CU4JiRHjKBBhgh3STtiIjmWhUfPqgf5hJE,697
-hud/datasets/parallel.py,sha256=hB_WSRWxgkfsPhbsiuM98IuR5Kln4B5iYOMxw86Yg8o,25441
-hud/datasets/runner.py,sha256=_BmQ4K-md3cyIs1YIiCH0HIjs_3Qwi0MAt4KJ8pLdTU,4625
+hud/datasets/parallel.py,sha256=3V2jCOQxh_JesUYyNv8kPjLpddL5zuvQfMswiVYEUgo,25429
+hud/datasets/runner.py,sha256=fHsblKzi0mEk5PDkpTWnRMjbCFrld4UcwrBCYIR4M1s,7517
 hud/datasets/utils.py,sha256=hdZfjWH5l3FVJaWBSHEEpjujAG7DqEam_vHgslL8MLs,4279
 hud/misc/__init__.py,sha256=m_pprQQ-G-Y0Sd0NEiR8MtAMbElnuFZ2OWT8TXrw7c4,43
 hud/misc/claude_plays_pokemon.py,sha256=IthAkjDVr2Q-GNvX-QLJyMzN7-0pHqqJbagGNv2m7yo,10453
@@ -110,9 +111,9 @@ hud/native/tests/test_comparator.py,sha256=pDch3r3xDi2o5YXF_bkoLfIdHcCjse3foAaqy
 hud/native/tests/test_native_init.py,sha256=Z-2dinbQYEkrbCcfBrBOLGdpXtWWOtkfPzp7ZKri68Y,2839
 hud/otel/__init__.py,sha256=ii17ayoWiS5vAhA7UAmZ8TkmP52gs2pWyHsD46-uYbE,1003
 hud/otel/collector.py,sha256=jLZymZ8r7xt2VDuWexfbnT7PY1-0aiyLMgjBy8KDY1M,4497
-hud/otel/config.py,sha256=PzyMOk_WYlJzVgZwQ_-F0DIlBbEy8Xt3ZA2e1QvFDes,6802
-hud/otel/context.py,sha256=9kJ_99KWGDvqU2O_DjmTHPXmhNDsMhbk_AHNvcHXR5Q,19116
-hud/otel/exporters.py,sha256=RLAjWa8b2DJEU21740Idq4fmeIuabLEqGGUspcFDcH4,14331
+hud/otel/config.py,sha256=rqw4gaQCqJNEFWO_AfyqwgZVia706qJP0UveNSgDVSg,7037
+hud/otel/context.py,sha256=DLP89aGlo-sTO0f4t80EvIXNaXA2FQXdOrlrkWJ6bVU,19102
+hud/otel/exporters.py,sha256=k0yfUppMbcJ3IfNH3cc_u1hdR54qnnkT7lQjxO3I06g,21470
 hud/otel/instrumentation.py,sha256=fsFG9W89RdewFDxWKN9Ft4GUb7WbIKpfucTc16WxaZU,5093
 hud/otel/processors.py,sha256=-gGRbwifplcExDQBLfx_9tqWreDImULJNcENgO9q7VU,4700
 hud/otel/tests/__init__.py,sha256=VNJKBMaxTtbn7trW-1Ph50zCvCok_wTSGcI1HD6GOLA,43
@@ -124,7 +125,7 @@ hud/rl/buffer.py,sha256=z47HOjOBJx3umUzzUfdtq_N4ZoJ8FMBPkX8YQKBtd3A,15457
 hud/rl/chat_template.jinja,sha256=XTdzI8oFGEcSA-exKxyHaprwRDmX5Am1KEb0VxvUc6U,4965
 hud/rl/config.py,sha256=sCU56mjtgJpu_C0TXqpT14v1LmZv0ntmUjgNkFamTPA,5713
 hud/rl/distributed.py,sha256=1k65Qhd9Coh4dyra8TMTExO-tabyRS_wZqh7SO3kuxo,3650
-hud/rl/learner.py,sha256=GegoJiKUJSWxw96VYiEvasOJJ4MnVRxW52uwBvo07Zo,27223
+hud/rl/learner.py,sha256=pdS_5_5KItQ8k0EVO4PX7ExDSDkQMoxAZJ5WvI-J7Kw,27239
 hud/rl/train.py,sha256=d0Ldg_ZXhiMwdpB5yidMK-LY6X__CWBhVqglQ-3EnvU,15589
 hud/rl/types.py,sha256=lrLKo7iaqodYth2EyeuOQfLiuzXfYM2eJjPmpObrD7c,3965
 hud/rl/utils.py,sha256=IsgVUUibxnUzb32a4mu1sYrgJC1CwoG9E-Dd5y5VDOA,19115
@@ -150,20 +151,21 @@ hud/server/tests/test_run_wrapper.py,sha256=EdwxMWCIHAp8t-l6VUeMOMhPRLTWjEVfTyys
 hud/server/tests/test_server_extra.py,sha256=MAmIrFwQWMFAP2AT6h1kbExZGxEvJZLq1bxvS7guQFE,5489
 hud/server/tests/test_sigterm_runner.py,sha256=HTM_0DAxA2exGYj7LK4udxMGXHZhY9LDZaKkHhQMu_Y,2610
 hud/shared/__init__.py,sha256=IPxPCqtPLguryN-nBq78Sakypw2bRiE2iHv3SXG8YRk,139
-hud/shared/exceptions.py,sha256=dhmt5KMciQ2fmV-yqXtTNO868k-WzuZae6w7dOk3M44,12144
+hud/shared/exceptions.py,sha256=seDNhHX12UptJlQgmNwBFrUHm3sIBXg_-waqyvNgFQk,11856
 hud/shared/hints.py,sha256=aa1CtBzsxLHRSZBFCXH00uY-1j2_7WLxYFwAy-neibE,5086
 hud/shared/requests.py,sha256=HWrPp7nBSK4jhv9wqZdFiNrVaaxV0vWS8fcgGtoztBc,9479
 hud/shared/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-hud/shared/tests/test_exceptions.py,sha256=bCKGqw2RwBmlGIMNpddzyfnWAOQwSIYB76r6iaBE2is,15761
+hud/shared/tests/test_exceptions.py,sha256=02UiYzqWqEtbrHxazZ1SBrOJL7-IXxWmeaU7n1Zf904,15565
 hud/shared/tests/test_requests.py,sha256=nKFcSN1sjrOouVU2xik9lE5Wxapy3EWsO8iIXrM_Sts,9114
-hud/telemetry/__init__.py,sha256=uWiloBMXgEzPRsRIOpiSBhcTxJDyHfBqTg7qi8kxSTc,683
+hud/telemetry/__init__.py,sha256=oYJ9fiaiwUGzfMH0qBixX74uSpAR_hyhl8Y-REjmNak,1396
+hud/telemetry/async_context.py,sha256=SjL1VNFdXmy_4t_gIbZslQ1Ea5vYC9Ibv9xeqr9elXw,11139
 hud/telemetry/instrument.py,sha256=m3u6YK02PTk39Jr4L3se7l-cYyKx0maCaqf5Z5JqWNA,14096
-hud/telemetry/job.py,sha256=LjspT-mSqQO2DnFL6h0ZkCkeMrrpjAuFVZnTJiOaDek,11585
+hud/telemetry/job.py,sha256=3W9KHLfBR13IYZvll496P9F6a9HUXCthFMY-MJLi7EA,13278
 hud/telemetry/replay.py,sha256=YW17s314s5Wy6Rl8MXHqg1FU8EF9_XcHBMJI0rrkyS4,2306
-hud/telemetry/trace.py,sha256=nUePF9LUtB4ao0832YoRxuakfGl8G1-NrxAyAXssnv4,4837
+hud/telemetry/trace.py,sha256=9zNfLIHM1UwfkUX23LmqUcIGX2hPzfHz_x2_G-DY-NA,5049
 hud/telemetry/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hud/telemetry/tests/test_replay.py,sha256=eREc6qgSJDRT1pOPdyhiEoEJ9H2yT1ospaU1RvTKlvg,1328
-hud/telemetry/tests/test_trace.py,sha256=0rxR77CjcStat3ILA9QAswieOJ3J_386QmjmNDp34oA,2486
+hud/telemetry/tests/test_trace.py,sha256=EJ5j8gbTTGdVm3NXhrC7r-5dwNSNfW0a-8-wxo-Jow4,2512
 hud/tools/__init__.py,sha256=i6lE0GxYcPnlLLd-55ryCCHo7o9anC4RfqkuYFXvzMQ,1009
 hud/tools/base.py,sha256=IxpnEHfxUtgz1k_w-09l65KKpDV3tQu4WdF3nZDsdwU,17854
 hud/tools/bash.py,sha256=1jl7cpB1ApGXn7Hy8zghJ2fXugEol6UeN0aYUSiM2EQ,5189
@@ -177,10 +179,10 @@ hud/tools/computer/__init__.py,sha256=QCdQG94XmKTCSOVkJI2e7NOP4bfpyAEkgEI9RQ-zbg
 hud/tools/computer/anthropic.py,sha256=oJfNMnjNFAn9mW1xY1nbWnTY2IqwFqdDR0mWSf8lu-s,17352
 hud/tools/computer/hud.py,sha256=v2VdLMsc8-3J4_k2vpcZRr_RJzXsxdVNs6IfbCL-WTc,16466
 hud/tools/computer/openai.py,sha256=QEsF45LWOHftDrAoIOnCFZZT1cL--s-ArSov5aluWb8,11189
-hud/tools/computer/qwen.py,sha256=6qtSRfxevkuot0YRZEmElvyexVI_M27vobKIar1b0OU,18475
+hud/tools/computer/qwen.py,sha256=K8FZfy664fQW6DXxeWTr2AorKIGxr9h_AeL-0vEYws8,18578
 hud/tools/computer/settings.py,sha256=GUvi9y3JtA2u-l1dkGcMH8kHSKOPlxTOnHaG0ocv_gI,3347
 hud/tools/executors/__init__.py,sha256=jHxfus9SLhkL6YGtebR5RyKYyVAix3yu5EkUp2Q27Kg,732
-hud/tools/executors/base.py,sha256=VP2SiIEBSXvklnkasGxVuy-OmDMd9rjuxZh_YuUQH7A,14481
+hud/tools/executors/base.py,sha256=mF_aiVoVkasoUi8hDyEtXMWbHKd4imKeFFJLJmQsjSY,14636
 hud/tools/executors/pyautogui.py,sha256=Gw3x2yw7x9xJ1uhYLxkOkArPnUQagUN1AZgBZ7YgbWo,22362
 hud/tools/executors/xdo.py,sha256=UF53DbMX-bRGiHd-O7cCJmCrVaYuP83xiJggER7HcDk,18137
 hud/tools/executors/tests/__init__.py,sha256=opFpGSH6cEqIZgt9izXd3Yt85pC7xkxiYmOZQTHf4AY,32
@@ -213,6 +215,7 @@ hud/utils/hud_console.py,sha256=ms2lXk3ZhYNmpGb0Dn902SKJBYfqo8Aa9DzkkC-xGSI,2359
 hud/utils/mcp.py,sha256=pMadd7A0DH6Y_aWywKU8jVYu2pRHGPEndV2ZQFrrj60,2888
 hud/utils/pretty_errors.py,sha256=WGeL4CTHtlA6KgPuV_JSX5l6H4-xbuTp6Y6tw1bkiFg,2430
 hud/utils/progress.py,sha256=suikwFM8sdSfkV10nAOEaInDhG4XKgOSvFePg4jSj1A,5927
+hud/utils/task_tracking.py,sha256=QsW8UqVqyjz5tl_qi65SaulOz7Zc-8hQMbafpVESyeU,8113
 hud/utils/tasks.py,sha256=7i36ck84gz1GZxhn9jryMBvKgMmcvLVu1YH5n3Y23-c,4985
 hud/utils/telemetry.py,sha256=hrVIx2rUjSGyy9IVxTZ_3Jii83PiHjyFRd5ls2whimM,1863
 hud/utils/tool_shorthand.py,sha256=_haLgK3yazLR2Y0jlEHUUQjw9uZCxi9yTipAwdOAJ70,2148
@@ -222,10 +225,10 @@ hud/utils/tests/test_init.py,sha256=2QLQSGgyP9wJhOvPCusm_zjJad0qApOZi1BXpxcdHXQ,
 hud/utils/tests/test_mcp.py,sha256=0pUa16mL-bqbZDXp5NHBnt1gO5o10BOg7zTMHZ1DNPM,4023
 hud/utils/tests/test_progress.py,sha256=QSF7Kpi03Ff_l3mAeqW9qs1nhK50j9vBiSobZq7T4f4,7394
 hud/utils/tests/test_telemetry.py,sha256=5jl7bEx8C8b-FfFUko5pf4UY-mPOR-9HaeL98dGtVHM,2781
-hud/utils/tests/test_version.py,sha256=EWbn6slnmozIIbiU4n3_1OoQoDYU37t3mOo2QHpIl7w,160
+hud/utils/tests/test_version.py,sha256=xbDp54OYQbjZS-l8yLfDbe9oY3L9FssclDNNznXVU2A,160
 hud/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-hud_python-0.4.51.dist-info/METADATA,sha256=4H8w-XvAHhtBJjP6kLlFEtUiXm4VcoR0zlahEzVPj7c,22275
-hud_python-0.4.51.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-hud_python-0.4.51.dist-info/entry_points.txt,sha256=jJbodNFg1m0-CDofe5AHvB4zKBq7sSdP97-ohaQ3ae4,63
-hud_python-0.4.51.dist-info/licenses/LICENSE,sha256=yIzBheVUf86FC1bztAcr7RYWWNxyd3B-UJQ3uddg1HA,1078
-hud_python-0.4.51.dist-info/RECORD,,
+hud_python-0.4.52.dist-info/METADATA,sha256=AeVDqBO6y829yPVcjpQv86phfPCcuuUTRhbuStw16go,22306
+hud_python-0.4.52.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hud_python-0.4.52.dist-info/entry_points.txt,sha256=jJbodNFg1m0-CDofe5AHvB4zKBq7sSdP97-ohaQ3ae4,63
+hud_python-0.4.52.dist-info/licenses/LICENSE,sha256=yIzBheVUf86FC1bztAcr7RYWWNxyd3B-UJQ3uddg1HA,1078
+hud_python-0.4.52.dist-info/RECORD,,