add better default value selection when creating pods (#25)

* add better default value selection when creating pods

* simplify connecting to machine and show private key path

* improve display of secure cloud

Author: JannikSt

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "prime-cli"
-version = "0.2.5"
+version = "0.2.6"
 description = "Prime Intellect CLI"
 readme = "README.md"
 requires-python = ">=3.8"

src/prime_cli/commands/availability.py

@@ -92,10 +92,10 @@ def list(
         table.add_column("Location", style="green")
         table.add_column("Stock", style="yellow")
         table.add_column("Price/Hr", style="magenta")
-        table.add_column("Memory (GB)", style="blue")
         table.add_column("Security", style="white")
         table.add_column("vCPUs", style="blue")
         table.add_column("RAM (GB)", style="blue")
+        table.add_column("Disk (GB)", style="blue")
 
         all_gpus: List[Dict[str, Any]] = []
         for gpu_type, gpus in availability_data.items():
@@ -113,6 +113,14 @@ def list(
                 location = f"{gpu.country or 'N/A'}"
 
                 short_id = generate_short_id(gpu)
+
+                disk_info: str = str(gpu.disk.default_count)
+                if (
+                    gpu.disk.max_count is not None
+                    and gpu.disk.max_count != gpu.disk.default_count
+                ):
+                    disk_info = f"{gpu.disk.default_count}+"
+
                 gpu_data = {
                     "short_id": short_id,
                     "cloud_id": gpu.cloud_id,
@@ -129,6 +137,7 @@ def list(
                     "vcpu": gpu.vcpu.default_count,
                     "memory": gpu.memory.default_count,
                     "is_spot": gpu.is_spot,
+                    "disk": disk_info,
                 }
                 all_gpus.append(gpu_data)
 
@@ -158,8 +167,14 @@ def list(
                     max_vcpu = max(g["vcpu"] for g in group)
                     min_mem = min(g["memory"] for g in group)
                     max_mem = max(g["memory"] for g in group)
-                    vcpu_range = f"{min_vcpu}-{max_vcpu}"
-                    memory_range = f"{min_mem}-{max_mem}"
+                    vcpu_range = (
+                        f"{min_vcpu}-{max_vcpu}"
+                        if min_vcpu != max_vcpu
+                        else str(min_vcpu)
+                    )
+                    memory_range = (
+                        f"{min_mem}-{max_mem}" if min_mem != max_mem else str(min_mem)
+                    )
                     base["vcpu"] = vcpu_range
                     base["memory"] = memory_range
                     filtered_gpus.append(base)
@@ -178,7 +193,7 @@ def list(
                 f"{gpu_entry['gpu_type']} (Spot)"
                 if gpu_entry["is_spot"]
                 else gpu_entry["gpu_type"]
-            )
+            ).replace("_", " ")
             table.add_row(
                 gpu_entry["short_id"],
                 gpu_type_display,
@@ -188,10 +203,12 @@ def list(
                 gpu_entry["location"],
                 gpu_entry["stock_status"],
                 gpu_entry["price"],
-                str(gpu_entry["gpu_memory"]),
-                gpu_entry["security"],
+                "community"
+                if gpu_entry["security"] == "community_cloud"
+                else "datacenter",
                 str(gpu_entry["vcpu"]),
                 str(gpu_entry["memory"]),
+                str(gpu_entry["disk"]),
             )
 
         console.print(table)

src/prime_cli/commands/pods.py

@@ -383,9 +383,12 @@ def create(
 
         if not name:
             while True:
+                gpu_name = selected_gpu.gpu_type.lower().split("_")[0]
+                default_name = f"{gpu_name}-{selected_gpu.gpu_count}"
                 name = typer.prompt(
                     "Pod name (alphanumeric and dashes only, must contain at least "
                     "1 letter)",
+                    default=default_name,
                 )
                 if (
                     name
@@ -468,21 +471,25 @@ def create(
                     raise typer.Exit(1)
 
         if not image and selected_gpu.images:
-            # Show available images
-            console.print("\n[bold]Available Images:[/bold]")
-            for idx, img in enumerate(selected_gpu.images):
-                console.print(f"{idx + 1}. {img}")
-
-            # Prompt for image selection
-            image_idx = typer.prompt(
-                "Select image number", type=int, default=1, show_default=False
-            )
+            if len(selected_gpu.images) == 1:
+                # If only one image available, use it directly
+                image = selected_gpu.images[0]
+            else:
+                # Show available images
+                console.print("\n[bold]Available Images:[/bold]")
+                for idx, img in enumerate(selected_gpu.images):
+                    console.print(f"{idx + 1}. {img}")
+
+                # Prompt for image selection
+                image_idx = typer.prompt(
+                    "Select image number", type=int, default=1, show_default=False
+                )
 
-            if image_idx < 1 or image_idx > len(selected_gpu.images):
-                console.print("[red]Invalid image selection[/red]")
-                raise typer.Exit(1)
+                if image_idx < 1 or image_idx > len(selected_gpu.images):
+                    console.print("[red]Invalid image selection[/red]")
+                    raise typer.Exit(1)
 
-            image = selected_gpu.images[image_idx - 1]
+                image = selected_gpu.images[image_idx - 1]
 
         # Get team ID from config if not provided
         if not team_id:
@@ -613,9 +620,10 @@ def terminate(pod_id: str) -> None:
         raise typer.Exit(1)
 
 
-@app.command()
-def ssh(pod_id: str) -> None:
-    """SSH into a pod using configured SSH key"""
+@app.command(name="connect")
+@app.command(name="ssh")
+def connect(pod_id: str) -> None:
+    """SSH / connect to a pod using configured SSH key"""
     try:
         base_client = APIClient()
         pods_client = PodsClient(base_client)
@@ -637,6 +645,11 @@ def ssh(pod_id: str) -> None:
             console.print(f"[red]SSH key not found at {ssh_key_path}[/red]")
             raise typer.Exit(1)
 
+        console.print(f"[blue]Using SSH key:[/blue] {ssh_key_path}")
+        console.print(
+            "[dim]To change SSH key path, use: prime config set-ssh-key-path[/dim]"
+        )
+
         ssh_conn = status.ssh_connection
         # Handle ssh_conn being either a string or list of strings
         connections: List[str] = []