Autobump new model groups' latest version when register to a new model

Signed-off-by: Sicheng Song <[email protected]>
opensearch-project · Oct 5, 2023 · f1758bd · f1758bd
1 parent 0b1c663
commit f1758bd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/common/src/main/java/org/opensearch/ml/common/MLModelGroup.java b/common/src/main/java/org/opensearch/ml/common/MLModelGroup.java
@@ -39,6 +39,7 @@ public class MLModelGroup implements ToXContentObject {
     @Setter
     private String name;
     private String description;
+    @Setter
     private int latestVersion;
     private List<String> backendRoles;
     private User owner;

diff --git a/plugin/src/main/java/org/opensearch/ml/action/models/UpdateModelTransportAction.java b/plugin/src/main/java/org/opensearch/ml/action/models/UpdateModelTransportAction.java
@@ -221,6 +221,7 @@ private void updateModelWithRegisteringNewModelGroup(
                     mlModelGroupManager.getModelGroup(newModelGroupId, ActionListener.wrap(newModelGroup -> {
                         String updatedVersion = incrementLatestVersion(newModelGroup);
                         updateModelInput.setVersion(updatedVersion);
+                        newModelGroup.setLatestVersion(Integer.parseInt(updatedVersion));
                         updateRequestConstructor(modelId, updateRequest, updateModelInput, actionListener);
                     },
                         exception -> actionListener