Skip to content

Commit 13c1699

Browse files
authored
Fix duplicate LLM answer (#1813)
* Fix duplicate LLM answer * Adjust tests
1 parent b4396ac commit 13c1699

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

src/Elastic.Documentation.Site/Assets/web-components/SearchOrAskAi/AskAi/AskAiAnswer.test.tsx

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ describe('AskAiAnswer Component', () => {
6363
{
6464
id: 'some-id-1',
6565
timestamp: 0,
66-
type: 'ai_message',
66+
type: 'ai_message_chunk',
6767
data: {
6868
content:
6969
'Elasticsearch is a distributed search engine...',
@@ -134,7 +134,7 @@ describe('AskAiAnswer Component', () => {
134134
{
135135
id: 'some-id-1',
136136
timestamp: 1,
137-
type: 'ai_message',
137+
type: 'ai_message_chunk',
138138
data: {
139139
content: 'Here is your answer about Elasticsearch.',
140140
},
@@ -188,7 +188,7 @@ describe('AskAiAnswer Component', () => {
188188
{
189189
id: 'some-id-1',
190190
timestamp: 12345,
191-
type: 'ai_message',
191+
type: 'ai_message_chunk',
192192
data: { content: 'Here is your answer.' },
193193
},
194194
]

src/Elastic.Documentation.Site/Assets/web-components/SearchOrAskAi/AskAi/AskAiAnswer.tsx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ import { v4 as uuidv4 } from 'uuid'
2222
// Helper function to accumulate AI message content
2323
const getAccumulatedContent = (messages: LlmGatewayMessage[]) => {
2424
return messages
25-
.filter((m) => m.type === 'ai_message' || m.type === 'ai_message_chunk')
25+
.filter((m) => m.type === 'ai_message_chunk')
2626
.map((m) => m.data.content)
2727
.join('')
2828
}

0 commit comments

Comments
 (0)