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.' },
                 },
             ]
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('')
 }