feat: Add HOST_POWER_CYCLE to remediations CLI modes and state (#444)

* feat(cli): sync beta cluster SDK updates

Co-authored-by: Blaine Kasten <blainekasten@gmail.com>

* test(cli): cover beta cluster SDK sync

Co-authored-by: Blaine Kasten <blainekasten@gmail.com>

* fix(cli): remove cluster storage project ID updates

Co-authored-by: Blaine Kasten <blainekasten@gmail.com>

---------

Co-authored-by: Cursor Agent <cursoragent@cursor.com>
Co-authored-by: Blaine Kasten <blainekasten@gmail.com>

Author: 3 people

src/together/lib/cli/api/beta/clusters/remediations/approve.py

@@ -17,6 +17,7 @@
         "HOST_AWARE",
         "EVICT_WITHOUT_REPLACEMENT",
         "REBOOT_VM",
+        "HOST_POWER_CYCLE",
     ],
     Parameter(help="Remediation mode to use after approval"),
 ]
@@ -39,6 +40,7 @@ async def approve(
                 "REMEDIATION_MODE_HOST_AWARE",
                 "REMEDIATION_MODE_EVICT_WITHOUT_REPLACEMENT",
                 "REMEDIATION_MODE_REBOOT_VM",
+                "REMEDIATION_MODE_HOST_POWER_CYCLE",
             ],
             f"REMEDIATION_MODE_{mode}",
         )

src/together/lib/cli/api/beta/clusters/remediations/create.py

@@ -16,6 +16,7 @@
         "HOST_AWARE",
         "EVICT_WITHOUT_REPLACEMENT",
         "REBOOT_VM",
+        "HOST_POWER_CYCLE",
     ],
     Parameter(help="The type of remediation to perform"),
 ]
@@ -37,6 +38,7 @@ async def create(
             "REMEDIATION_MODE_HOST_AWARE",
             "REMEDIATION_MODE_EVICT_WITHOUT_REPLACEMENT",
             "REMEDIATION_MODE_REBOOT_VM",
+            "REMEDIATION_MODE_HOST_POWER_CYCLE",
         ],
         f"REMEDIATION_MODE_{mode}",
     )

src/together/lib/cli/api/beta/clusters/remediations/list.py

@@ -20,6 +20,7 @@
                 "HOST_AWARE",
                 "EVICT_WITHOUT_REPLACEMENT",
                 "REBOOT_VM",
+                "HOST_POWER_CYCLE",
             ]
         ]
     ],
@@ -128,6 +129,8 @@ def _colorize(state: str) -> str:
         "FAILED": "red",
         "CANCELLED": "dim",
         "AUTO_RESOLVED": "green",
+        "QUARANTINING": "yellow",
+        "QUARANTINED": "yellow",
     }
     color = state_colors[state] if state in state_colors else "white"
     return f"[{color}]{state}[/{color}]"

tests/cli/test_beta_clusters.py

@@ -915,7 +915,7 @@ def test_remediations_create_json(self, respx_mock: MockRouter, cli_runner: CliR
                 "c1",
                 "i1",
                 "--mode",
-                "VM_ONLY",
+                "HOST_POWER_CYCLE",
                 "--reason",
                 "node unhealthy",
                 "--remediation-id",
@@ -928,7 +928,7 @@ def test_remediations_create_json(self, respx_mock: MockRouter, cli_runner: CliR
         request = cast(Call, route.calls[0]).request
         assert request.url.params["remediation_id"] == "rem-created"
         assert json.loads(request.content.decode()) == {
-            "mode": "REMEDIATION_MODE_VM_ONLY",
+            "mode": "REMEDIATION_MODE_HOST_POWER_CYCLE",
             "reason": "node unhealthy",
         }
         assert result.exit_code == 0
@@ -1001,7 +1001,9 @@ def test_remediations_list_accepts_filters(self, respx_mock: MockRouter, cli_run
                 "--mode",
                 "VM_ONLY",
                 "--mode",
-                "REBOOT_VM",
+                "HOST_POWER_CYCLE",
+                "--state",
+                "QUARANTINED",
                 "--state",
                 "PENDING_APPROVAL",
                 "--trigger",
@@ -1013,8 +1015,8 @@ def test_remediations_list_accepts_filters(self, respx_mock: MockRouter, cli_run
         )
 
         params = cast(Call, route.calls[0]).request.url.params
-        assert params["mode"] == "REMEDIATION_MODE_VM_ONLY,REMEDIATION_MODE_REBOOT_VM"
-        assert params["state"] == "PENDING_APPROVAL"
+        assert params["mode"] == "REMEDIATION_MODE_VM_ONLY,REMEDIATION_MODE_HOST_POWER_CYCLE"
+        assert params["state"] == "QUARANTINED,PENDING_APPROVAL"
         assert params["trigger"] == "REMEDIATION_TRIGGER_AUTOMATED"
         assert params["page_token"] == "next-token"
         assert result.exit_code == 0
@@ -1064,15 +1066,15 @@ def test_remediations_approve_resolves_cluster_and_instance(
                 "--comment",
                 "go",
                 "--mode",
-                "REBOOT_VM",
+                "HOST_POWER_CYCLE",
                 "--json",
             ]
         )
 
         assert json.loads(result.output)["state"] == "PENDING"
         assert json.loads(cast(Call, route.calls[0]).request.content.decode()) == {
             "comment": "go",
-            "mode": "REMEDIATION_MODE_REBOOT_VM",
+            "mode": "REMEDIATION_MODE_HOST_POWER_CYCLE",
         }
         assert result.exit_code == 0