Create custom organization policy constraints

This page shows you how to use Organization Policy Service custom constraints to restrict specific operations on the following Google Cloud resources:

  • notebooks.googleapis.com/Instance

To learn more about Organization Policy, see Custom organization policies.

About organization policies and constraints

The Google Cloud Organization Policy Service gives you centralized, programmatic control over your organization's resources. As the organization policy administrator, you can define an organization policy, which is a set of restrictions called constraints that apply to Google Cloud resources and descendants of those resources in the Google Cloud resource hierarchy. You can enforce organization policies at the organization, folder, or project level.

Organization Policy provides built-in managed constraints for various Google Cloud services. However, if you want more granular, customizable control over the specific fields that are restricted in your organization policies, you can also create custom constraints and use those custom constraints in an organization policy.

Policy inheritance

By default, organization policies are inherited by the descendants of the resources on which you enforce the policy. For example, if you enforce a policy on a folder, Google Cloud enforces the policy on all projects in the folder. To learn more about this behavior and how to change it, refer to Hierarchy evaluation rules.

Benefits

You can use custom organization policies to allow or deny specific configurations of Gemini Enterprise Agent Platform Workbench instances. For example, you can require that all new instances have Shielded VM features enabled, restrict the machine types that can be used, or require that instances don't have an external IP address. If a request to create or update an Agent Platform Workbench instance fails to satisfy the custom constraint set by your organization policy, the request fails and an error is returned to the caller.

Limitations

Like all organization policy constraints, policy changes don't apply retroactively to existing resources.

  • A new policy doesn't affect the configuration of existing Agent Platform Workbench instances.
  • An existing instance configuration remains valid even if it has non-compliant values, unless the instance is updated.

Before you begin

  1. Sign in to your Google Cloud account. If you're new to Google Cloud, create an account to evaluate how our products perform in real-world scenarios. New customers also get $300 in free credits to run, test, and deploy workloads.
  2. In the Google Cloud console, on the project selector page, select or create a Google Cloud project.

    Roles required to select or create a project

    • Select a project: Selecting a project doesn't require a specific IAM role—you can select any project that you've been granted a role on.
    • Create a project: To create a project, you need the Project Creator role (roles/resourcemanager.projectCreator), which contains the resourcemanager.projects.create permission. Learn how to grant roles.

    Go to project selector

  3. Verify that billing is enabled for your Google Cloud project.

  4. Install the Google Cloud CLI.

  5. If you're using an external identity provider (IdP), you must first sign in to the gcloud CLI with your federated identity.

  6. To initialize the gcloud CLI, run the following command:

    gcloud init
  7. In the Google Cloud console, on the project selector page, select or create a Google Cloud project.

    Roles required to select or create a project

    • Select a project: Selecting a project doesn't require a specific IAM role—you can select any project that you've been granted a role on.
    • Create a project: To create a project, you need the Project Creator role (roles/resourcemanager.projectCreator), which contains the resourcemanager.projects.create permission. Learn how to grant roles.

    Go to project selector

  8. Verify that billing is enabled for your Google Cloud project.

  9. Install the Google Cloud CLI.

  10. If you're using an external identity provider (IdP), you must first sign in to the gcloud CLI with your federated identity.

  11. To initialize the gcloud CLI, run the following command:

    gcloud init
  12. Ensure that you know your organization ID.

Required roles

To get the permissions that you need to manage custom organization policies, ask your administrator to grant you the Organization Policy Administrator (roles/orgpolicy.policyAdmin) IAM role on the organization resource. For more information about granting roles, see Manage access to projects, folders, and organizations.

You might also be able to get the required permissions through custom roles or other predefined roles.

Gemini Enterprise Agent Platform Workbench supported resources

The following table lists the fields of the notebooks.googleapis.com/Instance resource that you can reference in custom constraints.

Resource Field
notebooks.googleapis.com/Instance resource.disableProxyAccess
resource.enableDeletionProtection
resource.enableManagedEuc
resource.enableThirdPartyIdentity
resource.gceSetup.acceleratorConfigs.coreCount
resource.gceSetup.acceleratorConfigs.type
resource.gceSetup.bootDisk.diskEncryption
resource.gceSetup.bootDisk.diskSizeGb
resource.gceSetup.bootDisk.kmsKey
resource.gceSetup.confidentialInstanceConfig.confidentialInstanceType
resource.gceSetup.containerImage.repository
resource.gceSetup.containerImage.tag
resource.gceSetup.dataDisks.diskEncryption
resource.gceSetup.dataDisks.diskSizeGb
resource.gceSetup.dataDisks.kmsKey
resource.gceSetup.disablePublicIp
resource.gceSetup.enableIpForwarding
resource.gceSetup.gpuDriverConfig.customGpuDriverPath
resource.gceSetup.gpuDriverConfig.enableGpuDriver
resource.gceSetup.machineType
resource.gceSetup.minCpuPlatform
resource.gceSetup.networkInterfaces.accessConfigs.externalIp
resource.gceSetup.networkInterfaces.network
resource.gceSetup.networkInterfaces.nicType
resource.gceSetup.networkInterfaces.subnet
resource.gceSetup.reservationAffinity.consumeReservationType
resource.gceSetup.reservationAffinity.key
resource.gceSetup.reservationAffinity.values
resource.gceSetup.serviceAccounts.email
resource.gceSetup.shieldedInstanceConfig.enableIntegrityMonitoring
resource.gceSetup.shieldedInstanceConfig.enableSecureBoot
resource.gceSetup.shieldedInstanceConfig.enableVtpm
resource.gceSetup.systemMetadata

Set up a custom constraint

A custom constraint is defined in a YAML file by the resources, methods, conditions, and actions that are supported by the service on which you are enforcing the organization policy. Conditions for your custom constraints are defined using Common Expression Language (CEL). For more information about how to build conditions in custom constraints using CEL, see the CEL section of Creating and managing custom constraints.

Console

To create a custom constraint, do the following:

  1. In the Google Cloud console, go to the Organization policies page.

    Go to Organization policies

  2. From the project picker, select the project that you want to set the organization policy for.
  3. Click Custom constraint.
  4. In the Display name box, enter a human-readable name for the constraint. This name is used in error messages and can be used for identification and debugging. Don't use personally identifiable information (PII) or sensitive data in display names because this name could be exposed in error messages. This field can contain up to 200 characters.
  5. In the Constraint ID box, enter the ID that you want for your new custom constraint. A custom constraint can only contain letters (including upper and lowercase) or numbers, for example custom.restrictWorkbenchMachineType. This field can contain up to 70 characters, not counting the prefix (custom.), for example, organizations/123456789/customConstraints/custom. Don't include PII or sensitive data in your constraint ID, because it could be exposed in error messages.
  6. In the Description box, enter a human-readable description of the constraint. This description is used as an error message when the policy is violated. Include details about why the policy violation occurred and how to resolve the policy violation. Don't include PII or sensitive data in your description, because it could be exposed in error messages. This field can contain up to 2000 characters.
  7. In the Resource type box, select the name of the Google Cloud REST resource containing the object and field that you want to restrict—for example, container.googleapis.com/NodePool. Most resource types support up to 20 custom constraints. If you attempt to create more custom constraints, the operation fails.
  8. Under Enforcement method, select whether to enforce the constraint on a REST CREATE method or both CREATE and UPDATE methods. If you enforce the constraint with the UPDATE method on a resource that violates the constraint, changes to that resource are blocked by the organization policy unless the change resolves the violation.
  9. To see supported methods for each service, find the service in Services that support custom constraints.

  10. To define a condition, click Edit condition.
    1. In the Add condition panel, create a CEL condition that refers to a supported service resource, for example, resource.management.autoUpgrade == false. This field can contain up to 1000 characters. For details about CEL usage, see Common Expression Language. For more information about the service resources you can use in your custom constraints, see Custom constraint supported services.
    2. Click Save.
  11. Under Action, select whether to allow or deny the evaluated method if the condition is met.
  12. The deny action means that the operation to create or update the resource is blocked if the condition evaluates to true.

    The allow action means that the operation to create or update the resource is permitted only if the condition evaluates to true. Every other case except those explicitly listed in the condition is blocked.

  13. Click Create constraint.
  14. When you have entered a value into each field, the equivalent YAML configuration for this custom constraint appears on the right.

gcloud

  1. To create a custom constraint, create a YAML file using the following format:
  2. name: organizations/ORGANIZATION_ID/customConstraints/CONSTRAINT_NAME
    resourceTypes: RESOURCE_NAME
    methodTypes:
      - CREATE
    - UPDATE
    condition: "CONDITION" actionType: ACTION displayName: DISPLAY_NAME description: DESCRIPTION

    Replace the following:

    • ORGANIZATION_ID: your organization ID, such as 123456789.
    • CONSTRAINT_NAME: the name that you want for your new custom constraint. A custom constraint can only contain letters (including upper and lowercase) or numbers, for example, custom.restrictWorkbenchMachineType. This field can contain up to 70 characters, not counting the prefix (custom.)— for example, organizations/123456789/customConstraints/custom. Don't include PII or sensitive data in your constraint ID, because it could be exposed in error messages.
    • RESOURCE_NAME: the fully qualified name of the Google Cloud resource containing the object and field that you want to restrict. For example, notebooks.googleapis.com/Instance. Most resource types support up to 20 custom constraints. If you attempt to create more custom constraints, the operation fails.
    • methodTypes: the REST methods that the constraint is enforced on. Can be CREATE or both CREATE and UPDATE. If you enforce the constraint with the UPDATE method on a resource that violates the constraint, changes to that resource are blocked by the organization policy unless the change resolves the violation.
    • To see the supported methods for each service, find the service in Services that support custom constraints.

    • CONDITION: a CEL condition that is written against a representation of a supported service resource. This field can contain up to 1000 characters. For example, "resource.gceSetup.machineType.startsWith('n1-standard-')".
    • For more information about the resources available to write conditions against, see Supported resources.

    • ACTION: the action to take if the condition is met. Possible values are ALLOW and DENY.
    • The allow action means that if the condition evaluates to true, the operation to create or update the resource is permitted. This also means that every other case except the one explicitly listed in the condition is blocked.

      The deny action means that if the condition evaluates to true, the operation to create or update the resource is blocked.

    • DISPLAY_NAME: a human-readable name for the constraint. This name is used in error messages and can be used for identification and debugging. Don't use PII or sensitive data in display names because this name could be exposed in error messages. This field can contain up to 200 characters.
    • DESCRIPTION: a human-friendly description of the constraint to display as an error message when the policy is violated. This field can contain up to 2000 characters.
  3. After you have created the YAML file for a new custom constraint, you must set it up to make it available for organization policies in your organization. To set up a custom constraint, use the gcloud org-policies set-custom-constraint command:
  4. gcloud org-policies set-custom-constraint CONSTRAINT_PATH

    Replace CONSTRAINT_PATH with the full path to your custom constraint file. For example, /home/user/customconstraint.yaml.

    After this operation is complete, your custom constraints are available as organization policies in your list of Google Cloud organization policies.

  5. To verify that the custom constraint exists, use the gcloud org-policies list-custom-constraints command:
  6. gcloud org-policies list-custom-constraints --organization=ORGANIZATION_ID

    Replace ORGANIZATION_ID with the ID of your organization resource.

    For more information, see Viewing organization policies.

Enforce a custom organization policy

You can enforce a constraint by creating an organization policy that references it, and then applying that organization policy to a Google Cloud resource.

Console

  1. In the Google Cloud console, go to the Organization policies page.

    Go to Organization policies

  2. From the project picker, select the project that you want to set the organization policy for.
  3. From the list on the Organization policies page, select your constraint to view the Policy details page for that constraint.
  4. To configure the organization policy for this resource, click Manage policy.
  5. On the Edit policy page, select Override parent's policy.
  6. Click Add a rule.
  7. In the Enforcement section, select whether this organization policy is enforced or not.
  8. Optional: To make the organization policy conditional on a tag, click Add condition. Note that if you add a conditional rule to an organization policy, you must add at least one unconditional rule or the policy cannot be saved. For more information, see Scope organization policies with tags.
  9. Click Test changes to simulate the effect of the organization policy. For more information, see Test organization policy changes with Policy Simulator.
  10. To enforce the organization policy in dry-run mode, click Set dry run policy. For more information, see Test organization policies.
  11. After you verify that the organization policy in dry-run mode works as intended, set the live policy by clicking Set policy.

gcloud

  1. To create an organization policy with boolean rules, create a policy YAML file that references the constraint:
  2. name: projects/PROJECT_ID/policies/CONSTRAINT_NAME
    spec:
      rules:
      - enforce: true
    
    dryRunSpec:
      rules:
      - enforce: true

    Replace the following:

    • PROJECT_ID: the project that you want to enforce your constraint on.
    • CONSTRAINT_NAME: the name you defined for your custom constraint. For example, custom.restrictWorkbenchMachineType.
  3. To enforce the organization policy in dry-run mode, run the following command with the dryRunSpec flag:
  4. gcloud org-policies set-policy POLICY_PATH --update-mask=dryRunSpec

    Replace POLICY_PATH with the full path to your organization policy YAML file. The policy requires up to 15 minutes to take effect.

  5. After you verify that the organization policy in dry-run mode works as intended, set the live policy with the org-policies set-policy command and the spec flag:
  6. gcloud org-policies set-policy POLICY_PATH --update-mask=spec

    Replace POLICY_PATH with the full path to your organization policy YAML file. The policy requires up to 15 minutes to take effect.

Test the custom organization policy

The following example creates a custom constraint and policy that require all new Agent Platform Workbench instances, in a specific project, to use a machine type from the n1-standard family.

Before you begin, you must know the following:

  • Your organization ID
  • A project ID

Create the constraint

  1. Save the following file as constraint-restrict-machine-type.yaml:

    name: organizations/ORGANIZATION_ID/customConstraints/custom.restrictWorkbenchMachineType
    resourceTypes:
    - notebooks.googleapis.com/Instance
    methodTypes:
    - CREATE
    - UPDATE
    condition: "resource.gceSetup.machineType.startsWith('n1-standard-')"
    actionType: ALLOW
    displayName: Restrict Workbench instance machine types
    description: Workbench instances must use an n1-standard machine type.
    

    This defines a constraint where every new or updated instance must use a machine type in the n1-standard family, otherwise the operation is denied.

  2. Apply the constraint:

    gcloud org-policies set-custom-constraint ~/constraint-restrict-machine-type.yaml
    
  3. Verify that the constraint exists:

    gcloud org-policies list-custom-constraints --organization=ORGANIZATION_ID
    

    The output is similar to the following:

    CUSTOM_CONSTRAINT                       ACTION_TYPE  METHOD_TYPES   RESOURCE_TYPES                     DISPLAY_NAME
    custom.restrictWorkbenchMachineType     ALLOW        CREATE,UPDATE  notebooks.googleapis.com/Instance  Restrict Workbench instance machine types
    ...
    

Create the policy

  1. Save the following file as policy-restrict-machine-type.yaml:

    name: projects/PROJECT_ID/policies/custom.restrictWorkbenchMachineType
    spec:
      rules:
      - enforce: true
    

    Replace PROJECT_ID with your project ID.

  2. Apply the policy:

    gcloud org-policies set-policy ~/policy-restrict-machine-type.yaml
    
  3. Verify that the policy exists:

    gcloud org-policies list --project=PROJECT_ID
    

    The output is similar to the following:

    CONSTRAINT                             LIST_POLICY  BOOLEAN_POLICY  ETAG
    custom.restrictWorkbenchMachineType    -            SET             COCsm5QGENiXi2E=
    

After you apply the policy, wait for about two minutes for Google Cloud to start enforcing the policy.

Test the policy

Try to create an Agent Platform Workbench instance with a machine type that isn't in the n1-standard family in the project:

gcloud workbench instances create INSTANCE_NAME \
    --project=PROJECT_ID \
    --location=LOCATION \
    --machine-type=e2-standard-4

The operation is disallowed and the output is similar to the following:

ERROR: (gcloud.workbench.instances.create) Could not fetch resource:
– Operation denied by custom org policies: [customConstraints/custom.restrictWorkbenchMachineType]: Workbench instances must use an n1-standard machine type.

Example custom organization policies for common use cases

This table provides syntax examples for some common custom constraints.

Description Constraint syntax
Restrict machine types for Agent Platform Workbench instances
name: organizations/ORGANIZATION_ID/customConstraints/custom.restrictWorkbenchMachineType
resourceTypes:
- notebooks.googleapis.com/Instance
methodTypes:
- CREATE
- UPDATE
condition: "resource.gceSetup.machineType.startsWith('n1-standard-')"
actionType: ALLOW
displayName: Restrict Workbench instance machine types
description: Workbench instances must use an n1-standard machine type.
Require Secure Boot on Agent Platform Workbench instances
name: organizations/ORGANIZATION_ID/customConstraints/custom.workbenchRequireSecureBoot
resourceTypes:
- notebooks.googleapis.com/Instance
methodTypes:
- CREATE
- UPDATE
condition: "resource.gceSetup.shieldedInstanceConfig.enableSecureBoot == true"
actionType: ALLOW
displayName: Require Secure Boot on Workbench instances
description: All Workbench instances must have Secure Boot enabled.
Require vTPM and integrity monitoring on Agent Platform Workbench instances
name: organizations/ORGANIZATION_ID/customConstraints/custom.workbenchRequireShieldedVm
resourceTypes:
- notebooks.googleapis.com/Instance
methodTypes:
- CREATE
- UPDATE
condition: "resource.gceSetup.shieldedInstanceConfig.enableVtpm == true && resource.gceSetup.shieldedInstanceConfig.enableIntegrityMonitoring == true"
actionType: ALLOW
displayName: Require vTPM and integrity monitoring
description: All Workbench instances must have vTPM and integrity monitoring enabled.
Require that Agent Platform Workbench instances don't have an external IP address
name: organizations/ORGANIZATION_ID/customConstraints/custom.workbenchDisablePublicIp
resourceTypes:
- notebooks.googleapis.com/Instance
methodTypes:
- CREATE
- UPDATE
condition: "resource.gceSetup.disablePublicIp == true"
actionType: ALLOW
displayName: Disable external IP on Workbench instances
description: Workbench instances must not have an external IP address.
Require file downloads to be disabled on Agent Platform Workbench instances
name: organizations/ORGANIZATION_ID/customConstraints/custom.workbenchDisableDownloads
resourceTypes:
- notebooks.googleapis.com/Instance
methodTypes:
- CREATE
- UPDATE
condition: "'notebook-disable-downloads' in resource.gceSetup.systemMetadata && resource.gceSetup.systemMetadata['notebook-disable-downloads'] == 'true'"
actionType: ALLOW
displayName: Disable file downloads on Workbench instances
description: Workbench instances must have JupyterLab file downloads disabled.
Require idle shutdown to be enabled on Agent Platform Workbench instances
name: organizations/ORGANIZATION_ID/customConstraints/custom.workbenchRequireIdleShutdown
resourceTypes:
- notebooks.googleapis.com/Instance
methodTypes:
- CREATE
- UPDATE
condition: "'idle-timeout-seconds' in resource.gceSetup.systemMetadata && int(resource.gceSetup.systemMetadata['idle-timeout-seconds']) > 0"
actionType: ALLOW
displayName: Require idle shutdown on Workbench instances
description: Workbench instances must have idle shutdown enabled.

What's next