Remove and update default params for serving (#3608)

* Remove default params for serving

* remove changes for deprecated field

* Update default to false

* add PT acceptance test

* integration test needs to be run on Unity Catalog-enabled workspace

* Update to smaller model

* Fix model name

* Add computed and optional

---------

Co-authored-by: Alex Ott <[email protected]>

Author: arpitjasa-db

internal/acceptance/model_serving_test.go

@@ -81,3 +81,36 @@ func TestAccModelServing(t *testing.T) {
 		},
 	)
 }
+
+func TestUcAccModelServingProvisionedThroughput(t *testing.T) {
+	loadWorkspaceEnv(t)
+	if isGcp(t) {
+		skipf(t)("not available on GCP")
+	}
+
+	name := fmt.Sprintf("terraform-test-model-serving-pt-%s",
+		acctest.RandStringFromCharSet(5, acctest.CharSetAlphaNum))
+	unityWorkspaceLevel(t, step{
+		Template: fmt.Sprintf(`
+			resource "databricks_model_serving" "endpoint" {
+				name = "%s"
+				config {
+					served_entities{
+						name = "pt_model"
+						entity_name = "system.ai.mistral_7b_instruct_v0_1"
+						entity_version = "1"
+						min_provisioned_throughput = 0
+						max_provisioned_throughput = 970
+					}
+					traffic_config {
+						routes {
+							served_model_name = "pt_model"
+							traffic_percentage = 100
+						}
+					}
+				}
+			}
+		`, name),
+	},
+	)
+}

serving/resource_model_serving.go

@@ -39,9 +39,12 @@ func ResourceModelServing() common.Resource {
 
 			common.MustSchemaPath(m, "config", "served_entities", "scale_to_zero_enabled").Required = false
 			common.MustSchemaPath(m, "config", "served_entities", "scale_to_zero_enabled").Optional = true
-			common.MustSchemaPath(m, "config", "served_entities", "scale_to_zero_enabled").Default = true
+			common.MustSchemaPath(m, "config", "served_entities", "scale_to_zero_enabled").Default = false
 			common.MustSchemaPath(m, "config", "served_entities", "name").Computed = true
-			common.MustSchemaPath(m, "config", "served_entities", "workload_type").Default = "CPU"
+			common.MustSchemaPath(m, "config", "served_entities", "workload_size").Optional = true
+			common.MustSchemaPath(m, "config", "served_entities", "workload_size").Computed = true
+			common.MustSchemaPath(m, "config", "served_entities", "workload_type").Optional = true
+			common.MustSchemaPath(m, "config", "served_entities", "workload_type").Computed = true
 			common.MustSchemaPath(m, "config", "served_entities", "workload_type").DiffSuppressFunc = func(k, old, new string, d *schema.ResourceData) bool {
 				return old == "" && new == "CPU"
 			}