Fix model enable flag not loading (opensearch-project#2221)

Signed-off-by: Sicheng Song <sicheng.song@outlook.com>
zane-neo · Mar 19, 2024 · 8e86157 · 8e86157
1 parent a7d1ef0
commit 8e86157
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/plugin/src/main/java/org/opensearch/ml/model/MLModelManager.java b/plugin/src/main/java/org/opensearch/ml/model/MLModelManager.java
@@ -294,6 +294,7 @@ private void uploadMLModelMeta(MLRegisterModelMetaInput mlRegisterModelMetaInput
                     .modelGroupId(mlRegisterModelMetaInput.getModelGroupId())
                     .description(mlRegisterModelMetaInput.getDescription())
                     .rateLimiter(mlRegisterModelMetaInput.getRateLimiter())
+                    .isEnabled(mlRegisterModelMetaInput.getIsEnabled())
                     .modelFormat(mlRegisterModelMetaInput.getModelFormat())
                     .modelState(MLModelState.REGISTERING)
                     .modelConfig(mlRegisterModelMetaInput.getModelConfig())
@@ -525,6 +526,7 @@ private void indexRemoteModel(
                     .version(version)
                     .description(registerModelInput.getDescription())
                     .rateLimiter(registerModelInput.getRateLimiter())
+                    .isEnabled(registerModelInput.getIsEnabled())
                     .modelFormat(registerModelInput.getModelFormat())
                     .modelState(MLModelState.REGISTERED)
                     .connector(registerModelInput.getConnector())
@@ -590,6 +592,7 @@ void indexRemoteModel(MLRegisterModelInput registerModelInput, MLTask mlTask, St
                     .version(version)
                     .description(registerModelInput.getDescription())
                     .rateLimiter(registerModelInput.getRateLimiter())
+                    .isEnabled(registerModelInput.getIsEnabled())
                     .modelFormat(registerModelInput.getModelFormat())
                     .modelState(MLModelState.REGISTERED)
                     .connector(registerModelInput.getConnector())
@@ -657,6 +660,7 @@ private void registerModelFromUrl(MLRegisterModelInput registerModelInput, MLTas
                     .version(version)
                     .description(registerModelInput.getDescription())
                     .rateLimiter(registerModelInput.getRateLimiter())
+                    .isEnabled(registerModelInput.getIsEnabled())
                     .modelFormat(registerModelInput.getModelFormat())
                     .modelState(MLModelState.REGISTERING)
                     .modelConfig(registerModelInput.getModelConfig())