fall back to shadeform when cloud is excesssupply

Author: drewmalin

v1/providers/shadeform/instancetype.go

@@ -18,6 +18,7 @@ import (
 const (
 	UsdCurrentCode = "USD"
 	AllRegions     = "all"
+	excessSupply   = "excesssupply"
 )
 
 // TODO: We need to apply a filter to specifically limit the integration and api to selected clouds and shade instance types
@@ -176,6 +177,7 @@ func (c *ShadeformClient) convertShadeformInstanceTypeToV1InstanceType(shadeform
 
 	gpuName := shadeformGPUTypeToBrevGPUName(shadeformInstanceType.Configuration.GpuType)
 	gpuManufacturer := v1.GetManufacturer(shadeformInstanceType.Configuration.GpuManufacturer)
+	cloud := shadeformCloud(shadeformInstanceType.Cloud)
 
 	for _, region := range shadeformInstanceType.Availability {
 		instanceTypes = append(instanceTypes, v1.InstanceType{
@@ -205,7 +207,7 @@ func (c *ShadeformClient) convertShadeformInstanceTypeToV1InstanceType(shadeform
 			IsAvailable: region.Available,
 			Location:    region.Region,
 			Provider:    CloudProviderID,
-			Cloud:       string(shadeformInstanceType.Cloud),
+			Cloud:       cloud,
 		})
 	}
 
@@ -230,3 +232,16 @@ func shadeformGPUTypeToBrevGPUName(gpuType string) string {
 	gpuType = strings.Split(gpuType, "_")[0]
 	return gpuType
 }
+
+func shadeformCloud(cloud openapi.Cloud) string {
+	shadeformCloud := string(cloud)
+
+	// Shadeform will return the cloud as "excesssupply" if the instance type is retrieved
+	// from cloud partners and not a direct cloud provider. In this case, we should just return
+	// the Shadeform Cloud Provider ID.
+	if strings.EqualFold(shadeformCloud, excessSupply) {
+		return CloudProviderID
+	}
+
+	return shadeformCloud
+}