hud-python 0.4.1__py3-none-any.whl → 0.4.3__py3-none-any.whl

hud/telemetry/replay.py

@@ -1,74 +1,74 @@
-"""Trace retrieval and replay functionality.
-
-This module provides APIs to retrieve collected traces for analysis,
-debugging, and replay purposes.
-"""
-
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-from hud.otel.collector import clear_trace as _clear_trace
-from hud.otel.collector import get_trace as _get_trace
-
-if TYPE_CHECKING:
-    from hud.types import Trace
-
-__all__ = ["clear_trace", "get_trace"]
-
-
-def get_trace(task_run_id: str) -> Trace | None:
-    """Retrieve the collected trace for a task run.
-
-    Returns None if trace collection was disabled or the trace doesn't exist.
-
-    Args:
-        task_run_id: The task run ID to retrieve the trace for
-
-    Returns:
-        Trace object containing all collected steps, or None if not found
-
-    Usage:
-        import hud
-
-        # Run agent with tracing
-        with hud.trace() as task_run_id:
-            agent = MyAgent()
-            result = await agent.run("solve task")
-
-        # Get the trace for analysis
-        trace = hud.get_trace(task_run_id)
-        if trace:
-            print(f"Collected {len(trace.trace)} steps")
-
-            # Analyze agent vs MCP steps
-            agent_steps = [s for s in trace.trace if s.category == "agent"]
-            mcp_steps = [s for s in trace.trace if s.category == "mcp"]
-
-            print(f"Agent steps: {len(agent_steps)}")
-            print(f"MCP steps: {len(mcp_steps)}")
-
-            # Replay or analyze individual steps
-            for step in trace.trace:
-                if step.category == "agent" and step.result:
-                    print(f"Agent: {step.result.get('content') if isinstance(step.result, dict) else step.result}")
-                if step.category == "mcp" and step.request:
-                    print(f"MCP: {step.request.method if hasattr(step.request, 'method') else step.request}")
-    """  # noqa: E501
-    return _get_trace(task_run_id)
-
-
-def clear_trace(task_run_id: str) -> None:
-    """Clear the collected trace for a task run ID.
-
-    Useful for cleaning up memory after processing large traces.
-
-    Args:
-        task_run_id: The task run ID to clear the trace for
-
-    Usage:
-        trace = hud.get_trace(task_run_id)
-        # Process trace...
-        hud.clear_trace(task_run_id)  # Free memory
-    """
-    _clear_trace(task_run_id)
+"""Trace retrieval and replay functionality.
+
+This module provides APIs to retrieve collected traces for analysis,
+debugging, and replay purposes.
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from hud.otel.collector import clear_trace as _clear_trace
+from hud.otel.collector import get_trace as _get_trace
+
+if TYPE_CHECKING:
+    from hud.types import Trace
+
+__all__ = ["clear_trace", "get_trace"]
+
+
+def get_trace(task_run_id: str) -> Trace | None:
+    """Retrieve the collected trace for a task run.
+
+    Returns None if trace collection was disabled or the trace doesn't exist.
+
+    Args:
+        task_run_id: The task run ID to retrieve the trace for
+
+    Returns:
+        Trace object containing all collected steps, or None if not found
+
+    Usage:
+        import hud
+
+        # Run agent with tracing
+        with hud.trace() as task_run_id:
+            agent = MyAgent()
+            result = await agent.run("solve task")
+
+        # Get the trace for analysis
+        trace = hud.get_trace(task_run_id)
+        if trace:
+            print(f"Collected {len(trace.trace)} steps")
+
+            # Analyze agent vs MCP steps
+            agent_steps = [s for s in trace.trace if s.category == "agent"]
+            mcp_steps = [s for s in trace.trace if s.category == "mcp"]
+
+            print(f"Agent steps: {len(agent_steps)}")
+            print(f"MCP steps: {len(mcp_steps)}")
+
+            # Replay or analyze individual steps
+            for step in trace.trace:
+                if step.category == "agent" and step.result:
+                    print(f"Agent: {step.result.get('content') if isinstance(step.result, dict) else step.result}")
+                if step.category == "mcp" and step.request:
+                    print(f"MCP: {step.request.method if hasattr(step.request, 'method') else step.request}")
+    """  # noqa: E501
+    return _get_trace(task_run_id)
+
+
+def clear_trace(task_run_id: str) -> None:
+    """Clear the collected trace for a task run ID.
+
+    Useful for cleaning up memory after processing large traces.
+
+    Args:
+        task_run_id: The task run ID to clear the trace for
+
+    Usage:
+        trace = hud.get_trace(task_run_id)
+        # Process trace...
+        hud.clear_trace(task_run_id)  # Free memory
+    """
+    _clear_trace(task_run_id)

hud/telemetry/trace.py

@@ -1,83 +1,83 @@
-"""User-facing trace context manager for HUD telemetry.
-
-This module provides the simple trace() API that users interact with.
-The actual OpenTelemetry implementation is in hud.otel.
-"""
-
-from __future__ import annotations
-
-import uuid
-from contextlib import contextmanager
-from typing import TYPE_CHECKING, Any
-
-from hud.otel import configure_telemetry
-from hud.otel import trace as OtelTrace
-
-if TYPE_CHECKING:
-    from collections.abc import Generator
-
-__all__ = ["trace"]
-
-
-@contextmanager
-def trace(
-    name: str = "Test task from hud",
-    *,
-    root: bool = True,
-    attrs: dict[str, Any] | None = None,
-    job_id: str | None = None,
-    task_id: str | None = None,
-) -> Generator[str, None, None]:
-    """Start a HUD trace context.
-
-    A unique task_run_id is automatically generated for each trace.
-
-    Args:
-        name: Descriptive name for this trace/task
-        root: Whether this is a root trace (updates task status)
-        attrs: Additional attributes to attach to the trace
-        job_id: Optional job ID to associate with this trace
-
-    Yields:
-        str: The auto-generated task run ID
-
-    Usage:
-        import hud
-
-        with hud.trace("My Task") as task_run_id:
-            # Your code here
-            print(f"Running task: {task_run_id}")
-
-        # Or with default name:
-        with hud.trace() as task_run_id:
-            pass
-
-        # Or with job_id:
-        with hud.trace("My Task", job_id="550e8400-e29b-41d4-a716-446655440000") as task_run_id:
-            pass
-    """
-    # Ensure telemetry is configured
-    configure_telemetry()
-
-    # Only generate task_run_id if using HUD backend
-    # For custom OTLP backends, we don't need it
-    from hud.settings import get_settings
-
-    settings = get_settings()
-
-    if settings.telemetry_enabled and settings.api_key:
-        task_run_id = str(uuid.uuid4())
-    else:
-        # Use a placeholder for custom backends
-        task_run_id = "custom-otlp-trace"
-
-    # Delegate to OpenTelemetry implementation
-    with OtelTrace(
-        task_run_id,
-        is_root=root,
-        span_name=name,
-        attributes=attrs or {},
-        job_id=job_id,
-        task_id=task_id,
-    ) as run_id:
-        yield run_id
+"""User-facing trace context manager for HUD telemetry.
+
+This module provides the simple trace() API that users interact with.
+The actual OpenTelemetry implementation is in hud.otel.
+"""
+
+from __future__ import annotations
+
+import uuid
+from contextlib import contextmanager
+from typing import TYPE_CHECKING, Any
+
+from hud.otel import configure_telemetry
+from hud.otel import trace as OtelTrace
+
+if TYPE_CHECKING:
+    from collections.abc import Generator
+
+__all__ = ["trace"]
+
+
+@contextmanager
+def trace(
+    name: str = "Test task from hud",
+    *,
+    root: bool = True,
+    attrs: dict[str, Any] | None = None,
+    job_id: str | None = None,
+    task_id: str | None = None,
+) -> Generator[str, None, None]:
+    """Start a HUD trace context.
+
+    A unique task_run_id is automatically generated for each trace.
+
+    Args:
+        name: Descriptive name for this trace/task
+        root: Whether this is a root trace (updates task status)
+        attrs: Additional attributes to attach to the trace
+        job_id: Optional job ID to associate with this trace
+
+    Yields:
+        str: The auto-generated task run ID
+
+    Usage:
+        import hud
+
+        with hud.trace("My Task") as task_run_id:
+            # Your code here
+            print(f"Running task: {task_run_id}")
+
+        # Or with default name:
+        with hud.trace() as task_run_id:
+            pass
+
+        # Or with job_id:
+        with hud.trace("My Task", job_id="550e8400-e29b-41d4-a716-446655440000") as task_run_id:
+            pass
+    """
+    # Ensure telemetry is configured
+    configure_telemetry()
+
+    # Only generate task_run_id if using HUD backend
+    # For custom OTLP backends, we don't need it
+    from hud.settings import get_settings
+
+    settings = get_settings()
+
+    if settings.telemetry_enabled and settings.api_key:
+        task_run_id = str(uuid.uuid4())
+    else:
+        # Use a placeholder for custom backends
+        task_run_id = "custom-otlp-trace"
+
+    # Delegate to OpenTelemetry implementation
+    with OtelTrace(
+        task_run_id,
+        is_root=root,
+        span_name=name,
+        attributes=attrs or {},
+        job_id=job_id,
+        task_id=task_id,
+    ) as run_id:
+        yield run_id

hud/tools/__init__.py

@@ -1,33 +1,33 @@
-"""HUD tools for computer control, file editing, and bash commands."""
-
-from __future__ import annotations
-
-from typing import TYPE_CHECKING, Any
-
-from .base import BaseHub, BaseTool
-from .bash import BashTool
-from .edit import EditTool
-from .playwright import PlaywrightTool
-
-if TYPE_CHECKING:
-    from .computer import AnthropicComputerTool, HudComputerTool, OpenAIComputerTool
-
-__all__ = [
-    "AnthropicComputerTool",
-    "BaseHub",
-    "BaseTool",
-    "BashTool",
-    "EditTool",
-    "HudComputerTool",
-    "OpenAIComputerTool",
-    "PlaywrightTool",
-]
-
-
-def __getattr__(name: str) -> Any:
-    """Lazy import computer tools to avoid importing pyautogui unless needed."""
-    if name in ("AnthropicComputerTool", "HudComputerTool", "OpenAIComputerTool"):
-        from . import computer
-
-        return getattr(computer, name)
-    raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
+"""HUD tools for computer control, file editing, and bash commands."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
+from .base import BaseHub, BaseTool
+from .bash import BashTool
+from .edit import EditTool
+from .playwright import PlaywrightTool
+
+if TYPE_CHECKING:
+    from .computer import AnthropicComputerTool, HudComputerTool, OpenAIComputerTool
+
+__all__ = [
+    "AnthropicComputerTool",
+    "BaseHub",
+    "BaseTool",
+    "BashTool",
+    "EditTool",
+    "HudComputerTool",
+    "OpenAIComputerTool",
+    "PlaywrightTool",
+]
+
+
+def __getattr__(name: str) -> Any:
+    """Lazy import computer tools to avoid importing pyautogui unless needed."""
+    if name in ("AnthropicComputerTool", "HudComputerTool", "OpenAIComputerTool"):
+        from . import computer
+
+        return getattr(computer, name)
+    raise AttributeError(f"module '{__name__}' has no attribute '{name}'")