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

hud/agents/__init__.py

@@ -3,7 +3,6 @@ from __future__ import annotations
 from .art import ArtHUDAgent
 from .base import MCPAgent
 from .claude import ClaudeAgent
-from .langchain import LangChainAgent
 from .openai import OperatorAgent
 from .openai_chat_generic import GenericOpenAIChatAgent
 
@@ -11,7 +10,6 @@ __all__ = [
     "ArtHUDAgent",
     "ClaudeAgent",
     "GenericOpenAIChatAgent",
-    "LangChainAgent",
     "MCPAgent",
     "OperatorAgent",
 ]

hud/cli/init.py

@@ -37,7 +37,7 @@ version = "0.1.0"
 description = "A minimal HUD environment"
 requires-python = ">=3.11"
 dependencies = [
-    "hud-python @ git+https://github.com/hud-evals/hud-python.git",
+    "hud-python",
 ]
 
 [build-system]
@@ -259,7 +259,7 @@ def create_environment(name: str | None, directory: str, force: bool) -> None:
         console.print("   [cyan]hud dev[/cyan]")
     else:
         console.print("1. Enter the directory:")
-        console.print(f"   [cyan]cd {target_dir.relative_to(Path.cwd())}[/cyan]")
+        console.print(f"   [cyan]cd {target_dir}[/cyan]")
         console.print("\n2. Start development server:")
         console.print("   [cyan]hud dev[/cyan]")
 

hud/cli/push.py

@@ -348,15 +348,6 @@ def push_environment(
     console.print("Test locally:")
     console.print(f"  [cyan]hud run {image}[/cyan]\n")
 
-    console.print("Use in MCP configs:")
-    console.print("  Claude Desktop:")
-    console.print(
-        f'    [cyan]{{"docker": {{"image": "{pushed_digest}", "command": "auto"}}}}[/cyan]\n'
-    )
-
-    console.print("  Via HUD (recommended):")
-    console.print(f'    [cyan]{{"hud": {{"registry": "{pushed_digest}"}}}}[/cyan]\n')
-
     console.print("Share environment:")
     console.print(
         "  Share the updated [cyan]hud.lock.yaml[/cyan] for others to reproduce your exact environment"  # noqa: E501

hud/clients/base.py

@@ -128,8 +128,33 @@ class BaseHUDClient(AgentMCPClient):
 
         logger.debug("Initializing MCP client...")
 
-        # Subclasses implement connection
-        await self._connect(self._mcp_config)
+        try:
+            # Subclasses implement connection
+            await self._connect(self._mcp_config)
+        except RuntimeError as e:
+            # Re-raise authentication errors with clear message
+            if "Authentication failed" in str(e):
+                raise
+            raise
+        except Exception as e:
+            # Check for authentication errors in the exception chain
+            error_msg = str(e)
+            if "401" in error_msg or "Unauthorized" in error_msg:
+                # Check if connecting to HUD API
+                for server_config in self._mcp_config.values():
+                    url = server_config.get("url", "")
+                    if "mcp.hud.so" in url:
+                        raise RuntimeError(
+                            "Authentication failed for HUD API. "
+                            "Please ensure your HUD_API_KEY environment variable is set correctly. "
+                            "You can get an API key at https://app.hud.so"
+                        ) from e
+                # Generic 401 error
+                raise RuntimeError(
+                    f"Authentication failed (401 Unauthorized). "
+                    f"Please check your credentials or API key."
+                ) from e
+            raise
 
         # Common hud behavior - fetch telemetry
         await self._fetch_telemetry()

hud/clients/fastmcp.py

@@ -82,7 +82,27 @@ class FastMCPHUDClient(BaseHUDClient):
 
         if self._stack is None:
             self._stack = AsyncExitStack()
-            await self._stack.enter_async_context(self._client)
+            try:
+                await self._stack.enter_async_context(self._client)
+            except Exception as e:
+                # Check for authentication errors
+                error_msg = str(e)
+                if "401" in error_msg or "Unauthorized" in error_msg:
+                    # Check if connecting to HUD API
+                    for server_config in mcp_config.values():
+                        url = server_config.get("url", "")
+                        if "mcp.hud.so" in url:
+                            raise RuntimeError(
+                                "Authentication failed for HUD API. "
+                                "Please ensure your HUD_API_KEY environment variable is set correctly. "
+                                "You can get an API key at https://app.hud.so"
+                            ) from e
+                    # Generic 401 error
+                    raise RuntimeError(
+                        f"Authentication failed (401 Unauthorized). "
+                        f"Please check your credentials or API key."
+                    ) from e
+                raise
 
             # Configure validation for output schemas based on client setting
             from mcp.client.session import ValidationOptions

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.0"
+    assert hud.__version__ == "0.4.1"

hud/version.py

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

{hud_python-0.4.0.dist-info → hud_python-0.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hud-python
-Version: 0.4.0
+Version: 0.4.1
 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
@@ -129,11 +129,12 @@ OSS RL environment + evals toolkit. Wrap software as environments, run benchmark
 # Core installation - MCP servers, telemetry, basic tools for environment design
 pip install hud-python
 
-# Agent installation - Adds AI providers, telemetry, datasets
+# Agent installation - Adds AI providers, datasets
 pip install "hud-python[agent]"
 
 # CLI utilities
 uv tool install hud-python
+# uv tool update-shell
 
 # From source (latest)
 git clone https://github.com/hud-evals/hud-python
@@ -162,19 +163,20 @@ from hud.datasets import Task  # See docs: https://docs.hud.so/reference/tasks
 
 async def main() -> None:
     with hud.trace("Quick Start 2048"): # All telemetry works for any MCP-based agent (see https://app.hud.so)
-        task = Task(
-            prompt="Reach 64 in 2048.",
-            mcp_config={
+        task = {
+            "prompt": "Reach 64 in 2048.",
+            "mcp_config": {
                 "hud": {
-                    "url": "https://mcp.hud.so",  # HUD's cloud MCP server (see https://docs.hud.so/core-concepts/architecture)
+                    "url": "https://mcp.hud.so/v3/mcp",  # HUD's cloud MCP server (see https://docs.hud.so/core-concepts/architecture)
                     "headers": {
-                        "Authorization": f"Bearer {os.getenv('HUD_API_KEY')}",  # Get your key at https://app.hud.so
-                        "Mcp-Image": "hudpython/hud-text-2048:v1.1"  # Docker image from https://hub.docker.com/u/hudpython
+                        "Authorization": f"Bearer {settings.api_key}",  # Get your key at https://app.hud.so
+                        "Mcp-Image": "hudpython/hud-text-2048:v1.2"  # Docker image from https://hub.docker.com/u/hudpython
                     }
                 }
             },
-            evaluate_tool={"name": "evaluate", "arguments": {"name": "max_number", "target": 64}},
-        )
+            "evaluate_tool": {"name": "evaluate", "arguments": {"name": "max_number", "arguments": {"target": 64}}},
+        }
+        task = Task(**task)
 
         # 1. Define the client explicitly:
         client = MCPClient(mcp_config=task.mcp_config)
@@ -189,7 +191,7 @@ async def main() -> None:
         # result = await ClaudeAgent().run(task)
 
         print(f"Reward: {result.reward}")
-        await client.close()
+        await client.shutdown()
 
 asyncio.run(main())
 ```
@@ -346,7 +348,7 @@ result = await ClaudeAgent().run({  # See all agents: https://docs.hud.so/refere
     "prompt": "Please explore this environment",
     "mcp_config": {
         "my-environment": {
-            "url": "https://mcp.hud.so",
+            "url": "https://mcp.hud.so/v3/mcp",
             "headers": {
                 "Authorization": f"Bearer {os.getenv('HUD_API_KEY')}",
                 "Mcp-Image": "my-name/my-environment:latest"

{hud_python-0.4.0.dist-info → hud_python-0.4.1.dist-info}/RECORD

@@ -2,8 +2,8 @@ hud/__init__.py,sha256=g2ounE6q48MfCsMm-4UfoC__d0KLz1O3e_arCHFQgBk,419
 hud/datasets.py,sha256=bDIWbuu2eshsxl4RvtxZkb4k28NevIBaKUsk8v6pDQk,12214
 hud/settings.py,sha256=-GEBsb60PwhikCQlFgD1DiuNWgOjFmFVhqK7iTwcvac,2112
 hud/types.py,sha256=li7HCCzIH8FfIGEgO2GFmAHpwCCUfBT-NbXloVVkJAY,4373
-hud/version.py,sha256=BB6GRVXtk_lpQD1Yd5SnD0oneYg4_D8aTGe3eNbR3Jg,111
-hud/agents/__init__.py,sha256=S1hatsJF5stZymVknQlBmZCgP_WI2mc0efqSO_8PtuA,411
+hud/version.py,sha256=JvdmARn-wCmKr0LMLqNvVkCz8MF01kl1VDUCZC2oDgw,111
+hud/agents/__init__.py,sha256=2F7aXfCROLXreVTxWoJNlTZeP8MSY7eZUom1ZSITlsU,349
 hud/agents/art.py,sha256=GE31q7XcnP8qo7Uu1VJCCG5RaiKVf53WIubtdqUPmIU,3574
 hud/agents/base.py,sha256=CHZj73_gwcWcuUIv4fT_1PH2Ac5EnizYV4rBa10JdVE,22857
 hud/agents/claude.py,sha256=2EozkHZS9M2b7Z1sMb07CFeftfnHdSO8W1aDn-rGVGE,14646
@@ -26,11 +26,11 @@ hud/cli/clone.py,sha256=nc10s8qm4943fvC5f4ASDHVJ7-A_J6cWkyuwiPuBPbo,6284
 hud/cli/cursor.py,sha256=hyaArxYpiVENzjPYQAit7cwNjEdTDaCpftX0A1fC_zw,3101
 hud/cli/debug.py,sha256=ZAHMaNjyYosi40bhid_hrCaFCw4Uu7DrTn7F6uhgN54,14005
 hud/cli/docker_utils.py,sha256=FQLmSJZLLYl5fq4H0sekP0_0u6FK-53IXIbPn9zpAbI,2778
-hud/cli/init.py,sha256=5StiFHsl_Ogo03o85XCNcHixEwz1Jj482bR4VIGY5iM,8027
+hud/cli/init.py,sha256=AgUGFYyUqSlCNNJiXd3p7Z_eOL8nJIfx3ais-9qDucs,7953
 hud/cli/interactive.py,sha256=ewO-ipLNR9Q16OteAAouSa_2_FNSH3OO3vMxXBJQyQI,13216
 hud/cli/mcp_server.py,sha256=E5CsIHz5mxoFyfXfZ2UQxirlsLgF9bvzdbl_nu6LATw,28873
 hud/cli/pull.py,sha256=Y5XMOPPIgJ2agZyLgjHYMAJ_HDUiOE1LJka-s_5zMak,12275
-hud/cli/push.py,sha256=W6fkf0EqC-p7e_QhFIM3byo7nNvpIXASC9cp7RH07ZM,14875
+hud/cli/push.py,sha256=IkuIO6g6EmTzjCOB0eZ1DB5KLN7-yu5xQ2Cb2whVkJ8,14533
 hud/cli/remote_runner.py,sha256=8iH9ez11Xd9R2lsbArn517MVICogsYcyCMkXA_fe2TQ,9716
 hud/cli/runner.py,sha256=5526x2hC_cd7MSvDPtMVsDIAjVPXHMpre0KO3CChkQY,5048
 hud/cli/utils.py,sha256=rKnjcgJ3Fm844qardCrQPQ_PFVtad9JIXQhUEOHhl0A,9104
@@ -45,8 +45,8 @@ hud/cli/tests/test_mcp_server.py,sha256=6v9-5_iNDTf3WLFWUevS_brv1puG7U0GDpTx4szw
 hud/cli/tests/test_utils.py,sha256=DZfe-0xoX_VGQO-mSZr7xca0WBl6z_9Eai_7OxjzuUI,13854
 hud/clients/README.md,sha256=ypuNGE97Bv-1c--DIiy7mh-yUVITsEKn-CBgDvLs4J4,3969
 hud/clients/__init__.py,sha256=_mdxGF3pnuFB0tpVKpNZ9lz27iKSOa2-ck3NjHHFg_Y,344
-hud/clients/base.py,sha256=LU6LgOqhTbVwoDlzie6nPd6UeWI5mNPaDjMuugVcgVA,12668
-hud/clients/fastmcp.py,sha256=vPANIBMMLklRlhNgA_Dl6HdIETNJbM8zvEPb-i93F38,8327
+hud/clients/base.py,sha256=tYHwoCArNSdc6tCSbGoDDdKw_IVitNIHxF2mk94FB3Y,13876
+hud/clients/fastmcp.py,sha256=mF1ZZHburlpzNdZZ1ndZCaF_0jVT7-ot2sqmZXK49Vo,9389
 hud/clients/mcp_use.py,sha256=38_-yfNC-zhHbqmV2HRmBpMd-fTbB22DO0PnJPd7A50,11302
 hud/clients/tests/__init__.py,sha256=FSkZn2zN5p4DQTMcv6RppiboIxG-9wSUoPpB5kw06g8,34
 hud/clients/tests/test_client_integration.py,sha256=he-Vtmo84G7HuNSk1harEwcBli_K5T9ttQaH2MKGgDU,4309
@@ -123,10 +123,10 @@ hud/utils/tests/test_async_utils.py,sha256=Sraje4i83CcCEmMtLQQA5GKS8BKXsPZ1uPp-f
 hud/utils/tests/test_init.py,sha256=jKwlbj7tz9tOpgBe9cffFGAIsVS47KYkW7DnygcRqYA,400
 hud/utils/tests/test_progress.py,sha256=8kcinr6u6vgvLSWZ4rZlNWz4vL835SUIFwmZj11Q4Qg,7655
 hud/utils/tests/test_telemetry.py,sha256=DNoGqq7zs1fjKA9GsmvT1U8FlekHAVwYD4pxSJvLgYs,2863
-hud/utils/tests/test_version.py,sha256=FplWmgCztUyXs1t5-MMvXwM7uO0rAV0UJnmJIcX_WNs,167
+hud/utils/tests/test_version.py,sha256=vwgyzM9FUBbH9by_fUwTmBPLEoQ0cgqZZ_jtEsH-x2E,167
 hud/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-hud_python-0.4.0.dist-info/METADATA,sha256=4eQM1ezjKR2OogFJBB2MDvyJK1TCkKMYV54MOSV3du8,19485
-hud_python-0.4.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-hud_python-0.4.0.dist-info/entry_points.txt,sha256=jJbodNFg1m0-CDofe5AHvB4zKBq7sSdP97-ohaQ3ae4,63
-hud_python-0.4.0.dist-info/licenses/LICENSE,sha256=bskFRICmpDj1GtBpnuisjMR6mtQZtkPnRCtTz1CMMB0,1099
-hud_python-0.4.0.dist-info/RECORD,,
+hud_python-0.4.1.dist-info/METADATA,sha256=Re8FIOUusD1B4-jkdb3lyTJNOwXld4L-BRDMhs8H-CY,19554
+hud_python-0.4.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hud_python-0.4.1.dist-info/entry_points.txt,sha256=jJbodNFg1m0-CDofe5AHvB4zKBq7sSdP97-ohaQ3ae4,63
+hud_python-0.4.1.dist-info/licenses/LICENSE,sha256=bskFRICmpDj1GtBpnuisjMR6mtQZtkPnRCtTz1CMMB0,1099
+hud_python-0.4.1.dist-info/RECORD,,