Use current timestamps for registering agent

Signed-off-by: Daniel Widdis <widdis@gmail.com>
dbwiddis · Dec 27, 2023 · 3fadbd5 · 3fadbd5
1 parent 8fa9b3f
commit 3fadbd5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/src/main/java/org/opensearch/flowframework/workflow/RegisterAgentStep.java b/src/main/java/org/opensearch/flowframework/workflow/RegisterAgentStep.java
@@ -160,8 +160,8 @@ public void onFailure(Exception e) {
             List<MLToolSpec> toolsList = getTools(toolsOrder, previousNodeInputs, outputs);
             Map<String, String> parameters = getStringToStringMap(inputs.get(PARAMETERS_FIELD), PARAMETERS_FIELD);
             MLMemorySpec memory = getMLMemorySpec(inputs.get(MEMORY_FIELD));
-            Instant createdTime = Instant.ofEpochMilli((Long) inputs.get(CREATED_TIME));
-            Instant lastUpdateTime = Instant.ofEpochMilli((Long) inputs.get(LAST_UPDATED_TIME_FIELD));
+            Instant createdTime = Instant.now();
+            Instant lastUpdateTime = createdTime;
             String appType = (String) inputs.get(APP_TYPE_FIELD);
 
             // Case when modelId is present in previous node inputs

diff --git a/src/main/resources/mappings/workflow-steps.json b/src/main/resources/mappings/workflow-steps.json
@@ -141,8 +141,6 @@
             "tools",
             "parameters",
             "memory",
-            "created_time",
-            "last_updated_time",
             "app_type"
         ],
         "outputs":[