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

Fix block indexing in Google provider impl

Browse source
This commit is contained in:
Mario Zechner 2025-09-19 00:10:43 +02:00
parent 2296dc4052
commit 9e86079386
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
packages/ai/src/providers/google.ts
View file

@ -99,12 +99,13 @@ export const streamGoogle: StreamFunction<"google-generative-ai"> = (
}
if (isThinking) {
currentBlock = { type: "thinking", thinking: "", thinkingSignature: undefined };
output.content.push(currentBlock);
stream.push({ type: "thinking_start", contentIndex: blockIndex(), partial: output });
} else {
currentBlock = { type: "text", text: "" };
output.content.push(currentBlock);
stream.push({ type: "text_start", contentIndex: blockIndex(), partial: output });
}
output.content.push(currentBlock);
}
if (currentBlock.type === "thinking") {
currentBlock.thinking += part.text;