Skip to content

Commit 9041f1c

Browse files
committed
fix setMlguard
1 parent 0ddff57 commit 9041f1c

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/remote/RemoteConnectorExecutor.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ default void setRateLimiter(TokenBucket rateLimiter) {}
102102

103103
default void setUserRateLimiterMap(Map<String, TokenBucket> userRateLimiterMap) {}
104104

105-
default void setMLGuard(MLGuard mlGuard) {}
105+
default void setMlGuard(MLGuard mlGuard) {}
106106

107107
default void preparePayloadAndInvokeRemoteModel(MLInput mlInput, List<ModelTensors> tensorOutputs) {
108108
Connector connector = getConnector();

ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/remote/RemoteModel.java

+1-2
Original file line numberDiff line numberDiff line change
@@ -92,8 +92,7 @@ public void initModel(MLModel model, Map<String, Object> params, Encryptor encry
9292
this.connectorExecutor.setXContentRegistry((NamedXContentRegistry) params.get(XCONTENT_REGISTRY));
9393
this.connectorExecutor.setRateLimiter((TokenBucket) params.get(RATE_LIMITER));
9494
this.connectorExecutor.setUserRateLimiterMap((Map<String, TokenBucket>) params.get(USER_RATE_LIMITER_MAP));
95-
MLGuard mgd = (MLGuard) params.get(GUARDRAILS);
96-
this.connectorExecutor.setMLGuard(mgd);
95+
this.connectorExecutor.setMlGuard((MLGuard) params.get(GUARDRAILS));
9796
} catch (RuntimeException e) {
9897
log.error("Failed to init remote model.", e);
9998
throw e;

0 commit comments

Comments
 (0)