getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-17 10:02:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(ai): preserve mistral thinking replay

Browse source
This commit is contained in:
Mario Zechner 2026-03-05 23:49:18 +01:00
parent eb9f1183ad
commit 4c175790b1
1 changed files with 10 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

15
packages/ai/src/providers/mistral.ts
View file

@ -465,17 +465,22 @@ function toChatMessages(messages: Message[], supportsImages: boolean): ChatCompl
} }
if (msg.role === "assistant") { if (msg.role === "assistant") {
const textParts: Array<{ type: "text"; text: string }> = []; const contentParts: ContentChunk[] = [];
const toolCalls: Array<{ id: string; type: "function"; function: { name: string; arguments: string } }> = []; const toolCalls: Array<{ id: string; type: "function"; function: { name: string; arguments: string } }> = [];
for (const block of msg.content) { for (const block of msg.content) {
if (block.type === "text") { if (block.type === "text") {
if (block.text.trim().length > 0) textParts.push({ type: "text", text: sanitizeSurrogates(block.text) }); if (block.text.trim().length > 0) {
contentParts.push({ type: "text", text: sanitizeSurrogates(block.text) });
}
continue; continue;
} }
if (block.type === "thinking") { if (block.type === "thinking") {
if (block.thinking.trim().length > 0) { if (block.thinking.trim().length > 0) {
textParts.push({ type: "text", text: sanitizeSurrogates(block.thinking) }); contentParts.push({
type: "thinking",
thinking: [{ type: "text", text: sanitizeSurrogates(block.thinking) }],
});
} }
continue; continue;
} }
@ -487,9 +492,9 @@ function toChatMessages(messages: Message[], supportsImages: boolean): ChatCompl
} }
const assistantMessage: ChatCompletionStreamRequestMessages = { role: "assistant" }; const assistantMessage: ChatCompletionStreamRequestMessages = { role: "assistant" };
if (textParts.length > 0) assistantMessage.content = textParts; if (contentParts.length > 0) assistantMessage.content = contentParts;
if (toolCalls.length > 0) assistantMessage.toolCalls = toolCalls; if (toolCalls.length > 0) assistantMessage.toolCalls = toolCalls;
if (textParts.length > 0 || toolCalls.length > 0) result.push(assistantMessage); if (contentParts.length > 0 || toolCalls.length > 0) result.push(assistantMessage);
continue; continue;
} }