dayhoff-tools 1.13.6__py3-none-any.whl → 1.13.8__py3-none-any.whl

dayhoff_tools/cli/engines_studios/engine_commands.py

@@ -69,11 +69,15 @@ def _update_ssh_config_silent(client: StudioManagerClient, env: str) -> bool:
             if state != "running":
                 continue
 
+            # Map environment to AWS profile
+            profile = f"{env}-devaccess"
+
             config_entries.append(f"\nHost {name}\n")
             config_entries.append(f"    HostName {instance_id}\n")
             config_entries.append(f"    User {user}\n")
+            config_entries.append(f"    ForwardAgent yes\n")
             config_entries.append(
-                f"    ProxyCommand aws ssm start-session --target %h --document-name AWS-StartSSHSession --parameters 'portNumber=%p'\n"
+                f"    ProxyCommand aws ssm start-session --target %h --document-name AWS-StartSSHSession --parameters 'portNumber=%p' --profile {profile}\n"
             )
 
         config_entries.append(managed_end)
@@ -213,46 +217,71 @@ def launch_engine(
 
     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(
+                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(
+                "  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(
+                "     • 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)
+            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(
+                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(
+                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(
+                "  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():
+        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(
+                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(
+                "  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()
@@ -594,19 +623,24 @@ def engine_status(name_or_id: str, env: Optional[str]):
                     remaining = readiness["estimated_time_remaining_seconds"]
                     click.echo(f"Estimated Time Remaining: {remaining}s")
 
-        # Show attached studios (always, even without idle state)
-        attached_studios = status_data.get("attached_studios", [])
-        if attached_studios:
-            studio_names = ", ".join([f"\033[35m{s.get('user', 'unknown')}\033[0m" for s in attached_studios])
-            click.echo(f"\nAttached Studios: {studio_names}")
-        else:
-            click.echo(f"\nAttached Studios: None")
-        
         # Show idle state (only for running engines) - always detailed per user request
+        attached_studios = status_data.get("attached_studios", [])
         if status_data.get("idle_state"):
             click.echo(
                 f"\n{format_idle_state(status_data['idle_state'], detailed=True, attached_studios=attached_studios)}"
             )
+        else:
+            # If no idle state yet, still show attached studios
+            if attached_studios:
+                studio_names = ", ".join(
+                    [
+                        f"\033[35m{s.get('user', 'unknown')}\033[0m"
+                        for s in attached_studios
+                    ]
+                )
+                click.echo(f"\nAttached Studios: {studio_names}")
+            else:
+                click.echo(f"\nAttached Studios: None")
 
     except Exception as e:
         click.echo(f"✗ Error: {e}", err=True)
@@ -853,6 +887,7 @@ def config_ssh(clean: bool, all: bool, admin: bool, env: Optional[str]):
             config_entries.append(f"\nHost {name}\n")
             config_entries.append(f"    HostName {instance_id}\n")
             config_entries.append(f"    User {username}\n")
+            config_entries.append(f"    ForwardAgent yes\n")
             config_entries.append(
                 f"    ProxyCommand aws ssm start-session --target %h --document-name AWS-StartSSHSession --parameters 'portNumber=%p' --profile {aws_profile}\n"
             )

{dayhoff_tools-1.13.6.dist-info → dayhoff_tools-1.13.8.dist-info}/METADATA

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

{dayhoff_tools-1.13.6.dist-info → dayhoff_tools-1.13.8.dist-info}/RECORD

@@ -14,7 +14,7 @@ dayhoff_tools/cli/engines_studios/__init__.py,sha256=E6aG0C6qjJnJuClemSKRFlYvLUL
 dayhoff_tools/cli/engines_studios/api_client.py,sha256=McZyyh5H36gkrK6s6Z7s9hl7yLLSLrHjsBtR4Opg6Ko,13491
 dayhoff_tools/cli/engines_studios/auth.py,sha256=rwetV5hp4jSvK8FyvKgXCnezLOZx1aW8oiSDc6U83iE,5189
 dayhoff_tools/cli/engines_studios/engine-studio-cli.md,sha256=or4k7ZZKPMTkvu67PdcUTE2_cxjnj0HQxxTuJZR1uiA,29924
-dayhoff_tools/cli/engines_studios/engine_commands.py,sha256=sGxm1WG06WjzykIIeWi6cj2CjomBrVoJf-LNKOzAxCk,37669
+dayhoff_tools/cli/engines_studios/engine_commands.py,sha256=8utssuiLfxrff3Pk3PjMSUIVMU83mm9N8-Ny1uQr9es,38371
 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
@@ -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.6.dist-info/METADATA,sha256=zMLk8JgTbcMVO2r1RVgQKrTdjRvmS-r-6rG96L3kHq4,2980
-dayhoff_tools-1.13.6.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-dayhoff_tools-1.13.6.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
-dayhoff_tools-1.13.6.dist-info/RECORD,,
+dayhoff_tools-1.13.8.dist-info/METADATA,sha256=I4Yu04pcP7t7LZv_9Ilado80GtOBzURE8saHY-viyTk,2980
+dayhoff_tools-1.13.8.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+dayhoff_tools-1.13.8.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
+dayhoff_tools-1.13.8.dist-info/RECORD,,