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

hud/cli/hf.py

@@ -9,9 +9,9 @@ from pathlib import Path
 import typer
 
 from hud.cli.rl.utils import get_mcp_config_from_lock, read_lock_file, write_lock_file
-from hud.utils.design import HUDDesign
+from hud.utils.hud_console import HUDConsole
 
-design = HUDDesign()
+hud_console = HUDConsole()
 
 
 def hf_command(
@@ -37,11 +37,11 @@ def hf_command(
         hud hf tasks.json --name hud-evals/web-tasks --private
         hud hf tasks.json --name local-dataset --no-push
     """
-    design.header("HuggingFace Dataset Converter", icon="📊")
+    hud_console.header("HuggingFace Dataset Converter", icon="📊")
 
     # Auto-detect task file if not provided
     if tasks_file is None:
-        design.info("Looking for task files...")
+        hud_console.info("Looking for task files...")
 
         # Common task file patterns
         patterns = [
@@ -60,30 +60,30 @@ def hf_command(
         json_files = sorted(set(json_files))
 
         if not json_files:
-            design.error("No task files found in current directory")
-            design.info("Create a task JSON file (e.g., tasks.json) or specify the file path")
+            hud_console.error("No task files found in current directory")
+            hud_console.info("Create a task JSON file (e.g., tasks.json) or specify the file path")
             raise typer.Exit(1)
         elif len(json_files) == 1:
             tasks_file = json_files[0]
-            design.info(f"Found task file: {tasks_file}")
+            hud_console.info(f"Found task file: {tasks_file}")
         else:
             # Multiple files found, let user choose
-            design.info("Multiple task files found:")
-            file_choice = design.select(
+            hud_console.info("Multiple task files found:")
+            file_choice = hud_console.select(
                 "Select a task file to convert:",
                 choices=[str(f) for f in json_files],
             )
             tasks_file = Path(file_choice)
-            design.success(f"Selected: {tasks_file}")
+            hud_console.success(f"Selected: {tasks_file}")
 
     # Validate inputs
     if tasks_file and not tasks_file.exists():
-        design.error(f"Tasks file not found: {tasks_file}")
+        hud_console.error(f"Tasks file not found: {tasks_file}")
         raise typer.Exit(1)
 
     # Suggest dataset name if not provided
     if name is None:
-        design.info("Generating dataset name suggestion...")
+        hud_console.info("Generating dataset name suggestion...")
 
         # Try to get HF username from environment or git config
         hf_username = None
@@ -107,7 +107,7 @@ def hf_command(
                 if result.returncode == 0 and result.stdout.strip():
                     hf_username = result.stdout.strip().lower().replace(" ", "-")
             except Exception:
-                design.warning("Failed to get HF username from git config")
+                hud_console.warning("Failed to get HF username from git config")
 
         # Get environment name from current directory or lock file
         env_name = Path.cwd().name
@@ -126,7 +126,7 @@ def hf_command(
                         if image_name and image_name != "local":
                             env_name = image_name
             except Exception as e:
-                design.warning(f"Failed to get HF username from lock file: {e}")
+                hud_console.warning(f"Failed to get HF username from lock file: {e}")
 
         # Generate suggestions
         suggestions = []
@@ -137,60 +137,60 @@ def hf_command(
         suggestions.append(f"hud-evals/{env_name}-tasks")
 
         # Let user choose or enter custom
-        design.info("Dataset name suggestions:")
+        hud_console.info("Dataset name suggestions:")
         suggestions.append("Enter custom name...")
 
-        choice = design.select("Select or enter a dataset name:", choices=suggestions)
+        choice = hud_console.select("Select or enter a dataset name:", choices=suggestions)
 
         if choice == "Enter custom name...":
             name = typer.prompt("Enter dataset name (e.g., 'my-org/my-dataset')")
         else:
             name = choice
 
-        design.success(f"Using dataset name: {name}")
+        hud_console.success(f"Using dataset name: {name}")
 
     # Validate dataset name format
     if push and name and "/" not in name:
-        design.error("Dataset name must include organization (e.g., 'my-org/my-dataset')")
-        design.info("For local-only datasets, use --no-push")
+        hud_console.error("Dataset name must include organization (e.g., 'my-org/my-dataset')")
+        hud_console.info("For local-only datasets, use --no-push")
         raise typer.Exit(1)
 
     # Load tasks
-    design.info(f"Loading tasks from: {tasks_file}")
+    hud_console.info(f"Loading tasks from: {tasks_file}")
     try:
         if tasks_file is None:
             raise ValueError("Tasks file is required")
         with open(tasks_file) as f:
             tasks_data = json.load(f)
     except json.JSONDecodeError as e:
-        design.error(f"Invalid JSON file: {e}")
+        hud_console.error(f"Invalid JSON file: {e}")
         raise typer.Exit(1) from e
 
     # Handle both single task and list of tasks
     if isinstance(tasks_data, dict):
         tasks = [tasks_data]
-        design.info("Found 1 task")
+        hud_console.info("Found 1 task")
     elif isinstance(tasks_data, list):
         tasks = tasks_data
-        design.info(f"Found {len(tasks)} tasks")
+        hud_console.info(f"Found {len(tasks)} tasks")
     else:
-        design.error("Tasks file must contain a JSON object or array")
+        hud_console.error("Tasks file must contain a JSON object or array")
         raise typer.Exit(1)
 
     # Validate task format
     valid_tasks = []
     for i, task in enumerate(tasks):
         if not isinstance(task, dict):
-            design.warning(f"Skipping task {i}: not a JSON object")
+            hud_console.warning(f"Skipping task {i}: not a JSON object")
             continue
 
         # Required fields
         if "prompt" not in task:
-            design.warning(f"Skipping task {i}: missing 'prompt' field")
+            hud_console.warning(f"Skipping task {i}: missing 'prompt' field")
             continue
 
         if "evaluate_tool" not in task:
-            design.warning(f"Skipping task {i}: missing 'evaluate_tool' field")
+            hud_console.warning(f"Skipping task {i}: missing 'evaluate_tool' field")
             continue
 
         # Add default values
@@ -203,23 +203,23 @@ def hf_command(
             if mcp_config:
                 task["mcp_config"] = mcp_config
             else:
-                design.warning(f"Task {task['id']}: missing 'mcp_config' field")
+                hud_console.warning(f"Task {task['id']}: missing 'mcp_config' field")
                 continue
 
         valid_tasks.append(task)
 
     if not valid_tasks:
-        design.error("No valid tasks found")
+        hud_console.error("No valid tasks found")
         raise typer.Exit(1)
 
-    design.success(f"Validated {len(valid_tasks)} tasks")
+    hud_console.success(f"Validated {len(valid_tasks)} tasks")
 
     # Check if dataset is suitable for training
     if len(valid_tasks) < 4:
-        design.warning(
+        hud_console.warning(
             f"Dataset has only {len(valid_tasks)} task(s). RL training typically requires at least 4 tasks."  # noqa: E501
         )
-        use_for_training = design.select(
+        use_for_training = hud_console.select(
             "Will this dataset be used for RL training?",
             ["Yes, duplicate tasks to reach 4", "No, keep as is"],
         )
@@ -238,7 +238,7 @@ def hf_command(
                     )
                     valid_tasks.append(duplicated_task)
 
-            design.info(f"Duplicated tasks: {original_count} → {len(valid_tasks)}")
+            hud_console.info(f"Duplicated tasks: {original_count} → {len(valid_tasks)}")
 
     # Check if MCP configs should be converted to remote
     sample_mcp_config = valid_tasks[0].get("mcp_config", {})
@@ -266,10 +266,10 @@ def hf_command(
                 config_type = "local"
 
     if config_type == "remote" and remote_image:
-        design.info(f"Tasks already use remote MCP configs with image: {remote_image}")
+        hud_console.info(f"Tasks already use remote MCP configs with image: {remote_image}")
 
     if config_type == "local":
-        convert_to_remote = design.select(
+        convert_to_remote = hud_console.select(
             "Tasks use local MCP configs. Convert to remote configs for training?",
             ["Yes, convert to remote (requires public image)", "No, keep local configs"],
         )
@@ -281,38 +281,38 @@ def hf_command(
             image = get_image_from_lock()
 
             if not image:
-                design.error("No image found in hud.lock.yaml")
-                design.hint("Run 'hud build' first")
+                hud_console.error("No image found in hud.lock.yaml")
+                hud_console.hint("Run 'hud build' first")
                 raise typer.Exit(1)
 
             # Check if image contains registry prefix (indicates it's pushed)
             if "/" not in image or image.startswith("local/"):
                 # Clean up image name for display (remove SHA if present)
                 display_image = image.split("@")[0] if "@" in image else image
-                design.warning(f"Image '{display_image}' appears to be local only")
-                push_image = design.select(
+                hud_console.warning(f"Image '{display_image}' appears to be local only")
+                push_image = hud_console.select(
                     "Would you like to push the image to make it publicly available?",
                     ["Yes, push image", "No, cancel"],
                 )
 
                 if push_image == "Yes, push image":
-                    design.info("Running 'hud push' to publish image...")
+                    hud_console.info("Running 'hud push' to publish image...")
                     # Import here to avoid circular imports
                     from hud.cli.push import push_command
 
                     # Run push command (it's synchronous)
                     push_command(directory=".", yes=True)
-                    design.success("Image pushed successfully")
+                    hud_console.success("Image pushed successfully")
 
                     # Re-read the image name as it may have changed
                     image = get_image_from_lock()
                 else:
-                    design.info("Keeping local MCP configs")
+                    hud_console.info("Keeping local MCP configs")
                     convert_to_remote = None
 
             if convert_to_remote and image:
                 # Convert all task configs to remote
-                design.info(f"Converting MCP configs to use remote image: {image}")
+                hud_console.info(f"Converting MCP configs to use remote image: {image}")
 
                 for task in valid_tasks:
                     # Create remote MCP config
@@ -327,7 +327,7 @@ def hf_command(
                     }
                     task["mcp_config"] = remote_config
 
-                design.success("✓ Converted all tasks to use remote MCP configs")
+                hud_console.success("✓ Converted all tasks to use remote MCP configs")
 
     # Convert to HuggingFace format
     dataset_dict = {
@@ -352,11 +352,11 @@ def hf_command(
         try:
             from datasets import Dataset
         except ImportError as e:
-            design.error("datasets library not installed")
-            design.info("Install with: pip install datasets")
+            hud_console.error("datasets library not installed")
+            hud_console.info("Install with: pip install datasets")
             raise typer.Exit(1) from e
 
-        design.info(f"Creating HuggingFace dataset: {name}")
+        hud_console.info(f"Creating HuggingFace dataset: {name}")
         dataset = Dataset.from_dict(dataset_dict)
 
         # Set up HF token
@@ -365,15 +365,15 @@ def hf_command(
 
             os.environ["HF_TOKEN"] = token
 
-        design.info(f"Pushing to Hub (private={private})...")
+        hud_console.info(f"Pushing to Hub (private={private})...")
         try:
             if name is None:
                 raise ValueError("Dataset name is required")
             dataset.push_to_hub(name, private=private)
-            design.success(f"Dataset published: https://huggingface.co/datasets/{name}")
+            hud_console.success(f"Dataset published: https://huggingface.co/datasets/{name}")
         except Exception as e:
-            design.error(f"Failed to push to Hub: {e}")
-            design.hint("Make sure you're logged in: huggingface-cli login")
+            hud_console.error(f"Failed to push to Hub: {e}")
+            hud_console.hint("Make sure you're logged in: huggingface-cli login")
             raise typer.Exit(1) from e
     else:
         # Save locally
@@ -382,7 +382,7 @@ def hf_command(
         output_file = Path(f"{name.replace('/', '_')}_dataset.json")
         with open(output_file, "w") as f:
             json.dump(dataset_dict, f, indent=2)
-        design.success(f"Dataset saved locally: {output_file}")
+        hud_console.success(f"Dataset saved locally: {output_file}")
 
     # Update hud.lock.yaml if requested
     if update_lock:
@@ -403,4 +403,4 @@ def update_lock_file(dataset_name: str, task_count: int) -> None:
 
     # Write back
     if write_lock_file(lock_data):
-        design.success(f"Updated hud.lock.yaml with dataset: {dataset_name}")
+        hud_console.success(f"Updated hud.lock.yaml with dataset: {dataset_name}")

hud/cli/init.py

@@ -8,7 +8,7 @@ import typer
 from rich.panel import Panel
 from rich.syntax import Syntax
 
-from hud.utils.design import HUDDesign
+from hud.utils.hud_console import HUDConsole
 
 # Embedded templates
 DOCKERFILE_TEMPLATE = """FROM python:3.11-slim
@@ -541,7 +541,7 @@ def sanitize_name(name: str) -> str:
 def create_environment(name: str | None, directory: str, force: bool) -> None:
     """Create a new HUD environment from templates."""
 
-    design = HUDDesign()
+    hud_console = HUDConsole()
 
     # Determine environment name
     if name is None:
@@ -549,7 +549,7 @@ def create_environment(name: str | None, directory: str, force: bool) -> None:
         current_dir = Path.cwd()
         name = current_dir.name
         target_dir = current_dir
-        design.info(f"Using current directory name: {name}")
+        hud_console.info(f"Using current directory name: {name}")
     else:
         # Create new directory
         target_dir = Path(directory) / name
@@ -557,16 +557,16 @@ def create_environment(name: str | None, directory: str, force: bool) -> None:
     # Sanitize name for Python package
     package_name = sanitize_name(name)
     if package_name != name:
-        design.warning(f"Package name adjusted for Python: {name} → {package_name}")
+        hud_console.warning(f"Package name adjusted for Python: {name} → {package_name}")
 
     # Check if directory exists
     if target_dir.exists() and any(target_dir.iterdir()):
         if not force:
-            design.error(f"Directory {target_dir} already exists and is not empty")
-            design.info("Use --force to overwrite existing files")
+            hud_console.error(f"Directory {target_dir} already exists and is not empty")
+            hud_console.info("Use --force to overwrite existing files")
             raise typer.Exit(1)
         else:
-            design.warning(f"Overwriting existing files in {target_dir}")
+            hud_console.warning(f"Overwriting existing files in {target_dir}")
 
     # Create directory structure
     src_dir = target_dir / "src" / "controller"
@@ -633,38 +633,38 @@ def create_environment(name: str | None, directory: str, force: bool) -> None:
     files_created.append(".env")
 
     # Success message
-    design.header(f"Created HUD Environment: {name}")
+    hud_console.header(f"Created HUD Environment: {name}")
 
-    design.section_title("Files created")
+    hud_console.section_title("Files created")
     for file in files_created:
-        design.status_item(file, "created")
+        hud_console.status_item(file, "created")
 
-    design.section_title("Next steps")
+    hud_console.section_title("Next steps")
 
     # Show commands based on where we created the environment
     if target_dir == Path.cwd():
-        design.info("1. Start development server (with MCP inspector):")
-        design.command_example("hud dev --inspector")
+        hud_console.info("1. Start development server (with MCP inspector):")
+        hud_console.command_example("hud dev --inspector")
     else:
-        design.info("1. Enter the directory:")
-        design.command_example(f"cd {target_dir}")
-        design.info("\n2. Start development server (with MCP inspector):")
-        design.command_example("hud dev --inspector")
+        hud_console.info("1. Enter the directory:")
+        hud_console.command_example(f"cd {target_dir}")
+        hud_console.info("\n2. Start development server (with MCP inspector):")
+        hud_console.command_example("hud dev --inspector")
 
-    design.info("\n3. Connect from Cursor or test via the MCP inspector:")
-    design.info("   Follow the instructions shown by hud dev --inspector")
+    hud_console.info("\n3. Connect from Cursor or test via the MCP inspector:")
+    hud_console.info("   Follow the instructions shown by hud dev --inspector")
 
-    design.info("\n4. Test your environment:")
-    design.command_example("python test_task.py")
+    hud_console.info("\n4. Test your environment:")
+    hud_console.command_example("python test_task.py")
 
-    design.info("\n5. Customize your environment:")
-    design.info("   - Add tools to src/controller/server.py")
-    design.info("   - Add state to src/controller/env.py")
-    design.info("   - Modify tasks in tasks.json")
-    design.info("   - Experiment in test_env.ipynb")
+    hud_console.info("\n5. Customize your environment:")
+    hud_console.info("   - Add tools to src/controller/server.py")
+    hud_console.info("   - Add state to src/controller/env.py")
+    hud_console.info("   - Modify tasks in tasks.json")
+    hud_console.info("   - Experiment in test_env.ipynb")
 
     # Show a sample of the server code
-    design.section_title("Your MCP server")
+    hud_console.section_title("Your MCP server")
     sample_code = '''@mcp.tool()
 async def act() -> str:
     """Perform an action that changes the environment state."""
@@ -674,4 +674,4 @@ async def act() -> str:
     return f"Action #{count} performed. Current count: {count}"'''
 
     syntax = Syntax(sample_code, "python", theme="monokai", line_numbers=False)
-    design.console.print(Panel(syntax, border_style="dim"))
+    hud_console.console.print(Panel(syntax, border_style="dim"))

hud/cli/list_func.py

@@ -8,7 +8,7 @@ import typer
 import yaml
 from rich.table import Table
 
-from hud.utils.design import HUDDesign
+from hud.utils.hud_console import HUDConsole
 
 from .utils.registry import extract_name_and_tag, get_registry_dir, list_registry_entries
 
@@ -46,10 +46,10 @@ def list_environments(
     verbose: bool = False,
 ) -> None:
     """List all HUD environments in the local registry."""
-    design = HUDDesign()
+    hud_console = HUDConsole()
 
     if not json_output:
-        design.header("HUD Environment Registry")
+        hud_console.header("HUD Environment Registry")
 
     # Check for environment directory
     env_dir = get_registry_dir()
@@ -57,10 +57,10 @@ def list_environments(
         if json_output:
             print("[]")  # noqa: T201
         else:
-            design.info("No environments found in local registry.")
-            design.info("")
-            design.info("Pull environments with: [cyan]hud pull <org/name:tag>[/cyan]")
-            design.info("Build environments with: [cyan]hud build[/cyan]")
+            hud_console.info("No environments found in local registry.")
+            hud_console.info("")
+            hud_console.info("Pull environments with: [cyan]hud pull <org/name:tag>[/cyan]")
+            hud_console.info("Build environments with: [cyan]hud build[/cyan]")
         return
 
     # Collect all environments using the registry helper
@@ -103,7 +103,7 @@ def list_environments(
 
         except Exception as e:
             if verbose:
-                design.warning(f"Failed to read {lock_file}: {e}")
+                hud_console.warning(f"Failed to read {lock_file}: {e}")
 
     # Sort by pulled time (newest first)
     environments.sort(key=lambda x: x["pulled_time"], reverse=True)
@@ -129,10 +129,10 @@ def list_environments(
         return
 
     if not environments:
-        design.info("No environments found matching criteria.")
-        design.info("")
-        design.info("Pull environments with: [cyan]hud pull <org/name:tag>[/cyan]")
-        design.info("Build environments with: [cyan]hud build[/cyan]")
+        hud_console.info("No environments found matching criteria.")
+        hud_console.info("")
+        hud_console.info("Pull environments with: [cyan]hud pull <org/name:tag>[/cyan]")
+        hud_console.info("Build environments with: [cyan]hud build[/cyan]")
         return
 
     # Create table
@@ -169,26 +169,26 @@ def list_environments(
 
         table.add_row(*row)
 
-    design.print(table)  # type: ignore
-    design.info("")
+    hud_console.print(table)  # type: ignore
+    hud_console.info("")
 
     # Show usage hints
-    design.section_title("Usage")
+    hud_console.section_title("Usage")
     if environments:
         # Use the most recently pulled environment as example
         example_env = environments[0]
         example_ref = f"{example_env['name']}:{example_env['tag']}"
 
-        design.info(f"Run an environment: [cyan]hud run {example_ref}[/cyan]")
-        design.info(f"Analyze tools: [cyan]hud analyze {example_ref}[/cyan]")
-        design.info(f"Debug server: [cyan]hud debug {example_ref}[/cyan]")
+        hud_console.info(f"Run an environment: [cyan]hud run {example_ref}[/cyan]")
+        hud_console.info(f"Analyze tools: [cyan]hud analyze {example_ref}[/cyan]")
+        hud_console.info(f"Debug server: [cyan]hud debug {example_ref}[/cyan]")
 
-    design.info("Pull more environments: [cyan]hud pull <org/name:tag>[/cyan]")
-    design.info("Build new environments: [cyan]hud build[/cyan]")
+    hud_console.info("Pull more environments: [cyan]hud pull <org/name:tag>[/cyan]")
+    hud_console.info("Build new environments: [cyan]hud build[/cyan]")
 
     if verbose:
-        design.info("")
-        design.info(f"[dim]Registry location: {env_dir}[/dim]")
+        hud_console.info("")
+        hud_console.info(f"[dim]Registry location: {env_dir}[/dim]")
 
 
 def list_command(