google-cloud-workstations 0.3.0__tar.gz → 0.4.0__tar.gz

{google-cloud-workstations-0.3.0/google_cloud_workstations.egg-info → google-cloud-workstations-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: google-cloud-workstations
-Version: 0.3.0
+Version: 0.4.0
 Summary: Google Cloud Workstations API client library
 Home-page: https://github.com/googleapis/google-cloud-python
 Author: Google LLC

{google-cloud-workstations-0.3.0 → google-cloud-workstations-0.4.0}/google/cloud/workstations/gapic_version.py

@@ -14,4 +14,4 @@
 # limitations under the License.
 #
 
-__version__ = "0.3.0"  # {x-release-please-version}
+__version__ = "0.4.0"  # {x-release-please-version}

{google-cloud-workstations-0.3.0/google/cloud/workstations_v1beta → google-cloud-workstations-0.4.0/google/cloud/workstations_v1}/gapic_version.py

@@ -14,4 +14,4 @@
 # limitations under the License.
 #
 
-__version__ = "0.3.0"  # {x-release-please-version}
+__version__ = "0.4.0"  # {x-release-please-version}

{google-cloud-workstations-0.3.0/google/cloud/workstations_v1 → google-cloud-workstations-0.4.0/google/cloud/workstations_v1beta}/gapic_version.py

@@ -14,4 +14,4 @@
 # limitations under the License.
 #
 
-__version__ = "0.3.0"  # {x-release-please-version}
+__version__ = "0.4.0"  # {x-release-please-version}

{google-cloud-workstations-0.3.0 → google-cloud-workstations-0.4.0}/google/cloud/workstations_v1beta/services/workstations/async_client.py

@@ -1281,7 +1281,9 @@ class WorkstationsAsyncClient:
                 on the ``request`` instance; if ``request`` is provided, this
                 should not be set.
             workstation_config_id (:class:`str`):
-                Required. ID to use for the config.
+                Required. ID to use for the
+                workstation configuration.
+
                 This corresponds to the ``workstation_config_id`` field
                 on the ``request`` instance; if ``request`` is provided, this
                 should not be set.
@@ -1408,7 +1410,8 @@ class WorkstationsAsyncClient:
                 should not be set.
             update_mask (:class:`google.protobuf.field_mask_pb2.FieldMask`):
                 Required. Mask specifying which
-                fields in the config should be updated.
+                fields in the workstation configuration
+                should be updated.
 
                 This corresponds to the ``update_mask`` field
                 on the ``request`` instance; if ``request`` is provided, this
@@ -1530,8 +1533,8 @@ class WorkstationsAsyncClient:
                 The request object. Message for deleting a workstation
                 configuration.
             name (:class:`str`):
-                Required. Name of the config to
-                delete.
+                Required. Name of the workstation
+                configuration to delete.
 
                 This corresponds to the ``name`` field
                 on the ``request`` instance; if ``request`` is provided, this
@@ -1721,7 +1724,8 @@ class WorkstationsAsyncClient:
         timeout: Union[float, object] = gapic_v1.method.DEFAULT,
         metadata: Sequence[Tuple[str, str]] = (),
     ) -> pagers.ListWorkstationsAsyncPager:
-        r"""Returns all Workstations using the specified config.
+        r"""Returns all Workstations using the specified
+        workstation configuration.
 
         .. code-block:: python
 
@@ -1844,8 +1848,8 @@ class WorkstationsAsyncClient:
         timeout: Union[float, object] = gapic_v1.method.DEFAULT,
         metadata: Sequence[Tuple[str, str]] = (),
     ) -> pagers.ListUsableWorkstationsAsyncPager:
-        r"""Returns all Workstations using the specified config
-        on which the caller has the
+        r"""Returns all workstations using the specified
+        workstation configuration on which the caller has the
         "workstations.workstations.use" permission.
 
         .. code-block:: python
@@ -2146,7 +2150,8 @@ class WorkstationsAsyncClient:
                 should not be set.
             update_mask (:class:`google.protobuf.field_mask_pb2.FieldMask`):
                 Required. Mask specifying which
-                fields in the config should be updated.
+                fields in the workstation configuration
+                should be updated.
 
                 This corresponds to the ``update_mask`` field
                 on the ``request`` instance; if ``request`` is provided, this

{google-cloud-workstations-0.3.0 → google-cloud-workstations-0.4.0}/google/cloud/workstations_v1beta/services/workstations/client.py

@@ -1519,7 +1519,9 @@ class WorkstationsClient(metaclass=WorkstationsClientMeta):
                 on the ``request`` instance; if ``request`` is provided, this
                 should not be set.
             workstation_config_id (str):
-                Required. ID to use for the config.
+                Required. ID to use for the
+                workstation configuration.
+
                 This corresponds to the ``workstation_config_id`` field
                 on the ``request`` instance; if ``request`` is provided, this
                 should not be set.
@@ -1648,7 +1650,8 @@ class WorkstationsClient(metaclass=WorkstationsClientMeta):
                 should not be set.
             update_mask (google.protobuf.field_mask_pb2.FieldMask):
                 Required. Mask specifying which
-                fields in the config should be updated.
+                fields in the workstation configuration
+                should be updated.
 
                 This corresponds to the ``update_mask`` field
                 on the ``request`` instance; if ``request`` is provided, this
@@ -1772,8 +1775,8 @@ class WorkstationsClient(metaclass=WorkstationsClientMeta):
                 The request object. Message for deleting a workstation
                 configuration.
             name (str):
-                Required. Name of the config to
-                delete.
+                Required. Name of the workstation
+                configuration to delete.
 
                 This corresponds to the ``name`` field
                 on the ``request`` instance; if ``request`` is provided, this
@@ -1956,7 +1959,8 @@ class WorkstationsClient(metaclass=WorkstationsClientMeta):
         timeout: Union[float, object] = gapic_v1.method.DEFAULT,
         metadata: Sequence[Tuple[str, str]] = (),
     ) -> pagers.ListWorkstationsPager:
-        r"""Returns all Workstations using the specified config.
+        r"""Returns all Workstations using the specified
+        workstation configuration.
 
         .. code-block:: python
 
@@ -2070,8 +2074,8 @@ class WorkstationsClient(metaclass=WorkstationsClientMeta):
         timeout: Union[float, object] = gapic_v1.method.DEFAULT,
         metadata: Sequence[Tuple[str, str]] = (),
     ) -> pagers.ListUsableWorkstationsPager:
-        r"""Returns all Workstations using the specified config
-        on which the caller has the
+        r"""Returns all workstations using the specified
+        workstation configuration on which the caller has the
         "workstations.workstations.use" permission.
 
         .. code-block:: python
@@ -2363,7 +2367,8 @@ class WorkstationsClient(metaclass=WorkstationsClientMeta):
                 should not be set.
             update_mask (google.protobuf.field_mask_pb2.FieldMask):
                 Required. Mask specifying which
-                fields in the config should be updated.
+                fields in the workstation configuration
+                should be updated.
 
                 This corresponds to the ``update_mask`` field
                 on the ``request`` instance; if ``request`` is provided, this

{google-cloud-workstations-0.3.0 → google-cloud-workstations-0.4.0}/google/cloud/workstations_v1beta/services/workstations/transports/grpc.py

@@ -599,7 +599,8 @@ class WorkstationsGrpcTransport(WorkstationsTransport):
     ]:
         r"""Return a callable for the list workstations method over gRPC.
 
-        Returns all Workstations using the specified config.
+        Returns all Workstations using the specified
+        workstation configuration.
 
         Returns:
             Callable[[~.ListWorkstationsRequest],
@@ -628,8 +629,8 @@ class WorkstationsGrpcTransport(WorkstationsTransport):
     ]:
         r"""Return a callable for the list usable workstations method over gRPC.
 
-        Returns all Workstations using the specified config
-        on which the caller has the
+        Returns all workstations using the specified
+        workstation configuration on which the caller has the
         "workstations.workstations.use" permission.
 
         Returns:

{google-cloud-workstations-0.3.0 → google-cloud-workstations-0.4.0}/google/cloud/workstations_v1beta/services/workstations/transports/grpc_asyncio.py

@@ -615,7 +615,8 @@ class WorkstationsGrpcAsyncIOTransport(WorkstationsTransport):
     ]:
         r"""Return a callable for the list workstations method over gRPC.
 
-        Returns all Workstations using the specified config.
+        Returns all Workstations using the specified
+        workstation configuration.
 
         Returns:
             Callable[[~.ListWorkstationsRequest],
@@ -644,8 +645,8 @@ class WorkstationsGrpcAsyncIOTransport(WorkstationsTransport):
     ]:
         r"""Return a callable for the list usable workstations method over gRPC.
 
-        Returns all Workstations using the specified config
-        on which the caller has the
+        Returns all workstations using the specified
+        workstation configuration on which the caller has the
         "workstations.workstations.use" permission.
 
         Returns:

{google-cloud-workstations-0.3.0 → google-cloud-workstations-0.4.0}/google/cloud/workstations_v1beta/types/workstations.py

@@ -93,8 +93,8 @@ class WorkstationCluster(proto.Message):
             soft-deleted.
         etag (str):
             Checksum computed by the server. May be sent
-            on update and delete requests to ensure that the
-            client has an up-to-date value before
+            on update and delete requests to make sure that
+            the client has an up-to-date value before
             proceeding.
         network (str):
             Immutable. Name of the Compute Engine network
@@ -105,6 +105,13 @@ class WorkstationCluster(proto.Message):
             subnetwork in which instances associated with
             this cluster will be created. Must be part of
             the subnetwork specified for this cluster.
+        control_plane_ip (str):
+            Output only. The private IP address of the
+            control plane for this cluster. Workstation VMs
+            need access to this IP address to work with the
+            service, so make sure that your firewall rules
+            allow egress from the workstation VMs to this
+            address.
         private_cluster_config (google.cloud.workstations_v1beta.types.WorkstationCluster.PrivateClusterConfig):
             Configuration for private cluster.
         degraded (bool):
@@ -217,6 +224,10 @@ class WorkstationCluster(proto.Message):
         proto.STRING,
         number=11,
     )
+    control_plane_ip: str = proto.Field(
+        proto.STRING,
+        number=16,
+    )
     private_cluster_config: PrivateClusterConfig = proto.Field(
         proto.MESSAGE,
         number=12,
@@ -267,8 +278,8 @@ class WorkstationConfig(proto.Message):
             soft-deleted.
         etag (str):
             Checksum computed by the server. May be sent
-            on update and delete requests to ensure that the
-            client has an up-to-date value before
+            on update and delete requests to make sure that
+            the client has an up-to-date value before
             proceeding.
         idle_timeout (google.protobuf.duration_pb2.Duration):
             How long to wait before automatically
@@ -292,9 +303,9 @@ class WorkstationConfig(proto.Message):
             workstation using this configuration when that
             workstation is started.
         encryption_key (google.cloud.workstations_v1beta.types.WorkstationConfig.CustomerEncryptionKey):
-            Encrypts resources of this workstation
-            configuration using a customer-managed
-            encryption key.
+            Immutable. Encrypts resources of this
+            workstation configuration using a
+            customer-managed encryption key.
             If specified, the boot disk of the Compute
             Engine instance and the persistent disk are
             encrypted using this encryption key. If this
@@ -311,6 +322,15 @@ class WorkstationConfig(proto.Message):
             If the encryption key is revoked, the
             workstation session will automatically be
             stopped within 7 hours.
+
+            Immutable after the workstation configuration is
+            created.
+        readiness_checks (MutableSequence[google.cloud.workstations_v1beta.types.WorkstationConfig.ReadinessCheck]):
+            Readiness checks to perform when starting a
+            workstation using this workstation
+            configuration. Mark a workstation as running
+            only after all specified readiness checks return
+            200 status codes.
         degraded (bool):
             Output only. Whether this resource is in degraded mode, in
             which case it may require user action to restore full
@@ -319,6 +339,12 @@ class WorkstationConfig(proto.Message):
         conditions (MutableSequence[google.rpc.status_pb2.Status]):
             Output only. Status conditions describing the
             current resource state.
+        enable_audit_agent (bool):
+            Whether to enable linux auditd logging on the workstation.
+            When enabled, a service account must also be specified that
+            has logging.buckets.write permission on the project.
+            Operating system audit logging is distinct from `Cloud Audit
+            Logs <https://cloud.google.com/workstations/docs/audit-logging>`__.
     """
 
     class Host(proto.Message):
@@ -353,7 +379,11 @@ class WorkstationConfig(proto.Message):
                     machines backing the Workstations.
                 pool_size (int):
                     Number of instances to pool for faster
-                    workstation starup.
+                    workstation startup.
+                pooled_instances (int):
+                    Output only. Number of instances currently
+                    available in the pool for faster workstation
+                    startup.
                 disable_public_ip_addresses (bool):
                     Whether instances have no public IP address.
                 shielded_instance_config (google.cloud.workstations_v1beta.types.WorkstationConfig.Host.GceInstance.GceShieldedInstanceConfig):
@@ -363,7 +393,10 @@ class WorkstationConfig(proto.Message):
                     A set of Compute Engine Confidential VM
                     instance options.
                 boot_disk_size_gb (int):
-                    Size of the boot disk in GB.
+                    Size of the boot disk in GB. Defaults to 50.
+                accelerators (MutableSequence[google.cloud.workstations_v1beta.types.WorkstationConfig.Host.GceInstance.Accelerator]):
+                    A list of the type and count of accelerator
+                    cards attached to the instance.
             """
 
             class GceShieldedInstanceConfig(proto.Message):
@@ -406,6 +439,27 @@ class WorkstationConfig(proto.Message):
                     number=1,
                 )
 
+            class Accelerator(proto.Message):
+                r"""An accelerator card attached to the instance.
+
+                Attributes:
+                    type_ (str):
+                        Type of accelerator resource to attach to the
+                        instance, for example, "nvidia-tesla-p100".
+                    count (int):
+                        Number of accelerator cards exposed to the
+                        instance.
+                """
+
+                type_: str = proto.Field(
+                    proto.STRING,
+                    number=1,
+                )
+                count: int = proto.Field(
+                    proto.INT32,
+                    number=2,
+                )
+
             machine_type: str = proto.Field(
                 proto.STRING,
                 number=1,
@@ -422,6 +476,10 @@ class WorkstationConfig(proto.Message):
                 proto.INT32,
                 number=5,
             )
+            pooled_instances: int = proto.Field(
+                proto.INT32,
+                number=12,
+            )
             disable_public_ip_addresses: bool = proto.Field(
                 proto.BOOL,
                 number=6,
@@ -440,6 +498,13 @@ class WorkstationConfig(proto.Message):
                 proto.INT32,
                 number=9,
             )
+            accelerators: MutableSequence[
+                "WorkstationConfig.Host.GceInstance.Accelerator"
+            ] = proto.RepeatedField(
+                proto.MESSAGE,
+                number=11,
+                message="WorkstationConfig.Host.GceInstance.Accelerator",
+            )
 
         gce_instance: "WorkstationConfig.Host.GceInstance" = proto.Field(
             proto.MESSAGE,
@@ -471,13 +536,14 @@ class WorkstationConfig(proto.Message):
             Attributes:
                 size_gb (int):
                     Size of the disk in GB. Must be empty if source_snapshot is
-                    set.
+                    set. Defaults to 200.
                 fs_type (str):
                     Type of file system that the disk should be formatted with.
                     The workstation image must support this file system type.
-                    Must be empty if source_snapshot is set.
+                    Must be empty if source_snapshot is set. Defaults to ext4.
                 disk_type (str):
-                    Type of the disk to use.
+                    Type of the disk to use. Defaults to
+                    pd-standard.
                 source_snapshot (str):
                     Name of the snapshot to use as the source for the disk. If
                     set, size_gb and fs_type must be empty.
@@ -544,8 +610,8 @@ class WorkstationConfig(proto.Message):
         Attributes:
             image (str):
                 Docker image defining the container. This
-                image must be accessible by the config's service
-                account.
+                image must be accessible by the service account
+                specified in the workstation configuration.
             command (MutableSequence[str]):
                 If set, overrides the default ENTRYPOINT
                 specified by the image.
@@ -553,7 +619,7 @@ class WorkstationConfig(proto.Message):
                 Arguments passed to the entrypoint.
             env (MutableMapping[str, str]):
                 Environment variables passed to the
-                container.
+                container's entrypoint.
             working_dir (str):
                 If set, overrides the default DIR specified
                 by the image.
@@ -594,13 +660,14 @@ class WorkstationConfig(proto.Message):
 
         Attributes:
             kms_key (str):
-                The name of the Google Cloud KMS encryption key. For
-                example,
+                Immutable. The name of the Google Cloud KMS encryption key.
+                For example,
                 ``projects/PROJECT_ID/locations/REGION/keyRings/KEY_RING/cryptoKeys/KEY_NAME``.
             kms_key_service_account (str):
-                The service account to use with the specified KMS key. We
-                recommend that you use a separate service account and follow
-                KMS best practices. For more information, see `Separation of
+                Immutable. The service account to use with the specified KMS
+                key. We recommend that you use a separate service account
+                and follow KMS best practices. For more information, see
+                `Separation of
                 duties <https://cloud.google.com/kms/docs/separation-of-duties>`__
                 and ``gcloud kms keys add-iam-policy-binding``
                 ```--member`` <https://cloud.google.com/sdk/gcloud/reference/kms/keys/add-iam-policy-binding#--member>`__.
@@ -615,6 +682,25 @@ class WorkstationConfig(proto.Message):
             number=2,
         )
 
+    class ReadinessCheck(proto.Message):
+        r"""A readiness check to be performed on a workstation.
+
+        Attributes:
+            path (str):
+                Path to which the request should be sent.
+            port (int):
+                Port to which the request should be sent.
+        """
+
+        path: str = proto.Field(
+            proto.STRING,
+            number=1,
+        )
+        port: int = proto.Field(
+            proto.INT32,
+            number=2,
+        )
+
     name: str = proto.Field(
         proto.STRING,
         number=1,
@@ -690,6 +776,11 @@ class WorkstationConfig(proto.Message):
         number=17,
         message=CustomerEncryptionKey,
     )
+    readiness_checks: MutableSequence[ReadinessCheck] = proto.RepeatedField(
+        proto.MESSAGE,
+        number=19,
+        message=ReadinessCheck,
+    )
     degraded: bool = proto.Field(
         proto.BOOL,
         number=15,
@@ -699,6 +790,10 @@ class WorkstationConfig(proto.Message):
         number=16,
         message=status_pb2.Status,
     )
+    enable_audit_agent: bool = proto.Field(
+        proto.BOOL,
+        number=20,
+    )
 
 
 class Workstation(proto.Message):
@@ -734,8 +829,8 @@ class Workstation(proto.Message):
             soft-deleted.
         etag (str):
             Checksum computed by the server. May be sent
-            on update and delete requests to ensure that the
-            client has an up-to-date value before
+            on update and delete requests to make sure that
+            the client has an up-to-date value before
             proceeding.
         state (google.cloud.workstations_v1beta.types.Workstation.State):
             Output only. Current state of the
@@ -747,6 +842,9 @@ class Workstation(proto.Message):
             send traffic to a different port, clients may prefix the
             host with the destination port in the format
             ``{port}-{host}``.
+        env (MutableMapping[str, str]):
+            Environment variables passed to the
+            workstation container's entrypoint.
     """
 
     class State(proto.Enum):
@@ -828,6 +926,11 @@ class Workstation(proto.Message):
         proto.STRING,
         number=11,
     )
+    env: MutableMapping[str, str] = proto.MapField(
+        proto.STRING,
+        proto.STRING,
+        number=12,
+    )
 
 
 class GetWorkstationClusterRequest(proto.Message):
@@ -990,7 +1093,7 @@ class DeleteWorkstationClusterRequest(proto.Message):
         etag (str):
             If set, the request will be rejected if the
             latest version of the workstation cluster on the
-            server does not have this etag.
+            server does not have this ETag.
         force (bool):
             If set, any workstation configurations and
             workstations in the workstation cluster are also
@@ -1158,7 +1261,8 @@ class CreateWorkstationConfigRequest(proto.Message):
         parent (str):
             Required. Parent resource name.
         workstation_config_id (str):
-            Required. ID to use for the config.
+            Required. ID to use for the workstation
+            configuration.
         workstation_config (google.cloud.workstations_v1beta.types.WorkstationConfig):
             Required. Config to create.
         validate_only (bool):
@@ -1193,13 +1297,14 @@ class UpdateWorkstationConfigRequest(proto.Message):
             Required. Config to update.
         update_mask (google.protobuf.field_mask_pb2.FieldMask):
             Required. Mask specifying which fields in the
-            config should be updated.
+            workstation configuration should be updated.
         validate_only (bool):
             If set, validate the request and preview the
             review, but do not actually apply it.
         allow_missing (bool):
-            If set, and the config is not found, a new config will be
-            created. In this situation, update_mask is ignored.
+            If set and the workstation configuration is not found, a new
+            workstation configuration will be created. In this
+            situation, update_mask is ignored.
     """
 
     workstation_config: "WorkstationConfig" = proto.Field(
@@ -1227,18 +1332,20 @@ class DeleteWorkstationConfigRequest(proto.Message):
 
     Attributes:
         name (str):
-            Required. Name of the config to delete.
+            Required. Name of the workstation
+            configuration to delete.
         validate_only (bool):
             If set, validate the request and preview the
             review, but do not actually apply it.
         etag (str):
-            If set, the request will be rejected if the
-            latest version of the config on the server does
-            not have this etag.
+            If set, the request is rejected if the latest
+            version of the workstation configuration on the
+            server does not have this ETag.
         force (bool):
-            If set, any Workstations in the config will
-            also be deleted. Otherwise, the request will
-            work only if the config has no workstations.
+            If set, any workstations in the workstation
+            configuration are also deleted. Otherwise, the
+            request works only if the workstation
+            configuration has no workstations.
     """
 
     name: str = proto.Field(
@@ -1435,13 +1542,14 @@ class UpdateWorkstationRequest(proto.Message):
             Required. Workstation to update.
         update_mask (google.protobuf.field_mask_pb2.FieldMask):
             Required. Mask specifying which fields in the
-            config should be updated.
+            workstation configuration should be updated.
         validate_only (bool):
             If set, validate the request and preview the
             review, but do not actually apply it.
         allow_missing (bool):
-            If set, and the config is not found, a new config will be
-            created. In this situation, update_mask is ignored.
+            If set and the workstation configuration is not found, a new
+            workstation configuration is created. In this situation,
+            update_mask is ignored.
     """
 
     workstation: "Workstation" = proto.Field(
@@ -1476,7 +1584,7 @@ class DeleteWorkstationRequest(proto.Message):
         etag (str):
             If set, the request will be rejected if the
             latest version of the workstation on the server
-            does not have this etag.
+            does not have this ETag.
     """
 
     name: str = proto.Field(
@@ -1505,7 +1613,7 @@ class StartWorkstationRequest(proto.Message):
         etag (str):
             If set, the request will be rejected if the
             latest version of the workstation on the server
-            does not have this etag.
+            does not have this ETag.
     """
 
     name: str = proto.Field(
@@ -1534,7 +1642,7 @@ class StopWorkstationRequest(proto.Message):
         etag (str):
             If set, the request will be rejected if the
             latest version of the workstation on the server
-            does not have this etag.
+            does not have this ETag.
     """
 
     name: str = proto.Field(
@@ -1607,7 +1715,7 @@ class GenerateAccessTokenResponse(proto.Message):
         access_token (str):
             The generated bearer access token. To use this token,
             include it in an Authorization header of an HTTP request
-            sent to the associated workstation's hostname, for example,
+            sent to the associated workstation's hostname—for example,
             ``Authorization: Bearer <access_token>``.
         expire_time (google.protobuf.timestamp_pb2.Timestamp):
             Time at which the generated token will

{google-cloud-workstations-0.3.0 → google-cloud-workstations-0.4.0/google_cloud_workstations.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: google-cloud-workstations
-Version: 0.3.0
+Version: 0.4.0
 Summary: Google Cloud Workstations API client library
 Home-page: https://github.com/googleapis/google-cloud-python
 Author: Google LLC

{google-cloud-workstations-0.3.0 → google-cloud-workstations-0.4.0}/tests/unit/gapic/workstations_v1beta/test_workstations.py

@@ -738,6 +738,7 @@ def test_get_workstation_cluster(request_type, transport: str = "grpc"):
             etag="etag_value",
             network="network_value",
             subnetwork="subnetwork_value",
+            control_plane_ip="control_plane_ip_value",
             degraded=True,
         )
         response = client.get_workstation_cluster(request)
@@ -756,6 +757,7 @@ def test_get_workstation_cluster(request_type, transport: str = "grpc"):
     assert response.etag == "etag_value"
     assert response.network == "network_value"
     assert response.subnetwork == "subnetwork_value"
+    assert response.control_plane_ip == "control_plane_ip_value"
     assert response.degraded is True
 
 
@@ -805,6 +807,7 @@ async def test_get_workstation_cluster_async(
                 etag="etag_value",
                 network="network_value",
                 subnetwork="subnetwork_value",
+                control_plane_ip="control_plane_ip_value",
                 degraded=True,
             )
         )
@@ -824,6 +827,7 @@ async def test_get_workstation_cluster_async(
     assert response.etag == "etag_value"
     assert response.network == "network_value"
     assert response.subnetwork == "subnetwork_value"
+    assert response.control_plane_ip == "control_plane_ip_value"
     assert response.degraded is True
 
 
@@ -2212,6 +2216,7 @@ def test_get_workstation_config(request_type, transport: str = "grpc"):
             reconciling=True,
             etag="etag_value",
             degraded=True,
+            enable_audit_agent=True,
         )
         response = client.get_workstation_config(request)
 
@@ -2228,6 +2233,7 @@ def test_get_workstation_config(request_type, transport: str = "grpc"):
     assert response.reconciling is True
     assert response.etag == "etag_value"
     assert response.degraded is True
+    assert response.enable_audit_agent is True
 
 
 def test_get_workstation_config_empty_call():
@@ -2275,6 +2281,7 @@ async def test_get_workstation_config_async(
                 reconciling=True,
                 etag="etag_value",
                 degraded=True,
+                enable_audit_agent=True,
             )
         )
         response = await client.get_workstation_config(request)
@@ -2292,6 +2299,7 @@ async def test_get_workstation_config_async(
     assert response.reconciling is True
     assert response.etag == "etag_value"
     assert response.degraded is True
+    assert response.enable_audit_agent is True
 
 
 @pytest.mark.asyncio
@@ -6739,6 +6747,7 @@ def test_get_workstation_cluster_rest(request_type):
             etag="etag_value",
             network="network_value",
             subnetwork="subnetwork_value",
+            control_plane_ip="control_plane_ip_value",
             degraded=True,
         )
 
@@ -6761,6 +6770,7 @@ def test_get_workstation_cluster_rest(request_type):
     assert response.etag == "etag_value"
     assert response.network == "network_value"
     assert response.subnetwork == "subnetwork_value"
+    assert response.control_plane_ip == "control_plane_ip_value"
     assert response.degraded is True
 
 
@@ -7371,6 +7381,7 @@ def test_create_workstation_cluster_rest(request_type):
         "etag": "etag_value",
         "network": "network_value",
         "subnetwork": "subnetwork_value",
+        "control_plane_ip": "control_plane_ip_value",
         "private_cluster_config": {
             "enable_private_endpoint": True,
             "cluster_hostname": "cluster_hostname_value",
@@ -7615,6 +7626,7 @@ def test_create_workstation_cluster_rest_bad_request(
         "etag": "etag_value",
         "network": "network_value",
         "subnetwork": "subnetwork_value",
+        "control_plane_ip": "control_plane_ip_value",
         "private_cluster_config": {
             "enable_private_endpoint": True,
             "cluster_hostname": "cluster_hostname_value",
@@ -7746,6 +7758,7 @@ def test_update_workstation_cluster_rest(request_type):
         "etag": "etag_value",
         "network": "network_value",
         "subnetwork": "subnetwork_value",
+        "control_plane_ip": "control_plane_ip_value",
         "private_cluster_config": {
             "enable_private_endpoint": True,
             "cluster_hostname": "cluster_hostname_value",
@@ -7974,6 +7987,7 @@ def test_update_workstation_cluster_rest_bad_request(
         "etag": "etag_value",
         "network": "network_value",
         "subnetwork": "subnetwork_value",
+        "control_plane_ip": "control_plane_ip_value",
         "private_cluster_config": {
             "enable_private_endpoint": True,
             "cluster_hostname": "cluster_hostname_value",
@@ -8390,6 +8404,7 @@ def test_get_workstation_config_rest(request_type):
             reconciling=True,
             etag="etag_value",
             degraded=True,
+            enable_audit_agent=True,
         )
 
         # Wrap the value into a proper Response obj
@@ -8410,6 +8425,7 @@ def test_get_workstation_config_rest(request_type):
     assert response.reconciling is True
     assert response.etag == "etag_value"
     assert response.degraded is True
+    assert response.enable_audit_agent is True
 
 
 def test_get_workstation_config_rest_required_fields(
@@ -9397,6 +9413,7 @@ def test_create_workstation_config_rest(request_type):
                 "service_account": "service_account_value",
                 "tags": ["tags_value1", "tags_value2"],
                 "pool_size": 980,
+                "pooled_instances": 1706,
                 "disable_public_ip_addresses": True,
                 "shielded_instance_config": {
                     "enable_secure_boot": True,
@@ -9405,6 +9422,7 @@ def test_create_workstation_config_rest(request_type):
                 },
                 "confidential_instance_config": {"enable_confidential_compute": True},
                 "boot_disk_size_gb": 1792,
+                "accelerators": [{"type_": "type__value", "count": 553}],
             }
         },
         "persistent_directories": [
@@ -9431,6 +9449,7 @@ def test_create_workstation_config_rest(request_type):
             "kms_key": "kms_key_value",
             "kms_key_service_account": "kms_key_service_account_value",
         },
+        "readiness_checks": [{"path": "path_value", "port": 453}],
         "degraded": True,
         "conditions": [
             {
@@ -9444,6 +9463,7 @@ def test_create_workstation_config_rest(request_type):
                 ],
             }
         ],
+        "enable_audit_agent": True,
     }
     request = request_type(**request_init)
 
@@ -9677,6 +9697,7 @@ def test_create_workstation_config_rest_bad_request(
                 "service_account": "service_account_value",
                 "tags": ["tags_value1", "tags_value2"],
                 "pool_size": 980,
+                "pooled_instances": 1706,
                 "disable_public_ip_addresses": True,
                 "shielded_instance_config": {
                     "enable_secure_boot": True,
@@ -9685,6 +9706,7 @@ def test_create_workstation_config_rest_bad_request(
                 },
                 "confidential_instance_config": {"enable_confidential_compute": True},
                 "boot_disk_size_gb": 1792,
+                "accelerators": [{"type_": "type__value", "count": 553}],
             }
         },
         "persistent_directories": [
@@ -9711,6 +9733,7 @@ def test_create_workstation_config_rest_bad_request(
             "kms_key": "kms_key_value",
             "kms_key_service_account": "kms_key_service_account_value",
         },
+        "readiness_checks": [{"path": "path_value", "port": 453}],
         "degraded": True,
         "conditions": [
             {
@@ -9724,6 +9747,7 @@ def test_create_workstation_config_rest_bad_request(
                 ],
             }
         ],
+        "enable_audit_agent": True,
     }
     request = request_type(**request_init)
 
@@ -9844,6 +9868,7 @@ def test_update_workstation_config_rest(request_type):
                 "service_account": "service_account_value",
                 "tags": ["tags_value1", "tags_value2"],
                 "pool_size": 980,
+                "pooled_instances": 1706,
                 "disable_public_ip_addresses": True,
                 "shielded_instance_config": {
                     "enable_secure_boot": True,
@@ -9852,6 +9877,7 @@ def test_update_workstation_config_rest(request_type):
                 },
                 "confidential_instance_config": {"enable_confidential_compute": True},
                 "boot_disk_size_gb": 1792,
+                "accelerators": [{"type_": "type__value", "count": 553}],
             }
         },
         "persistent_directories": [
@@ -9878,6 +9904,7 @@ def test_update_workstation_config_rest(request_type):
             "kms_key": "kms_key_value",
             "kms_key_service_account": "kms_key_service_account_value",
         },
+        "readiness_checks": [{"path": "path_value", "port": 453}],
         "degraded": True,
         "conditions": [
             {
@@ -9891,6 +9918,7 @@ def test_update_workstation_config_rest(request_type):
                 ],
             }
         ],
+        "enable_audit_agent": True,
     }
     request = request_type(**request_init)
 
@@ -10106,6 +10134,7 @@ def test_update_workstation_config_rest_bad_request(
                 "service_account": "service_account_value",
                 "tags": ["tags_value1", "tags_value2"],
                 "pool_size": 980,
+                "pooled_instances": 1706,
                 "disable_public_ip_addresses": True,
                 "shielded_instance_config": {
                     "enable_secure_boot": True,
@@ -10114,6 +10143,7 @@ def test_update_workstation_config_rest_bad_request(
                 },
                 "confidential_instance_config": {"enable_confidential_compute": True},
                 "boot_disk_size_gb": 1792,
+                "accelerators": [{"type_": "type__value", "count": 553}],
             }
         },
         "persistent_directories": [
@@ -10140,6 +10170,7 @@ def test_update_workstation_config_rest_bad_request(
             "kms_key": "kms_key_value",
             "kms_key_service_account": "kms_key_service_account_value",
         },
+        "readiness_checks": [{"path": "path_value", "port": 453}],
         "degraded": True,
         "conditions": [
             {
@@ -10153,6 +10184,7 @@ def test_update_workstation_config_rest_bad_request(
                 ],
             }
         ],
+        "enable_audit_agent": True,
     }
     request = request_type(**request_init)
 
@@ -11539,6 +11571,7 @@ def test_create_workstation_rest(request_type):
         "etag": "etag_value",
         "state": 1,
         "host": "host_value",
+        "env": {},
     }
     request = request_type(**request_init)
 
@@ -11763,6 +11796,7 @@ def test_create_workstation_rest_bad_request(
         "etag": "etag_value",
         "state": 1,
         "host": "host_value",
+        "env": {},
     }
     request = request_type(**request_init)
 
@@ -11877,6 +11911,7 @@ def test_update_workstation_rest(request_type):
         "etag": "etag_value",
         "state": 1,
         "host": "host_value",
+        "env": {},
     }
     request = request_type(**request_init)
 
@@ -12086,6 +12121,7 @@ def test_update_workstation_rest_bad_request(
         "etag": "etag_value",
         "state": 1,
         "host": "host_value",
+        "env": {},
     }
     request = request_type(**request_init)