add kg config to discovery search engine (GoogleCloudPlatform#16262)

Author: kefang2

mmv1/products/discoveryengine/SearchEngine.yaml

@@ -173,3 +173,47 @@ properties:
 
       If this field is set and processed successfully, the Engine will be
       protected by the KMS key, as indicated in the cmek_config field.
+  - name: 'knowledgeGraphConfig'
+    type: NestedObject
+    description: |
+      Configurations for the Knowledge Graph.
+    properties:
+      - name: 'enableCloudKnowledgeGraph'
+        type: Boolean
+        default_from_api: true
+        description: |
+          Whether to enable the Cloud Knowledge Graph for the engine.
+      - name: 'cloudKnowledgeGraphTypes'
+        type: Array
+        description: |
+          Specify entity types to support.
+        item_type:
+          type: String
+          description: |
+            Specify entity types to support.
+      - name: 'enablePrivateKnowledgeGraph'
+        type: Boolean
+        default_from_api: true
+        description: |
+          Whether to enable the Private Knowledge Graph for the engine.
+      - name: 'featureConfig'
+        type: NestedObject
+        description: |
+          Feature config for the Knowledge Graph.
+        properties:
+          - name: 'disablePrivateKgQueryUnderstanding'
+            type: Boolean
+            description: |
+              Whether to disable the private KG query understanding for the engine.
+          - name: 'disablePrivateKgEnrichment'
+            type: Boolean
+            description: |
+              Whether to disable the private KG enrichment for the engine.
+          - name: 'disablePrivateKgAutoComplete'
+            type: Boolean
+            description: |
+              Whether to disable the private KG auto complete for the engine.
+          - name: 'disablePrivateKgQueryUiChips'
+            type: Boolean
+            description: |
+              Whether to disable the private KG for query UI chips.

mmv1/templates/terraform/examples/discoveryengine_searchengine_agentspace_basic.tf.tmpl

@@ -17,4 +17,6 @@ resource "google_discovery_engine_search_engine" "agentspace_basic" {
   app_type                    = "APP_TYPE_INTRANET"
   search_engine_config {
   }
+  knowledge_graph_config {
+  }
 }

mmv1/third_party/terraform/services/discoveryengine/resource_discovery_engine_search_engine_test.go

@@ -12,7 +12,6 @@ func TestAccDiscoveryEngineSearchEngine_discoveryengineSearchengineBasicExample_
 
 	context := map[string]interface{}{
 		"random_suffix": acctest.RandString(t, 10),
-		"kms_key_name":  acctest.BootstrapKMSKeyWithPurposeInLocationAndName(t, "ENCRYPT_DECRYPT", "us", "tftest-shared-key-6").CryptoKey.Name,
 	}
 
 	acctest.VcrTest(t, resource.TestCase{
@@ -78,7 +77,10 @@ resource "google_discovery_engine_search_engine" "basic" {
     search_tier = "SEARCH_TIER_ENTERPRISE"
     search_add_ons = ["SEARCH_ADD_ON_LLM"]
   }
-  kms_key_name = "%{kms_key_name}"
+  knowledge_graph_config {
+    enable_cloud_knowledge_graph = true
+    enable_private_knowledge_graph = true
+  }
 }
 `, context)
 }
@@ -120,7 +122,17 @@ resource "google_discovery_engine_search_engine" "basic" {
   features = {
     feedback = "FEATURE_STATE_OFF"
   }
-  kms_key_name = "%{kms_key_name}"
+  knowledge_graph_config {
+    enable_cloud_knowledge_graph = false
+    cloud_knowledge_graph_types = ["foobar"]
+    enable_private_knowledge_graph = false
+    feature_config {
+      disable_private_kg_query_understanding = true
+      disable_private_kg_enrichment = true
+      disable_private_kg_auto_complete = true
+      disable_private_kg_query_ui_chips = true
+    }
+  }
 }
 `, context)
 }