AIR CLI Integration: Implement the air get command

acceptance/experimental/air/get/output.txt

@@ -0,0 +1,52 @@
+
+=== get (text)
+>>> [CLI] experimental air get run 123
+Job Link: [DATABRICKS_URL]/jobs/runs/123?o=[NUMID]
+
+Run ID:       123
+Status:       SUCCESS
+Submitted:    2023-11-14 22:13 UTC
+Retries:      0
+Duration:     12s
+Experiment:   my-exp
+MLflow Run:   my-run
+User:         user@example.com
+Accelerators: 8x H100
+
+=== get (json)
+>>> [CLI] experimental air get run 123 -o json
+{
+  "v": 1,
+  "ts": "[TIMESTAMP]",
+  "data": {
+    "run_id": "123",
+    "status": "SUCCESS",
+    "started_at": "[TIMESTAMP]",
+    "duration_seconds": 12,
+    "attempt_number": 0,
+    "experiment_name": "my-exp",
+    "dashboard_url": "[DATABRICKS_URL]/jobs/runs/123?o=[NUMID]",
+    "mlflow_url": "[DATABRICKS_URL]/ml/experiments/exp1/runs/run1/artifacts/logs/node_0"
+  }
+}
+
+=== invalid run id
+>>> [CLI] experimental air get run notanumber
+Error: invalid JOB_RUN_ID "notanumber": must be a positive integer
+
+Exit code: 1
+
+=== invalid run id (json)
+>>> [CLI] experimental air get run notanumber -o json
+{
+  "v": 1,
+  "ts": "[TIMESTAMP]",
+  "error": {
+    "code": "INVALID_ARGS",
+    "kind": "PERMANENT",
+    "message": "invalid JOB_RUN_ID \"notanumber\": must be a positive integer",
+    "retryable": false
+  }
+}
+
+Exit code: 1

acceptance/experimental/air/get/script

@@ -0,0 +1,11 @@
+title "get (text)"
+trace $CLI experimental air get run 123
+
+title "get (json)"
+trace $CLI experimental air get run 123 -o json
+
+title "invalid run id"
+errcode trace $CLI experimental air get run notanumber
+
+title "invalid run id (json)"
+errcode trace $CLI experimental air get run notanumber -o json

acceptance/experimental/air/get/test.toml

@@ -0,0 +1,47 @@
+# This command does not deploy a bundle, so no engine matrix is needed.
+[EnvMatrix]
+DATABRICKS_BUNDLE_ENGINE = []
+
+# The SDK occasionally probes host reachability with a HEAD request; stub it so
+# the test is deterministic.
+[[Server]]
+Pattern = "HEAD /"
+Response.Body = ''
+
+# A single GenAI-compute run with an experiment, GPUs, and a creator.
+[[Server]]
+Pattern = "GET /api/2.2/jobs/runs/get"
+Response.Body = '''
+{
+  "run_id": 123,
+  "run_page_url": "https://my-workspace.cloud.databricks.test/jobs/runs/123",
+  "creator_user_name": "user@example.com",
+  "start_time": 1700000000000,
+  "end_time": 1700000012000,
+  "state": {"life_cycle_state": "TERMINATED", "result_state": "SUCCESS"},
+  "tasks": [
+    {
+      "task_key": "train",
+      "attempt_number": 0,
+      "gen_ai_compute_task": {
+        "mlflow_experiment_name": "/Users/user@example.com/my-exp",
+        "compute": {"gpu_type": "GPU_8xH100", "num_gpus": 8}
+      }
+    }
+  ]
+}
+'''
+
+# MLflow identifiers for the deep-link (runs/get-output is not modeled by the typed SDK).
+[[Server]]
+Pattern = "GET /api/2.2/jobs/runs/get-output"
+Response.Body = '''
+{"gen_ai_compute_output": {"run_info": {"mlflow_experiment_id": "exp1", "mlflow_run_id": "run1"}}}
+'''
+
+# The MLflow Run cell shows the run's name, fetched from the MLflow REST API.
+[[Server]]
+Pattern = "GET /api/2.0/mlflow/runs/get"
+Response.Body = '''
+{"run": {"info": {"run_name": "my-run"}}}
+'''

acceptance/experimental/air/help/output.txt

@@ -11,7 +11,7 @@ Usage:
 
 Available Commands:
   cancel         Cancel one or more runs
-  get            Show details for a run
+  get            Show details for a specific resource
   list           List recent runs
   logs           Stream or fetch logs for a run
   register-image Mirror a Docker image into the workspace registry

acceptance/experimental/air/unimplemented/output.txt

@@ -5,12 +5,6 @@ Error: `air run` is not implemented yet
 
 Exit code: 1
 
-=== get
->>> [CLI] experimental air get 123
-Error: `air get` is not implemented yet
-
-Exit code: 1
-
 === list
 >>> [CLI] experimental air list
 Error: `air list` is not implemented yet

acceptance/experimental/air/unimplemented/script

@@ -3,9 +3,6 @@
 title "run"
 errcode trace $CLI experimental air run
 
-title "get"
-errcode trace $CLI experimental air get 123
-
 title "list"
 errcode trace $CLI experimental air list