addressed review comments

Author: Sita04

model_armor/create_template.py

@@ -19,9 +19,13 @@ def create_model_armor_template(project_id: str, location: str, template_id: str
     # [START modelarmor_create_template]
 
     from google.api_core.client_options import ClientOptions
-    from google.cloud import modelarmor_v1
-    from google.cloud.modelarmor_v1 import Template, DetectionConfidenceLevel, FilterConfig, PiAndJailbreakFilterSettings, \
-        MaliciousUriFilterSettings
+    from google.cloud.modelarmor_v1 import (
+        Template,
+        DetectionConfidenceLevel,
+        FilterConfig,
+        PiAndJailbreakFilterSettings,
+        MaliciousUriFilterSettings,
+    )
 
     client = modelarmor_v1.ModelArmorClient(
         transport="rest",
@@ -34,7 +38,6 @@ def create_model_armor_template(project_id: str, location: str, template_id: str
     # template_id = "template_id"
 
     template = Template(
-        name=f"projects/{project_id}/locations/{location}/templates/{template_id}",
         filter_config=FilterConfig(
             pi_and_jailbreak_filter_settings=PiAndJailbreakFilterSettings(
                 filter_enforcement=PiAndJailbreakFilterSettings.PiAndJailbreakFilterEnforcement.ENABLED,

model_armor/delete_template.py

@@ -17,15 +17,18 @@ def delete_model_armor_template(project_id: str, location: str, template_id: str
     # [START modelarmor_delete_template]
 
     from google.api_core.client_options import ClientOptions
-    from google.cloud import modelarmor_v1
+    from google.cloud.modelarmor_v1 import (
+        ModelArmorClient,
+        DeleteTemplateRequest,
+    )
 
     client = modelarmor_v1.ModelArmorClient(
         transport="rest",
         client_options=ClientOptions(api_endpoint=f"modelarmor.{location}.rep.googleapis.com"),
     )
 
     # TODO(Developer): Uncomment these variables and initialize
-    # project_id = "your-google-cloud-project-id"
+    # project_id = "YOUR_PROJECT_ID"
     # location = "us-central1"
     # template_id = "template_id"
 

model_armor/get_template.py

@@ -27,7 +27,7 @@ def get_model_armor_template(project_id: str, location: str, template_id: str) -
     )
 
     # TODO(Developer): Uncomment these variables and initialize
-    # project_id = "your-google-cloud-project-id"
+    # project_id = "YOUR_PROJECT_ID"
     # location = "us-central1"
     # template_id = "template_id"
 

model_armor/list_templates.py

@@ -26,7 +26,7 @@ def list_model_armor_templates(project_id: str, location: str) -> ListTemplatesP
     )
 
     # TODO(Developer): Uncomment these variables and initialize
-    # project_id = "your-google-cloud-project-id"
+    # project_id = "YOUR_PROJECT_ID"
     # location = "us-central1"
 
     # Initialize request argument(s)

model_armor/sanitize_user_prompt.py

@@ -29,16 +29,17 @@ def sanitize_user_prompt(
     )
 
     # TODO(Developer): Uncomment these variables and initialize
-    # project_id = "your-google-cloud-project-id"
+    # project_id = "YOUR_PROJECT_ID"
     # location = "us-central1"
     # template_id = "template_id"
 
     # Define the prompt
     user_prompt = "Can you describe this link? https://testsafebrowsing.appspot.com/s/malware.html"
 
     # Initialize request argument(s)
-    user_prompt_data = modelarmor_v1.DataItem()
-    user_prompt_data.text = user_prompt
+    user_prompt_data = modelarmor_v1.DataItem(
+        text=user_prompt
+    )
 
     request = modelarmor_v1.SanitizeUserPromptRequest(
         name=f"projects/{project_id}/locations/{location}/templates/{template_id}",

model_armor/update_template.py

@@ -29,7 +29,7 @@ def update_model_armor_template(project_id: str, location: str, template_id: str
     )
 
     # TODO(Developer): Uncomment these variables and initialize
-    # project_id = "your-google-cloud-project-id"
+    # project_id = "YOUR_PROJECT_ID"
     # location = "us-central1"
     # template_id = "template_id"