Skip to content

Commit b10a848

Browse files
committed
Adapt to changes
Signed-off-by: Craig Perkins <[email protected]>
1 parent 482382a commit b10a848

File tree

3 files changed

+23
-3
lines changed

3 files changed

+23
-3
lines changed

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

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
package org.opensearch.ml.action.model_group;
77

88
import static org.opensearch.ml.action.handler.MLSearchHandler.wrapRestActionListener;
9-
import static org.opensearch.ml.common.CommonValue.ML_MODEL_GROUP_INDEX;
109
import static org.opensearch.ml.common.CommonValue.ML_MODEL_GROUP_RESOURCE_TYPE;
1110
import static org.opensearch.ml.helper.ModelAccessControlHelper.shouldUseResourceAuthz;
1211
import static org.opensearch.ml.utils.RestActionUtils.wrapListenerToHandleSearchIndexNotFound;

plugin/src/main/java/org/opensearch/ml/resources/MLResourceSharingExtension.java

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,18 @@ public class MLResourceSharingExtension implements ResourceSharingExtension {
1919

2020
@Override
2121
public Set<ResourceProvider> getResourceProviders() {
22-
return Set.of(new ResourceProvider(ML_MODEL_GROUP_RESOURCE_TYPE, ML_MODEL_GROUP_INDEX));
22+
return Set.of(new ResourceProvider() {
23+
24+
@Override
25+
public String resourceType() {
26+
return ML_MODEL_GROUP_RESOURCE_TYPE;
27+
}
28+
29+
@Override
30+
public String resourceIndexName() {
31+
return ML_MODEL_GROUP_INDEX;
32+
}
33+
});
2334
}
2435

2536
@Override

plugin/src/test/java/org/opensearch/ml/resources/MLResourceSharingExtensionTests.java

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,17 @@ public void testGetResourceProviders_returnsUnmodifiableSet() {
6767
Set<ResourceProvider> providers = ext.getResourceProviders();
6868

6969
// Attempt to modify — Set.of(...) should be unmodifiable and throw
70-
providers.add(new ResourceProvider("some.Type", "some-index"));
70+
providers.add(new ResourceProvider() {
71+
@Override
72+
public String resourceType() {
73+
return "exampleType";
74+
}
75+
76+
@Override
77+
public String resourceIndexName() {
78+
return "some-index";
79+
}
80+
});
7181
}
7282

7383
@Test

0 commit comments

Comments
 (0)