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

[ML] External inference service rolling upgrade tests #107619

Merged
merged 8 commits into from Apr 24, 2024

Conversation

davidkyle
Copy link
Member

Rolling upgrade tests for OpenAI, Azure, Cohere and Hugging Face. The services are tested by setting the services' url to a local mock web server and mocking the responses.

The tests exposed 1 issue for OpenAI where an older model that does not have the dimensionsSetByUser field throws an validation error.

@davidkyle davidkyle added >test Issues or PRs that are addressing/adding tests :ml Machine learning test-full-bwc Trigger full BWC version matrix tests auto-backport-and-merge Automatically create backport pull requests and merge when ready v8.14.0 v8.15.0 labels Apr 18, 2024
@davidkyle davidkyle requested a review from a team as a code owner April 18, 2024 14:05
@elasticsearchmachine elasticsearchmachine added the Team:ML Meta label for the ML team label Apr 18, 2024
@elasticsearchmachine
Copy link
Collaborator

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

x-pack/plugin/inference/qa/rolling-upgrade/build.gradle Outdated Show resolved Hide resolved
}

@SuppressWarnings("unchecked")
@AwaitsFix(bugUrl = "Cannot set the URL in the tests")
Copy link
Contributor

Choose a reason for hiding this comment

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

Yeah that's a bummer, is there a way to set some config flag for a test build of elasticsearch that we can then expose a url field? That way in production builds we won't allow it

}

@SuppressWarnings("unchecked")
@AwaitsFix(bugUrl = "Cannot set the URL in the tests")
Copy link
Member Author

Choose a reason for hiding this comment

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

The Azure OpenAI config does not allow a user settable URL which means it can't be tested with the mock web server. Either we enable that for these tests or find some other way of setting the URL

x-pack/plugin/inference/qa/rolling-upgrade/build.gradle Outdated Show resolved Hide resolved
@davidkyle
Copy link
Member Author

@elasticmachine update branch

@davidkyle
Copy link
Member Author

@elasticmachine update branch

@davidkyle davidkyle merged commit dd741ba into elastic:main Apr 24, 2024
14 checks passed
davidkyle added a commit to davidkyle/elasticsearch that referenced this pull request Apr 24, 2024
Rolling upgrade tests for OpenAI, Cohere and Hugging Face. 
The services are tested by setting the services' url to a local mock
 web server and mocking the responses.
@elasticsearchmachine
Copy link
Collaborator

💚 Backport successful

Status Branch Result
8.14

elasticsearchmachine pushed a commit that referenced this pull request Apr 25, 2024
#107868)

* [ML] External inference service rolling upgrade tests (#107619)

Rolling upgrade tests for OpenAI, Cohere and Hugging Face. 
The services are tested by setting the services' url to a local mock
 web server and mocking the responses.

* fix compilation

* Reduce inference rolling upgrade test parallelism

---------

Co-authored-by: Elastic Machine <elasticmachine@users.noreply.github.com>
Co-authored-by: Mark Vieira <portugee@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
auto-backport-and-merge Automatically create backport pull requests and merge when ready :ml Machine learning Team:ML Meta label for the ML team >test Issues or PRs that are addressing/adding tests test-full-bwc Trigger full BWC version matrix tests v8.14.0 v8.15.0
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

5 participants