[Security solution] Fix regenerate button with LangGraph (#188249)

elastic · Jul 13, 2024 · f72a321 · f72a321
1 parent c299737
commit f72a321
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/...server/lib/langchain/graphs/default_assistant_graph/nodes/persist_conversation_changes.ts b/...server/lib/langchain/graphs/default_assistant_graph/nodes/persist_conversation_changes.ts
@@ -49,6 +49,19 @@ export const persistConversationChanges = async ({
     });
   }
 
+  const lastMessage = state.conversation.messages
+    ? state.conversation.messages[state.conversation.messages.length - 1]
+    : undefined;
+  if (lastMessage && lastMessage.content === state.input && lastMessage.role === 'user') {
+    // this is a regenerated message, do not update the conversation again
+    const langChainMessages = getLangChainMessages(state.conversation.messages ?? []);
+    const messages = langChainMessages.slice(0, -1); // all but the last message
+    return {
+      conversation: state.conversation,
+      messages,
+    };
+  }
+
   const updatedConversation = await conversationsDataClient?.appendConversationMessages({
     existingConversation: conversation ? conversation : state.conversation,
     messages: [