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: Add PrivateEndpoint class and HTTP methods #1033

Merged
merged 176 commits into from
Jun 28, 2022
Merged

feat: Add PrivateEndpoint class and HTTP methods #1033

merged 176 commits into from
Jun 28, 2022

Conversation

vinnysenthil
Copy link
Contributor

@vinnysenthil vinnysenthil commented Feb 23, 2022

Summary of Changes

  • Adds PrivateEndpoint class which subclasses Endpoint
  • New create, deploy methods to specify network
  • Adds new predict, explain and health_check methods to make unauthenticated REST HTTP calls to model server within network
  • Add network to aiplatform.init() and associated classes, reading the field in Model.deploy & Endpoint.create
  • Add urllib3 dependency requirement to library
  • Add unit tests and system tests for PrivateEndpoint

Fixes #708, b/194725522 , b/211371627 (partial), b/220936610, b/230003208, b/233758779 🦕

@nayaknishant nayaknishant added the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Mar 22, 2022
@yoshi-kokoro yoshi-kokoro removed the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Mar 22, 2022
@nayaknishant nayaknishant added the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Jun 16, 2022
@yoshi-kokoro yoshi-kokoro removed the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Jun 16, 2022
@nayaknishant nayaknishant added the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Jun 21, 2022
@yoshi-kokoro yoshi-kokoro removed the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Jun 21, 2022
@nayaknishant nayaknishant added the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Jun 22, 2022
@yoshi-kokoro yoshi-kokoro removed the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Jun 22, 2022
google/cloud/aiplatform/models.py Outdated Show resolved Hide resolved
google/cloud/aiplatform/models.py Outdated Show resolved Hide resolved
google/cloud/aiplatform/models.py Outdated Show resolved Hide resolved
@nayaknishant nayaknishant added the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Jun 27, 2022
@yoshi-kokoro yoshi-kokoro removed the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Jun 27, 2022
@nayaknishant nayaknishant added the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Jun 28, 2022
@yoshi-kokoro yoshi-kokoro removed the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Jun 28, 2022
@nayaknishant nayaknishant added the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Jun 28, 2022
@yoshi-kokoro yoshi-kokoro removed the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Jun 28, 2022
@nayaknishant nayaknishant merged commit 425a32f into main Jun 28, 2022
@nayaknishant nayaknishant deleted the private-ep branch June 28, 2022 22:23
sararob pushed a commit to sararob/python-aiplatform that referenced this pull request Jun 29, 2022
* Add `network` to SDK initializer

* Add urllib3 to library requirements

* Fix network init() docstrings

* Update Model/Endpoint docs to use top namespace

* Add core Private Endpoint wrapper

* Drop logs, add URI props

* Lint, update typing for Prediction class

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* Add bug refs, test stubs, minor fixes

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* test commit

* adding examples, slight changes

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* adding deployed model check:
q

* adding deployed model check

* minor changes, adding error catching for explanations

* testing to get other fields

* removing edge case

* adding print statement for debugging explain

* using GET instead of POST for explain

* using GET instead of POST for explain

* using GET instead of POST for explain

* removing explain for now, adding model id

* removing explain for now, adding model id

* removing explain for now, adding model id

* cleaned and added docstrings

* adding explain back

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* removing explain until working

* cleaning up docstrings

* create test done, working on predict

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* added unit tests for PrivateEndpoint

* test debugging

* fixing unit tests

* adding fixes

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* formatting encryption_spec_key_name

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* fixing comments

* adjusting traffic for private endpoint

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* adjusting traffic for private endpoint

* adjusting traffic for private endpoint

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* adding fixes

* added delete for private Endpoint

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* fixing traffic percentage

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* using network instead of class, moving exceptions

* cleaning up docstrings

* adding fixes, delete testing

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* adding fixes

* adding Raises section to private Endpoint docstrings

* added private Endpoint check in init, added to testing

* added private Endpoint check in init, added to testing

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* adding try/except inside functions, fixing tests, misc fixes

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* reverting _validate_deploy_args method

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* fixed type hint

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* lint issue

* lint issue

* adding system tests

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* adding system test changes

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* adding shared_state to system test

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* trying outside import

* putting imports back inside

* added system test changes

* added system test changes

* adding fixes

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* fixing system test and init fixes

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* adding fixes

* adding fixes

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* adding fixes

* adding fixes

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* added global network arg to associated classes

* style formatting

* docstring fixes

* adding extra info to network arg docstring

* adding extra info to network arg docstring

* removing changes to noxfile

* reverting noxfile

* reverting noxfile

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* reverting job files to not include global network

* reverting initalizer to not include global network config

* removing global network config

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* added fix and test to models

* added fix and test to models

* simplifying delete method

Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
Co-authored-by: nayaknishant <nishantnayak@google.com>
Co-authored-by: sasha-gitg <44654632+sasha-gitg@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
api: vertex-ai Issues related to the googleapis/python-aiplatform API. size: xl Pull request size is extra large.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Private endpoints are not supported
7 participants