fix(nebius): filter ARM64 images from boot disk selection

Author: mmoulikk

v1/providers/nebius/instance.go

@@ -1209,7 +1209,7 @@ func (c *NebiusClient) buildDiskCreateRequest(ctx context.Context, diskName stri
 		publicImagesParent := c.getPublicImagesParent()
 
 		// Skip validation for known-good common families to speed up instance start
-		knownFamilies := []string{"ubuntu22.04-cuda12", "mk8s-worker-node-v-1-32-ubuntu24.04", "mk8s-worker-node-v-1-32-ubuntu24.04-cuda12.8"}
+		knownFamilies := []string{"ubuntu24.04-cuda13.0", "ubuntu24.04-cuda12", "ubuntu22.04-cuda12", "mk8s-worker-node-v-1-32-ubuntu24.04", "mk8s-worker-node-v-1-32-ubuntu24.04-cuda12.8"}
 		isKnownFamily := false
 		for _, known := range knownFamilies {
 			if imageFamily == known {
@@ -1230,21 +1230,24 @@ func (c *NebiusClient) buildDiskCreateRequest(ctx context.Context, diskName stri
 			return baseReq, nil
 		}
 
-		// For unknown families, validate first
-		_, err := c.sdk.Services().Compute().V1().Image().GetLatestByFamily(ctx, &compute.GetImageLatestByFamilyRequest{
+		// For unknown families, validate first and check architecture
+		latestImage, err := c.sdk.Services().Compute().V1().Image().GetLatestByFamily(ctx, &compute.GetImageLatestByFamilyRequest{
 			ParentId:    publicImagesParent,
 			ImageFamily: imageFamily,
 		})
 		if err == nil {
-			// Family works, use it
-			baseReq.Spec.Source = &compute.DiskSpec_SourceImageFamily{
-				SourceImageFamily: &compute.SourceImageFamily{
-					ImageFamily: imageFamily,
-					ParentId:    publicImagesParent,
-				},
+			isARM64 := latestImage.Spec != nil && latestImage.Spec.GetCpuArchitecture() == compute.ImageSpec_ARM64
+			if !isARM64 {
+				baseReq.Spec.Source = &compute.DiskSpec_SourceImageFamily{
+					SourceImageFamily: &compute.SourceImageFamily{
+						ImageFamily: imageFamily,
+						ParentId:    publicImagesParent,
+					},
+				}
+				baseReq.Metadata.Labels["image-family"] = imageFamily
+				return baseReq, nil
 			}
-			baseReq.Metadata.Labels["image-family"] = imageFamily
-			return baseReq, nil
+			// ARM64 family — fall through to getWorkingPublicImageID which filters by architecture
 		}
 	}
 
@@ -1290,6 +1293,10 @@ func (c *NebiusClient) getWorkingPublicImageID(ctx context.Context, requestedIma
 			continue
 		}
 
+		if image.Spec != nil && image.Spec.GetCpuArchitecture() == compute.ImageSpec_ARM64 {
+			continue
+		}
+
 		imageName := strings.ToLower(image.Metadata.Name)
 
 		// Set fallback to first available image
@@ -1583,6 +1590,9 @@ func (c *NebiusClient) parseInstanceType(ctx context.Context, instanceTypeID str
 func (c *NebiusClient) resolveImageFamily(ctx context.Context, imageID string) (string, error) {
 	// Common Nebius image families - if ImageID matches one of these, use it directly
 	commonFamilies := []string{
+		"ubuntu24.04-cuda13.0",
+		"ubuntu24.04-cuda12",
+		"ubuntu24.04-driverless",
 		"ubuntu22.04-cuda12",
 		"mk8s-worker-node-v-1-32-ubuntu24.04",
 		"mk8s-worker-node-v-1-32-ubuntu24.04-cuda12.8",