Skip to content

Commit 7cfa4df

Browse files
Fix failing tests
Signed-off-by: rithin-pullela-aws <rithinp@amazon.com>
1 parent cbeed7a commit 7cfa4df

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

plugin/src/test/java/org/opensearch/ml/settings/MLFeatureEnabledSettingTests.java

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,11 +11,13 @@
1111
import static org.mockito.Mockito.times;
1212
import static org.mockito.Mockito.verify;
1313
import static org.mockito.Mockito.when;
14+
import static org.opensearch.ml.common.settings.MLCommonsSettings.ML_COMMONS_AGENTIC_SEARCH_ENABLED;
1415
import static org.opensearch.ml.common.settings.MLCommonsSettings.ML_COMMONS_AGENT_FRAMEWORK_ENABLED;
1516
import static org.opensearch.ml.common.settings.MLCommonsSettings.ML_COMMONS_CONNECTOR_PRIVATE_IP_ENABLED;
1617
import static org.opensearch.ml.common.settings.MLCommonsSettings.ML_COMMONS_CONTROLLER_ENABLED;
1718
import static org.opensearch.ml.common.settings.MLCommonsSettings.ML_COMMONS_EXECUTE_TOOL_ENABLED;
1819
import static org.opensearch.ml.common.settings.MLCommonsSettings.ML_COMMONS_LOCAL_MODEL_ENABLED;
20+
import static org.opensearch.ml.common.settings.MLCommonsSettings.ML_COMMONS_MCP_CONNECTOR_ENABLED;
1921
import static org.opensearch.ml.common.settings.MLCommonsSettings.ML_COMMONS_MCP_SERVER_ENABLED;
2022
import static org.opensearch.ml.common.settings.MLCommonsSettings.ML_COMMONS_METRIC_COLLECTION_ENABLED;
2123
import static org.opensearch.ml.common.settings.MLCommonsSettings.ML_COMMONS_MULTI_TENANCY_ENABLED;
@@ -72,7 +74,9 @@ public void setUp() {
7274
ML_COMMONS_RAG_PIPELINE_FEATURE_ENABLED,
7375
ML_COMMONS_METRIC_COLLECTION_ENABLED,
7476
ML_COMMONS_STATIC_METRIC_COLLECTION_ENABLED,
75-
ML_COMMONS_EXECUTE_TOOL_ENABLED
77+
ML_COMMONS_EXECUTE_TOOL_ENABLED,
78+
ML_COMMONS_AGENTIC_SEARCH_ENABLED,
79+
ML_COMMONS_MCP_CONNECTOR_ENABLED
7680
)
7781
)
7882
);

0 commit comments

Comments
 (0)