use capacity ID

Author: drewmalin

v1/providers/sfcomputev2/brev_constants.go

@@ -14,10 +14,10 @@ const (
 // Brev environment config for SFCompute V2.
 // TODO: source these from environment variables rather than hardcoding them here.
 const (
-	// BrevProductionCapacityID is the SFCompute V2 capacity ID for Brev production instances.
-	BrevProductionCapacityID = "brev-production-capacity"
+	// BrevDefaultCapacityID is the SFCompute V2 capacity ID for Brev production instances.
+	BrevDefaultCapacityID = "brev-default-capacity"
 
-	// BrevProductionImageID is the SFCompute image for Brev production instances
+	// BrevDefaultImageID is the default SFCompute image for Brev instances
 	// (ubuntu-24.04.4-cuda-12.8, vm_images.vm_image_id).
-	BrevProductionImageID = "vmi_4GwEvmclFURy7ztFQjOdr"
+	BrevDefaultImageID = "vmi_4GwEvmclFURy7ztFQjOdr"
 )

v1/providers/sfcomputev2/instance.go

@@ -29,8 +29,8 @@ func (c *SFCClientV2) CreateInstance(ctx context.Context, attrs v1.CreateInstanc
 
 	cloudInit := sshKeyCloudInit(attrs.PublicKey)
 	resp, err := c.client.Instances.Create(ctx, components.CreateInstanceRequest{
-		Capacity:          BrevProductionCapacityID,
-		Image:             BrevProductionImageID,
+		Capacity:          BrevDefaultCapacityID,
+		Image:             BrevDefaultImageID,
 		CloudInitUserData: &cloudInit,
 		Tags:              optionalnullable.From(&tags),
 		Name:              optionalnullable.From(&attrs.Name),
@@ -95,7 +95,7 @@ func (c *SFCClientV2) ListInstances(ctx context.Context, args v1.ListInstancesAr
 		v1.LogField("location", c.location),
 	)
 
-	capacityID := BrevProductionCapacityID
+	capacityID := BrevDefaultCapacityID
 	resp, err := c.client.Instances.List(ctx, operations.ListInstancesRequest{
 		Capacity: &capacityID,
 	})

v1/providers/sfcomputev2/instancetype.go

@@ -150,7 +150,7 @@ func (c *SFCClientV2) availableSlots(ctx context.Context) (int, error) {
 // currentCapacityAllocation returns the NodeAllocation from the most recent schedule entry
 // in BrevProductionCapacityID that is currently in effect (EffectiveAt <= now).
 func (c *SFCClientV2) currentCapacityAllocation(ctx context.Context) (int, error) {
-	resp, err := c.client.Capacities.Fetch(ctx, BrevProductionCapacityID, nil, nil)
+	resp, err := c.client.Capacities.Fetch(ctx, BrevDefaultCapacityID, nil, nil)
 	if err != nil {
 		return 0, errors.WrapAndTrace(err)
 	}
@@ -173,7 +173,7 @@ func (c *SFCClientV2) currentCapacityAllocation(ctx context.Context) (int, error
 // activeInstanceCount returns the number of non-terminated instances in BrevProductionCapacityID.
 // All non-terminated instances occupy a slot in the capacity, including failed ones.
 func (c *SFCClientV2) activeInstanceCount(ctx context.Context) (int, error) {
-	capacityID := BrevProductionCapacityID
+	capacityID := BrevDefaultCapacityID
 	resp, err := c.client.Instances.List(ctx, operations.ListInstancesRequest{
 		Capacity: &capacityID,
 	})