Optimize the way of deploying local models for integration tests

[martin-gaievski]

There is a problem with the current approach of loading and deploying local ML model for integration tests. Current approach is to deploy one model per test method. At the end of test method execution that model is undeployed. Such approach leads to multiple deploy/undeploy calls on a single test cluster.

Currently we're using ml-commons to deploy the model. As per ml-commons team the engine they are using (PyTorch) is not optimized for recurring model redeployments.

In environment with limited memory that may lead to high memory consumption, and in such case Native Memory Circuit Breaker in ml-commons will be opened. In such case no new model deployment is possible and CB exception will be returned.

Suggested approach is to change paradigm from model per test case to a shared models for all test suite. This way models can be deployed once during the cluster setup, used by the test and then undeployed in the tear down phase. This seems feasible as models are using in a read-only mode and there is limited number of different local models. Currently there are 3 different models that are used in integ tests (https://github.com/opensearch-project/neural-search/tree/main/src/test/resources/processor):

• bert sentence_transformer traced model, used for most of tests https://github.com/opensearch-project/neural-search/blob/main/src/test/resources/processor/UploadModelRequestBody.json
• model for sparce encoding used only for Sparce encoding processor https://github.com/opensearch-project/neural-search/blob/main/src/test/resources/processor/UploadSparseEncodingModelRequestBody.json
• TinyBERT-L-2-v2 text similarity model, used by reranker https://github.com/opensearch-project/neural-search/blob/main/src/test/resources/processor/UploadTextSimilarityModelRequestBody.json

Ref:

• Issue opened for ml-commons repo https://github.com/opensearch-project/ml-commons/issues/2308
• Short term fix in neural-search https://github.com/opensearch-project/neural-search/pull/683
• Meta issue for the repo https://github.com/opensearch-project/neural-search/issues/667

[martin-gaievski]

Ml-commons team has implemented fix that allows memory CB to be disabled - https://github.com/opensearch-project/ml-commons/pull/2469). Fix on neural-search side is to disable the CB for integ and BWC tests