Add tests for endpoints json mode

Author: blainekasten

src/together/lib/cli/api/endpoints/create.py

@@ -149,8 +149,8 @@ def create(
             or "the selected configuration" in error_msg
             or "hardware is required" in error_msg
         ):
-            click.secho("Invalid hardware selected.", fg="red", err=True)
-            click.echo("\nAvailable hardware options:")
+            click.secho("Invalid hardware selected.", fg="red", err=True, file=sys.stderr)
+            click.echo("\nAvailable hardware options:", file=sys.stderr)
             ctx.invoke(list_hardware, available=True, model=model, json=False)
             sys.exit(1)
         elif "model" in error_msg and (
@@ -162,14 +162,17 @@ def create(
             click.echo(
                 f"Error: Model '{model}' was not found or is not available for dedicated endpoints.",
                 err=True,
+                file=sys.stderr,
             )
             click.echo(
                 "Please check that the model name is correct and that it supports dedicated endpoint deployment.",
                 err=True,
+                file=sys.stderr,
             )
             click.echo(
                 "You can browse available models at: https://api.together.ai/models",
                 err=True,
+                file=sys.stderr,
             )
             sys.exit(1)
         raise e

tests/cli/test_endpoints.py

@@ -0,0 +1,45 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+import subprocess
+import json
+import os
+
+base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
+
+def run_command(command: str) -> subprocess.CompletedProcess[str]:
+    return subprocess.run(["together", "--base-url", base_url, "endpoints", *command.split(" "), "--json"], capture_output=True, text=True)
+
+def assert_command_returns_json(command: str) -> None:
+    result = run_command(command)
+    assert result.returncode == 0
+    assert result.stdout is not None
+    assert json.loads(result.stdout) is not None
+
+class TestEndpoints:
+    def test_json_mode_is_supported(self) -> None:
+        assert_command_returns_json("hardware")
+        assert_command_returns_json("hardware --model deepseek-ai/DeepSeek-R1")
+        assert_command_returns_json("availability-zones")
+        assert_command_returns_json("list")
+        assert_command_returns_json("list --type dedicated")
+        assert_command_returns_json("list --usage-type on-demand")
+        assert_command_returns_json("list --usage-type reserved")
+        assert_command_returns_json("list --mine")
+        assert_command_returns_json("create --model deepseek-ai/DeepSeek-R1 --hardware 1x_nvidia_a100_80gb_sxm")
+        assert_command_returns_json("delete endpoint-123")
+        assert_command_returns_json("start endpoint-123")
+        assert_command_returns_json("stop endpoint-123")
+        assert_command_returns_json("retrieve endpoint-123")
+        assert_command_returns_json("update endpoint-123 --min-replicas 2 --max-replicas 4 --inactive-timeout 60")
+    
+    def test_create_requires_model(self) -> None:
+        result = run_command("create")
+        assert result.returncode == 2
+        assert "Error: Missing option '--model'." in result.stderr
+
+    def test_create_requires_hardware(self) -> None:
+        result = run_command("create --model deepseek-ai/DeepSeek-R1 --hardware ''")
+        assert result.returncode == 1
+        assert "Invalid hardware selected." in result.stderr