dayhoff-tools 1.3.14__tar.gz → 1.3.16__tar.gz

{dayhoff_tools-1.3.14 → dayhoff_tools-1.3.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: dayhoff-tools
-Version: 1.3.14
+Version: 1.3.16
 Summary: Common tools for all the repos at Dayhoff Labs
 Author: Daniel Martin-Alarcon
 Author-email: dma@dayhofflabs.com

{dayhoff_tools-1.3.14 → dayhoff_tools-1.3.16}/dayhoff_tools/cli/engine_commands.py

@@ -388,6 +388,12 @@ def launch_engine(
         help="Engine type: cpu, cpumax, t4, a10g, a100",
     ),
     user: Optional[str] = typer.Option(None, "--user", "-u", help="Override username"),
+    boot_disk_size: Optional[int] = typer.Option(
+        None,
+        "--size",
+        "-s",
+        help="Boot disk size in GB (default: 50GB, min: 20GB, max: 1000GB)",
+    ),
 ):
     """Launch a new engine instance."""
     username = check_aws_sso()
@@ -401,9 +407,19 @@ def launch_engine(
         console.print(f"Valid types: {', '.join(valid_types)}")
         raise typer.Exit(1)
 
+    # Validate boot disk size
+    if boot_disk_size is not None:
+        if boot_disk_size < 20:
+            console.print("[red]❌ Boot disk size must be at least 20GB[/red]")
+            raise typer.Exit(1)
+        if boot_disk_size > 1000:
+            console.print("[red]❌ Boot disk size cannot exceed 1000GB[/red]")
+            raise typer.Exit(1)
+
     cost = HOURLY_COSTS.get(engine_type, 0)
+    disk_info = f" with {boot_disk_size}GB boot disk" if boot_disk_size else ""
     console.print(
-        f"Launching [cyan]{name}[/cyan] ({engine_type}) for ${cost:.2f}/hour..."
+        f"Launching [cyan]{name}[/cyan] ({engine_type}){disk_info} for ${cost:.2f}/hour..."
     )
 
     with Progress(
@@ -413,17 +429,23 @@ def launch_engine(
     ) as progress:
         progress.add_task("Creating engine...", total=None)
 
-        response = make_api_request(
-            "POST",
-            "/engines",
-            json_data={"name": name, "user": username, "engine_type": engine_type},
-        )
+        request_data = {
+            "name": name,
+            "user": username,
+            "engine_type": engine_type,
+        }
+        if boot_disk_size is not None:
+            request_data["boot_disk_size"] = boot_disk_size
+
+        response = make_api_request("POST", "/engines", json_data=request_data)
 
     if response.status_code == 201:
         data = response.json()
         console.print(f"[green]✓ Engine launched successfully![/green]")
         console.print(f"Instance ID: [cyan]{data['instance_id']}[/cyan]")
         console.print(f"Type: {data['instance_type']} (${cost:.2f}/hour)")
+        if boot_disk_size:
+            console.print(f"Boot disk: {boot_disk_size}GB")
         console.print("\nThe engine is initializing. This may take a few minutes.")
         console.print(f"Check status with: [cyan]dh engine status {name}[/cyan]")
     else:
@@ -1104,8 +1126,9 @@ def resize_engine(
         elif online and engine["state"].lower() == "running":
             console.print("\n[yellow]⚠️  Online resize complete. You must now expand the filesystem:[/yellow]")
             console.print(f"1. SSH into the engine: [cyan]ssh {engine['name']}[/cyan]")
-            console.print("2. Run: [cyan]sudo growpart /dev/nvme0n1 1[/cyan]")
-            console.print("3. Run: [cyan]sudo xfs_growfs /[/cyan]")
+            console.print("2. Find the root device: [cyan]lsblk[/cyan]")
+            console.print("3. Expand the partition: [cyan]sudo growpart /dev/nvme0n1 1[/cyan] (adjust device name as needed)")
+            console.print("4. Expand the filesystem: [cyan]sudo xfs_growfs /[/cyan]")
         
     except ClientError as e:
         console.print(f"[red]❌ Failed to resize engine: {e}[/red]")

{dayhoff_tools-1.3.14 → dayhoff_tools-1.3.16}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "dayhoff-tools"
-version = "1.3.14"
+version = "1.3.16"
 description = "Common tools for all the repos at Dayhoff Labs"
 authors = [
     {name = "Daniel Martin-Alarcon", email = "dma@dayhofflabs.com"}