Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat(aiplatform): add ComputeTokens API #8999

Merged
merged 19 commits into from
Nov 15, 2023
Merged
Changes from all commits
Commits
Show all changes
19 commits
Select commit Hold shift + click to select a range
57eae5f
feat: Add a new field `token_type` to `TokenOptions` message proto
gcf-owl-bot[bot] Nov 10, 2023
2ec3c85
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Nov 10, 2023
7acc3da
feat: add protected_artifact_location_id to CustomJob
gcf-owl-bot[bot] Nov 10, 2023
e54055a
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Nov 10, 2023
87e43ac
feat: add protected_artifact_location_id to CustomJob
gcf-owl-bot[bot] Nov 10, 2023
7714a4d
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Nov 10, 2023
9844bf5
build: Change language namespaces
gcf-owl-bot[bot] Nov 14, 2023
010422b
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Nov 14, 2023
6512ca1
feat: add ComputeTokens and CountTokens API
gcf-owl-bot[bot] Nov 14, 2023
108f55d
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Nov 14, 2023
25c6323
feat: add deployment_timeout to UploadModel ModelContainerSpec
gcf-owl-bot[bot] Nov 14, 2023
7fd2970
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Nov 14, 2023
57eb56b
feat: add deployment_timeout to UploadModel ModelContainerSpec
gcf-owl-bot[bot] Nov 14, 2023
f254d2d
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Nov 14, 2023
f600432
feat: Added REGIONAL tier support in the v1 API
gcf-owl-bot[bot] Nov 15, 2023
f89718c
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Nov 15, 2023
251c3cb
feat: add ComputeTokens API
gcf-owl-bot[bot] Nov 15, 2023
bc588f6
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Nov 15, 2023
5a4d573
Merge branch 'main' into owl-bot-copy
codyoss Nov 15, 2023
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
213 changes: 115 additions & 98 deletions aiplatform/apiv1/aiplatformpb/custom_job.pb.go

Large diffs are not rendered by default.

218 changes: 122 additions & 96 deletions aiplatform/apiv1/aiplatformpb/feature.pb.go

Large diffs are not rendered by default.

403 changes: 403 additions & 0 deletions aiplatform/apiv1/aiplatformpb/feature_group.pb.go

Large diffs are not rendered by default.

552 changes: 552 additions & 0 deletions aiplatform/apiv1/aiplatformpb/feature_online_store.pb.go

Large diffs are not rendered by default.

2,840 changes: 2,840 additions & 0 deletions aiplatform/apiv1/aiplatformpb/feature_online_store_admin_service.pb.go

Large diffs are not rendered by default.

788 changes: 788 additions & 0 deletions aiplatform/apiv1/aiplatformpb/feature_online_store_service.pb.go

Large diffs are not rendered by default.

1,622 changes: 1,622 additions & 0 deletions aiplatform/apiv1/aiplatformpb/feature_registry_service.pb.go

Large diffs are not rendered by default.

645 changes: 645 additions & 0 deletions aiplatform/apiv1/aiplatformpb/feature_view.pb.go

Large diffs are not rendered by default.

249 changes: 249 additions & 0 deletions aiplatform/apiv1/aiplatformpb/feature_view_sync.pb.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

39 changes: 32 additions & 7 deletions aiplatform/apiv1/aiplatformpb/featurestore_service.pb.go
388 changes: 320 additions & 68 deletions aiplatform/apiv1/aiplatformpb/index.pb.go

Large diffs are not rendered by default.

9 changes: 5 additions & 4 deletions aiplatform/apiv1/aiplatformpb/index_endpoint.pb.go
531 changes: 531 additions & 0 deletions aiplatform/apiv1/aiplatformpb/llm_utility_service.pb.go

Large diffs are not rendered by default.

181 changes: 97 additions & 84 deletions aiplatform/apiv1/aiplatformpb/machine_resources.pb.go

Large diffs are not rendered by default.

391 changes: 322 additions & 69 deletions aiplatform/apiv1/aiplatformpb/model.pb.go

Large diffs are not rendered by default.

234 changes: 199 additions & 35 deletions aiplatform/apiv1/aiplatformpb/prediction_service.pb.go
1,217 changes: 774 additions & 443 deletions aiplatform/apiv1/aiplatformpb/study.pb.go

Large diffs are not rendered by default.

915 changes: 840 additions & 75 deletions aiplatform/apiv1/auxiliary.go

Large diffs are not rendered by default.

1,051 changes: 1,051 additions & 0 deletions aiplatform/apiv1/feature_online_store_admin_client.go

Large diffs are not rendered by default.

672 changes: 672 additions & 0 deletions aiplatform/apiv1/feature_online_store_admin_client_example_test.go

Large diffs are not rendered by default.

523 changes: 523 additions & 0 deletions aiplatform/apiv1/feature_online_store_client.go

Large diffs are not rendered by default.

328 changes: 328 additions & 0 deletions aiplatform/apiv1/feature_online_store_client_example_test.go
944 changes: 944 additions & 0 deletions aiplatform/apiv1/feature_registry_client.go

Large diffs are not rendered by default.

591 changes: 591 additions & 0 deletions aiplatform/apiv1/feature_registry_client_example_test.go

Large diffs are not rendered by default.

366 changes: 366 additions & 0 deletions aiplatform/apiv1/gapic_metadata.json
Original file line number Diff line number Diff line change
@@ -253,6 +253,303 @@
}
}
},
"FeatureOnlineStoreAdminService": {
"clients": {
"grpc": {
"libraryClient": "FeatureOnlineStoreAdminClient",
"rpcs": {
"CancelOperation": {
"methods": [
"CancelOperation"
]
},
"CreateFeatureOnlineStore": {
"methods": [
"CreateFeatureOnlineStore"
]
},
"CreateFeatureView": {
"methods": [
"CreateFeatureView"
]
},
"DeleteFeatureOnlineStore": {
"methods": [
"DeleteFeatureOnlineStore"
]
},
"DeleteFeatureView": {
"methods": [
"DeleteFeatureView"
]
},
"DeleteOperation": {
"methods": [
"DeleteOperation"
]
},
"GetFeatureOnlineStore": {
"methods": [
"GetFeatureOnlineStore"
]
},
"GetFeatureView": {
"methods": [
"GetFeatureView"
]
},
"GetFeatureViewSync": {
"methods": [
"GetFeatureViewSync"
]
},
"GetIamPolicy": {
"methods": [
"GetIamPolicy"
]
},
"GetLocation": {
"methods": [
"GetLocation"
]
},
"GetOperation": {
"methods": [
"GetOperation"
]
},
"ListFeatureOnlineStores": {
"methods": [
"ListFeatureOnlineStores"
]
},
"ListFeatureViewSyncs": {
"methods": [
"ListFeatureViewSyncs"
]
},
"ListFeatureViews": {
"methods": [
"ListFeatureViews"
]
},
"ListLocations": {
"methods": [
"ListLocations"
]
},
"ListOperations": {
"methods": [
"ListOperations"
]
},
"SetIamPolicy": {
"methods": [
"SetIamPolicy"
]
},
"SyncFeatureView": {
"methods": [
"SyncFeatureView"
]
},
"TestIamPermissions": {
"methods": [
"TestIamPermissions"
]
},
"UpdateFeatureOnlineStore": {
"methods": [
"UpdateFeatureOnlineStore"
]
},
"UpdateFeatureView": {
"methods": [
"UpdateFeatureView"
]
},
"WaitOperation": {
"methods": [
"WaitOperation"
]
}
}
}
}
},
"FeatureOnlineStoreService": {
"clients": {
"grpc": {
"libraryClient": "FeatureOnlineStoreClient",
"rpcs": {
"CancelOperation": {
"methods": [
"CancelOperation"
]
},
"DeleteOperation": {
"methods": [
"DeleteOperation"
]
},
"FetchFeatureValues": {
"methods": [
"FetchFeatureValues"
]
},
"GetIamPolicy": {
"methods": [
"GetIamPolicy"
]
},
"GetLocation": {
"methods": [
"GetLocation"
]
},
"GetOperation": {
"methods": [
"GetOperation"
]
},
"ListLocations": {
"methods": [
"ListLocations"
]
},
"ListOperations": {
"methods": [
"ListOperations"
]
},
"SetIamPolicy": {
"methods": [
"SetIamPolicy"
]
},
"TestIamPermissions": {
"methods": [
"TestIamPermissions"
]
},
"WaitOperation": {
"methods": [
"WaitOperation"
]
}
}
}
}
},
"FeatureRegistryService": {
"clients": {
"grpc": {
"libraryClient": "FeatureRegistryClient",
"rpcs": {
"CancelOperation": {
"methods": [
"CancelOperation"
]
},
"CreateFeature": {
"methods": [
"CreateFeature"
]
},
"CreateFeatureGroup": {
"methods": [
"CreateFeatureGroup"
]
},
"DeleteFeature": {
"methods": [
"DeleteFeature"
]
},
"DeleteFeatureGroup": {
"methods": [
"DeleteFeatureGroup"
]
},
"DeleteOperation": {
"methods": [
"DeleteOperation"
]
},
"GetFeature": {
"methods": [
"GetFeature"
]
},
"GetFeatureGroup": {
"methods": [
"GetFeatureGroup"
]
},
"GetIamPolicy": {
"methods": [
"GetIamPolicy"
]
},
"GetLocation": {
"methods": [
"GetLocation"
]
},
"GetOperation": {
"methods": [
"GetOperation"
]
},
"ListFeatureGroups": {
"methods": [
"ListFeatureGroups"
]
},
"ListFeatures": {
"methods": [
"ListFeatures"
]
},
"ListLocations": {
"methods": [
"ListLocations"
]
},
"ListOperations": {
"methods": [
"ListOperations"
]
},
"SetIamPolicy": {
"methods": [
"SetIamPolicy"
]
},
"TestIamPermissions": {
"methods": [
"TestIamPermissions"
]
},
"UpdateFeature": {
"methods": [
"UpdateFeature"
]
},
"UpdateFeatureGroup": {
"methods": [
"UpdateFeatureGroup"
]
},
"WaitOperation": {
"methods": [
"WaitOperation"
]
}
}
}
}
},
"FeaturestoreOnlineServingService": {
"clients": {
"grpc": {
@@ -918,6 +1215,75 @@
}
}
},
"LlmUtilityService": {
"clients": {
"grpc": {
"libraryClient": "LlmUtilityClient",
"rpcs": {
"CancelOperation": {
"methods": [
"CancelOperation"
]
},
"ComputeTokens": {
"methods": [
"ComputeTokens"
]
},
"CountTokens": {
"methods": [
"CountTokens"
]
},
"DeleteOperation": {
"methods": [
"DeleteOperation"
]
},
"GetIamPolicy": {
"methods": [
"GetIamPolicy"
]
},
"GetLocation": {
"methods": [
"GetLocation"
]
},
"GetOperation": {
"methods": [
"GetOperation"
]
},
"ListLocations": {
"methods": [
"ListLocations"
]
},
"ListOperations": {
"methods": [
"ListOperations"
]
},
"SetIamPolicy": {
"methods": [
"SetIamPolicy"
]
},
"TestIamPermissions": {
"methods": [
"TestIamPermissions"
]
},
"WaitOperation": {
"methods": [
"WaitOperation"
]
}
}
}
}
},
"MatchService": {
"clients": {
"grpc": {
553 changes: 553 additions & 0 deletions aiplatform/apiv1/llm_utility_client.go

Large diffs are not rendered by default.

353 changes: 353 additions & 0 deletions aiplatform/apiv1/llm_utility_client_example_test.go
223 changes: 120 additions & 103 deletions aiplatform/apiv1beta1/aiplatformpb/custom_job.pb.go

Large diffs are not rendered by default.

55 changes: 39 additions & 16 deletions aiplatform/apiv1beta1/aiplatformpb/feature.pb.go
678 changes: 436 additions & 242 deletions aiplatform/apiv1beta1/aiplatformpb/feature_online_store_service.pb.go

Large diffs are not rendered by default.

18 changes: 15 additions & 3 deletions aiplatform/apiv1beta1/aiplatformpb/featurestore_service.pb.go
395 changes: 324 additions & 71 deletions aiplatform/apiv1beta1/aiplatformpb/index.pb.go

Large diffs are not rendered by default.

9 changes: 5 additions & 4 deletions aiplatform/apiv1beta1/aiplatformpb/index_endpoint.pb.go
470 changes: 470 additions & 0 deletions aiplatform/apiv1beta1/aiplatformpb/llm_utility_service.pb.go

Large diffs are not rendered by default.

184 changes: 98 additions & 86 deletions aiplatform/apiv1beta1/aiplatformpb/machine_resources.pb.go

Large diffs are not rendered by default.

397 changes: 326 additions & 71 deletions aiplatform/apiv1beta1/aiplatformpb/model.pb.go

Large diffs are not rendered by default.

642 changes: 431 additions & 211 deletions aiplatform/apiv1beta1/aiplatformpb/pipeline_service.pb.go

Large diffs are not rendered by default.

1,318 changes: 825 additions & 493 deletions aiplatform/apiv1beta1/aiplatformpb/study.pb.go

Large diffs are not rendered by default.

64 changes: 64 additions & 0 deletions aiplatform/apiv1beta1/auxiliary.go
134 changes: 134 additions & 0 deletions aiplatform/apiv1beta1/gapic_metadata.json
Original file line number Diff line number Diff line change
@@ -2559,6 +2559,130 @@
}
}
},
"LlmUtilityService": {
"clients": {
"grpc": {
"libraryClient": "LlmUtilityClient",
"rpcs": {
"CancelOperation": {
"methods": [
"CancelOperation"
]
},
"ComputeTokens": {
"methods": [
"ComputeTokens"
]
},
"DeleteOperation": {
"methods": [
"DeleteOperation"
]
},
"GetIamPolicy": {
"methods": [
"GetIamPolicy"
]
},
"GetLocation": {
"methods": [
"GetLocation"
]
},
"GetOperation": {
"methods": [
"GetOperation"
]
},
"ListLocations": {
"methods": [
"ListLocations"
]
},
"ListOperations": {
"methods": [
"ListOperations"
]
},
"SetIamPolicy": {
"methods": [
"SetIamPolicy"
]
},
"TestIamPermissions": {
"methods": [
"TestIamPermissions"
]
},
"WaitOperation": {
"methods": [
"WaitOperation"
]
}
}
},
"rest": {
"libraryClient": "LlmUtilityClient",
"rpcs": {
"CancelOperation": {
"methods": [
"CancelOperation"
]
},
"ComputeTokens": {
"methods": [
"ComputeTokens"
]
},
"DeleteOperation": {
"methods": [
"DeleteOperation"
]
},
"GetIamPolicy": {
"methods": [
"GetIamPolicy"
]
},
"GetLocation": {
"methods": [
"GetLocation"
]
},
"GetOperation": {
"methods": [
"GetOperation"
]
},
"ListLocations": {
"methods": [
"ListLocations"
]
},
"ListOperations": {
"methods": [
"ListOperations"
]
},
"SetIamPolicy": {
"methods": [
"SetIamPolicy"
]
},
"TestIamPermissions": {
"methods": [
"TestIamPermissions"
]
},
"WaitOperation": {
"methods": [
"WaitOperation"
]
}
}
}
}
},
"MatchService": {
"clients": {
"grpc": {
@@ -3858,6 +3982,11 @@
"grpc": {
"libraryClient": "PipelineClient",
"rpcs": {
"BatchDeletePipelineJobs": {
"methods": [
"BatchDeletePipelineJobs"
]
},
"CancelOperation": {
"methods": [
"CancelOperation"
@@ -3963,6 +4092,11 @@
"rest": {
"libraryClient": "PipelineClient",
"rpcs": {
"BatchDeletePipelineJobs": {
"methods": [
"BatchDeletePipelineJobs"
]
},
"CancelOperation": {
"methods": [
"CancelOperation"
1,304 changes: 1,304 additions & 0 deletions aiplatform/apiv1beta1/llm_utility_client.go

Large diffs are not rendered by default.

345 changes: 345 additions & 0 deletions aiplatform/apiv1beta1/llm_utility_client_example_test.go
Loading