chore(deploy): install agent-sandbox warm-pool extensions in dev and e2e clusters

Apply extensions.yaml alongside manifest.yaml when bootstrapping the local
k3d dev cluster and the e2e kube harness, reusing the pinned
AGENT_SANDBOX_VERSION already used for core. This installs the
SandboxTemplate / SandboxWarmPool / SandboxClaim CRDs and reconfigures the
existing agent-sandbox-controller, so clusters are ready for the warm-pooled
sandbox path (RFC 0005).

extensions.yaml rolls the controller deployment, so the e2e harness waits for
the rollout after both applies and for the new extension CRDs to be
Established. No gateway behavior changes yet.

Signed-off-by: Roshni Malani <rmalani@nvidia.com>

Author: rmalani-nv

e2e/with-kube-gateway.sh

@@ -533,10 +533,18 @@ fi
 # The Kubernetes compute driver creates and watches Sandbox CRs reconciled
 # by the upstream agent-sandbox-controller. Without the CRD + controller,
 # every gateway K8s call 404s and CreateSandbox never produces a Pod.
-echo "Installing agent-sandbox CRDs and controller (${AGENT_SANDBOX_VERSION})..."
+# The warm-pool extensions (SandboxTemplate / SandboxWarmPool / SandboxClaim) are
+# applied alongside core so the e2e cluster matches the dev cluster and is ready
+# for warm-pooled sandbox coverage. extensions.yaml reconfigures the
+# agent-sandbox-controller deployment, so wait for the rollout after both applies.
+echo "Installing agent-sandbox CRDs, controller, and warm-pool extensions (${AGENT_SANDBOX_VERSION})..."
 _agent_sandbox_base="https://github.com/kubernetes-sigs/agent-sandbox/releases/download/${AGENT_SANDBOX_VERSION}"
 kctl apply -f "${_agent_sandbox_base}/manifest.yaml"
+kctl apply -f "${_agent_sandbox_base}/extensions.yaml"
 kctl wait --for=condition=Established crd/sandboxes.agents.x-k8s.io --timeout=120s
+kctl wait --for=condition=Established crd/sandboxtemplates.extensions.agents.x-k8s.io --timeout=120s
+kctl wait --for=condition=Established crd/sandboxwarmpools.extensions.agents.x-k8s.io --timeout=120s
+kctl wait --for=condition=Established crd/sandboxclaims.extensions.agents.x-k8s.io --timeout=120s
 kctl -n agent-sandbox-system rollout status deployment/agent-sandbox-controller --timeout=300s
 
 helm_extra_args=()

tasks/scripts/helm-k3s-local.sh

@@ -143,6 +143,11 @@ apply_base_manifests() {
   local base="https://github.com/kubernetes-sigs/agent-sandbox/releases/download/${AGENT_SANDBOX_VERSION}"
   echo "Applying agent-sandbox manifest (${AGENT_SANDBOX_VERSION})..."
   kubectl --kubeconfig="${KUBECONFIG_TARGET}" apply -f "${base}/manifest.yaml"
+  # Warm-pool extensions (SandboxTemplate / SandboxWarmPool / SandboxClaim) so the
+  # cluster is ready for warm-pooled sandboxes. extensions.yaml reconfigures the
+  # existing agent-sandbox-controller deployment rather than adding a new one.
+  echo "Applying agent-sandbox warm-pool extensions (${AGENT_SANDBOX_VERSION})..."
+  kubectl --kubeconfig="${KUBECONFIG_TARGET}" apply -f "${base}/extensions.yaml"
 }
 
 configure_ghcr_credentials() {