Skip to content

Add enterprise license check for Inference API actions #119893

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

Merged
merged 9 commits into from
Jan 13, 2025

Conversation

dan-rubinstein
Copy link
Member

@dan-rubinstein dan-rubinstein commented Jan 9, 2025

The Inference API is meant to only be used by users with an enterprise license agreement. Up to this point we have not enforced this requirement so this change adds license checks to the inference APIs transport actions. The GET/DELETE model APIs and the GET inference usage actions have been excluded from this check as we would like to still allow users who have let their enterprise subscription expire to check their remaining inference endpoints through the GET action and clean them up through the DELETE action. This change also adds integration tests to check that the actions are properly blocking on invalid license types. Note that the tests use the trial license type instead of the enterprise type as it has the same permissions and is the preferred type to use in testing as enterprise licensing requires more complicated setup.

Note: Before this change is merged we should confirm whether the GET inference services API needs the license check. The change currently has it checking for valid license type but there may be some use case in the kibana UI that still needs to make the call even for users who do not have enterprise licenses.

@dan-rubinstein dan-rubinstein added >enhancement :ml Machine learning Team:ML Meta label for the ML team v9.0.0 v8.18.0 labels Jan 9, 2025
@elasticsearchmachine elasticsearchmachine added the external-contributor Pull request authored by a developer outside the Elasticsearch team label Jan 9, 2025
@dan-rubinstein dan-rubinstein removed the external-contributor Pull request authored by a developer outside the Elasticsearch team label Jan 9, 2025
@elasticsearchmachine
Copy link
Collaborator

Hi @dan-rubinstein, I've created a changelog YAML for you.

@dan-rubinstein
Copy link
Member Author

@elasticmachine merge upstream

@dan-rubinstein dan-rubinstein marked this pull request as ready for review January 10, 2025 21:32
@elasticsearchmachine
Copy link
Collaborator

Pinging @elastic/ml-core (Team:ML)

Copy link
Member

@davidkyle davidkyle left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@dan-rubinstein
Copy link
Member Author

@elasticmachine merge upstream

@dan-rubinstein dan-rubinstein added the auto-backport Automatically create backport pull requests when merged label Jan 13, 2025
@dan-rubinstein
Copy link
Member Author

@elasticmachine merge upstream

@dan-rubinstein dan-rubinstein merged commit d79edcb into elastic:main Jan 13, 2025
16 checks passed
@elasticsearchmachine
Copy link
Collaborator

💚 Backport successful

Status Branch Result
8.x

dan-rubinstein added a commit to dan-rubinstein/elasticsearch that referenced this pull request Jan 13, 2025
* Add enterprise license check for Inference API actions

* Update docs/changelog/119893.yaml

* Adding missing plugin to ModelRegistryIT and removing license check from get inference services API

* Fix tests

* Fix basic license test

* Removing unused feature flag from InferenceUpgradeTestCase

---------

Co-authored-by: Elastic Machine <[email protected]>
martijnvg pushed a commit to martijnvg/elasticsearch that referenced this pull request Jan 14, 2025
* Add enterprise license check for Inference API actions

* Update docs/changelog/119893.yaml

* Adding missing plugin to ModelRegistryIT and removing license check from get inference services API

* Fix tests

* Fix basic license test

* Removing unused feature flag from InferenceUpgradeTestCase

---------

Co-authored-by: Elastic Machine <[email protected]>
elasticsearchmachine pushed a commit that referenced this pull request Jan 15, 2025
…0066)

* Add enterprise license check for Inference API actions

* Update docs/changelog/119893.yaml

* Adding missing plugin to ModelRegistryIT and removing license check from get inference services API

* Fix tests

* Fix basic license test

* Removing unused feature flag from InferenceUpgradeTestCase

---------

Co-authored-by: Elastic Machine <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
auto-backport Automatically create backport pull requests when merged >enhancement :ml Machine learning Team:ML Meta label for the ML team v8.18.0 v9.0.0
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants