From d95b97e9775fdffad8ef8ba57e49ba5d1fb8c4e9 Mon Sep 17 00:00:00 2001 From: Jan Calanog Date: Tue, 2 Sep 2025 12:11:50 +0200 Subject: [PATCH 1/2] Fix duplicate LLM answer --- .../Assets/web-components/SearchOrAskAi/AskAi/AskAiAnswer.tsx | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/Elastic.Documentation.Site/Assets/web-components/SearchOrAskAi/AskAi/AskAiAnswer.tsx b/src/Elastic.Documentation.Site/Assets/web-components/SearchOrAskAi/AskAi/AskAiAnswer.tsx index c269afcec..e24191b14 100644 --- a/src/Elastic.Documentation.Site/Assets/web-components/SearchOrAskAi/AskAi/AskAiAnswer.tsx +++ b/src/Elastic.Documentation.Site/Assets/web-components/SearchOrAskAi/AskAi/AskAiAnswer.tsx @@ -22,7 +22,7 @@ import { v4 as uuidv4 } from 'uuid' // Helper function to accumulate AI message content const getAccumulatedContent = (messages: LlmGatewayMessage[]) => { return messages - .filter((m) => m.type === 'ai_message' || m.type === 'ai_message_chunk') + .filter((m) => m.type === 'ai_message_chunk') .map((m) => m.data.content) .join('') } From 1095b386ec3789bbd5a8e5642f9e551ecab8cc5b Mon Sep 17 00:00:00 2001 From: Jan Calanog Date: Tue, 2 Sep 2025 12:14:13 +0200 Subject: [PATCH 2/2] Adjust tests --- .../web-components/SearchOrAskAi/AskAi/AskAiAnswer.test.tsx | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/Elastic.Documentation.Site/Assets/web-components/SearchOrAskAi/AskAi/AskAiAnswer.test.tsx b/src/Elastic.Documentation.Site/Assets/web-components/SearchOrAskAi/AskAi/AskAiAnswer.test.tsx index 4b957c8c2..67d909c57 100644 --- a/src/Elastic.Documentation.Site/Assets/web-components/SearchOrAskAi/AskAi/AskAiAnswer.test.tsx +++ b/src/Elastic.Documentation.Site/Assets/web-components/SearchOrAskAi/AskAi/AskAiAnswer.test.tsx @@ -63,7 +63,7 @@ describe('AskAiAnswer Component', () => { { id: 'some-id-1', timestamp: 0, - type: 'ai_message', + type: 'ai_message_chunk', data: { content: 'Elasticsearch is a distributed search engine...', @@ -134,7 +134,7 @@ describe('AskAiAnswer Component', () => { { id: 'some-id-1', timestamp: 1, - type: 'ai_message', + type: 'ai_message_chunk', data: { content: 'Here is your answer about Elasticsearch.', }, @@ -188,7 +188,7 @@ describe('AskAiAnswer Component', () => { { id: 'some-id-1', timestamp: 12345, - type: 'ai_message', + type: 'ai_message_chunk', data: { content: 'Here is your answer.' }, }, ]