From a0103c5d56e099393ab7de56d1903e5f632da201 Mon Sep 17 00:00:00 2001 From: "gcf-owl-bot[bot]" <78513119+gcf-owl-bot[bot]@users.noreply.github.com> Date: Mon, 6 Nov 2023 14:20:42 -0800 Subject: [PATCH] Copybara import of the project: MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit -- 222f6ead640d3895646fd74ab264f265704742d1 by Owl Bot : feat: add UpdateExplanationDataset to aiplatform PiperOrigin-RevId: 541213716 Source-Link: https://github.com/googleapis/googleapis/commit/4a858c4e033098aaecd29dc9302f406d9eb39684 Source-Link: https://github.com/googleapis/googleapis-gen/commit/e4c3eab1fda49c5d9619bb82872a0a7e81f83528 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiZTRjM2VhYjFmZGE0OWM1ZDk2MTliYjgyODcyYTBhN2U4MWY4MzUyOCJ9 -- 75bf15d7eca8342b2d1f95aff153e57a6e4a120a by Owl Bot : 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- c8f2c7e5841c94080f30715eef278952214ea7de by Owl Bot : chore: Generate Ruby documentai wrapper from v1 instead of v1beta3 PiperOrigin-RevId: 542603310 Source-Link: https://github.com/googleapis/googleapis/commit/45001d80e079b14e4c808a93f99849a9325835c0 Source-Link: https://github.com/googleapis/googleapis-gen/commit/ad6882bb648d1170a5414e0e9999c84a43f515f0 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiYWQ2ODgyYmI2NDhkMTE3MGE1NDE0ZTBlOTk5OWM4NGE0M2Y1MTVmMCJ9 -- df2e2d6f5e529714118e51eb04ad6c9811df94dd by Owl Bot : 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- eb4f7ab64b9d2391adb54b504b90259379a08c27 by Owl Bot : docs: tightened requirements on cloud storage subscription filename suffixes PiperOrigin-RevId: 543503563 Source-Link: https://github.com/googleapis/googleapis/commit/212ecef96e3ea7c67465ff3f52680301ad0ea1f9 Source-Link: https://github.com/googleapis/googleapis-gen/commit/532d27eea6ff128fdcbb3332176e894e76dcc685 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNTMyZDI3ZWVhNmZmMTI4ZmRjYmIzMzMyMTc2ZTg5NGU3NmRjYzY4NSJ9 -- 8b9d874407ebb8b1603429ca2ccb8212b448eb56 by Owl Bot : 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 1ff05f7fded898f9b59d3402b73c7596af0fda19 by Owl Bot : fix: Add async context manager return types chore: Mock return_value should not populate oneof message fields chore: Support snippet generation for services that only support REST transport chore: Update gapic-generator-python to v1.11.0 PiperOrigin-RevId: 545430278 Source-Link: https://github.com/googleapis/googleapis/commit/601b5326107eeb74800b426d1f9933faa233258a Source-Link: https://github.com/googleapis/googleapis-gen/commit/b3f18d0f6560a855022fd058865e7620479d7af9 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiYjNmMThkMGY2NTYwYTg1NTAyMmZkMDU4ODY1ZTc2MjA0NzlkN2FmOSJ9 -- 79469acf5f4554ed68d11a07aa5b8d72f6b97d06 by Owl Bot : 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 9c3dfe85da0aeafb6edea9c9b40a065aafa21247 by Owl Bot : chore: Update gapic-generator-python to v1.11.1 PiperOrigin-RevId: 545769621 Source-Link: https://github.com/googleapis/googleapis/commit/cc2f031396e0cda5dd89f191493fe9eddff47233 Source-Link: https://github.com/googleapis/googleapis-gen/commit/1a46ae7652ef98a86c2bc232b6da0412e9f30bdb Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMWE0NmFlNzY1MmVmOThhODZjMmJjMjMyYjZkYTA0MTJlOWYzMGJkYiJ9 -- f221309d4d81139d65e69e2f2fd7fd79c37d230f by Owl Bot : 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 043107e4022d97566e0deb31877b6cb4c092bdbe by Owl Bot : feat: added GitIntegrationSettings to the Agent PiperOrigin-RevId: 546355900 Source-Link: https://github.com/googleapis/googleapis/commit/dc3d3a2bf6b082b22fefa27119caaaf74e2fce8e Source-Link: https://github.com/googleapis/googleapis-gen/commit/2480a9f8bfbc7987f583d70e199584a46b52133f Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMjQ4MGE5ZjhiZmJjNzk4N2Y1ODNkNzBlMTk5NTg0YTQ2YjUyMTMzZiJ9 -- 5112a1a767467861040ff7f1f93f4fa04e808477 by Owl Bot : 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 57b17e5d95f378bb9a10f94e5c66f83ebfedf7d5 by Owl Bot : chore: Update gapic-generator-python to v1.11.2 PiperOrigin-RevId: 546510849 Source-Link: https://github.com/googleapis/googleapis/commit/736073ad9a9763a170eceaaa54519bcc0ea55a5e Source-Link: https://github.com/googleapis/googleapis-gen/commit/deb64e8ec19d141e31089fe932b3a997ad541c4d Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiZGViNjRlOGVjMTlkMTQxZTMxMDg5ZmU5MzJiM2E5OTdhZDU0MWM0ZCJ9 -- 195ff4e30e442bf440fb2343db0fc5d03fd1ac31 by Owl Bot : 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 4c05c4104eeb05c3bad280c7438b8c9759429278 by Owl Bot : chore: Update gapic-generator-python to v1.11.3 PiperOrigin-RevId: 546899192 Source-Link: https://github.com/googleapis/googleapis/commit/e6b16918b98fe1a35f725b56537354f22b6cdc48 Source-Link: https://github.com/googleapis/googleapis-gen/commit/0b3917c421cbda7fcb67092e16c33f3ea46f4bc7 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMGIzOTE3YzQyMWNiZGE3ZmNiNjcwOTJlMTZjMzNmM2VhNDZmNGJjNyJ9 -- e0cd53327bc3c360b0fcabb5d11b9b36794af78a by Owl Bot : 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 6e490f4c7efa741ac1bf6e067b6f48a75a48cb31 by Owl Bot : feat: Add routing information in Cloud Build GRPC clients PiperOrigin-RevId: 547283196 Source-Link: https://github.com/googleapis/googleapis/commit/64d54ff78cfe7cdc2f4b95717dc9afa3ef32a29a Source-Link: https://github.com/googleapis/googleapis-gen/commit/1de18221109b4a025c0b2e8c25324cb01b6df626 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMWRlMTgyMjExMDliNGEwMjVjMGIyZThjMjUzMjRjYjAxYjZkZjYyNiJ9 -- 57a75ff402a4c3d19889cc85c294db7364953d2b by Owl Bot : 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 1f1e8a469dc7a220fb288a54cdb6fb4506fa17e0 by Owl Bot : feat: add data_item_count to Dataset feat: add DeleteSavedQuery to DatasetService feat: add JOB_STATE_PARTIALLY_SUCCEEDED to JobState feat: add pipeline_job to Model feat: add reserved_ip_ranges to PipelineJob PiperOrigin-RevId: 547343567 Source-Link: https://github.com/googleapis/googleapis/commit/72b20cc483d055fd061ce9857e7ade6482ec8056 Source-Link: https://github.com/googleapis/googleapis-gen/commit/4d84e76df6f5a361eca3573a0f69221b421e7359 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNGQ4NGU3NmRmNmY1YTM2MWVjYTM1NzNhMGY2OTIyMWI0MjFlNzM1OSJ9 -- 61d7cf5dead4299f0d5457450e582f7384c7825e by Owl Bot : 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 330ca6bb9bead38189bec8696550c3fb6dfc300c by Owl Bot : feat: add data_item_count to Dataset feat: add DeleteSavedQuery to DatasetService feat: add ExportEndpointOperationMetadata and ExportEndpointResponse to Endpoint feat: add JOB_STATE_PARTIALLY_SUCCEEDED to JobState feat: add pipeline_job to Model feat: add persistent_resource.proto and persistent_resource_service.proto feat: add reserved_ip_ranges to PipelineJob feat: add ReadTensorboardSize to TensorboardService PiperOrigin-RevId: 547519544 Source-Link: https://github.com/googleapis/googleapis/commit/4f40e536783917770dc9b1ccd286d10c8430c50e Source-Link: https://github.com/googleapis/googleapis-gen/commit/cf937c7ecbe8ceeaa031f1389934149bdadf04be Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiY2Y5MzdjN2VjYmU4Y2VlYWEwMzFmMTM4OTkzNDE0OWJkYWRmMDRiZSJ9 -- acb959dc9d28c20857176aac4b7f90b8c5de91ff by Owl Bot : 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 216a83e289cd62765091dc28a65d43a564925549 by Owl Bot : chore: Update gapic-generator-python to v1.11.4 PiperOrigin-RevId: 547897126 Source-Link: https://github.com/googleapis/googleapis/commit/c09c75e087d8f9a2d466b4aaad7dd2926b5ead5a Source-Link: https://github.com/googleapis/googleapis-gen/commit/45e0ec4343517cd0aa66b5ca64232a1802c2f945 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNDVlMGVjNDM0MzUxN2NkMGFhNjZiNWNhNjQyMzJhMTgwMmMyZjk0NSJ9 -- 09bc5c60c2cb45a4c73c1a0d137355225174c510 by Owl Bot : 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- ec20ac38b2c7c0ff35fbb42b341a4faf70670eed by Owl Bot : docs: small documentation updates chore: add enums for additional platform logs PiperOrigin-RevId: 549453285 Source-Link: https://github.com/googleapis/googleapis/commit/b9fb22c5183a7d8753a7e2c633b4f2e50da5b50a Source-Link: https://github.com/googleapis/googleapis-gen/commit/ea3ae98030b5e04e7288a2a9bd3eca81663491be Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiZWEzYWU5ODAzMGI1ZTA0ZTcyODhhMmE5YmQzZWNhODE2NjM0OTFiZSJ9 -- 8ff3cd7e6e5f739b8384c602148a67c758925baf by Owl Bot : 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 8e541cf78323b8b9c93048ba7c26beef9190c52c by Owl Bot : feat: ScheduleService (schedule_service.proto) creates and manages Schedule resources to launch scheduled pipelines runs feat: Schedule (schedule.proto) periodically schedules runs to make API calls PiperOrigin-RevId: 550058610 Source-Link: https://github.com/googleapis/googleapis/commit/5b24e63727233503ff65adada22af98365fd37f4 Source-Link: https://github.com/googleapis/googleapis-gen/commit/4c5f6f9c4292c14287217b56f2dcb04cd7727832 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNGM1ZjZmOWM0MjkyYzE0Mjg3MjE3YjU2ZjJkY2IwNGNkNzcyNzgzMiJ9 -- 8173e334d97e7711e5d9fb7c069657ac16e279d1 by Owl Bot : 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 6a3491c67d8c430c0fb93e710db8e40af054341d by Owl Bot : feat: add policy_name to PlacementPolicy message within a node pool feat: add support for HostMaintenancePolicy feat: add support for AdditionalPodNetworkConfig and AdditionalNodeNetworkConfig feat: add enable_multi_networking to NetworkConfig PiperOrigin-RevId: 550168784 Source-Link: https://github.com/googleapis/googleapis/commit/b53aeac1895ce28b78a8295ee6e04187f4268f43 Source-Link: https://github.com/googleapis/googleapis-gen/commit/6e0edf48c9d991a2d02aa94e723bbd07e45e14d4 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNmUwZWRmNDhjOWQ5OTFhMmQwMmFhOTRlNzIzYmJkMDdlNDVlMTRkNCJ9 -- 72fbdb994f7a4960034b1ac4a0fba77aeac39d2b by Owl Bot : 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- ef2bc05d3b9d7de71a9c2576b7b0862d70d199d1 by nayaknishant : reverting Owlbot flake8 changes -- ddfe45068187dfac28c1f318fa469f67aac77d14 by nayaknishant : removing merge HEAD comment COPYBARA_INTEGRATE_REVIEW=https://github.com/googleapis/python-aiplatform/pull/2810 from googleapis:owl-bot-copy ddfe45068187dfac28c1f318fa469f67aac77d14 PiperOrigin-RevId: 579956341 --- docs/aiplatform_v1/services_.rst | 23 +++++ docs/aiplatform_v1/types_.rst | 6 ++ docs/aiplatform_v1beta1/services_.rst | 28 +++++++ docs/aiplatform_v1beta1/types_.rst | 6 ++ docs/definition_v1/services_.rst | 4 + docs/definition_v1/types_.rst | 6 ++ docs/definition_v1beta1/services_.rst | 4 + docs/definition_v1beta1/types_.rst | 6 ++ docs/instance_v1/services_.rst | 4 + docs/instance_v1/types_.rst | 6 ++ docs/instance_v1beta1/services_.rst | 4 + docs/instance_v1beta1/types_.rst | 6 ++ docs/params_v1/services_.rst | 4 + docs/params_v1/types_.rst | 6 ++ docs/params_v1beta1/services_.rst | 4 + docs/params_v1beta1/types_.rst | 6 ++ docs/prediction_v1/services_.rst | 4 + docs/prediction_v1/types_.rst | 6 ++ docs/prediction_v1beta1/services_.rst | 4 + docs/prediction_v1beta1/types_.rst | 6 ++ .../cloud/aiplatform_v1/gapic_metadata.json | 84 +++++++++++++++++++ .../services/migration_service/client.py | 18 ++-- google/cloud/aiplatform_v1/types/__init__.py | 24 ++++++ .../feature_registry_service/async_client.py | 2 +- .../feature_registry_service/client.py | 2 +- .../services/migration_service/client.py | 26 +++++- .../types/feature_online_store.py | 46 ++++++++++ .../types/prediction_service.py | 47 +++++++++++ .../aiplatform_v1/test_migration_service.py | 26 +++--- ...test_feature_online_store_admin_service.py | 1 + .../test_migration_service.py | 72 +++++++++++++--- 31 files changed, 455 insertions(+), 36 deletions(-) create mode 100644 docs/aiplatform_v1/services_.rst create mode 100644 docs/aiplatform_v1/types_.rst create mode 100644 docs/aiplatform_v1beta1/services_.rst create mode 100644 docs/aiplatform_v1beta1/types_.rst create mode 100644 docs/definition_v1/services_.rst create mode 100644 docs/definition_v1/types_.rst create mode 100644 docs/definition_v1beta1/services_.rst create mode 100644 docs/definition_v1beta1/types_.rst create mode 100644 docs/instance_v1/services_.rst create mode 100644 docs/instance_v1/types_.rst create mode 100644 docs/instance_v1beta1/services_.rst create mode 100644 docs/instance_v1beta1/types_.rst create mode 100644 docs/params_v1/services_.rst create mode 100644 docs/params_v1/types_.rst create mode 100644 docs/params_v1beta1/services_.rst create mode 100644 docs/params_v1beta1/types_.rst create mode 100644 docs/prediction_v1/services_.rst create mode 100644 docs/prediction_v1/types_.rst create mode 100644 docs/prediction_v1beta1/services_.rst create mode 100644 docs/prediction_v1beta1/types_.rst diff --git a/docs/aiplatform_v1/services_.rst b/docs/aiplatform_v1/services_.rst new file mode 100644 index 0000000000..93afd80841 --- /dev/null +++ b/docs/aiplatform_v1/services_.rst @@ -0,0 +1,23 @@ +Services for Google Cloud Aiplatform v1 API +=========================================== +.. toctree:: + :maxdepth: 2 + + dataset_service + endpoint_service + featurestore_online_serving_service + featurestore_service + index_endpoint_service + index_service + job_service + match_service + metadata_service + migration_service + model_garden_service + model_service + pipeline_service + prediction_service + schedule_service + specialist_pool_service + tensorboard_service + vizier_service diff --git a/docs/aiplatform_v1/types_.rst b/docs/aiplatform_v1/types_.rst new file mode 100644 index 0000000000..da19f0e39e --- /dev/null +++ b/docs/aiplatform_v1/types_.rst @@ -0,0 +1,6 @@ +Types for Google Cloud Aiplatform v1 API +======================================== + +.. automodule:: google.cloud.aiplatform_v1.types + :members: + :show-inheritance: diff --git a/docs/aiplatform_v1beta1/services_.rst b/docs/aiplatform_v1beta1/services_.rst new file mode 100644 index 0000000000..809b60c29c --- /dev/null +++ b/docs/aiplatform_v1beta1/services_.rst @@ -0,0 +1,28 @@ +Services for Google Cloud Aiplatform v1beta1 API +================================================ +.. toctree:: + :maxdepth: 2 + + dataset_service + deployment_resource_pool_service + endpoint_service + feature_online_store_admin_service + feature_online_store_service + feature_registry_service + featurestore_online_serving_service + featurestore_service + index_endpoint_service + index_service + job_service + match_service + metadata_service + migration_service + model_garden_service + model_service + persistent_resource_service + pipeline_service + prediction_service + schedule_service + specialist_pool_service + tensorboard_service + vizier_service diff --git a/docs/aiplatform_v1beta1/types_.rst b/docs/aiplatform_v1beta1/types_.rst new file mode 100644 index 0000000000..19bab68ada --- /dev/null +++ b/docs/aiplatform_v1beta1/types_.rst @@ -0,0 +1,6 @@ +Types for Google Cloud Aiplatform v1beta1 API +============================================= + +.. automodule:: google.cloud.aiplatform_v1beta1.types + :members: + :show-inheritance: diff --git a/docs/definition_v1/services_.rst b/docs/definition_v1/services_.rst new file mode 100644 index 0000000000..ba6b1940e8 --- /dev/null +++ b/docs/definition_v1/services_.rst @@ -0,0 +1,4 @@ +Services for Google Cloud Aiplatform V1 Schema Trainingjob Definition v1 API +============================================================================ +.. toctree:: + :maxdepth: 2 diff --git a/docs/definition_v1/types_.rst b/docs/definition_v1/types_.rst new file mode 100644 index 0000000000..0add260eee --- /dev/null +++ b/docs/definition_v1/types_.rst @@ -0,0 +1,6 @@ +Types for Google Cloud Aiplatform V1 Schema Trainingjob Definition v1 API +========================================================================= + +.. automodule:: google.cloud.aiplatform.v1.schema.trainingjob.definition_v1.types + :members: + :show-inheritance: diff --git a/docs/definition_v1beta1/services_.rst b/docs/definition_v1beta1/services_.rst new file mode 100644 index 0000000000..5f1ed5f2b7 --- /dev/null +++ b/docs/definition_v1beta1/services_.rst @@ -0,0 +1,4 @@ +Services for Google Cloud Aiplatform V1beta1 Schema Trainingjob Definition v1beta1 API +====================================================================================== +.. toctree:: + :maxdepth: 2 diff --git a/docs/definition_v1beta1/types_.rst b/docs/definition_v1beta1/types_.rst new file mode 100644 index 0000000000..3f351d03fc --- /dev/null +++ b/docs/definition_v1beta1/types_.rst @@ -0,0 +1,6 @@ +Types for Google Cloud Aiplatform V1beta1 Schema Trainingjob Definition v1beta1 API +=================================================================================== + +.. automodule:: google.cloud.aiplatform.v1beta1.schema.trainingjob.definition_v1beta1.types + :members: + :show-inheritance: diff --git a/docs/instance_v1/services_.rst b/docs/instance_v1/services_.rst new file mode 100644 index 0000000000..50c011c69a --- /dev/null +++ b/docs/instance_v1/services_.rst @@ -0,0 +1,4 @@ +Services for Google Cloud Aiplatform V1 Schema Predict Instance v1 API +====================================================================== +.. toctree:: + :maxdepth: 2 diff --git a/docs/instance_v1/types_.rst b/docs/instance_v1/types_.rst new file mode 100644 index 0000000000..81597999f2 --- /dev/null +++ b/docs/instance_v1/types_.rst @@ -0,0 +1,6 @@ +Types for Google Cloud Aiplatform V1 Schema Predict Instance v1 API +=================================================================== + +.. automodule:: google.cloud.aiplatform.v1.schema.predict.instance_v1.types + :members: + :show-inheritance: diff --git a/docs/instance_v1beta1/services_.rst b/docs/instance_v1beta1/services_.rst new file mode 100644 index 0000000000..941dbcca59 --- /dev/null +++ b/docs/instance_v1beta1/services_.rst @@ -0,0 +1,4 @@ +Services for Google Cloud Aiplatform V1beta1 Schema Predict Instance v1beta1 API +================================================================================ +.. toctree:: + :maxdepth: 2 diff --git a/docs/instance_v1beta1/types_.rst b/docs/instance_v1beta1/types_.rst new file mode 100644 index 0000000000..c52ae4800c --- /dev/null +++ b/docs/instance_v1beta1/types_.rst @@ -0,0 +1,6 @@ +Types for Google Cloud Aiplatform V1beta1 Schema Predict Instance v1beta1 API +============================================================================= + +.. automodule:: google.cloud.aiplatform.v1beta1.schema.predict.instance_v1beta1.types + :members: + :show-inheritance: diff --git a/docs/params_v1/services_.rst b/docs/params_v1/services_.rst new file mode 100644 index 0000000000..bf08ea6e98 --- /dev/null +++ b/docs/params_v1/services_.rst @@ -0,0 +1,4 @@ +Services for Google Cloud Aiplatform V1 Schema Predict Params v1 API +==================================================================== +.. toctree:: + :maxdepth: 2 diff --git a/docs/params_v1/types_.rst b/docs/params_v1/types_.rst new file mode 100644 index 0000000000..afc962c218 --- /dev/null +++ b/docs/params_v1/types_.rst @@ -0,0 +1,6 @@ +Types for Google Cloud Aiplatform V1 Schema Predict Params v1 API +================================================================= + +.. automodule:: google.cloud.aiplatform.v1.schema.predict.params_v1.types + :members: + :show-inheritance: diff --git a/docs/params_v1beta1/services_.rst b/docs/params_v1beta1/services_.rst new file mode 100644 index 0000000000..b3b897a0f4 --- /dev/null +++ b/docs/params_v1beta1/services_.rst @@ -0,0 +1,4 @@ +Services for Google Cloud Aiplatform V1beta1 Schema Predict Params v1beta1 API +============================================================================== +.. toctree:: + :maxdepth: 2 diff --git a/docs/params_v1beta1/types_.rst b/docs/params_v1beta1/types_.rst new file mode 100644 index 0000000000..ce7a29cb01 --- /dev/null +++ b/docs/params_v1beta1/types_.rst @@ -0,0 +1,6 @@ +Types for Google Cloud Aiplatform V1beta1 Schema Predict Params v1beta1 API +=========================================================================== + +.. automodule:: google.cloud.aiplatform.v1beta1.schema.predict.params_v1beta1.types + :members: + :show-inheritance: diff --git a/docs/prediction_v1/services_.rst b/docs/prediction_v1/services_.rst new file mode 100644 index 0000000000..ad6f034387 --- /dev/null +++ b/docs/prediction_v1/services_.rst @@ -0,0 +1,4 @@ +Services for Google Cloud Aiplatform V1 Schema Predict Prediction v1 API +======================================================================== +.. toctree:: + :maxdepth: 2 diff --git a/docs/prediction_v1/types_.rst b/docs/prediction_v1/types_.rst new file mode 100644 index 0000000000..739ca93799 --- /dev/null +++ b/docs/prediction_v1/types_.rst @@ -0,0 +1,6 @@ +Types for Google Cloud Aiplatform V1 Schema Predict Prediction v1 API +===================================================================== + +.. automodule:: google.cloud.aiplatform.v1.schema.predict.prediction_v1.types + :members: + :show-inheritance: diff --git a/docs/prediction_v1beta1/services_.rst b/docs/prediction_v1beta1/services_.rst new file mode 100644 index 0000000000..6de5e17520 --- /dev/null +++ b/docs/prediction_v1beta1/services_.rst @@ -0,0 +1,4 @@ +Services for Google Cloud Aiplatform V1beta1 Schema Predict Prediction v1beta1 API +================================================================================== +.. toctree:: + :maxdepth: 2 diff --git a/docs/prediction_v1beta1/types_.rst b/docs/prediction_v1beta1/types_.rst new file mode 100644 index 0000000000..cdbe7f2842 --- /dev/null +++ b/docs/prediction_v1beta1/types_.rst @@ -0,0 +1,6 @@ +Types for Google Cloud Aiplatform V1beta1 Schema Predict Prediction v1beta1 API +=============================================================================== + +.. automodule:: google.cloud.aiplatform.v1beta1.schema.predict.prediction_v1beta1.types + :members: + :show-inheritance: diff --git a/google/cloud/aiplatform_v1/gapic_metadata.json b/google/cloud/aiplatform_v1/gapic_metadata.json index 481256e086..7137ea8899 100644 --- a/google/cloud/aiplatform_v1/gapic_metadata.json +++ b/google/cloud/aiplatform_v1/gapic_metadata.json @@ -1975,6 +1975,90 @@ } } }, + "ScheduleService": { + "clients": { + "grpc": { + "libraryClient": "ScheduleServiceClient", + "rpcs": { + "CreateSchedule": { + "methods": [ + "create_schedule" + ] + }, + "DeleteSchedule": { + "methods": [ + "delete_schedule" + ] + }, + "GetSchedule": { + "methods": [ + "get_schedule" + ] + }, + "ListSchedules": { + "methods": [ + "list_schedules" + ] + }, + "PauseSchedule": { + "methods": [ + "pause_schedule" + ] + }, + "ResumeSchedule": { + "methods": [ + "resume_schedule" + ] + }, + "UpdateSchedule": { + "methods": [ + "update_schedule" + ] + } + } + }, + "grpc-async": { + "libraryClient": "ScheduleServiceAsyncClient", + "rpcs": { + "CreateSchedule": { + "methods": [ + "create_schedule" + ] + }, + "DeleteSchedule": { + "methods": [ + "delete_schedule" + ] + }, + "GetSchedule": { + "methods": [ + "get_schedule" + ] + }, + "ListSchedules": { + "methods": [ + "list_schedules" + ] + }, + "PauseSchedule": { + "methods": [ + "pause_schedule" + ] + }, + "ResumeSchedule": { + "methods": [ + "resume_schedule" + ] + }, + "UpdateSchedule": { + "methods": [ + "update_schedule" + ] + } + } + } + } + }, "SpecialistPoolService": { "clients": { "grpc": { diff --git a/google/cloud/aiplatform_v1/services/migration_service/client.py b/google/cloud/aiplatform_v1/services/migration_service/client.py index ec924555fe..6f737a3810 100644 --- a/google/cloud/aiplatform_v1/services/migration_service/client.py +++ b/google/cloud/aiplatform_v1/services/migration_service/client.py @@ -230,40 +230,40 @@ def parse_dataset_path(path: str) -> Dict[str, str]: @staticmethod def dataset_path( project: str, + location: str, dataset: str, ) -> str: """Returns a fully-qualified dataset string.""" - return "projects/{project}/datasets/{dataset}".format( + return "projects/{project}/locations/{location}/datasets/{dataset}".format( project=project, + location=location, dataset=dataset, ) @staticmethod def parse_dataset_path(path: str) -> Dict[str, str]: """Parses a dataset path into its component segments.""" - m = re.match(r"^projects/(?P.+?)/datasets/(?P.+?)$", path) + m = re.match( + r"^projects/(?P.+?)/locations/(?P.+?)/datasets/(?P.+?)$", + path, + ) return m.groupdict() if m else {} @staticmethod def dataset_path( project: str, - location: str, dataset: str, ) -> str: """Returns a fully-qualified dataset string.""" - return "projects/{project}/locations/{location}/datasets/{dataset}".format( + return "projects/{project}/datasets/{dataset}".format( project=project, - location=location, dataset=dataset, ) @staticmethod def parse_dataset_path(path: str) -> Dict[str, str]: """Parses a dataset path into its component segments.""" - m = re.match( - r"^projects/(?P.+?)/locations/(?P.+?)/datasets/(?P.+?)$", - path, - ) + m = re.match(r"^projects/(?P.+?)/datasets/(?P.+?)$", path) return m.groupdict() if m else {} @staticmethod diff --git a/google/cloud/aiplatform_v1/types/__init__.py b/google/cloud/aiplatform_v1/types/__init__.py index 14603a1447..7709740465 100644 --- a/google/cloud/aiplatform_v1/types/__init__.py +++ b/google/cloud/aiplatform_v1/types/__init__.py @@ -559,6 +559,30 @@ from .service_networking import ( PrivateServiceConnectConfig, ) + +from .publisher_model import ( + PublisherModel, +) +from .saved_query import ( + SavedQuery, +) +from .schedule import ( + Schedule, +) +from .schedule_service import ( + CreateScheduleRequest, + DeleteScheduleRequest, + GetScheduleRequest, + ListSchedulesRequest, + ListSchedulesResponse, + PauseScheduleRequest, + ResumeScheduleRequest, + UpdateScheduleRequest, +) +from .service_networking import ( + PrivateServiceConnectConfig, +) + from .specialist_pool import ( SpecialistPool, ) diff --git a/google/cloud/aiplatform_v1beta1/services/feature_registry_service/async_client.py b/google/cloud/aiplatform_v1beta1/services/feature_registry_service/async_client.py index f64a8e0510..00135fb2f7 100644 --- a/google/cloud/aiplatform_v1beta1/services/feature_registry_service/async_client.py +++ b/google/cloud/aiplatform_v1beta1/services/feature_registry_service/async_client.py @@ -1035,7 +1035,7 @@ async def sample_create_feature(): response, self._client._transport.operations_client, gca_feature.Feature, - metadata_type=feature_registry_service.CreateRegistryFeatureOperationMetadata, + metadata_type=featurestore_service.CreateFeatureOperationMetadata, ) # Done; return the response. diff --git a/google/cloud/aiplatform_v1beta1/services/feature_registry_service/client.py b/google/cloud/aiplatform_v1beta1/services/feature_registry_service/client.py index b7a71b4f6d..c1ef0d14c3 100644 --- a/google/cloud/aiplatform_v1beta1/services/feature_registry_service/client.py +++ b/google/cloud/aiplatform_v1beta1/services/feature_registry_service/client.py @@ -1281,7 +1281,7 @@ def sample_create_feature(): response, self._transport.operations_client, gca_feature.Feature, - metadata_type=feature_registry_service.CreateRegistryFeatureOperationMetadata, + metadata_type=featurestore_service.CreateFeatureOperationMetadata, ) # Done; return the response. diff --git a/google/cloud/aiplatform_v1beta1/services/migration_service/client.py b/google/cloud/aiplatform_v1beta1/services/migration_service/client.py index 55a32d1c1b..31dce2992e 100644 --- a/google/cloud/aiplatform_v1beta1/services/migration_service/client.py +++ b/google/cloud/aiplatform_v1beta1/services/migration_service/client.py @@ -227,6 +227,23 @@ def parse_dataset_path(path: str) -> Dict[str, str]: ) return m.groupdict() if m else {} + @staticmethod + def dataset_path( + project: str, + dataset: str, + ) -> str: + """Returns a fully-qualified dataset string.""" + return "projects/{project}/datasets/{dataset}".format( + project=project, + dataset=dataset, + ) + + @staticmethod + def parse_dataset_path(path: str) -> Dict[str, str]: + """Parses a dataset path into its component segments.""" + m = re.match(r"^projects/(?P.+?)/datasets/(?P.+?)$", path) + return m.groupdict() if m else {} + @staticmethod def dataset_path( project: str, @@ -252,18 +269,23 @@ def parse_dataset_path(path: str) -> Dict[str, str]: @staticmethod def dataset_path( project: str, + location: str, dataset: str, ) -> str: """Returns a fully-qualified dataset string.""" - return "projects/{project}/datasets/{dataset}".format( + return "projects/{project}/locations/{location}/datasets/{dataset}".format( project=project, + location=location, dataset=dataset, ) @staticmethod def parse_dataset_path(path: str) -> Dict[str, str]: """Parses a dataset path into its component segments.""" - m = re.match(r"^projects/(?P.+?)/datasets/(?P.+?)$", path) + m = re.match( + r"^projects/(?P.+?)/locations/(?P.+?)/datasets/(?P.+?)$", + path, + ) return m.groupdict() if m else {} @staticmethod diff --git a/google/cloud/aiplatform_v1beta1/types/feature_online_store.py b/google/cloud/aiplatform_v1beta1/types/feature_online_store.py index e15989a3c5..9f35aa29d6 100644 --- a/google/cloud/aiplatform_v1beta1/types/feature_online_store.py +++ b/google/cloud/aiplatform_v1beta1/types/feature_online_store.py @@ -19,6 +19,7 @@ import proto # type: ignore +from google.cloud.aiplatform_v1beta1.types import service_networking from google.protobuf import timestamp_pb2 # type: ignore @@ -35,6 +36,10 @@ class FeatureOnlineStore(proto.Message): repository for serving ML features and embedding indexes at low latency. The Feature Online Store is a top-level container. + This message has `oneof`_ fields (mutually exclusive fields). + For each oneof, at most one member field can be set at the same time. + Setting any member of the oneof automatically clears all other + members. .. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields @@ -45,6 +50,16 @@ class FeatureOnlineStore(proto.Message): featureValues for all FeatureViews under this FeatureOnlineStore. + This field is a member of `oneof`_ ``storage_type``. + optimized (google.cloud.aiplatform_v1beta1.types.FeatureOnlineStore.Optimized): + Contains settings for the Optimized store that will be + created to serve featureValues for all FeatureViews under + this FeatureOnlineStore. When choose Optimized storage type, + need to set + [PrivateServiceConnectConfig.enable_private_service_connect][google.cloud.aiplatform.v1beta1.PrivateServiceConnectConfig.enable_private_service_connect] + to use private endpoint. Otherwise will use public endpoint + by default. + This field is a member of `oneof`_ ``storage_type``. name (str): Output only. Name of the FeatureOnlineStore. Format: @@ -161,20 +176,45 @@ class AutoScaling(proto.Message): message="FeatureOnlineStore.Bigtable.AutoScaling", ) + class Optimized(proto.Message): + r"""Optimized storage type to replace lightning""" + class DedicatedServingEndpoint(proto.Message): r"""The dedicated serving endpoint for this FeatureOnlineStore. + Only need to set when you choose Optimized storage type or + enable EmbeddingManagement. Will use public endpoint by default. Attributes: public_endpoint_domain_name (str): Output only. This field will be populated with the domain name to use for this FeatureOnlineStore + private_service_connect_config (google.cloud.aiplatform_v1beta1.types.PrivateServiceConnectConfig): + Optional. Private service connect config. If + [PrivateServiceConnectConfig.enable_private_service_connect][google.cloud.aiplatform.v1beta1.PrivateServiceConnectConfig.enable_private_service_connect] + set to true, customers will use private service connection + to send request. Otherwise, the connection will set to + public endpoint. + service_attachment (str): + Output only. The name of the service + attachment resource. Populated if private + service connect is enabled and after + FeatureViewSync is created. """ public_endpoint_domain_name: str = proto.Field( proto.STRING, number=2, ) + private_service_connect_config: service_networking.PrivateServiceConnectConfig = proto.Field( + proto.MESSAGE, + number=3, + message=service_networking.PrivateServiceConnectConfig, + ) + service_attachment: str = proto.Field( + proto.STRING, + number=4, + ) class EmbeddingManagement(proto.Message): r"""Contains settings for embedding management. @@ -198,6 +238,12 @@ class EmbeddingManagement(proto.Message): oneof="storage_type", message=Bigtable, ) + optimized: Optimized = proto.Field( + proto.MESSAGE, + number=12, + oneof="storage_type", + message=Optimized, + ) name: str = proto.Field( proto.STRING, number=1, diff --git a/google/cloud/aiplatform_v1beta1/types/prediction_service.py b/google/cloud/aiplatform_v1beta1/types/prediction_service.py index fd6b807e14..7b9e12444d 100644 --- a/google/cloud/aiplatform_v1beta1/types/prediction_service.py +++ b/google/cloud/aiplatform_v1beta1/types/prediction_service.py @@ -290,6 +290,14 @@ class ExplainRequest(proto.Message): methods to reduce approximate errors; - Using different baselines for explaining the prediction results. + concurrent_explanation_spec_override (MutableMapping[str, google.cloud.aiplatform_v1beta1.types.ExplanationSpecOverride]): + Optional. This field is the same as the one above, but + supports multiple explanations to occur in parallel. The key + can be any string. Each override will be run against the + model, then its explanations will be grouped together. + + Note - these explanations are run **In Addition** to the + default Explanation in the deployed model. deployed_model_id (str): If specified, this ExplainRequest will be served by the chosen DeployedModel, overriding @@ -315,6 +323,14 @@ class ExplainRequest(proto.Message): number=5, message=explanation.ExplanationSpecOverride, ) + concurrent_explanation_spec_override: MutableMapping[ + str, explanation.ExplanationSpecOverride + ] = proto.MapField( + proto.STRING, + proto.MESSAGE, + number=6, + message=explanation.ExplanationSpecOverride, + ) deployed_model_id: str = proto.Field( proto.STRING, number=3, @@ -333,6 +349,10 @@ class ExplainResponse(proto.Message): It has the same number of elements as [instances][google.cloud.aiplatform.v1beta1.ExplainRequest.instances] to be explained. + concurrent_explanations (MutableMapping[str, google.cloud.aiplatform_v1beta1.types.ExplainResponse.ConcurrentExplanation]): + This field stores the results of the + explanations run in parallel with the default + explanation strategy/method. deployed_model_id (str): ID of the Endpoint's DeployedModel that served this explanation. @@ -342,11 +362,38 @@ class ExplainResponse(proto.Message): [PredictResponse.predictions][google.cloud.aiplatform.v1beta1.PredictResponse.predictions]. """ + class ConcurrentExplanation(proto.Message): + r"""This message is a wrapper grouping Concurrent Explanations. + + Attributes: + explanations (MutableSequence[google.cloud.aiplatform_v1beta1.types.Explanation]): + The explanations of the Model's + [PredictResponse.predictions][google.cloud.aiplatform.v1beta1.PredictResponse.predictions]. + + It has the same number of elements as + [instances][google.cloud.aiplatform.v1beta1.ExplainRequest.instances] + to be explained. + """ + + explanations: MutableSequence[explanation.Explanation] = proto.RepeatedField( + proto.MESSAGE, + number=1, + message=explanation.Explanation, + ) + explanations: MutableSequence[explanation.Explanation] = proto.RepeatedField( proto.MESSAGE, number=1, message=explanation.Explanation, ) + concurrent_explanations: MutableMapping[ + str, ConcurrentExplanation + ] = proto.MapField( + proto.STRING, + proto.MESSAGE, + number=4, + message=ConcurrentExplanation, + ) deployed_model_id: str = proto.Field( proto.STRING, number=2, diff --git a/tests/unit/gapic/aiplatform_v1/test_migration_service.py b/tests/unit/gapic/aiplatform_v1/test_migration_service.py index ed1cd95f39..c85deda21c 100644 --- a/tests/unit/gapic/aiplatform_v1/test_migration_service.py +++ b/tests/unit/gapic/aiplatform_v1/test_migration_service.py @@ -2032,19 +2032,22 @@ def test_parse_dataset_path(): def test_dataset_path(): project = "squid" - dataset = "clam" - expected = "projects/{project}/datasets/{dataset}".format( + location = "clam" + dataset = "whelk" + expected = "projects/{project}/locations/{location}/datasets/{dataset}".format( project=project, + location=location, dataset=dataset, ) - actual = MigrationServiceClient.dataset_path(project, dataset) + actual = MigrationServiceClient.dataset_path(project, location, dataset) assert expected == actual def test_parse_dataset_path(): expected = { - "project": "whelk", - "dataset": "octopus", + "project": "octopus", + "location": "oyster", + "dataset": "nudibranch", } path = MigrationServiceClient.dataset_path(**expected) @@ -2054,22 +2057,19 @@ def test_parse_dataset_path(): def test_dataset_path(): - project = "oyster" - location = "nudibranch" - dataset = "cuttlefish" - expected = "projects/{project}/locations/{location}/datasets/{dataset}".format( + project = "cuttlefish" + dataset = "mussel" + expected = "projects/{project}/datasets/{dataset}".format( project=project, - location=location, dataset=dataset, ) - actual = MigrationServiceClient.dataset_path(project, location, dataset) + actual = MigrationServiceClient.dataset_path(project, dataset) assert expected == actual def test_parse_dataset_path(): expected = { - "project": "mussel", - "location": "winkle", + "project": "winkle", "dataset": "nautilus", } path = MigrationServiceClient.dataset_path(**expected) diff --git a/tests/unit/gapic/aiplatform_v1beta1/test_feature_online_store_admin_service.py b/tests/unit/gapic/aiplatform_v1beta1/test_feature_online_store_admin_service.py index 60647315dd..b11bba9ffa 100644 --- a/tests/unit/gapic/aiplatform_v1beta1/test_feature_online_store_admin_service.py +++ b/tests/unit/gapic/aiplatform_v1beta1/test_feature_online_store_admin_service.py @@ -62,6 +62,7 @@ from google.cloud.aiplatform_v1beta1.types import feature_view as gca_feature_view from google.cloud.aiplatform_v1beta1.types import feature_view_sync from google.cloud.aiplatform_v1beta1.types import operation as gca_operation +from google.cloud.aiplatform_v1beta1.types import service_networking from google.cloud.location import locations_pb2 from google.iam.v1 import iam_policy_pb2 # type: ignore from google.iam.v1 import options_pb2 # type: ignore diff --git a/tests/unit/gapic/aiplatform_v1beta1/test_migration_service.py b/tests/unit/gapic/aiplatform_v1beta1/test_migration_service.py index b287e8fe7f..c56b99ac1b 100644 --- a/tests/unit/gapic/aiplatform_v1beta1/test_migration_service.py +++ b/tests/unit/gapic/aiplatform_v1beta1/test_migration_service.py @@ -2034,8 +2034,31 @@ def test_parse_dataset_path(): def test_dataset_path(): project = "squid" - location = "clam" - dataset = "whelk" + dataset = "clam" + expected = "projects/{project}/datasets/{dataset}".format( + project=project, + dataset=dataset, + ) + actual = MigrationServiceClient.dataset_path(project, dataset) + assert expected == actual + + +def test_parse_dataset_path(): + expected = { + "project": "whelk", + "dataset": "octopus", + } + path = MigrationServiceClient.dataset_path(**expected) + + # Check that the path construction is reversible. + actual = MigrationServiceClient.parse_dataset_path(path) + assert expected == actual + + +def test_dataset_path(): + project = "oyster" + location = "nudibranch" + dataset = "cuttlefish" expected = "projects/{project}/locations/{location}/datasets/{dataset}".format( project=project, location=location, @@ -2047,9 +2070,9 @@ def test_dataset_path(): def test_parse_dataset_path(): expected = { - "project": "octopus", - "location": "oyster", - "dataset": "nudibranch", + "project": "mussel", + "location": "winkle", + "dataset": "nautilus", } path = MigrationServiceClient.dataset_path(**expected) @@ -2059,19 +2082,48 @@ def test_parse_dataset_path(): def test_dataset_path(): - project = "cuttlefish" - dataset = "mussel" - expected = "projects/{project}/datasets/{dataset}".format( + project = "scallop" + location = "abalone" + dataset = "squid" + expected = "projects/{project}/locations/{location}/datasets/{dataset}".format( project=project, + location=location, dataset=dataset, ) - actual = MigrationServiceClient.dataset_path(project, dataset) + actual = MigrationServiceClient.dataset_path(project, location, dataset) assert expected == actual def test_parse_dataset_path(): expected = { - "project": "winkle", + "project": "clam", + "location": "whelk", + "dataset": "octopus", + } + path = MigrationServiceClient.dataset_path(**expected) + + # Check that the path construction is reversible. + actual = MigrationServiceClient.parse_dataset_path(path) + assert expected == actual + + +def test_dataset_path(): + project = "oyster" + location = "nudibranch" + dataset = "cuttlefish" + expected = "projects/{project}/locations/{location}/datasets/{dataset}".format( + project=project, + location=location, + dataset=dataset, + ) + actual = MigrationServiceClient.dataset_path(project, location, dataset) + assert expected == actual + + +def test_parse_dataset_path(): + expected = { + "project": "mussel", + "location": "winkle", "dataset": "nautilus", } path = MigrationServiceClient.dataset_path(**expected)