Skip to content

Commit

Permalink
feat(tpu): update the API
Browse files Browse the repository at this point in the history
#### tpu:v1alpha1

The following keys were changed:
- resources.projects.resources.locations.resources.operations.methods.cancel.description
- schemas.Node.properties.state.enum
- schemas.Node.properties.state.enumDescriptions

#### tpu:v2alpha1

The following keys were added:
- resources.projects.resources.locations.resources.nodes.methods.performMaintenance.description
- resources.projects.resources.locations.resources.nodes.methods.performMaintenance.flatPath
- resources.projects.resources.locations.resources.nodes.methods.performMaintenance.httpMethod
- resources.projects.resources.locations.resources.nodes.methods.performMaintenance.id
- resources.projects.resources.locations.resources.nodes.methods.performMaintenance.parameterOrder
- resources.projects.resources.locations.resources.nodes.methods.performMaintenance.parameters.name.description
- resources.projects.resources.locations.resources.nodes.methods.performMaintenance.parameters.name.location
- resources.projects.resources.locations.resources.nodes.methods.performMaintenance.parameters.name.pattern
- resources.projects.resources.locations.resources.nodes.methods.performMaintenance.parameters.name.required
- resources.projects.resources.locations.resources.nodes.methods.performMaintenance.parameters.name.type
- resources.projects.resources.locations.resources.nodes.methods.performMaintenance.path
- resources.projects.resources.locations.resources.nodes.methods.performMaintenance.request.$ref
- resources.projects.resources.locations.resources.nodes.methods.performMaintenance.response.$ref
- resources.projects.resources.locations.resources.nodes.methods.performMaintenance.scopes
- resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.description
- resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.flatPath
- resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.httpMethod
- resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.id
- resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.parameterOrder
- resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.parameters.name.description
- resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.parameters.name.location
- resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.parameters.name.pattern
- resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.parameters.name.required
- resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.parameters.name.type
- resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.path
- resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.request.$ref
- resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.response.$ref
- resources.projects.resources.locations.resources.queuedResources.methods.performMaintenanceQueuedResource.scopes
- schemas.MultiNodeParams.properties.workloadType.description
- schemas.MultiNodeParams.properties.workloadType.enum
- schemas.MultiNodeParams.properties.workloadType.enumDescriptions
- schemas.MultiNodeParams.properties.workloadType.type
- schemas.Node.properties.upcomingMaintenance.$ref
- schemas.Node.properties.upcomingMaintenance.description
- schemas.Node.properties.upcomingMaintenance.readOnly
- schemas.PerformMaintenanceQueuedResourceRequest.description
- schemas.PerformMaintenanceQueuedResourceRequest.id
- schemas.PerformMaintenanceQueuedResourceRequest.properties.nodeNames.description
- schemas.PerformMaintenanceQueuedResourceRequest.properties.nodeNames.items.type
- schemas.PerformMaintenanceQueuedResourceRequest.properties.nodeNames.type
- schemas.PerformMaintenanceQueuedResourceRequest.type
- schemas.PerformMaintenanceRequest.description
- schemas.PerformMaintenanceRequest.id
- schemas.PerformMaintenanceRequest.type
- schemas.Reservation.properties.standard.description
- schemas.Standard.description
- schemas.Standard.properties.capacityUnits.description
- schemas.Standard.properties.usage.description
- schemas.UpcomingMaintenance.description
- schemas.UpcomingMaintenance.id
- schemas.UpcomingMaintenance.properties.canReschedule.description
- schemas.UpcomingMaintenance.properties.canReschedule.type
- schemas.UpcomingMaintenance.properties.latestWindowStartTime.description
- schemas.UpcomingMaintenance.properties.latestWindowStartTime.type
- schemas.UpcomingMaintenance.properties.maintenanceStatus.description
- schemas.UpcomingMaintenance.properties.maintenanceStatus.enum
- schemas.UpcomingMaintenance.properties.maintenanceStatus.enumDescriptions
- schemas.UpcomingMaintenance.properties.maintenanceStatus.type
- schemas.UpcomingMaintenance.properties.type.description
- schemas.UpcomingMaintenance.properties.type.enum
- schemas.UpcomingMaintenance.properties.type.enumDescriptions
- schemas.UpcomingMaintenance.properties.type.type
- schemas.UpcomingMaintenance.properties.windowEndTime.description
- schemas.UpcomingMaintenance.properties.windowEndTime.type
- schemas.UpcomingMaintenance.properties.windowStartTime.description
- schemas.UpcomingMaintenance.properties.windowStartTime.type
- schemas.UpcomingMaintenance.type
- schemas.Usage.description

The following keys were changed:
- resources.projects.resources.locations.resources.operations.methods.cancel.description
- resources.projects.resources.locations.resources.reservations.methods.list.parameters.pageSize.description
- schemas.AcceleratorConfig.properties.type.enum
- schemas.AcceleratorConfig.properties.type.enumDescriptions
- schemas.Node.properties.networkConfig.description
- schemas.Node.properties.networkConfigs.description
- schemas.Node.properties.state.enum
- schemas.Node.properties.state.enumDescriptions

#### tpu:v1

The following keys were changed:
- resources.projects.resources.locations.resources.operations.methods.cancel.description
- schemas.Node.properties.state.enum
- schemas.Node.properties.state.enumDescriptions

#### tpu:v2

The following keys were added:
- schemas.Node.properties.networkConfigs.description
- schemas.Node.properties.networkConfigs.items.$ref
- schemas.Node.properties.networkConfigs.type

The following keys were changed:
- resources.projects.resources.locations.resources.operations.methods.cancel.description
- schemas.AcceleratorConfig.properties.type.enum
- schemas.AcceleratorConfig.properties.type.enumDescriptions
- schemas.MultisliceParams.properties.nodeCount.description
- schemas.Node.properties.networkConfig.description
- schemas.Node.properties.state.enum
- schemas.Node.properties.state.enumDescriptions
  • Loading branch information
yoshi-automation authored and sofisl committed Jan 29, 2025
1 parent 0e6f86a commit 5c0a386
Show file tree
Hide file tree
Showing 8 changed files with 478 additions and 36 deletions.
10 changes: 6 additions & 4 deletions discovery/tpu-v1.json
Original file line number Diff line number Diff line change
Expand Up @@ -462,7 +462,7 @@
"operations": {
"methods": {
"cancel": {
"description": "Starts asynchronous cancellation on a long-running operation. The server makes a best effort to cancel the operation, but success is not guaranteed. If the server doesn't support this method, it returns `google.rpc.Code.UNIMPLEMENTED`. Clients can use Operations.GetOperation or other methods to check whether the cancellation succeeded or whether the operation completed despite cancellation. On successful cancellation, the operation is not deleted; instead, it becomes an operation with an Operation.error value with a google.rpc.Status.code of 1, corresponding to `Code.CANCELLED`.",
"description": "Starts asynchronous cancellation on a long-running operation. The server makes a best effort to cancel the operation, but success is not guaranteed. If the server doesn't support this method, it returns `google.rpc.Code.UNIMPLEMENTED`. Clients can use Operations.GetOperation or other methods to check whether the cancellation succeeded or whether the operation completed despite cancellation. On successful cancellation, the operation is not deleted; instead, it becomes an operation with an Operation.error value with a google.rpc.Status.code of `1`, corresponding to `Code.CANCELLED`.",
"flatPath": "v1/projects/{projectsId}/locations/{locationsId}/operations/{operationsId}:cancel",
"httpMethod": "POST",
"id": "tpu.projects.locations.operations.cancel",
Expand Down Expand Up @@ -659,7 +659,7 @@
}
}
},
"revision": "20230803",
"revision": "20241126",
"rootUrl": "https://tpu.googleapis.com/",
"schemas": {
"AcceleratorType": {
Expand Down Expand Up @@ -968,7 +968,8 @@
"TERMINATED",
"HIDING",
"HIDDEN",
"UNHIDING"
"UNHIDING",
"UNKNOWN"
],
"enumDescriptions": [
"TPU node state is not known/set.",
Expand All @@ -985,7 +986,8 @@
"TPU node has been terminated due to maintenance or has reached the end of its life cycle (for preemptible nodes).",
"TPU node is currently hiding.",
"TPU node has been hidden.",
"TPU node is currently unhiding."
"TPU node is currently unhiding.",
"TPU node has unknown state after a failed repair."
],
"readOnly": true,
"type": "string"
Expand Down
10 changes: 6 additions & 4 deletions discovery/tpu-v1alpha1.json
Original file line number Diff line number Diff line change
Expand Up @@ -472,7 +472,7 @@
"operations": {
"methods": {
"cancel": {
"description": "Starts asynchronous cancellation on a long-running operation. The server makes a best effort to cancel the operation, but success is not guaranteed. If the server doesn't support this method, it returns `google.rpc.Code.UNIMPLEMENTED`. Clients can use Operations.GetOperation or other methods to check whether the cancellation succeeded or whether the operation completed despite cancellation. On successful cancellation, the operation is not deleted; instead, it becomes an operation with an Operation.error value with a google.rpc.Status.code of 1, corresponding to `Code.CANCELLED`.",
"description": "Starts asynchronous cancellation on a long-running operation. The server makes a best effort to cancel the operation, but success is not guaranteed. If the server doesn't support this method, it returns `google.rpc.Code.UNIMPLEMENTED`. Clients can use Operations.GetOperation or other methods to check whether the cancellation succeeded or whether the operation completed despite cancellation. On successful cancellation, the operation is not deleted; instead, it becomes an operation with an Operation.error value with a google.rpc.Status.code of `1`, corresponding to `Code.CANCELLED`.",
"flatPath": "v1alpha1/projects/{projectsId}/locations/{locationsId}/operations/{operationsId}:cancel",
"httpMethod": "POST",
"id": "tpu.projects.locations.operations.cancel",
Expand Down Expand Up @@ -669,7 +669,7 @@
}
}
},
"revision": "20230803",
"revision": "20241126",
"rootUrl": "https://tpu.googleapis.com/",
"schemas": {
"AcceleratorType": {
Expand Down Expand Up @@ -978,7 +978,8 @@
"TERMINATED",
"HIDING",
"HIDDEN",
"UNHIDING"
"UNHIDING",
"UNKNOWN"
],
"enumDescriptions": [
"TPU node state is not known/set.",
Expand All @@ -995,7 +996,8 @@
"TPU node has been terminated due to maintenance or has reached the end of its life cycle (for preemptible nodes).",
"TPU node is currently hiding.",
"TPU node has been hidden.",
"TPU node is currently unhiding."
"TPU node is currently unhiding.",
"TPU node has unknown state after a failed repair."
],
"readOnly": true,
"type": "string"
Expand Down
27 changes: 19 additions & 8 deletions discovery/tpu-v2.json
Original file line number Diff line number Diff line change
Expand Up @@ -524,7 +524,7 @@
"operations": {
"methods": {
"cancel": {
"description": "Starts asynchronous cancellation on a long-running operation. The server makes a best effort to cancel the operation, but success is not guaranteed. If the server doesn't support this method, it returns `google.rpc.Code.UNIMPLEMENTED`. Clients can use Operations.GetOperation or other methods to check whether the cancellation succeeded or whether the operation completed despite cancellation. On successful cancellation, the operation is not deleted; instead, it becomes an operation with an Operation.error value with a google.rpc.Status.code of 1, corresponding to `Code.CANCELLED`.",
"description": "Starts asynchronous cancellation on a long-running operation. The server makes a best effort to cancel the operation, but success is not guaranteed. If the server doesn't support this method, it returns `google.rpc.Code.UNIMPLEMENTED`. Clients can use Operations.GetOperation or other methods to check whether the cancellation succeeded or whether the operation completed despite cancellation. On successful cancellation, the operation is not deleted; instead, it becomes an operation with an Operation.error value with a google.rpc.Status.code of `1`, corresponding to `Code.CANCELLED`.",
"flatPath": "v2/projects/{projectsId}/locations/{locationsId}/operations/{operationsId}:cancel",
"httpMethod": "POST",
"id": "tpu.projects.locations.operations.cancel",
Expand Down Expand Up @@ -887,7 +887,7 @@
}
}
},
"revision": "20240613",
"revision": "20250116",
"rootUrl": "https://tpu.googleapis.com/",
"schemas": {
"AcceleratorConfig": {
Expand All @@ -906,15 +906,17 @@
"V3",
"V4",
"V5LITE_POD",
"V5P"
"V5P",
"V6E"
],
"enumDescriptions": [
"Unspecified version.",
"TPU v2.",
"TPU v3.",
"TPU v4.",
"TPU v5lite pod.",
"TPU v5p"
"TPU v5p.",
"TPU v6e."
],
"type": "string"
}
Expand Down Expand Up @@ -1323,7 +1325,7 @@
"id": "MultisliceParams",
"properties": {
"nodeCount": {
"description": "Required. Number of nodes with this spec. The system will attempt to provison \"node_count\" nodes as part of the request. This needs to be > 1.",
"description": "Required. Number of nodes with this spec. The system will attempt to provision \"node_count\" nodes as part of the request. This needs to be > 1.",
"format": "int32",
"type": "integer"
},
Expand Down Expand Up @@ -1489,7 +1491,14 @@
},
"networkConfig": {
"$ref": "NetworkConfig",
"description": "Network configurations for the TPU node."
"description": "Network configurations for the TPU node. network_config and network_configs are mutually exclusive, you can only specify one of them. If both are specified, an error will be returned."
},
"networkConfigs": {
"description": "Optional. Repeated network configurations for the TPU node. This field is used to specify multiple networks configs for the TPU node. network_config and network_configs are mutually exclusive, you can only specify one of them. If both are specified, an error will be returned.",
"items": {
"$ref": "NetworkConfig"
},
"type": "array"
},
"networkEndpoints": {
"description": "Output only. The network endpoints where TPU workers can be accessed and sent work. It is recommended that runtime clients of the node reach out to the 0th entry in this map first.",
Expand Down Expand Up @@ -1537,7 +1546,8 @@
"TERMINATED",
"HIDING",
"HIDDEN",
"UNHIDING"
"UNHIDING",
"UNKNOWN"
],
"enumDescriptions": [
"TPU node state is not known/set.",
Expand All @@ -1554,7 +1564,8 @@
"TPU node has been terminated due to maintenance or has reached the end of its life cycle (for preemptible nodes).",
"TPU node is currently hiding.",
"TPU node has been hidden.",
"TPU node is currently unhiding."
"TPU node is currently unhiding.",
"TPU node has unknown state after a failed repair."
],
"readOnly": true,
"type": "string"
Expand Down
Loading

0 comments on commit 5c0a386

Please sign in to comment.