hud-python 0.4.21__py3-none-any.whl → 0.4.23__py3-none-any.whl

hud/cli/remove.py

@@ -6,7 +6,7 @@ import shutil
 
 import typer
 
-from hud.utils.design import HUDDesign
+from hud.utils.hud_console import HUDConsole
 
 from .utils.registry import get_registry_dir, list_registry_entries, load_from_registry
 
@@ -17,8 +17,8 @@ def remove_environment(
     verbose: bool = False,
 ) -> None:
     """Remove an environment from the local registry."""
-    design = HUDDesign()
-    design.header("HUD Environment Removal")
+    hud_console = HUDConsole()
+    hud_console.header("HUD Environment Removal")
 
     # Find the environment to remove
     found_entry = None
@@ -45,12 +45,12 @@ def remove_environment(
                         found_digest = digest
                         break
             except Exception as e:
-                design.error(f"Error loading lock file: {e}")
+                hud_console.error(f"Error loading lock file: {e}")
                 continue
 
     if not found_entry:
-        design.error(f"Environment not found: {target}")
-        design.info("Use 'hud list' to see available environments")
+        hud_console.error(f"Environment not found: {target}")
+        hud_console.info("Use 'hud list' to see available environments")
         raise typer.Exit(1)
 
     # Load and display environment info
@@ -63,37 +63,39 @@ def remove_environment(
             metadata = lock_data.get("metadata", {})
             description = metadata.get("description", "No description")
 
-            design.section_title("Environment Details")
-            design.status_item("Image", image)
-            design.status_item("Digest", found_digest)
-            design.status_item("Description", description)
-            design.status_item("Location", str(found_entry.parent))
+            hud_console.section_title("Environment Details")
+            hud_console.status_item("Image", image)
+            hud_console.status_item("Digest", found_digest)
+            hud_console.status_item("Description", description)
+            hud_console.status_item("Location", str(found_entry.parent))
     except Exception as e:
         if verbose:
-            design.warning(f"Could not read environment details: {e}")
+            hud_console.warning(f"Could not read environment details: {e}")
 
     # Confirm deletion
     if not yes:
-        design.info("")
+        hud_console.info("")
         if not typer.confirm(f"Remove environment {found_digest}?"):
-            design.info("Aborted")
+            hud_console.info("Aborted")
             raise typer.Exit(0)
 
     # Remove the environment directory
     try:
         env_dir = found_entry.parent
         shutil.rmtree(env_dir)
-        design.success(f"Removed environment: {found_digest}")
+        hud_console.success(f"Removed environment: {found_digest}")
 
         # Check if the image is still available locally
         if lock_data:
             image = lock_data.get("image", "")
             if image:
-                design.info("")
-                design.info("Note: The Docker image may still exist locally.")
-                design.info(f"To remove it, run: [cyan]docker rmi {image.split('@')[0]}[/cyan]")
+                hud_console.info("")
+                hud_console.info("Note: The Docker image may still exist locally.")
+                hud_console.info(
+                    f"To remove it, run: [cyan]docker rmi {image.split('@')[0]}[/cyan]"
+                )
     except Exception as e:
-        design.error(f"Failed to remove environment: {e}")
+        hud_console.error(f"Failed to remove environment: {e}")
         raise typer.Exit(1) from e
 
 
@@ -102,38 +104,38 @@ def remove_all_environments(
     verbose: bool = False,
 ) -> None:
     """Remove all environments from the local registry."""
-    design = HUDDesign()
-    design.header("Remove All HUD Environments")
+    hud_console = HUDConsole()
+    hud_console.header("Remove All HUD Environments")
 
     registry_dir = get_registry_dir()
     if not registry_dir.exists():
-        design.info("No environments found in local registry.")
+        hud_console.info("No environments found in local registry.")
         return
 
     # Count environments
     entries = list(list_registry_entries())
     if not entries:
-        design.info("No environments found in local registry.")
+        hud_console.info("No environments found in local registry.")
         return
 
-    design.warning(f"This will remove {len(entries)} environment(s) from the local registry!")
+    hud_console.warning(f"This will remove {len(entries)} environment(s) from the local registry!")
 
     # List environments that will be removed
-    design.section_title("Environments to Remove")
+    hud_console.section_title("Environments to Remove")
     for digest, _ in entries:
         try:
             lock_data = load_from_registry(digest)
             if lock_data:
                 image = lock_data.get("image", "unknown")
-                design.info(f"  • {digest[:12]} - {image}")
+                hud_console.info(f"  • {digest[:12]} - {image}")
         except Exception:
-            design.info(f"  • {digest[:12]}")
+            hud_console.info(f"  • {digest[:12]}")
 
     # Confirm deletion
     if not yes:
-        design.info("")
+        hud_console.info("")
         if not typer.confirm("Remove ALL environments?", default=False):
-            design.info("Aborted")
+            hud_console.info("Aborted")
             raise typer.Exit(0)
 
     # Remove all environments
@@ -146,21 +148,21 @@ def remove_all_environments(
             shutil.rmtree(env_dir)
             removed += 1
             if verbose:
-                design.success(f"Removed: {digest}")
+                hud_console.success(f"Removed: {digest}")
         except Exception as e:
             failed += 1
             if verbose:
-                design.error(f"Failed to remove {digest}: {e}")
+                hud_console.error(f"Failed to remove {digest}: {e}")
 
-    design.info("")
+    hud_console.info("")
     if failed == 0:
-        design.success(f"Successfully removed {removed} environment(s)")
+        hud_console.success(f"Successfully removed {removed} environment(s)")
     else:
-        design.warning(f"Removed {removed} environment(s), failed to remove {failed}")
+        hud_console.warning(f"Removed {removed} environment(s), failed to remove {failed}")
 
-    design.info("")
-    design.info("Note: Docker images may still exist locally.")
-    design.info("To remove them, use: [cyan]docker image prune[/cyan]")
+    hud_console.info("")
+    hud_console.info("Note: Docker images may still exist locally.")
+    hud_console.info("To remove them, use: [cyan]docker image prune[/cyan]")
 
 
 def remove_command(
@@ -183,9 +185,9 @@ def remove_command(
         hud remove all --yes           # Remove all without confirmation
     """
     if not target:
-        design = HUDDesign()
-        design.error("Please specify an environment to remove or 'all'")
-        design.info("Use 'hud list' to see available environments")
+        hud_console = HUDConsole()
+        hud_console.error("Please specify an environment to remove or 'all'")
+        hud_console.info("Use 'hud list' to see available environments")
         raise typer.Exit(1)
 
     if target.lower() == "all":

hud/cli/rl/__init__.py

@@ -6,7 +6,7 @@ from pathlib import Path  # noqa: TC003
 
 import typer
 
-from hud.utils.design import HUDDesign
+from hud.utils.hud_console import HUDConsole
 
 # Create the RL subcommand app
 rl_app = typer.Typer(
@@ -15,7 +15,7 @@ rl_app = typer.Typer(
     rich_markup_mode="rich",
 )
 
-design = HUDDesign()
+hud_console = HUDConsole()
 
 
 @rl_app.callback(invoke_without_command=True)

hud/cli/rl/init.py

@@ -10,14 +10,14 @@ import typer
 import yaml
 
 from hud.clients import MCPClient
-from hud.utils.design import HUDDesign
+from hud.utils.hud_console import HUDConsole
 
-design = HUDDesign()
+hud_console = HUDConsole()
 
 
 def init_command_wrapper(directory: str, output: Path | None, force: bool, build: bool) -> None:
     """Wrapper to handle interactive prompts before entering async context."""
-    design.header("RL Config Generator", icon="🔧")
+    hud_console.header("RL Config Generator", icon="🔧")
 
     # Determine if this is a directory or Docker image
     path = Path(directory)
@@ -30,7 +30,7 @@ def init_command_wrapper(directory: str, output: Path | None, force: bool, build
         if not lock_path.exists():
             if build:
                 # Auto-build was requested
-                design.info("Building environment...")
+                hud_console.info("Building environment...")
                 from hud.cli.build import build_command
 
                 build_command(str(directory), None, False, False, {})
@@ -42,15 +42,15 @@ def init_command_wrapper(directory: str, output: Path | None, force: bool, build
                 image, source = get_image_name(directory)
 
                 if not (source == "cache" and image_exists(image)):
-                    design.warning(f"No hud.lock.yaml found in {directory}")
+                    hud_console.warning(f"No hud.lock.yaml found in {directory}")
                     # Need to handle interactive prompt here, before async
-                    action = design.select(
+                    action = hud_console.select(
                         "No lock file found. Would you like to:",
                         ["Build the environment", "Use Docker image directly", "Cancel"],
                     )
 
                     if action == "Build the environment":
-                        design.info("Building environment...")
+                        hud_console.info("Building environment...")
                         from hud.cli.build import build_command
 
                         build_command(str(directory), None, False, False, {})
@@ -78,11 +78,11 @@ async def init_command(directory: str, output: Path | None, force: bool, build:
         lock_path = path / "hud.lock.yaml"
 
         if lock_path.exists():
-            design.info(f"Found lock file: {lock_path}")
+            hud_console.info(f"Found lock file: {lock_path}")
             lock_data = read_lock_file_path(lock_path)
 
             if not lock_data:
-                design.error("Failed to read lock file")
+                hud_console.error("Failed to read lock file")
                 raise typer.Exit(1)
 
             # Get image and tools from lock file
@@ -90,13 +90,13 @@ async def init_command(directory: str, output: Path | None, force: bool, build:
             tools = lock_data.get("tools", [])
 
             if not image:
-                design.error("No image found in lock file")
-                design.hint("Run 'hud build' to create a proper lock file")
+                hud_console.error("No image found in lock file")
+                hud_console.hint("Run 'hud build' to create a proper lock file")
                 raise typer.Exit(1)
 
             if not tools:
-                design.error("No tools found in lock file")
-                design.hint("Lock file may be outdated. Run 'hud build' to regenerate")
+                hud_console.error("No tools found in lock file")
+                hud_console.hint("Lock file may be outdated. Run 'hud build' to regenerate")
                 raise typer.Exit(1)
 
             # Use lock file data to generate config
@@ -111,11 +111,11 @@ async def init_command(directory: str, output: Path | None, force: bool, build:
 
             if source == "cache" and image_exists(image):
                 # Found cached image in pyproject.toml
-                design.info(f"Using cached image: {image}")
+                hud_console.info(f"Using cached image: {image}")
                 await analyze_and_generate(image, output, force)
             else:
                 # This should have been handled in the wrapper
-                design.error("No valid image or lock file found")
+                hud_console.error("No valid image or lock file found")
                 raise typer.Exit(1)
 
     else:
@@ -130,7 +130,7 @@ def read_lock_file_path(lock_path: Path) -> dict[str, Any]:
         with open(lock_path) as f:
             return yaml.safe_load(f) or {}
     except Exception as e:
-        design.error(f"Failed to read lock file: {e}")
+        hud_console.error(f"Failed to read lock file: {e}")
         return {}
 
 
@@ -148,8 +148,8 @@ async def generate_from_lock(
 
     # Check if file exists
     if output.exists() and not force:
-        design.error(f"Config file already exists: {output}")
-        design.info("Use --force to overwrite")
+        hud_console.error(f"Config file already exists: {output}")
+        hud_console.info("Use --force to overwrite")
         raise typer.Exit(1)
 
     # Create output directory if needed
@@ -178,18 +178,18 @@ async def generate_from_lock(
     with open(output, "w") as f:  # noqa: ASYNC230
         yaml.dump(config, f, default_flow_style=False, sort_keys=False)
 
-    design.success(f"Generated config: {output}")
+    hud_console.success(f"Generated config: {output}")
 
     # Show summary
-    design.section_title("📋 Generated Configuration")
-    design.info("Source: hud.lock.yaml")
-    design.info(f"Image: {image}")
-    design.info(f"System prompt: {len(config['system_prompt'])} characters")
-    design.info(f"Action mappings: {len(config['action_mappings'])} tools")
-    design.info("")
-    design.info("Next steps:")
-    design.command_example("hud hf tasks.json --name my-tasks", "Create dataset")
-    design.command_example(f"hud rl --config {output}", "Start training")
+    hud_console.section_title("📋 Generated Configuration")
+    hud_console.info("Source: hud.lock.yaml")
+    hud_console.info(f"Image: {image}")
+    hud_console.info(f"System prompt: {len(config['system_prompt'])} characters")
+    hud_console.info(f"Action mappings: {len(config['action_mappings'])} tools")
+    hud_console.info("")
+    hud_console.info("Next steps:")
+    hud_console.command_example("hud hf tasks.json --name my-tasks", "Create dataset")
+    hud_console.command_example(f"hud rl --config {output}", "Start training")
 
 
 async def analyze_and_generate(image: str, output: Path | None, force: bool) -> None:
@@ -202,14 +202,14 @@ async def analyze_and_generate(image: str, output: Path | None, force: bool) ->
 
     # Check if file exists
     if output.exists() and not force:
-        design.error(f"Config file already exists: {output}")
-        design.info("Use --force to overwrite")
+        hud_console.error(f"Config file already exists: {output}")
+        hud_console.info("Use --force to overwrite")
         raise typer.Exit(1)
 
     # Create output directory if needed
     output.parent.mkdir(parents=True, exist_ok=True)
 
-    design.info(f"Analyzing environment: {image}")
+    hud_console.info(f"Analyzing environment: {image}")
 
     # Analyze the environment
     try:
@@ -231,23 +231,23 @@ async def analyze_and_generate(image: str, output: Path | None, force: bool) ->
             with open(output, "w") as f:  # noqa: ASYNC230
                 yaml.dump(config, f, default_flow_style=False, sort_keys=False)
 
-            design.success(f"Generated config: {output}")
+            hud_console.success(f"Generated config: {output}")
 
             # Show summary
-            design.section_title("📋 Generated Configuration")
-            design.info(f"System prompt: {len(config['system_prompt'])} characters")
-            design.info(f"Action mappings: {len(config['action_mappings'])} tools")
-            design.info("")
-            design.info("Next steps:")
-            design.command_example("hud hf tasks.json --name my-tasks", "Create dataset")
-            design.command_example(f"hud rl --config {output}", "Start training")
+            hud_console.section_title("📋 Generated Configuration")
+            hud_console.info(f"System prompt: {len(config['system_prompt'])} characters")
+            hud_console.info(f"Action mappings: {len(config['action_mappings'])} tools")
+            hud_console.info("")
+            hud_console.info("Next steps:")
+            hud_console.command_example("hud hf tasks.json --name my-tasks", "Create dataset")
+            hud_console.command_example(f"hud rl --config {output}", "Start training")
 
         finally:
             await client.shutdown()
 
     except Exception as e:
-        design.error(f"Failed to analyze environment: {e}")
-        design.hint("Make sure the Docker image exists and contains a valid MCP server")
+        hud_console.error(f"Failed to analyze environment: {e}")
+        hud_console.hint("Make sure the Docker image exists and contains a valid MCP server")
         raise typer.Exit(1) from e