addressed review comments

Author: Sita04

model_armor/create_template.py

@@ -55,9 +55,9 @@ def create_model_armor_template(project_id: str, location: str, template_id: str
 
     # Make the request
     response = client.create_template(request=request)
-
     # Response
-    return response
-
+    print(response.name)
 
 # [END modelarmor_create_template]
+
+    return response

model_armor/get_template.py

@@ -38,9 +38,9 @@ def get_model_armor_template(project_id: str, location: str, template_id: str) -
 
     # Make the request
     response = client.get_template(request=request)
+    print(response.name)
+
+# [END modelarmor_get_template]
 
     # Handle the response
     return response
-
-
-# [END modelarmor_get_template]

model_armor/list_templates.py

@@ -39,8 +39,7 @@ def list_model_armor_templates(project_id: str, location: str) -> ListTemplatesP
     for template in response:
         print(template.name)
 
+# [END modelarmor_list_templates]
+
     # Handle the response
     return response
-
-
-# [END modelarmor_list_templates]

model_armor/sanitize_user_prompt.py

@@ -50,8 +50,7 @@ def sanitize_user_prompt(
     # Match state is TRUE when the prompt is caught by one of the safety policies in the template.
     print(response.sanitization_result.filter_match_state)
 
+# [END modelarmor_sanitize_user_prompt]
+
     # Handle the response
     return response
-
-
-# [END modelarmor_sanitize_user_prompt]

model_armor/update_template.py

@@ -51,9 +51,10 @@ def update_model_armor_template(project_id: str, location: str, template_id: str
 
     # Make the request
     response = client.update_template(request=request)
+    # Print the updated config
+    print(response.filter_config)
+
+# [END modelarmor_update_template]
 
     # Response
     return response
-
-
-# [END modelarmor_update_template]