Skip to content

Commit 30a9ebb

Browse files
committed
Get all instances
Signed-off-by: Craig Perkins <[email protected]>
1 parent f836ba5 commit 30a9ebb

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

plugin/src/main/java/org/opensearch/ml/action/handler/MLSearchHandler.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77

88
import static org.opensearch.core.rest.RestStatus.BAD_REQUEST;
99
import static org.opensearch.core.rest.RestStatus.INTERNAL_SERVER_ERROR;
10+
import static org.opensearch.ml.common.CommonValue.ML_MODEL_GROUP_RESOURCE_TYPE;
1011
import static org.opensearch.ml.utils.RestActionUtils.wrapListenerToHandleSearchIndexNotFound;
1112

1213
import java.util.ArrayList;
@@ -152,7 +153,7 @@ public void search(SdkClient sdkClient, SearchRequest request, String tenantId,
152153
if (rsClientPresent && user != null && modelAccessControlHelper.modelAccessControlEnabled() && hasModelGroupIndex) {
153154
// RSC fast-path: get accessible group IDs → gate models (IDs or missing)
154155
ResourceSharingClient rsc = ResourceSharingClientAccessor.getInstance().getResourceSharingClient();
155-
rsc.getAccessibleResourceIds(CommonValue.ML_MODEL_GROUP_INDEX, ActionListener.wrap(ids -> {
156+
rsc.getAccessibleResourceIds(ML_MODEL_GROUP_RESOURCE_TYPE, ActionListener.wrap(ids -> {
156157
SearchSourceBuilder gated = Optional.ofNullable(request.source()).orElseGet(SearchSourceBuilder::new);
157158
gated.query(rewriteQueryBuilderRSC(gated.query(), ids)); // ids may be empty → "missing only"
158159
request.source(gated);

plugin/src/main/java/org/opensearch/ml/action/model_group/SearchModelGroupTransportAction.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77

88
import static org.opensearch.ml.action.handler.MLSearchHandler.wrapRestActionListener;
99
import static org.opensearch.ml.common.CommonValue.ML_MODEL_GROUP_INDEX;
10+
import static org.opensearch.ml.common.CommonValue.ML_MODEL_GROUP_RESOURCE_TYPE;
1011
import static org.opensearch.ml.utils.RestActionUtils.wrapListenerToHandleSearchIndexNotFound;
1112

1213
import java.util.Collections;
@@ -115,7 +116,7 @@ private void addAccessibleModelGroupsFilterAndSearch(
115116
SearchSourceBuilder sourceBuilder = request.source() != null ? request.source() : new SearchSourceBuilder();
116117
ResourceSharingClient rsc = ResourceSharingClientAccessor.getInstance().getResourceSharingClient();
117118
// filter by accessible model-groups
118-
rsc.getAccessibleResourceIds(ML_MODEL_GROUP_INDEX, ActionListener.wrap(ids -> {
119+
rsc.getAccessibleResourceIds(ML_MODEL_GROUP_RESOURCE_TYPE, ActionListener.wrap(ids -> {
119120
sourceBuilder.query(modelAccessControlHelper.mergeWithAccessFilter(sourceBuilder.query(), ids));
120121
request.source(sourceBuilder);
121122
search(tenantId, request, wrappedListener);

0 commit comments

Comments
 (0)