Skip to content
Merged
Show file tree
Hide file tree
Changes from 5 commits
Commits
Show all changes
23 commits
Select commit Hold shift + click to select a range
874955b
Added CRUD code snippets with codeowners file
rudrakhsha-crest Mar 28, 2025
8270ed7
Solved linting errors
rudrakhsha-crest Mar 28, 2025
3824b65
Added header comment
rudrakhsha-crest Mar 28, 2025
0b9faee
Removed hardcoded value from test
rudrakhsha-crest Mar 28, 2025
16111a5
Added floor settings snippets
rudrakhsha-crest Mar 28, 2025
d15d006
Removed crud specific files
rudrakhsha-crest Mar 28, 2025
313bd64
Updated test to disable floor settings in the after code
rudrakhsha-crest Apr 18, 2025
441fe5d
Merge branch 'main' into model-armor-floor-settings
rudrakhsha-crest Apr 18, 2025
8bd4854
Updated variable names in tests for org id and folder id
rudrakhsha-crest Apr 21, 2025
f145160
Merge branch 'model-armor-floor-settings' of github.com:rudrakhsha-cr…
rudrakhsha-crest Apr 21, 2025
e73984e
Updated model armor code snippets to use module exports
rudrakhsha-crest May 6, 2025
5f9ff8a
add-endofline-and-fix-copyright-year
harshnasitcrest May 6, 2025
b14a42a
Merge branch 'main' into model-armor-floor-settings
harshnasitcrest May 7, 2025
da8baec
Merge branch 'main' into model-armor-floor-settings
harshnasitcrest May 19, 2025
65f9c1e
resolve-merge-conflicts
harshnasitcrest May 19, 2025
f91919c
fix-package-json-formatting
harshnasitcrest May 19, 2025
49f96b8
change-metadata-in-create-template-snippet
harshnasitcrest May 22, 2025
7ee95d2
Merge branch 'main' into model-armor-floor-settings
harshnasitcrest May 23, 2025
13dd5d1
Merge branch 'main' into model-armor-floor-settings
glasnt May 30, 2025
544e43c
Merge branch 'main' into model-armor-floor-settings
harshnasitcrest Jun 6, 2025
1dd05f1
add testing envvars
glasnt Jun 10, 2025
832d423
Removed floor setting tests
harshnasitcrest Jun 11, 2025
d0e1df5
Merge branch 'main' into model-armor-floor-settings
harshnasitcrest Jun 11, 2025
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions .github/blunderbuss.yml
Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,10 @@ assign_issues_by:
- "api: parametermanager"
to:
- GoogleCloudPlatform/cloud-parameters-team
- labels:
- "api: modelarmor"
to:
- GoogleCloudPlatform/cloud-modelarmor-team

assign_prs_by:
- labels:
Expand Down Expand Up @@ -77,3 +81,7 @@ assign_prs_by:
- "api: parametermanager"
to:
- GoogleCloudPlatform/cloud-parameters-team
- labels:
- "api: modelarmor"
to:
- GoogleCloudPlatform/cloud-modelarmor-team
1 change: 1 addition & 0 deletions CODEOWNERS
Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,7 @@ document-warehouse @GoogleCloudPlatform/nodejs-samples-reviewers @GoogleCloudPla
ai-platform @GoogleCloudPlatform/nodejs-samples-reviewers @GoogleCloudPlatform/text-embedding @GoogleCloudPlatform/cloud-samples-reviewers
asset @GoogleCloudPlatform/cloud-asset-analysis-team @GoogleCloudPlatform/nodejs-samples-reviewers @GoogleCloudPlatform/cloud-samples-reviewers
dlp @GoogleCloudPlatform/googleapis-dlp @GoogleCloudPlatform/nodejs-samples-reviewers @GoogleCloudPlatform/cloud-samples-reviewers
model-armor @GoogleCloudPlatform/nodejs-samples-reviewers @GoogleCloudPlatform/cloud-samples-reviewers @GoogleCloudPlatform/cloud-modelarmor-team
security-center @GoogleCloudPlatform/gcp-security-command-center @GoogleCloudPlatform/nodejs-samples-reviewers @GoogleCloudPlatform/cloud-samples-reviewers
retail @GoogleCloudPlatform/cloud-retail-team @GoogleCloudPlatform/nodejs-samples-reviewers @GoogleCloudPlatform/cloud-samples-reviewers
media @GoogleCloudPlatform/cloud-media-team @GoogleCloudPlatform/nodejs-samples-reviewers @GoogleCloudPlatform/cloud-samples-reviewers
Expand Down
26 changes: 26 additions & 0 deletions model-armor/package.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
{
"name": "nodejs-model-armor-samples",
"private": true,
"license": "Apache-2.0",
"files": [
"*.js"
],
"author": "Google LLC",
"repository": "googleapis/nodejs-model-armor",
"engines": {
"node": ">=16.0.0"
},
"scripts": {
"test": "c8 mocha -p -j 2 --recursive test/ --timeout=60000"
},
"dependencies": {
"@google-cloud/modelarmor": "^0.1.0"
},
"devDependencies": {
"c8": "^10.0.0",
"chai": "^4.5.0",
"mocha": "^10.0.0",
"uuid": "^10.0.0"
}
}

92 changes: 92 additions & 0 deletions model-armor/snippets/createTemplate.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,92 @@
// Copyright 2025 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
//
// https://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.

'use strict';

/**
* Creates a Model Armor template with Responsible AI (RAI) filters.
*
* This function creates a template that can be used for sanitizing user prompts and model responses.
*
* @param {string} projectId - Google Cloud project ID where the template will be created.
* @param {string} locationId - Google Cloud location (region) for the template, e.g., 'us-central1'.
* @param {string} templateId - Unique identifier for the new template.
*/
async function main(projectId, locationId, templateId) {
// [START modelarmor_create_template]
/**
* TODO(developer): Uncomment these variables before running the sample.
*/
// const projectId = 'your-project-id';
// const locationId = 'us-central1';
// const templateId = 'your-template-id';

const parent = `projects/${projectId}/locations/${locationId}`;

// Imports the Model Armor library
const modelarmor = require('@google-cloud/modelarmor');
const {ModelArmorClient} = modelarmor.v1;
const {protos} = modelarmor;

// Instantiates a client
const client = new ModelArmorClient({
apiEndpoint: `modelarmor.${locationId}.rep.googleapis.com`,
});

async function createTemplate() {
/** Build the Model Armor template with your preferred filters.
For more details on filters, please refer to the following doc:
https://cloud.google.com/security-command-center/docs/key-concepts-model-armor#ma-filters
*/
const templateConfig = {
filterConfig: {
raiSettings: {
raiFilters: [
{
filterType:
protos.google.cloud.modelarmor.v1.RaiFilterType.HATE_SPEECH,
confidenceLevel:
protos.google.cloud.modelarmor.v1.DetectionConfidenceLevel.HIGH,
},
{
filterType:
protos.google.cloud.modelarmor.v1.RaiFilterType
.SEXUALLY_EXPLICIT,
confidenceLevel:
protos.google.cloud.modelarmor.v1.DetectionConfidenceLevel
.MEDIUM_AND_ABOVE,
},
],
},
},
};

// Construct request
const request = {
parent,
templateId,
template: templateConfig,
};

// Create the template
const [response] = await client.createTemplate(request);
console.log(`Created template: ${response.name}`);
}

createTemplate();
// [END modelarmor_create_template]
}

const args = process.argv.slice(2);
main(...args).catch(console.error);
129 changes: 129 additions & 0 deletions model-armor/snippets/createTemplateWithAdvancedSdp.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,129 @@
// Copyright 2025 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
//
// https://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.

'use strict';

/**
* Creates a new model armor template with advanced SDP settings enabled.
*
* @param {string} projectId - Google Cloud project ID where the template will be created.
* @param {string} locationId - Google Cloud location where the template will be created.
* @param {string} templateId - ID for the template to create.
* @param {string} inspectTemplate - Optional. Sensitive Data Protection inspect template resource name.
If only inspect template is provided (de-identify template
not provided), then Sensitive Data Protection InspectContent
action is performed during Sanitization. All Sensitive Data
Protection findings identified during inspection will be
returned as SdpFinding in SdpInsepctionResult e.g.
`organizations/{organization}/inspectTemplates/{inspect_template}`,
`projects/{project}/inspectTemplates/{inspect_template}`
`organizations/{organization}/locations/{location}/inspectTemplates/{inspect_template}`
`projects/{project}/locations/{location}/inspectTemplates/{inspect_template}`
* @param {string} deidentifyTemplate - Optional. Optional Sensitive Data Protection Deidentify template resource name.
If provided then DeidentifyContent action is performed
during Sanitization using this template and inspect
template. The De-identified data will be returned in
SdpDeidentifyResult. Note that all info-types present in the
deidentify template must be present in inspect template.
e.g.
`organizations/{organization}/deidentifyTemplates/{deidentify_template}`,
`projects/{project}/deidentifyTemplates/{deidentify_template}`
`organizations/{organization}/locations/{location}/deidentifyTemplates/{deidentify_template}`
`projects/{project}/locations/{location}/deidentifyTemplates/{deidentify_template}`
*/
async function main(
projectId,
locationId,
templateId,
inspectTemplate,
deidentifyTemplate
) {
// [START modelarmor_create_template_with_advanced_sdp]
/**
* TODO(developer): Uncomment these variables before running the sample.
*/
// const projectId = 'your-project-id';
// const locationId = 'us-central1';
// const templateId = 'template-id';
// const inspectTemplate = `projects/${projectId}/locations/${locationId}/inspectTemplates/inspect-template-id`;
// const deidentifyTemplate = `projects/${projectId}/locations/${locationId}/deidentifyTemplates/deidentify-template-id`;

const parent = `projects/${projectId}/locations/${locationId}`;

// Imports the Model Armor library
const modelarmor = require('@google-cloud/modelarmor');
const {ModelArmorClient} = modelarmor.v1;
const {protos} = modelarmor;

const RaiFilterType = protos.google.cloud.modelarmor.v1.RaiFilterType;
const DetectionConfidenceLevel =
protos.google.cloud.modelarmor.v1.DetectionConfidenceLevel;

// Instantiates a client
const client = new ModelArmorClient({
apiEndpoint: `modelarmor.${locationId}.rep.googleapis.com`,
});

async function createTemplateWithAdvancedSdp() {
// Configuration for the template with advanced SDP settings
const templateConfig = {
filterConfig: {
raiSettings: {
raiFilters: [
{
filterType: RaiFilterType.DANGEROUS,
confidenceLevel: DetectionConfidenceLevel.HIGH,
},
{
filterType: RaiFilterType.HARASSMENT,
confidenceLevel: DetectionConfidenceLevel.MEDIUM_AND_ABOVE,
},
{
filterType: RaiFilterType.HATE_SPEECH,
confidenceLevel: DetectionConfidenceLevel.HIGH,
},
{
filterType: RaiFilterType.SEXUALLY_EXPLICIT,
confidenceLevel: DetectionConfidenceLevel.HIGH,
},
],
},
sdpSettings: {
advancedConfig: {
inspectTemplate: inspectTemplate,
deidentifyTemplate: deidentifyTemplate,
},
},
},
};

// Construct request
const request = {
parent,
templateId,
template: templateConfig,
};

// Create the template
const [response] = await client.createTemplate(request);
console.log(`Created template: ${response.name}`);
}

createTemplateWithAdvancedSdp();
// [END modelarmor_create_template_with_advanced_sdp]
}

// Check if this script is being run directly
const args = process.argv.slice(2);
main(...args).catch(console.error);
99 changes: 99 additions & 0 deletions model-armor/snippets/createTemplateWithBasicSdp.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,99 @@
// Copyright 2025 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
//
// https://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.

'use strict';

/**
* Creates a new model armor template with basic SDP settings enabled.
*
* @param {string} projectId - Google Cloud project ID where the template will be created.
* @param {string} locationId - Google Cloud location where the template will be created.
* @param {string} templateId - ID for the template to create.
*/
async function main(projectId, locationId, templateId) {
// [START modelarmor_create_template_with_basic_sdp]
/**
* TODO(developer): Uncomment these variables before running the sample.
*/
// const projectId = 'your-project-id';
// const locationId = 'us-central1';
// const templateId = 'template-id';

const parent = `projects/${projectId}/locations/${locationId}`;

// Imports the Model Armor library
const modelarmor = require('@google-cloud/modelarmor');
const {ModelArmorClient} = modelarmor.v1;
const {protos} = modelarmor;

const RaiFilterType = protos.google.cloud.modelarmor.v1.RaiFilterType;
const DetectionConfidenceLevel =
protos.google.cloud.modelarmor.v1.DetectionConfidenceLevel;
const SdpBasicConfigEnforcement =
protos.google.cloud.modelarmor.v1.SdpBasicConfig.SdpBasicConfigEnforcement;

// Instantiates a client
const client = new ModelArmorClient({
apiEndpoint: `modelarmor.${locationId}.rep.googleapis.com`,
});

async function createTemplateWithBasicSdp() {
// Configuration for the template with basic SDP settings
const templateConfig = {
filterConfig: {
raiSettings: {
raiFilters: [
{
filterType: RaiFilterType.DANGEROUS,
confidenceLevel: DetectionConfidenceLevel.HIGH,
},
{
filterType: RaiFilterType.HARASSMENT,
confidenceLevel: DetectionConfidenceLevel.MEDIUM_AND_ABOVE,
},
{
filterType: RaiFilterType.HATE_SPEECH,
confidenceLevel: DetectionConfidenceLevel.HIGH,
},
{
filterType: RaiFilterType.SEXUALLY_EXPLICIT,
confidenceLevel: DetectionConfidenceLevel.HIGH,
},
],
},
sdpSettings: {
basicConfig: {
filterEnforcement: SdpBasicConfigEnforcement.ENABLED,
},
},
},
};

// Construct request
const request = {
parent,
templateId,
template: templateConfig,
};

const [response] = await client.createTemplate(request);
console.log(`Created template: ${response.name}`);
}

return createTemplateWithBasicSdp();
// [END modelarmor_create_template_with_basic_sdp]
}

const args = process.argv.slice(2);
main(...args).catch(console.error);
Loading