dayhoff-tools 1.13.8__tar.gz → 1.13.10__tar.gz

{dayhoff_tools-1.13.8 → dayhoff_tools-1.13.10}/PKG-INFO

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

{dayhoff_tools-1.13.8 → dayhoff_tools-1.13.10}/dayhoff_tools/cli/engines_studios/api_client.py

@@ -120,7 +120,7 @@ class StudioManagerClient:
         """
         url = f"{self.api_url}{path}"
         response = requests.request(method, url, **kwargs)
-        
+
         # Parse error body if request failed
         if not response.ok:
             try:
@@ -128,10 +128,10 @@ class StudioManagerClient:
                 error_message = error_body.get("error", response.text)
             except Exception:
                 error_message = response.text or f"HTTP {response.status_code}"
-            
+
             # Raise exception with the actual error message from API
             raise RuntimeError(error_message)
-            
+
         return response.json()
 
     # Engine operations

{dayhoff_tools-1.13.8 → dayhoff_tools-1.13.10}/dayhoff_tools/cli/engines_studios/progress.py

@@ -207,8 +207,12 @@ def format_idle_state(
 
                     if isinstance(value, list):
                         if value:  # Only show non-empty lists
-                            # All lists shown with bullets, no header for containers/connections/sessions
-                            if key in ["containers", "connections", "sessions"]:
+                            # Show workload containers with clear header
+                            if key == "containers":
+                                # Show actual workload container names that are keeping engine active
+                                for item in value[:5]:
+                                    lines.append(f"    • {item}")
+                            elif key in ["connections", "sessions"]:
                                 # Just show the items with bullets, no header
                                 for item in value[:5]:
                                     lines.append(f"    • {item}")

{dayhoff_tools-1.13.8 → dayhoff_tools-1.13.10}/dayhoff_tools/cli/engines_studios/simulators/studio_list_simulator.py

@@ -20,7 +20,9 @@ def colorize(text: str, color_code: str) -> str:
     return f"\033[{color_code}m{text}\033[0m"
 
 
-def format_list_output(studios: list[dict[str, Any]], engines_map: dict[str, str], env: str = "dev") -> None:
+def format_list_output(
+    studios: list[dict[str, Any]], engines_map: dict[str, str], env: str = "dev"
+) -> None:
     """Format and print studio list output matching the actual CLI."""
 
     # Header with blue account name
@@ -31,7 +33,9 @@ def format_list_output(studios: list[dict[str, Any]], engines_map: dict[str, str
         return
 
     # Calculate dynamic width for User column (longest user + 2 for padding)
-    max_user_len = max((len(studio.get("user", "unknown")) for studio in studios), default=4)
+    max_user_len = max(
+        (len(studio.get("user", "unknown")) for studio in studios), default=4
+    )
     user_width = max(max_user_len + 2, len("User") + 2)
 
     # Calculate dynamic width for Attached To column
@@ -41,7 +45,9 @@ def format_list_output(studios: list[dict[str, Any]], engines_map: dict[str, str
             instance_id = studio["attached_to"]
             engine_name = engines_map.get(instance_id, "unknown")
             max_attached_len = max(max_attached_len, len(engine_name))
-    attached_width = max(max_attached_len + 2, len("Attached To") + 2, 3)  # At least 3 for "-"
+    attached_width = max(
+        max_attached_len + 2, len("Attached To") + 2, 3
+    )  # At least 3 for "-"
 
     # Fixed widths for other columns
     status_width = 12
@@ -49,13 +55,39 @@ def format_list_output(studios: list[dict[str, Any]], engines_map: dict[str, str
     id_width = 25
 
     # Table top border
-    print("╭" + "─" * (user_width + 1) + "┬" + "─" * (status_width + 1) + "┬" + "─" * (attached_width + 1) + "┬" + "─" * (size_width + 1) + "┬" + "─" * (id_width + 1) + "╮")
+    print(
+        "╭"
+        + "─" * (user_width + 1)
+        + "┬"
+        + "─" * (status_width + 1)
+        + "┬"
+        + "─" * (attached_width + 1)
+        + "┬"
+        + "─" * (size_width + 1)
+        + "┬"
+        + "─" * (id_width + 1)
+        + "╮"
+    )
 
     # Table header
-    print(f"│ {'User':<{user_width}}│ {'Status':<{status_width}}│ {'Attached To':<{attached_width}}│ {'Size':<{size_width}}│ {'Studio ID':<{id_width}}│")
+    print(
+        f"│ {'User':<{user_width}}│ {'Status':<{status_width}}│ {'Attached To':<{attached_width}}│ {'Size':<{size_width}}│ {'Studio ID':<{id_width}}│"
+    )
 
     # Header separator
-    print("├" + "─" * (user_width + 1) + "┼" + "─" * (status_width + 1) + "┼" + "─" * (attached_width + 1) + "┼" + "─" * (size_width + 1) + "┼" + "─" * (id_width + 1) + "┤")
+    print(
+        "├"
+        + "─" * (user_width + 1)
+        + "┼"
+        + "─" * (status_width + 1)
+        + "┼"
+        + "─" * (attached_width + 1)
+        + "┼"
+        + "─" * (size_width + 1)
+        + "┼"
+        + "─" * (id_width + 1)
+        + "┤"
+    )
 
     # Table rows
     for studio in studios:
@@ -67,7 +99,7 @@ def format_list_output(studios: list[dict[str, Any]], engines_map: dict[str, str
 
         # Truncate if needed
         if len(user) > user_width - 1:
-            user = user[:user_width - 1]
+            user = user[: user_width - 1]
 
         # Color the user (blue)
         user_display = colorize(f"{user:<{user_width}}", "34")
@@ -75,7 +107,9 @@ def format_list_output(studios: list[dict[str, Any]], engines_map: dict[str, str
         # Format status - display "in-use" as "attached" in purple
         if status == "in-use":
             display_status = "attached"
-            status_display = colorize(f"{display_status:<{status_width}}", "35")  # Purple
+            status_display = colorize(
+                f"{display_status:<{status_width}}", "35"
+            )  # Purple
         elif status == "available":
             status_display = colorize(f"{status:<{status_width}}", "32")  # Green
         elif status in ["attaching", "detaching"]:
@@ -99,10 +133,24 @@ def format_list_output(studios: list[dict[str, Any]], engines_map: dict[str, str
         # Color the studio ID (grey)
         studio_id_display = colorize(f"{studio_id:<{id_width}}", "90")
 
-        print(f"│ {user_display}│ {status_display}│ {attached_display}│ {size:<{size_width}}│ {studio_id_display}│")
+        print(
+            f"│ {user_display}│ {status_display}│ {attached_display}│ {size:<{size_width}}│ {studio_id_display}│"
+        )
 
     # Table bottom border
-    print("╰" + "─" * (user_width + 1) + "┴" + "─" * (status_width + 1) + "┴" + "─" * (attached_width + 1) + "┴" + "─" * (size_width + 1) + "┴" + "─" * (id_width + 1) + "╯")
+    print(
+        "╰"
+        + "─" * (user_width + 1)
+        + "┴"
+        + "─" * (status_width + 1)
+        + "┴"
+        + "─" * (attached_width + 1)
+        + "┴"
+        + "─" * (size_width + 1)
+        + "┴"
+        + "─" * (id_width + 1)
+        + "╯"
+    )
 
     print(f"Total: {len(studios)}\n")
 
@@ -309,7 +357,9 @@ def main():
             print("=" * 80 + "\n")
 
             env = args.env if args.env else scenario_data["env"]
-            format_list_output(scenario_data["studios"], scenario_data["engines_map"], env)
+            format_list_output(
+                scenario_data["studios"], scenario_data["engines_map"], env
+            )
             print()  # Extra newline between scenarios
     else:
         # Show specific scenario
@@ -322,4 +372,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-

{dayhoff_tools-1.13.8 → dayhoff_tools-1.13.10}/dayhoff_tools/cli/engines_studios/studio_commands.py

@@ -222,9 +222,20 @@ def studio_status(user: Optional[str], env: Optional[str]):
         click.echo(f"User: {studio['user']}")
         # Status in blue
         click.echo(f"Status: \033[34m{studio['status']}\033[0m")
-        # Attached to in blue (if present)
+        # Attached to in blue (if present) - resolve instance ID to engine name
         if studio.get("attached_to"):
-            click.echo(f"Attached to: \033[34m{studio['attached_to']}\033[0m")
+            instance_id = studio["attached_to"]
+            # Try to resolve instance ID to engine name by searching engines list
+            engine_name = instance_id  # Default to instance ID if not found
+            try:
+                engines_result = client.list_engines()
+                for engine in engines_result.get("engines", []):
+                    if engine.get("instance_id") == instance_id:
+                        engine_name = engine.get("name", instance_id)
+                        break
+            except Exception:
+                pass  # Fall back to instance ID
+            click.echo(f"Attached to: \033[34m{engine_name}\033[0m")
         click.echo(f"Account: {env}")
         click.echo(f"Size: {studio['size_gb']}GB")
         if studio.get("created_at"):
@@ -272,7 +283,9 @@ def list_studios(env: Optional[str]):
             engines_map[engine["instance_id"]] = engine["name"]
 
         # Calculate dynamic width for User column (longest user + 2 for padding)
-        max_user_len = max((len(studio.get("user", "unknown")) for studio in studios), default=4)
+        max_user_len = max(
+            (len(studio.get("user", "unknown")) for studio in studios), default=4
+        )
         user_width = max(max_user_len + 2, len("User") + 2)
 
         # Calculate dynamic width for Attached To column
@@ -282,7 +295,9 @@ def list_studios(env: Optional[str]):
                 instance_id = studio["attached_to"]
                 engine_name = engines_map.get(instance_id, "unknown")
                 max_attached_len = max(max_attached_len, len(engine_name))
-        attached_width = max(max_attached_len + 2, len("Attached To") + 2, 3)  # At least 3 for "-"
+        attached_width = max(
+            max_attached_len + 2, len("Attached To") + 2, 3
+        )  # At least 3 for "-"
 
         # Fixed widths for other columns - reordered to [User, Status, Attached To, Size, Studio ID]
         status_width = 12
@@ -290,7 +305,19 @@ def list_studios(env: Optional[str]):
         id_width = 25
 
         # Table top border
-        click.echo("╭" + "─" * (user_width + 1) + "┬" + "─" * (status_width + 1) + "┬" + "─" * (attached_width + 1) + "┬" + "─" * (size_width + 1) + "┬" + "─" * (id_width + 1) + "╮")
+        click.echo(
+            "╭"
+            + "─" * (user_width + 1)
+            + "┬"
+            + "─" * (status_width + 1)
+            + "┬"
+            + "─" * (attached_width + 1)
+            + "┬"
+            + "─" * (size_width + 1)
+            + "┬"
+            + "─" * (id_width + 1)
+            + "╮"
+        )
 
         # Table header - reordered to [User, Status, Attached To, Size, Studio ID]
         click.echo(
@@ -298,7 +325,19 @@ def list_studios(env: Optional[str]):
         )
 
         # Header separator
-        click.echo("├" + "─" * (user_width + 1) + "┼" + "─" * (status_width + 1) + "┼" + "─" * (attached_width + 1) + "┼" + "─" * (size_width + 1) + "┼" + "─" * (id_width + 1) + "┤")
+        click.echo(
+            "├"
+            + "─" * (user_width + 1)
+            + "┼"
+            + "─" * (status_width + 1)
+            + "┼"
+            + "─" * (attached_width + 1)
+            + "┼"
+            + "─" * (size_width + 1)
+            + "┼"
+            + "─" * (id_width + 1)
+            + "┤"
+        )
 
         # Table rows
         for studio in studios:
@@ -310,7 +349,7 @@ def list_studios(env: Optional[str]):
 
             # Truncate if needed
             if len(user) > user_width - 1:
-                user = user[:user_width - 1]
+                user = user[: user_width - 1]
 
             # Color the user (blue)
             user_display = f"\033[34m{user:<{user_width}}\033[0m"
@@ -318,7 +357,9 @@ def list_studios(env: Optional[str]):
             # Format status - display "in-use" as "attached" in purple
             if status == "in-use":
                 display_status = "attached"
-                status_display = f"\033[35m{display_status:<{status_width}}\033[0m"  # Purple
+                status_display = (
+                    f"\033[35m{display_status:<{status_width}}\033[0m"  # Purple
+                )
             elif status == "available":
                 status_display = f"\033[32m{status:<{status_width}}\033[0m"  # Green
             elif status in ["attaching", "detaching"]:
@@ -326,9 +367,13 @@ def list_studios(env: Optional[str]):
             elif status == "attached":
                 status_display = f"\033[35m{status:<{status_width}}\033[0m"  # Purple
             elif status == "error":
-                status_display = f"\033[31m{status:<{status_width}}\033[0m"  # Red for error
+                status_display = (
+                    f"\033[31m{status:<{status_width}}\033[0m"  # Red for error
+                )
             else:
-                status_display = f"{status:<{status_width}}"  # No color for other states
+                status_display = (
+                    f"{status:<{status_width}}"  # No color for other states
+                )
 
             # Format Attached To column
             if attached_to:
@@ -347,7 +392,19 @@ 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(
+            "╰"
+            + "─" * (user_width + 1)
+            + "┴"
+            + "─" * (status_width + 1)
+            + "┴"
+            + "─" * (attached_width + 1)
+            + "┴"
+            + "─" * (size_width + 1)
+            + "┴"
+            + "─" * (id_width + 1)
+            + "╯"
+        )
 
         click.echo(f"Total: {len(studios)}\n")
 

{dayhoff_tools-1.13.8 → dayhoff_tools-1.13.10}/pyproject.toml

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