Removing suffix from simulated deploy model step

Signed-off-by: Joshua Palis <[email protected]>
opensearch-project · Jan 5, 2024 · aa3611b · aa3611b
1 parent b970504
commit aa3611b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/main/java/org/opensearch/flowframework/workflow/RegisterLocalModelStep.java b/src/main/java/org/opensearch/flowframework/workflow/RegisterLocalModelStep.java
@@ -184,7 +184,7 @@ public CompletableFuture<WorkflowData> execute(
                             // Simulate Model deployment step and update resources created
                             flowFrameworkIndicesHandler.updateResourceInStateIndex(
                                 currentNodeInputs.getWorkflowId(),
-                                currentNodeId + "-deploy",
+                                currentNodeId,
                                 DeployModelStep.NAME,
                                 id,
                                 ActionListener.wrap(deployUpdateResponse -> {

diff --git a/src/main/java/org/opensearch/flowframework/workflow/RegisterRemoteModelStep.java b/src/main/java/org/opensearch/flowframework/workflow/RegisterRemoteModelStep.java
@@ -119,7 +119,7 @@ public void onResponse(MLRegisterModelResponse mlRegisterModelResponse) {
                                 if (Boolean.TRUE.equals(deploy)) {
                                     flowFrameworkIndicesHandler.updateResourceInStateIndex(
                                         currentNodeInputs.getWorkflowId(),
-                                        currentNodeId + "-deploy",
+                                        currentNodeId,
                                         DeployModelStep.NAME,
                                         mlRegisterModelResponse.getModelId(),
                                         ActionListener.wrap(deployUpdateResponse -> {