hud-python 0.2.9__py3-none-any.whl → 0.3.0__py3-none-any.whl

hud/__init__.py

@@ -10,24 +10,31 @@ from .job import create_job, load_job, run_job
 from .job import job as register_job
 from .task import Task
 from .taskset import load_taskset
-from .telemetry import flush, init_telemetry, trace
+from .telemetry import flush, trace, trace_open
 from .version import __version__
 
-if settings.settings.telemetry_enabled:
-    init_telemetry()
+
+def init_telemetry() -> None:
+    from .telemetry import init_telemetry as _init_telemetry
+
+    _init_telemetry()
+
 
 if settings.settings.fancy_logging:
     import logging
+    import sys
 
     hud_logger = logging.getLogger("hud")
-    # TODO: Make this configurable
     hud_logger.setLevel(logging.INFO)
 
     if not hud_logger.handlers:
-        handler = logging.StreamHandler()
+        # Use the configured stream (defaults to stderr)
+        stream = sys.stderr if settings.settings.log_stream.lower() == "stderr" else sys.stdout
+        handler = logging.StreamHandler(stream)
         formatter = logging.Formatter("[%(levelname)s] %(asctime)s | %(name)s | %(message)s")
         handler.setFormatter(formatter)
         hud_logger.addHandler(handler)
+        hud_logger.propagate = False
 
 __all__ = [
     "Response",
@@ -38,6 +45,7 @@ __all__ = [
     "env",
     "flush",
     "gym",
+    "init_telemetry",
     "load_job",
     "load_taskset",
     "register_job",
@@ -46,6 +54,7 @@ __all__ = [
     "task",
     "taskset",
     "trace",
+    "trace_open",
     "types",
     "utils",
 ]

hud/env/docker_client.py

@@ -262,7 +262,7 @@ class DockerClient(Client):
         self,
         command: list[str],
         *,
-        timeout: int | None = None,
+        timeout: int | None = None,  # noqa: ASYNC109
     ) -> ExecuteResult:
         """
         Execute a command in the environment. May not be supported by all environments.

hud/env/environment.py

@@ -194,7 +194,9 @@ class Environment(BaseModel):
 
         observation = Observation.model_validate(result["observation"], strict=True)
 
-        return observation, 0, False, {}
+        info = result.get("info", {})
+
+        return observation, 0, False, info
 
     def _maybe_store_response(self, actions: list[CLA]) -> bool:
         """Store the final response into the environment.
@@ -376,13 +378,16 @@ def create_remote_config(
 
     # Case 1: Explicit config provided
     if config:
-        expanded_configs = expand_config(config)
-        if env and env.final_response and expanded_configs[0].args[0] in LOCAL_EVALUATORS:
-            # Ensure args is a list before appending
-            if not isinstance(expanded_configs[0].args, list):
-                expanded_configs[0].args = [expanded_configs[0].args]
-            expanded_configs[0].args.append(env.final_response)  # for remote responses
-        return [FunctionConfig(function=function, args=expanded_configs, metadata=metadata)]
+        if not isinstance(config, dict):
+            expanded_configs = expand_config(config)
+            if env and env.final_response and expanded_configs[0].args[0] in LOCAL_EVALUATORS:
+                # Ensure args is a list before appending
+                if not isinstance(expanded_configs[0].args, list):
+                    expanded_configs[0].args = [expanded_configs[0].args]
+                expanded_configs[0].args.append(env.final_response)  # for remote responses
+            return [FunctionConfig(function=function, args=expanded_configs, metadata=metadata)]
+        else:
+            return [FunctionConfig(function=function, args=[config], metadata=metadata)]
 
     # Otherwise, use the environment's task
     task = env.task if env else None

hud/env/local_docker_client.py

@@ -222,7 +222,7 @@ class LocalDockerClient(DockerClient):
         self,
         command: list[str],
         *,
-        timeout: int | None = None,
+        timeout: int | None = None,  # noqa: ASYNC109
     ) -> ExecuteResult:
         """
         Execute a command in the container.

hud/env/remote_client.py

@@ -147,7 +147,7 @@ class RemoteClient(Client):
         command: list[str],
         *,
         workdir: str | None = None,
-        timeout: float | None = None,
+        timeout: float | None = None,  # noqa: ASYNC109
     ) -> ExecuteResult:
         """
         Execute a command in the environment.

hud/env/remote_docker_client.py

@@ -23,7 +23,7 @@ logger = logging.getLogger("hud.env.remote_env_client")
 async def upload_bytes_to_presigned_url(
     presigned_url: str,
     data_bytes: bytes,
-    timeout: float = 600,
+    timeout: float = 600,  # noqa: ASYNC109
 ) -> None:
     try:
         async with httpx.AsyncClient() as client:
@@ -206,7 +206,7 @@ class RemoteDockerClient(DockerClient):
         command: list[str],
         *,
         workdir: str | None = None,
-        timeout: float | None = None,
+        timeout: float | None = None,  # noqa: ASYNC109
     ) -> ExecuteResult:
         """
         Execute a command in the environment.

hud/exceptions.py

@@ -4,8 +4,9 @@ import logging
 from typing import TYPE_CHECKING, Any
 
 if TYPE_CHECKING:
+    from typing import Self
+
     import httpx
-    from typing_extensions import Self
 
 logger = logging.getLogger(__name__)
 

hud/mcp_agent/__init__.py

@@ -0,0 +1,15 @@
+"""MCP Agent implementations for HUD."""
+
+from __future__ import annotations
+
+from .base import BaseMCPAgent
+from .claude import ClaudeMCPAgent
+from .langchain import LangChainMCPAgent
+from .openai import OpenAIMCPAgent
+
+__all__ = [
+    "BaseMCPAgent",
+    "ClaudeMCPAgent",
+    "LangChainMCPAgent",
+    "OpenAIMCPAgent",
+]