dayhoff-tools 1.4.18__tar.gz → 1.4.19__tar.gz

{dayhoff_tools-1.4.18 → dayhoff_tools-1.4.19}/PKG-INFO

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

{dayhoff_tools-1.4.18 → dayhoff_tools-1.4.19}/dayhoff_tools/cli/engine_commands.py

@@ -686,9 +686,8 @@ def engine_status(
         status_lines.append("")
         status_lines.append("[bold]Attached Studios:[/bold]")
         for studio in attached_studios:
-            attach_time = studio.get("attach_time", "Unknown")
             status_lines.append(
-                f"  • {studio['user']} ({studio['studio_id']}) - attached {attach_time}"
+                f"  • {studio['user']} ({studio['studio_id']})"
             )
 
     console.print(Panel("\n".join(status_lines), title="Engine Status", border_style="blue"))
@@ -1273,13 +1272,14 @@ def create_ami(
         console.print("Cleaning up instance for AMI creation...")
         ssm = boto3.client("ssm", region_name="us-east-1")
 
-        # Stop SSM, wipe registration, remove sentinel, and clear logs/history
+        # Clean up instance state, stopping SSM agent last to allow proper status reporting
         cleanup_commands = [
-            "sudo systemctl stop amazon-ssm-agent",
-            "sudo rm -rf /var/lib/amazon/ssm/* /etc/amazon/ssm/*",
             "sudo rm -f /opt/dayhoff/first_boot_complete.sentinel",
             "history -c",
             "sudo rm -rf /tmp/* /var/log/messages /var/log/cloud-init.log",
+            "sudo rm -rf /var/lib/amazon/ssm/* /etc/amazon/ssm/*",
+            # Stop SSM agent last with a delay to allow status reporting
+            "sleep 2 && sudo systemctl stop amazon-ssm-agent &",
         ]
 
         cleanup_response = ssm.send_command(
@@ -1302,9 +1302,14 @@ def create_ami(
             if result["Status"] in ["Success", "Failed"]:
                 break
 
-        if result["Status"] != "Success":
+        # Note: InProgress status is expected when SSM agent stops itself
+        if result["Status"] not in ["Success", "InProgress"]:
+            console.print(
+                f"[yellow]⚠️  Warning: Cleanup command status: {result['Status']}[/yellow]"
+            )
+        elif result["Status"] == "InProgress":
             console.print(
-                "[yellow]⚠️  Warning: Cleanup command may have failed[/yellow]"
+                "[dim]ℹ️  Cleanup command still in progress (expected when SSM agent stops itself)[/dim]"
             )
 
         # Get instance details to find volumes to exclude

{dayhoff_tools-1.4.18 → dayhoff_tools-1.4.19}/pyproject.toml

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