mark/vibn-frontend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: parse thoughtSignature correctly to support reasoning-to-text promotion

Browse Source
This commit is contained in:
mawkone
2026-06-01 14:46:45 -07:00
parent fbb542a3c7
commit 5a8787dbea
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
vibn-frontend/lib/ai/gemini-chat.ts
View File

@@ -151,10 +151,13 @@ export async function callGeminiChat(opts: {
const toolCalls: ToolCall[] = []; const toolCalls: ToolCall[] = [];
const parts = response.candidates?.[0]?.content?.parts ?? []; const parts = response.candidates?.[0]?.content?.parts ?? [];
const isPartThought = (p: Record<string, unknown>) =>
Boolean(p.thought || p.thoughtSignature);
for (const part of parts) { for (const part of parts) {
if (part.text) { if (part.text) {
if ((part as { thought?: unknown }).thought) thoughts += part.text; if (isPartThought(part as Record<string, unknown>))
thoughts += part.text;
else text += part.text; else text += part.text;
} }
if (part.functionCall) { if (part.functionCall) {
@@ -219,12 +222,14 @@ export async function* streamGeminiChat(opts: {
}); });
console.log("[GeminiChat] Stream request initiated"); console.log("[GeminiChat] Stream request initiated");
const isPartThought = (p: Record<string, unknown>) =>
Boolean(p.thought || p.thoughtSignature);
for await (const chunk of streamResult) { for await (const chunk of streamResult) {
const parts = chunk.candidates?.[0]?.content?.parts ?? []; const parts = chunk.candidates?.[0]?.content?.parts ?? [];
for (const part of parts) { for (const part of parts) {
if (part.text) { if (part.text) {
yield (part as { thought?: unknown }).thought yield isPartThought(part as Record<string, unknown>)
? { type: "thinking", text: part.text } ? { type: "thinking", text: part.text }
: { type: "text", text: part.text }; : { type: "text", text: part.text };
} }