dayhoff-tools 1.5.3__tar.gz → 1.5.5__tar.gz

{dayhoff_tools-1.5.3 → dayhoff_tools-1.5.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: dayhoff-tools
-Version: 1.5.3
+Version: 1.5.5
 Summary: Common tools for all the repos at Dayhoff Labs
 Author: Daniel Martin-Alarcon
 Author-email: dma@dayhofflabs.com
@@ -52,7 +52,9 @@ The base package includes minimal dependencies required for core CLI functionali
 ```bash
 pip install dayhoff-tools
 # or
-poetry add dayhoff-tools
+uv add dayhoff-tools
+# or, for the special case of installing it in the DHT repo itself,
+uv pip install -e .[full]
 ```
 
 ### Optional Dependencies

{dayhoff_tools-1.5.3 → dayhoff_tools-1.5.5}/README.md

@@ -9,7 +9,9 @@ The base package includes minimal dependencies required for core CLI functionali
 ```bash
 pip install dayhoff-tools
 # or
-poetry add dayhoff-tools
+uv add dayhoff-tools
+# or, for the special case of installing it in the DHT repo itself,
+uv pip install -e .[full]
 ```
 
 ### Optional Dependencies

{dayhoff_tools-1.5.3 → dayhoff_tools-1.5.5}/dayhoff_tools/cli/engine_commands.py

@@ -391,8 +391,15 @@ def check_session_manager_plugin():
     return True
 
 
-def update_ssh_config_entry(engine_name: str, instance_id: str, ssh_user: str):
-    """Add or update a single SSH config entry for the given SSH user."""
+def update_ssh_config_entry(engine_name: str, instance_id: str, ssh_user: str, idle_timeout: int = 600):
+    """Add or update a single SSH config entry for the given SSH user.
+
+    Args:
+        engine_name:  Host alias to write into ~/.ssh/config
+        instance_id:  EC2 instance-id (used by the proxy command)
+        ssh_user:     Username to place into the SSH stanza
+        idle_timeout: Idle timeout **in seconds** to pass to the SSM port-forward. 600 = 10 min.
+    """
     config_path = Path.home() / ".ssh" / "config"
     config_path.parent.mkdir(mode=0o700, exist_ok=True)
 
@@ -408,7 +415,7 @@ def update_ssh_config_entry(engine_name: str, instance_id: str, ssh_user: str):
 Host {engine_name} {SSH_MANAGED_COMMENT}
     HostName {instance_id}
     User {ssh_user}
-    ProxyCommand sh -c \"AWS_SSM_IDLE_TIMEOUT=600 aws ssm start-session --target %h --document-name AWS-StartSSHSession --parameters 'portNumber=%p'\"\n"""
+    ProxyCommand sh -c \"AWS_SSM_IDLE_TIMEOUT={idle_timeout} aws ssm start-session --target %h --document-name AWS-StartSSHSession --parameters 'portNumber=%p,idleTimeout={idle_timeout}'\"\n"""
 
     # Check if entry already exists
     host_line = f"Host {engine_name} {SSH_MANAGED_COMMENT}"
@@ -511,7 +518,7 @@ def launch_engine(
             "engine_type": engine_type,
         }
         if boot_disk_size is not None:
-            request_data["boot_disk_size"] = str(boot_disk_size)
+            request_data["boot_disk_size"] = boot_disk_size
         if availability_zone:
             request_data["availability_zone"] = availability_zone
 
@@ -899,6 +906,9 @@ def ssh_engine(
     admin: bool = typer.Option(
         False, "--admin", help="Connect as ec2-user instead of the engine owner user"
     ),
+    idle_timeout: int = typer.Option(
+        600, "--idle-timeout", help="Idle timeout (seconds) for the SSM port-forward (0 = disable)"
+    ),
 ):
     """Connect to an engine via SSH.
 
@@ -931,7 +941,7 @@ def ssh_engine(
     console.print(
         f"Updating SSH config for [cyan]{engine['name']}[/cyan] (user: {ssh_user})..."
     )
-    update_ssh_config_entry(engine["name"], engine["instance_id"], ssh_user)
+    update_ssh_config_entry(engine["name"], engine["instance_id"], ssh_user, idle_timeout)
 
     # Connect
     console.print(f"[green]✓ Connecting to {engine['name']}...[/green]")
@@ -1019,7 +1029,7 @@ def config_ssh(
                     f"Host {engine['name']} {SSH_MANAGED_COMMENT}",
                     f"    HostName {engine['instance_id']}",
                     f"    User {ssh_user}",
-                    f"    ProxyCommand sh -c \"AWS_SSM_IDLE_TIMEOUT=600 aws ssm start-session --target %h --document-name AWS-StartSSHSession --parameters 'portNumber=%p'\"",
+                    f"    ProxyCommand sh -c \"AWS_SSM_IDLE_TIMEOUT=600 aws ssm start-session --target %h --document-name AWS-StartSSHSession --parameters 'portNumber=%p,idleTimeout=600'\"",
                 ]
             )
 

{dayhoff_tools-1.5.3 → dayhoff_tools-1.5.5}/pyproject.toml

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