hud-python 0.5.34__tar.gz → 0.5.35__tar.gz

{hud_python-0.5.34 → hud_python-0.5.35}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hud-python
-Version: 0.5.34
+Version: 0.5.35
 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

{hud_python-0.5.34 → hud_python-0.5.35}/hud/agents/base.py

@@ -336,12 +336,15 @@ class MCPAgent(ABC):
                 f"Available tools: {sorted(available_tool_names)}"
             )
 
-        self.console.debug(
-            f"Discovered {len(self._available_tools)} tools from environment: "
-            f"{', '.join([t.name for t in self._available_tools])}"
+        self._categorized_tools = self.categorize_tools()
+
+        # Show tool discovery table (visible at INFO level)
+        self.console.format_tool_discovery(
+            tools=self._available_tools,
+            native_tools=self._categorized_tools.native + self._categorized_tools.hosted,
+            skipped=self._categorized_tools.skipped,
         )
 
-        self._categorized_tools = self.categorize_tools()
         for tool, reason in self._categorized_tools.skipped:
             logger.debug("Skipping tool %s: %s", tool.name, reason)
 
@@ -574,17 +577,13 @@ class MCPAgent(ABC):
                     tool_messages = await self.format_tool_results(tool_calls, tool_results)
                     messages.extend(tool_messages)
 
-                    # Compact step completion display
-                    step_info = f"\n[bold]Step {step_count}"
-                    if max_steps != -1:
-                        step_info += f"/{max_steps}"
-                    step_info += "[/bold]"
-
-                    # Show tool calls and results in compact format
-                    for call, result in zip(tool_calls, tool_results, strict=False):
-                        step_info += f"\n{call}\n{result}"
-
-                    self.console.info_log(step_info)
+                    if logger.isEnabledFor(logging.INFO):
+                        self.console.format_step(
+                            step=step_count,
+                            max_steps=max_steps,
+                            tool_calls=tool_calls,
+                            tool_results=tool_results,
+                        )
 
                 except Exception as e:
                     self.console.error_log(f"Step failed: {e}")

{hud_python-0.5.34 → hud_python-0.5.35}/hud/agents/claude.py

@@ -145,8 +145,12 @@ class ClaudeAgent(MCPAgent):
                 model_client = AsyncAnthropic(api_key=settings.anthropic_api_key)
             else:
                 raise ValueError(
-                    "No API key found. Set HUD_API_KEY for HUD gateway, "
-                    "or ANTHROPIC_API_KEY for direct Anthropic access."
+                    "No API key found for Claude.\n"
+                    "  • Set HUD_API_KEY to use HUD Gateway"
+                    " (add your Anthropic key at"
+                    " hud.ai/project/secrets for BYOK)\n"
+                    "  • Or set ANTHROPIC_API_KEY for direct"
+                    " access"
                 )
 
         self.anthropic_client: AsyncAnthropic | AsyncAnthropicBedrock = model_client

{hud_python-0.5.34 → hud_python-0.5.35}/hud/agents/gemini.py

@@ -94,8 +94,12 @@ class GeminiAgent(MCPAgent):
                         raise ValueError(f"Gemini API key is invalid: {e}") from e
             else:
                 raise ValueError(
-                    "No API key found. Set HUD_API_KEY for HUD gateway, "
-                    "or GEMINI_API_KEY for direct Gemini access."
+                    "No API key found for Gemini.\n"
+                    "  • Set HUD_API_KEY to use HUD Gateway"
+                    " (add your Gemini key at"
+                    " hud.ai/project/secrets for BYOK)\n"
+                    "  • Or set GEMINI_API_KEY for direct"
+                    " access"
                 )
 
         self.gemini_client: genai.Client = model_client

{hud_python-0.5.34 → hud_python-0.5.35}/hud/agents/openai.py

@@ -119,8 +119,12 @@ class OpenAIAgent(MCPAgent):
                         raise ValueError(f"OpenAI API key is invalid: {exc}") from exc
             else:
                 raise ValueError(
-                    "No API key found. Set HUD_API_KEY for HUD gateway, "
-                    "or OPENAI_API_KEY for direct OpenAI access."
+                    "No API key found for OpenAI.\n"
+                    "  • Set HUD_API_KEY to use HUD Gateway"
+                    " (add your OpenAI key at"
+                    " hud.ai/project/secrets for BYOK)\n"
+                    "  • Or set OPENAI_API_KEY for direct"
+                    " access"
                 )
 
         self.openai_client: AsyncOpenAI = model_client

{hud_python-0.5.34 → hud_python-0.5.35}/hud/cli/__init__.py

@@ -40,6 +40,7 @@ from .init import init_command  # noqa: E402
 from .link import link_command  # noqa: E402
 from .models import models_command  # noqa: E402
 from .push import push_command  # noqa: E402
+from .rl import rl_run_command, rl_status_command  # noqa: E402
 from .scenario import scenario_app  # noqa: E402
 from .sync import sync_app  # noqa: E402
 
@@ -114,6 +115,12 @@ app.add_typer(scenario_app, name="scenario")
 # Sync subcommand group
 app.add_typer(sync_app, name="sync")
 
+# RL subcommand group
+rl_app = typer.Typer(help="🚀 RL training commands\n\nExample: hud rl run my-taskset -m <model-id>")
+rl_app.command("run")(rl_run_command)
+rl_app.command("status")(rl_status_command)
+app.add_typer(rl_app, name="rl")
+
 
 # ---------------------------------------------------------------------------
 # Entry point

{hud_python-0.5.34 → hud_python-0.5.35}/hud/cli/analyze.py

@@ -4,8 +4,9 @@ from __future__ import annotations
 
 import asyncio
 import json
+import time
 from pathlib import Path  # noqa: TC003
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 import typer
 from rich.console import Console
@@ -16,6 +17,9 @@ from rich.tree import Tree
 
 from hud.utils.hud_console import HUDConsole
 
+if TYPE_CHECKING:
+    from collections.abc import Mapping
+
 console = Console()
 hud_console = HUDConsole()
 
@@ -112,19 +116,26 @@ async def analyze_environment(docker_cmd: list[str], output_format: str, verbose
 
         from fastmcp import Client as FastMCPClient
 
-        from hud.cli.utils.mcp import analyze_environment as mcp_analyze
+        from hud.cli.utils.analysis import analyze_environment as mcp_analyze
 
         client = FastMCPClient(transport=mcp_config)
         # Extract server name for display (first key in mcp_config)
         server_name = next(iter(mcp_config.keys()), None)
 
         try:
+            start_time = time.time()
             await client.__aenter__()
+            initialize_ms = int((time.time() - start_time) * 1000)
             progress.update(task, description="[green]✓ Client initialized[/green]")
 
             # Analyze environment
             progress.update(task, description="Analyzing environment...")
-            analysis = await mcp_analyze(client, verbose, server_name=server_name)
+            analysis = await mcp_analyze(
+                client,
+                verbose,
+                server_name=server_name,
+                initialize_ms=initialize_ms,
+            )
             progress.update(task, description="[green]✓ Analysis complete[/green]")
 
         except Exception as e:
@@ -154,7 +165,7 @@ async def analyze_environment(docker_cmd: list[str], output_format: str, verbose
         display_interactive(analysis)
 
 
-def display_interactive(analysis: dict) -> None:
+def display_interactive(analysis: Mapping[str, Any]) -> None:
     """Display analysis results in interactive format."""
     # Server metadata
     hud_console.section_title("📊 Environment Overview")
@@ -201,13 +212,13 @@ def display_interactive(analysis: dict) -> None:
     hud_console.section_title("🔧 Available Tools")
     tools_tree = Tree("[bold bright_white]Tools[/bold bright_white]")
 
-    # Check if we have hub_tools info (live analysis) or not (metadata-only)
-    if "hub_tools" in analysis:
+    # Check if we have hubTools info (live analysis) or not (metadata-only)
+    if "hubTools" in analysis:
         # Live analysis format - separate regular and hub tools
         # Regular tools
         regular_tools = tools_tree.add("[bright_white]Regular Tools[/bright_white]")
         for tool in analysis["tools"]:
-            if tool["name"] not in analysis["hub_tools"]:
+            if tool["name"] not in analysis["hubTools"]:
                 tool_node = regular_tools.add(f"[bright_white]{tool['name']}[/bright_white]")
                 if tool["description"]:
                     tool_node.add(f"[bright_black]{tool['description']}[/bright_black]")
@@ -219,9 +230,9 @@ def display_interactive(analysis: dict) -> None:
                     tool_node.add(syntax)
 
         # Hub tools
-        if analysis["hub_tools"]:
+        if analysis["hubTools"]:
             hub_tools = tools_tree.add("[bright_white]Hub Tools[/bright_white]")
-            for hub_name, functions in analysis["hub_tools"].items():
+            for hub_name, functions in analysis["hubTools"].items():
                 hub_node = hub_tools.add(f"[rgb(181,137,0)]{hub_name}[/rgb(181,137,0)]")
                 for func in functions:
                     hub_node.add(f"[bright_white]{func}[/bright_white]")
@@ -314,7 +325,7 @@ def display_interactive(analysis: dict) -> None:
         console.print(env_table)
 
 
-def display_markdown(analysis: dict) -> None:
+def display_markdown(analysis: Mapping[str, Any]) -> None:
     """Display analysis results in markdown format."""
     md = []
     md.append("# MCP Environment Analysis\n")
@@ -344,19 +355,19 @@ def display_markdown(analysis: dict) -> None:
     # Tools
     md.append("## Available Tools\n")
 
-    # Check if we have hub_tools info (live analysis) or not (metadata-only)
-    if "hub_tools" in analysis:
+    # Check if we have hubTools info (live analysis) or not (metadata-only)
+    if "hubTools" in analysis:
         # Regular tools
         md.append("### Regular Tools")
         for tool in analysis["tools"]:
-            if tool["name"] not in analysis["hub_tools"]:
+            if tool["name"] not in analysis["hubTools"]:
                 md.extend([f"- **{tool['name']}**: {tool.get('description', 'No description')}"])
         md.append("")
 
         # Hub tools
-        if analysis["hub_tools"]:
+        if analysis["hubTools"]:
             md.append("### Hub Tools")
-            for hub_name, functions in analysis["hub_tools"].items():
+            for hub_name, functions in analysis["hubTools"].items():
                 md.extend([f"- **{hub_name}**"])
                 for func in functions:
                     md.extend([f"  - {func}"])
@@ -469,18 +480,26 @@ async def _analyze_with_config(
 
         from fastmcp import Client as FastMCPClient
 
-        from hud.cli.utils.mcp import analyze_environment as mcp_analyze
+        from hud.cli.utils.analysis import analyze_environment as mcp_analyze
 
         config = _prepare_mcp_config(mcp_config)
         client = FastMCPClient(transport=config)
+        server_name = next(iter(config.keys()), None)
 
         try:
+            start_time = time.time()
             await client.__aenter__()
+            initialize_ms = int((time.time() - start_time) * 1000)
             progress.update(task, description="[green]✓ Client initialized[/green]")
 
             # Analyze environment
             progress.update(task, description="Analyzing environment...")
-            analysis = await mcp_analyze(client, verbose)
+            analysis = await mcp_analyze(
+                client,
+                verbose,
+                server_name=server_name,
+                initialize_ms=initialize_ms,
+            )
             progress.update(task, description="[green]✓ Analysis complete[/green]")
 
         except Exception as e: