Fix merge conflict

Signed-off-by: Arjun kumar Giri <arjung@amazon.com>
arjunkumargiri · Dec 5, 2023 · 371b685 · 371b685
1 parent 15d1f72
commit 371b685
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/agent/MLChatAgentRunner.java b/ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/agent/MLChatAgentRunner.java
@@ -382,6 +382,22 @@ private void runReAct(
                             .build();
                         conversationIndexMemory.save(msgTemp, parentInteractionId, traceNumber.addAndGet(1), null);
                     }
+                    if (finalAnswer != null) {
+                        finalAnswer = finalAnswer.trim();
+                        if (conversationIndexMemory != null) {
+                            String finalAnswer1 = finalAnswer;
+                            // Create final trace message.
+                            ConversationIndexMessage msgTemp = ConversationIndexMessage
+                                .conversationIndexMessageBuilder()
+                                .type("ReAct")
+                                .question(question)
+                                .response(finalAnswer1)
+                                .finalAnswer(true)
+                                .sessionId(sessionId)
+                                .build();
+                            conversationIndexMemory.save(msgTemp, parentInteractionId, traceNumber.addAndGet(1), null);
+                        }
+                    }
                     if (finalAnswer != null) {
                         finalAnswer = finalAnswer.trim();
                         if (conversationIndexMemory != null) {
@@ -401,7 +417,7 @@ private void runReAct(
                                 .getMemoryManager()
                                 .updateInteraction(
                                     parentInteractionId,
-                                    ImmutableMap.of(AI_RESPONSE_FIELD, finalAnswer1),
+                                    ImmutableMap.of(AI_RESPONSE_FIELD, finalAnswer1, ADDITIONAL_INFO_FIELD, additionalInfo),
                                     ActionListener.<UpdateResponse>wrap(updateResponse -> {
                                         log.info("Updated final answer into interaction id: {}", parentInteractionId);
                                         log.info("Final answer: {}", finalAnswer1);