dayhoff-tools 1.13.2__py3-none-any.whl → 1.13.4__py3-none-any.whl

dayhoff_tools/cli/engines_studios/engine-studio-cli.md

@@ -330,7 +330,7 @@ Engines for AWS Account dev
 │ bob-training │ running     │ bob         │ a10g        │ i-0fedcba987654321  │
 │ batch-worker │ stopped     │ charlie     │ cpumax      │ i-0abc123def456789  │
 ╰──────────────┴─────────────┴─────────────┴─────────────┴─────────────────────╯
-Total: 3 engine(s)
+Total: 3
 
 ```
 
@@ -764,7 +764,7 @@ Studios for AWS Account dev
 │ bob    │ available    │ -            │ 200GB     │ vol-0fedcba987654321      │
 │ carol  │ attaching    │ carol-gpu    │ 150GB     │ vol-0abc123def456789      │
 ╰────────┴──────────────┴──────────────┴───────────┴───────────────────────────╯
-Total: 3 studio(s)
+Total: 3
 
 ```
 

dayhoff_tools/cli/engines_studios/engine_commands.py

@@ -123,7 +123,9 @@ def engine_cli():
 @click.option(
     "--skip-ssh-config", is_flag=True, help="Don't automatically update SSH config"
 )
-@click.option("--yes", "-y", is_flag=True, help="Skip confirmation for non-dev environments")
+@click.option(
+    "--yes", "-y", is_flag=True, help="Skip confirmation for non-dev environments"
+)
 @click.option(
     "--env",
     default=None,
@@ -210,6 +212,48 @@ def launch_engine(
             click.echo(f"  dh engine status {name}")
 
     except Exception as e:
+        error_msg = str(e)
+        
+        # Check for quota/limit errors
+        if "VcpuLimitExceeded" in error_msg or "vCPU limit" in error_msg:
+            click.echo(f"✗ Failed to launch engine: vCPU quota exceeded", err=True)
+            click.echo("", err=True)
+            click.echo(f"The {env} AWS account has insufficient vCPU quota for {engine_type} instances.", err=True)
+            click.echo("", err=True)
+            click.echo("Solutions:", err=True)
+            click.echo("  1. Use a different instance type (e.g., --type cpu)", err=True)
+            click.echo("  2. Request a quota increase:", err=True)
+            click.echo("     • AWS Console → Service Quotas → Amazon EC2", err=True)
+            click.echo("     • Find quota for the instance family", err=True)
+            click.echo("     • Request increase (typically approved within 24h)", err=True)
+            click.echo("", err=True)
+            click.echo("For testing infrastructure, use CPU instances instead of GPU.", err=True)
+            raise click.Abort()
+        
+        # Check for insufficient capacity errors
+        if "InsufficientInstanceCapacity" in error_msg:
+            click.echo(f"✗ Failed to launch engine: insufficient EC2 capacity", err=True)
+            click.echo("", err=True)
+            click.echo(f"AWS does not have available {engine_type} capacity in your region/AZ.", err=True)
+            click.echo("", err=True)
+            click.echo("Solutions:", err=True)
+            click.echo("  1. Try again in a few minutes (capacity fluctuates)", err=True)
+            click.echo("  2. Use a different instance type", err=True)
+            click.echo("  3. Contact AWS support for capacity reservations", err=True)
+            raise click.Abort()
+        
+        # Check for instance limit errors
+        if "InstanceLimitExceeded" in error_msg or "instance limit" in error_msg.lower():
+            click.echo(f"✗ Failed to launch engine: instance limit exceeded", err=True)
+            click.echo("", err=True)
+            click.echo(f"You have reached the maximum number of running instances in {env}.", err=True)
+            click.echo("", err=True)
+            click.echo("Solutions:", err=True)
+            click.echo("  1. Terminate unused engines: dh engine2 list --env {env}", err=True)
+            click.echo("  2. Request a limit increase via AWS Service Quotas", err=True)
+            raise click.Abort()
+        
+        # Generic error
         click.echo(f"✗ Failed to launch engine: {e}", err=True)
         raise click.Abort()
 
@@ -222,7 +266,9 @@ def launch_engine(
 @click.option(
     "--skip-ssh-config", is_flag=True, help="Don't automatically update SSH config"
 )
-@click.option("--yes", "-y", is_flag=True, help="Skip confirmation for non-dev environments")
+@click.option(
+    "--yes", "-y", is_flag=True, help="Skip confirmation for non-dev environments"
+)
 @click.option(
     "--env",
     default=None,
@@ -345,7 +391,9 @@ def start_engine(
 
 @engine_cli.command("stop")
 @click.argument("name_or_id")
-@click.option("--yes", "-y", is_flag=True, help="Skip confirmation for non-dev environments")
+@click.option(
+    "--yes", "-y", is_flag=True, help="Skip confirmation for non-dev environments"
+)
 @click.option(
     "--env",
     default=None,
@@ -588,7 +636,9 @@ def list_engines(env: Optional[str]):
             return
 
         # Calculate dynamic width for Name column (longest name + 2 for padding)
-        max_name_len = max((len(engine.get("name", "unknown")) for engine in engines), default=4)
+        max_name_len = max(
+            (len(engine.get("name", "unknown")) for engine in engines), default=4
+        )
         name_width = max(max_name_len + 2, len("Name") + 2)
 
         # Fixed widths for other columns
@@ -597,11 +647,20 @@ def list_engines(env: Optional[str]):
         type_width = 12
         id_width = 20
 
-        # Calculate total width for separator line
-        total_width = name_width + state_width + user_width + type_width + id_width + 9  # +9 for separators and spaces
-
         # Table top border
-        click.echo("╭" + "─" * (name_width + 1) + "┬" + "─" * (state_width + 1) + "┬" + "─" * (user_width + 1) + "┬" + "─" * (type_width + 1) + "┬" + "─" * (id_width + 1) + "╮")
+        click.echo(
+            "╭"
+            + "─" * (name_width + 1)
+            + "┬"
+            + "─" * (state_width + 1)
+            + "┬"
+            + "─" * (user_width + 1)
+            + "┬"
+            + "─" * (type_width + 1)
+            + "┬"
+            + "─" * (id_width + 1)
+            + "╮"
+        )
 
         # Table header
         click.echo(
@@ -609,7 +668,19 @@ def list_engines(env: Optional[str]):
         )
 
         # Header separator
-        click.echo("├" + "─" * (name_width + 1) + "┼" + "─" * (state_width + 1) + "┼" + "─" * (user_width + 1) + "┼" + "─" * (type_width + 1) + "┼" + "─" * (id_width + 1) + "┤")
+        click.echo(
+            "├"
+            + "─" * (name_width + 1)
+            + "┼"
+            + "─" * (state_width + 1)
+            + "┼"
+            + "─" * (user_width + 1)
+            + "┼"
+            + "─" * (type_width + 1)
+            + "┼"
+            + "─" * (id_width + 1)
+            + "┤"
+        )
 
         # Table rows
         for engine in engines:
@@ -621,11 +692,11 @@ def list_engines(env: Optional[str]):
 
             # Truncate if needed
             if len(name) > name_width - 1:
-                name = name[:name_width - 1]
+                name = name[: name_width - 1]
             if len(user) > user_width - 1:
-                user = user[:user_width - 1]
+                user = user[: user_width - 1]
             if len(engine_type) > type_width - 1:
-                engine_type = engine_type[:type_width - 1]
+                engine_type = engine_type[: type_width - 1]
 
             # Color the name (blue)
             name_display = f"\033[34m{name:<{name_width}}\033[0m"
@@ -648,9 +719,21 @@ def list_engines(env: Optional[str]):
             )
 
         # Table bottom border
-        click.echo("╰" + "─" * (name_width + 1) + "┴" + "─" * (state_width + 1) + "┴" + "─" * (user_width + 1) + "┴" + "─" * (type_width + 1) + "┴" + "─" * (id_width + 1) + "╯")
+        click.echo(
+            "╰"
+            + "─" * (name_width + 1)
+            + "┴"
+            + "─" * (state_width + 1)
+            + "┴"
+            + "─" * (user_width + 1)
+            + "┴"
+            + "─" * (type_width + 1)
+            + "┴"
+            + "─" * (id_width + 1)
+            + "╯"
+        )
 
-        click.echo(f"Total: {len(engines)} engine(s)\n")
+        click.echo(f"Total: {len(engines)}\n")
 
     except Exception as e:
         click.echo(f"✗ Error: {e}", err=True)

dayhoff_tools/cli/engines_studios/simulators/engine_list_simulator.py

@@ -128,7 +128,7 @@ def format_list_output(engines: list[dict[str, Any]], env: str = "dev") -> None:
         + "╯"
     )
 
-    print(f"Total: {len(engines)} engine(s)\n")
+    print(f"Total: {len(engines)}\n")
 
 
 def generate_scenarios() -> dict[str, dict[str, Any]]:

dayhoff_tools/cli/engines_studios/simulators/studio_list_simulator.py

@@ -104,7 +104,7 @@ def format_list_output(studios: list[dict[str, Any]], engines_map: dict[str, str
     # Table bottom border
     print("╰" + "─" * (user_width + 1) + "┴" + "─" * (status_width + 1) + "┴" + "─" * (attached_width + 1) + "┴" + "─" * (size_width + 1) + "┴" + "─" * (id_width + 1) + "╯")
 
-    print(f"Total: {len(studios)} studio(s)\n")
+    print(f"Total: {len(studios)}\n")
 
 
 def generate_scenarios() -> dict[str, dict[str, Any]]:

dayhoff_tools/cli/engines_studios/studio_commands.py

@@ -349,7 +349,7 @@ def list_studios(env: Optional[str]):
         # Table bottom border
         click.echo("╰" + "─" * (user_width + 1) + "┴" + "─" * (status_width + 1) + "┴" + "─" * (attached_width + 1) + "┴" + "─" * (size_width + 1) + "┴" + "─" * (id_width + 1) + "╯")
 
-        click.echo(f"Total: {len(studios)} studio(s)\n")
+        click.echo(f"Total: {len(studios)}\n")
 
     except Exception as e:
         click.echo(f"✗ Error: {e}", err=True)

{dayhoff_tools-1.13.2.dist-info → dayhoff_tools-1.13.4.dist-info}/METADATA

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

{dayhoff_tools-1.13.2.dist-info → dayhoff_tools-1.13.4.dist-info}/RECORD

@@ -13,18 +13,18 @@ dayhoff_tools/cli/engine1/studio_commands.py,sha256=VwTQujz32-uMcYusDRE73SdzRpgv
 dayhoff_tools/cli/engines_studios/__init__.py,sha256=E6aG0C6qjJnJuClemSKRFlYvLUL49MQZOvfqNQ7SDKs,159
 dayhoff_tools/cli/engines_studios/api_client.py,sha256=9I55_Ns8VHxndGjvSt_c5ZohSqMOeywQlLjyuoDEqCQ,13039
 dayhoff_tools/cli/engines_studios/auth.py,sha256=rwetV5hp4jSvK8FyvKgXCnezLOZx1aW8oiSDc6U83iE,5189
-dayhoff_tools/cli/engines_studios/engine-studio-cli.md,sha256=vtFuQDi41DP_MAzxiQiyXWC396pwHnab4uil8kCrehk,29944
-dayhoff_tools/cli/engines_studios/engine_commands.py,sha256=qFoDJODwwQilflQZXNy-D-tcHvLi36AgLCNqqSQaj1w,34563
+dayhoff_tools/cli/engines_studios/engine-studio-cli.md,sha256=or4k7ZZKPMTkvu67PdcUTE2_cxjnj0HQxxTuJZR1uiA,29924
+dayhoff_tools/cli/engines_studios/engine_commands.py,sha256=ZvUt9Gfl1-mhTu-jWAP3w6C_JVcrwOqfJV_jG0-_YGA,37330
 dayhoff_tools/cli/engines_studios/progress.py,sha256=SMahdG2YmO5bEPSONrfAXVTdS6m_69Ep02t3hc2DdKQ,9264
 dayhoff_tools/cli/engines_studios/simulators/cli-simulators.md,sha256=FZJl6nehdr2Duht2cx3yijcak0yKyOaHTrTzvFTAfZs,4976
 dayhoff_tools/cli/engines_studios/simulators/demo.sh,sha256=8tYABSCxLNXqGs-4r071V9mpKNZ5DTQ34WZ-v3d5s94,5364
-dayhoff_tools/cli/engines_studios/simulators/engine_list_simulator.py,sha256=NJOO0CZFRJuIBS1xpYYIU9FW2-hQs8oq2BxwMifUqtY,9535
+dayhoff_tools/cli/engines_studios/simulators/engine_list_simulator.py,sha256=vUrB6gV9UX74L5uCRMTVcPmk_1ZOuP1jYJf6cMP7dOE,9525
 dayhoff_tools/cli/engines_studios/simulators/engine_status_simulator.py,sha256=KUm3gA2MiRgGrQV7KURhb5zabM18-30z_ugRjiq5iso,13024
 dayhoff_tools/cli/engines_studios/simulators/idle_status_simulator.py,sha256=F_MfEXdPKNVDCKgJV72QyU2oMG8hLt-Bwic4yFadRXE,17570
 dayhoff_tools/cli/engines_studios/simulators/simulator_utils.py,sha256=HA08pIMJWV3OFrWj3Ca8GldvgJZfFoTOloyLK0UWMgA,6729
-dayhoff_tools/cli/engines_studios/simulators/studio_list_simulator.py,sha256=MVQf_CshYqPuldW_VXE8LijSm26yYPWMg6QJLuRTobg,11207
+dayhoff_tools/cli/engines_studios/simulators/studio_list_simulator.py,sha256=CHqxeTApdYneOngfGCAZiNxi9al_yAP1hNhVsQJ57V4,11197
 dayhoff_tools/cli/engines_studios/simulators/studio_status_simulator.py,sha256=6WvpnRawJVaQf_H81zuR1_66igRRVxPxjAt8e69xjp4,5394
-dayhoff_tools/cli/engines_studios/studio_commands.py,sha256=RbXayvlrydEt5ao7OjYNAkyaVgAkukre3Iq1VrolKk8,24401
+dayhoff_tools/cli/engines_studios/studio_commands.py,sha256=0x0x2DZ7LWXQZOq0rKWdguVp2QQCrkSUjK43O6jnLOM,24391
 dayhoff_tools/cli/main.py,sha256=Nz_jtbppmvWKHZydQ0nkt_eejccJE90ces8xCGrerdY,7086
 dayhoff_tools/cli/swarm_commands.py,sha256=5EyKj8yietvT5lfoz8Zx0iQvVaNgc3SJX1z2zQR6o6M,5614
 dayhoff_tools/cli/utility_commands.py,sha256=e2P4dCCtoqMUGNyb0lFBZ6GZpl5Zslm1qqE5qIvsy38,50765
@@ -48,7 +48,7 @@ dayhoff_tools/intake/uniprot.py,sha256=BZYJQF63OtPcBBnQ7_P9gulxzJtqyorgyuDiPeOJq
 dayhoff_tools/logs.py,sha256=DKdeP0k0kliRcilwvX0mUB2eipO5BdWUeHwh-VnsICs,838
 dayhoff_tools/sqlite.py,sha256=jV55ikF8VpTfeQqqlHSbY8OgfyfHj8zgHNpZjBLos_E,18672
 dayhoff_tools/warehouse.py,sha256=UETBtZD3r7WgvURqfGbyHlT7cxoiVq8isjzMuerKw8I,24475
-dayhoff_tools-1.13.2.dist-info/METADATA,sha256=3yH2B0VHCo_8AfohB70VdnxLjHuK_zWklWcW9zN62qk,2980
-dayhoff_tools-1.13.2.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-dayhoff_tools-1.13.2.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
-dayhoff_tools-1.13.2.dist-info/RECORD,,
+dayhoff_tools-1.13.4.dist-info/METADATA,sha256=sqkHAmS0J1THxON0ivHLsbJTVrfi5i8rkScs1WtD9qc,2980
+dayhoff_tools-1.13.4.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+dayhoff_tools-1.13.4.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
+dayhoff_tools-1.13.4.dist-info/RECORD,,