fix container env build not found

cli/cmd/deploy/create.go

@@ -284,12 +284,20 @@ func (c *CreateAgent) CreateFromDocker(
 			env = map[string]string{}
 		}
 
-		buildEnv, err := GetNestedMap(mergedValues, "container", "env", "build")
+		envConfig, err := GetNestedMap(mergedValues, "container", "env")
 
 		if err == nil {
-			for key, val := range buildEnv {
-				if valStr, ok := val.(string); ok {
-					env[key] = valStr
+			_, exists := envConfig["build"]
+
+			if exists {
+				buildEnv, err := GetNestedMap(mergedValues, "container", "env", "build")
+
+				if err == nil {
+					for key, val := range buildEnv {
+						if valStr, ok := val.(string); ok {
+							env[key] = valStr
+						}
+					}
 				}
 			}
 		}

cli/cmd/deploy/deploy.go

@@ -166,12 +166,20 @@ func (d *DeployAgent) GetBuildEnv(opts *GetBuildEnvOpts) (map[string]string, err
 		return nil, err
 	}
 
-	buildEnv, err := GetNestedMap(conf, "container", "env", "build")
+	envConfig, err := GetNestedMap(conf, "container", "env")
 
 	if err == nil {
-		for key, val := range buildEnv {
-			if valStr, ok := val.(string); ok {
-				env[key] = valStr
+		_, exists := envConfig["build"]
+
+		if exists {
+			buildEnv, err := GetNestedMap(conf, "container", "env", "build")
+
+			if err == nil {
+				for key, val := range buildEnv {
+					if valStr, ok := val.(string); ok {
+						env[key] = valStr
+					}
+				}
 			}
 		}
 	}

cli/cmd/preview/build_image_driver.go

@@ -254,12 +254,20 @@ func (d *BuildDriver) Apply(resource *models.Resource) (*models.Resource, error)
 			env = map[string]string{}
 		}
 
-		buildEnv, err := deploy.GetNestedMap(mergedValues, "container", "env", "build")
+		envConfig, err := deploy.GetNestedMap(mergedValues, "container", "env")
 
 		if err == nil {
-			for key, val := range buildEnv {
-				if valStr, ok := val.(string); ok {
-					env[key] = valStr
+			_, exists := envConfig["build"]
+
+			if exists {
+				buildEnv, err := deploy.GetNestedMap(mergedValues, "container", "env", "build")
+
+				if err == nil {
+					for key, val := range buildEnv {
+						if valStr, ok := val.(string); ok {
+							env[key] = valStr
+						}
+					}
 				}
 			}
 		}