Skip to content

Commit 5ed38b8

Browse files
committed
Fix mistralAI completer
1 parent 844b36e commit 5ed38b8

File tree

1 file changed

+17
-3
lines changed

1 file changed

+17
-3
lines changed

src/default-providers/MistralAI/completer.ts

Lines changed: 17 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,9 +36,23 @@ export class CodestralCompleter extends BaseCompleter {
3636
false
3737
);
3838
const items = response.choices.map(choice => {
39-
const content = choice.message.content
40-
.replace(CODE_BLOCK_START_REGEX, '')
41-
.replace(CODE_BLOCK_END_REGEX, '');
39+
const messageContent = choice.message.content;
40+
let content = '';
41+
42+
if (typeof messageContent === 'string') {
43+
content = messageContent
44+
.replace(CODE_BLOCK_START_REGEX, '')
45+
.replace(CODE_BLOCK_END_REGEX, '');
46+
} else if (Array.isArray(messageContent)) {
47+
// Handle ContentChunk[] case - extract text content
48+
content = messageContent
49+
.filter(chunk => chunk.type === 'text')
50+
.map(chunk => chunk.text || '')
51+
.join('')
52+
.replace(CODE_BLOCK_START_REGEX, '')
53+
.replace(CODE_BLOCK_END_REGEX, '');
54+
}
55+
4256
return {
4357
insertText: content
4458
};

0 commit comments

Comments
 (0)