docs(discoveryengine): Add Cancel Operation Sample

Author: holtskinner

discoveryengine/cancel_operation_sample.py

@@ -0,0 +1,36 @@
+# Copyright 2023 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+# [START genappbuilder_cancel_operation]
+from google.cloud import discoveryengine
+from google.longrunning import operations_pb2
+
+# TODO(developer): Uncomment these variables before running the sample.
+# Example: `projects/{project}/locations/{location}/collections/{default_collection}/dataStores/{search_engine_id}/branches/{0}/operations/{operation_id}`
+# operation_name = "YOUR_OPERATION_NAME"
+
+
+def cancel_operation_sample(operation_name: str) -> None:
+    # Create a client
+    client = discoveryengine.DocumentServiceClient()
+
+    # Make CancelOperation request
+    request = operations_pb2.CancelOperationRequest(name=operation_name)
+    client.cancel_operation(request=request)
+
+    return
+
+
+# [END genappbuilder_cancel_operation]

discoveryengine/operations_sample_test.py

@@ -18,6 +18,7 @@
 from discoveryengine import get_operation_sample
 from discoveryengine import list_operations_sample
 from discoveryengine import poll_operation_sample
+from discoveryengine import cancel_operation_sample
 
 from google.api_core.exceptions import NotFound
 
@@ -59,3 +60,11 @@ def test_poll_operation():
     except NotFound as e:
         print(e.message)
         pass
+
+
+def test_cancel_operation():
+    try:
+        cancel_operation_sample.cancel_operation_sample(operation_name=operation_name)
+    except NotFound as e:
+        print(e.message)
+        pass

discoveryengine/requirements.txt

@@ -1 +1 @@
-google-cloud-discoveryengine==0.13.4
+google-cloud-discoveryengine==0.13.8