Update TemplateMetadata for log_template_operation flag

Author: mihirvala08

model_armor/snippets/create_template_with_metadata.py

@@ -76,7 +76,9 @@ def create_model_armor_template_with_metadata(
         # For more details on template metadata, please refer to the following doc:
         # https://cloud.google.com/security-command-center/docs/reference/model-armor/rest/v1/projects.locations.templates#templatemetadata
         template_metadata=modelarmor_v1.Template.TemplateMetadata(
-            ignore_partial_invocation_failures=True, log_sanitize_operations=True
+            ignore_partial_invocation_failures=True,
+            log_sanitize_operations=True,
+            log_template_operations=True,
         ),
     )
 

model_armor/snippets/snippets_test.py

@@ -561,6 +561,7 @@ def test_create_model_armor_template_with_metadata(
     )
 
     assert created_template.template_metadata.ignore_partial_invocation_failures
+    assert created_template.template_metadata.log_template_operations
     assert created_template.template_metadata.log_sanitize_operations
 
 
@@ -624,6 +625,7 @@ def test_update_model_armor_template_metadata(
     )
 
     assert updated_template.template_metadata.ignore_partial_invocation_failures
+    assert updated_template.template_metadata.log_template_operations
     assert updated_template.template_metadata.log_sanitize_operations
 
 

model_armor/snippets/update_template_metadata.py

@@ -95,7 +95,9 @@ def update_model_armor_template_metadata(
         # For more details on template metadata, please refer to the following doc:
         # https://cloud.google.com/security-command-center/docs/reference/model-armor/rest/v1/projects.locations.templates#templatemetadata
         template_metadata=modelarmor_v1.Template.TemplateMetadata(
-            ignore_partial_invocation_failures=True, log_sanitize_operations=True
+            ignore_partial_invocation_failures=True,
+            log_sanitize_operations=True,
+            log_template_operations=True,
         ),
     )