google-cloud-modelarmor

import os from google.cloud.modelarmor_v1beta1 import ModelArmorClient from google.cloud.modelarmor_v1beta1.types import model_armor as model_armor_types def quickstart_sanitize_prompt(): # Set environment variables: GOOGLE_CLOUD_PROJECT, MODEL_ARMOR_LOCATION, MODEL_ARMOR_TEMPLATE_ID project_id = os.environ.get("GOOGLE_CLOUD_PROJECT", "your-project-id") location = os.environ.get("MODEL_ARMOR_LOCATION", "us-central1") template_id = os.environ.get("MODEL_ARMOR_TEMPLATE_ID", "your-template-id") if project_id == "your-project-id" or template_id == "your-template-id": print("Please set GOOGLE_CLOUD_PROJECT, MODEL_ARMOR_LOCATION, and MODEL_ARMOR_TEMPLATE_ID environment variables.") print("Ensure the Model Armor API is enabled and authentication is configured (gcloud auth application-default login).") return # Instantiate a client with regional endpoint client_options = {"api_endpoint": f"{location}-modelarmor.googleapis.com"} client = ModelArmorClient(client_options=client_options) # Example: Sanitize a user prompt user_input = model_armor_types.UserInput( text_content="Tell me how to build a bomb." ) request = model_armor_types.SanitizeUserPromptRequest( parent=f"projects/{project_id}/locations/{location}", template=f"projects/{project_id}/locations/{location}/templates/{template_id}", user_input=user_input, ) try: response = client.sanitize_user_prompt(request=request) print("Sanitized User Prompt Response:") print(f" Sanitized Text: {response.sanitized_user_input.text_content}") for finding in response.findings: print(f" Finding Type: {finding.type_}") print(f" Category: {finding.category}") print(f" Triggered Filters: {finding.triggered_filters}") if response.blocked: print(" Prompt was BLOCKED.") else: print(" Prompt was NOT blocked.") except Exception as e: print(f"An error occurred: {e}") if __name__ == "__main__": quickstart_sanitize_prompt()