feat: add gpu values for app install

pkg/apiserver/api/errors.go

@@ -5,4 +5,5 @@ import "errors"
 var (
 	// ErrResourceNotFound indicates that a resource is not found.
 	ErrResourceNotFound = errors.New("resource not found")
+	ErrGPUNodeNotFound  = errors.New("no available gpu node found")
 )

pkg/apiserver/handler_webhook.go

@@ -7,6 +7,7 @@ import (
 	"fmt"
 	"strings"
 
+	"bytetrade.io/web3os/app-service/pkg/apiserver/api"
 	"bytetrade.io/web3os/app-service/pkg/appinstaller"
 	"bytetrade.io/web3os/app-service/pkg/constants"
 	"bytetrade.io/web3os/app-service/pkg/provider"
@@ -281,13 +282,20 @@ func (h *Handler) gpuLimitMutate(ctx context.Context, req *admissionv1.Admission
 	}
 
 	gpuRequired := appcfg.Requirement.GPU
-	if gpuRequired == nil || gpuRequired.IsZero() {
+	if gpuRequired == nil {
 		return resp
 	}
 	GPUType, err := h.findNvidiaGpuFromNodes(ctx)
-	if err != nil {
+	if err != nil && !errors.Is(err, api.ErrGPUNodeNotFound) {
 		return h.sidecarWebhook.AdmissionError(req.UID, err)
 	}
+	// no gpu found and gpu required is zero, no need to inject env, just return.
+	if gpuRequired.IsZero() && GPUType == "" {
+		return resp
+	}
+	if !gpuRequired.IsZero() && GPUType == "" {
+		return h.sidecarWebhook.AdmissionError(req.UID, api.ErrGPUNodeNotFound)
+	}
 
 	patchBytes, err := webhook.CreatePatchForDeployment(tpl, req.Namespace, gpuRequired, GPUType)
 	if err != nil {
@@ -325,7 +333,7 @@ func (h *Handler) findNvidiaGpuFromNodes(ctx context.Context) (string, error) {
 		return gtype, nil
 	}
 
-	return "", errors.New("no available gpu node found")
+	return "", api.ErrGPUNodeNotFound
 }
 
 func (h *Handler) providerRegistryValidate(req *restful.Request, resp *restful.Response) {

pkg/appinstaller/helm.go

@@ -451,6 +451,11 @@ func (h *HelmOps) setValues() (values map[string]interface{}, err error) {
 		klog.Errorf("Failed to get gpuType err=%v", err)
 		return values, err
 	}
+	values["GPU"] = map[string]interface{}{
+		"Type": gpuType,
+		"Cuda": os.Getenv("CUDA_VERSION"),
+	}
+
 	values["gpu"] = gpuType
 
 	if h.app.OIDC.Enabled {

pkg/utils/helm.go

@@ -149,6 +149,7 @@ func GetResourceListFromChart(chartPath string) (resources kube.ResourceList, er
 		"subjects": map[string]interface{}{},
 		"refs":     map[string]interface{}{},
 	}
+	values["GPU"] = map[string]interface{}{}
 
 	ret, err := instAction.RunWithContext(context.Background(), chartRequested, values)
 	if err != nil {