hud-python 0.4.45__py3-none-any.whl → 0.5.1__py3-none-any.whl

hud/cli/analyze.py

@@ -12,7 +12,6 @@ from rich.syntax import Syntax
 from rich.table import Table
 from rich.tree import Tree
 
-from hud.clients import MCPClient
 from hud.utils.hud_console import HUDConsole
 
 console = Console()
@@ -45,7 +44,10 @@ async def analyze_environment(docker_cmd: list[str], output_format: str, verbose
     ) as progress:
         task = progress.add_task("Initializing MCP client...", total=None)
 
-        client = MCPClient(mcp_config=mcp_config, verbose=verbose, auto_trace=False)
+        # Use FastMCP client directly - no mcp_use deprecation warnings
+        from hud.clients.fastmcp import FastMCPHUDClient
+
+        client = FastMCPHUDClient(mcp_config=mcp_config, verbose=verbose, auto_trace=False)
 
         try:
             await client.initialize()
@@ -141,8 +143,8 @@ def display_interactive(analysis: dict) -> None:
                     tool_node.add(f"[bright_black]{tool['description']}[/bright_black]")
 
                 # Show input schema if verbose
-                if analysis.get("verbose") and tool.get("input_schema"):
-                    schema_str = json.dumps(tool["input_schema"], indent=2)
+                if analysis.get("verbose") and tool.get("inputSchema"):
+                    schema_str = json.dumps(tool["inputSchema"], indent=2)
                     syntax = Syntax(schema_str, "json", theme="monokai", line_numbers=False)
                     tool_node.add(syntax)
 
@@ -168,6 +170,28 @@ def display_interactive(analysis: dict) -> None:
 
     console.print(tools_tree)
 
+    # Scenarios (Environment scripts exposed as prompt+resource)
+    if analysis.get("scenarios"):
+        hud_console.section_title("🎬 Scenarios")
+        scenarios_table = Table()
+        scenarios_table.add_column("Scenario", style="bright_white")
+        scenarios_table.add_column("Env", style="bright_black")
+        scenarios_table.add_column("Setup/Eval", style="bright_black")
+
+        for s in analysis["scenarios"][:20]:
+            setup = "✓" if s.get("has_setup_prompt") else "✗"
+            eval_ = "✓" if s.get("has_evaluate_resource") else "✗"
+            scenarios_table.add_row(
+                str(s.get("name", "")),
+                str(s.get("env", "")),
+                f"setup {setup} / eval {eval_}",
+            )
+
+        console.print(scenarios_table)
+        if len(analysis["scenarios"]) > 20:
+            remaining = len(analysis["scenarios"]) - 20
+            console.print(f"[bright_black]... and {remaining} more scenarios[/bright_black]")
+
     # Resources
     if analysis["resources"]:
         hud_console.section_title("📚 Available Resources")
@@ -283,6 +307,17 @@ def display_markdown(analysis: dict) -> None:
             md.extend([f"| {uri} | {name} | {mime_type} |"])
         md.append("")
 
+    # Scenarios
+    if analysis.get("scenarios"):
+        md.append("## Scenarios\n")
+        for s in analysis["scenarios"]:
+            name = s.get("name", "")
+            env = s.get("env", "")
+            setup = "✓" if s.get("has_setup_prompt") else "✗"
+            eval_ = "✓" if s.get("has_evaluate_resource") else "✗"
+            md.append(f"- **{name}** ({env}) — setup {setup} / eval {eval_}")
+        md.append("")
+
     # Telemetry (only for live analysis)
     if analysis.get("telemetry"):
         md.append("## Telemetry")
@@ -344,7 +379,10 @@ async def _analyze_with_config(
     ) as progress:
         task = progress.add_task("Initializing MCP client...", total=None)
 
-        client = MCPClient(mcp_config=mcp_config, verbose=verbose)
+        # Use FastMCP client directly - no mcp_use deprecation warnings
+        from hud.clients.fastmcp import FastMCPHUDClient
+
+        client = FastMCPHUDClient(mcp_config=mcp_config, verbose=verbose)
 
         try:
             await client.initialize()