Skip to content

Add enterprise license check for Inference API actions#119893

Merged
dan-rubinstein merged 9 commits intoelastic:mainfrom
dan-rubinstein:license-check
Jan 13, 2025
Merged

Add enterprise license check for Inference API actions#119893
dan-rubinstein merged 9 commits intoelastic:mainfrom
dan-rubinstein:license-check

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

@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 <elasticmachine@users.noreply.github.com>
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 <elasticmachine@users.noreply.github.com>
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 <elasticmachine@users.noreply.github.com>
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

4 participants