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

Enable thinking stream in frontend lib/ai/gemini-chat.ts

Browse Source
This commit is contained in:
mawkone
2026-06-15 11:46:20 -07:00
parent e58972d594
commit dfb79f3acd
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -124,6 +124,13 @@ export async function callGeminiChat(opts: {
maxOutputTokens: 8192, maxOutputTokens: 8192,
}; };
if (GEMINI_MODEL.includes("thinking") || GEMINI_MODEL.includes("pro")) {
config.thinkingConfig = {
thinkingBudget: 1024,
includeThoughts: true,
};
}
if (opts.systemPrompt) { if (opts.systemPrompt) {
config.systemInstruction = opts.systemPrompt; config.systemInstruction = opts.systemPrompt;
} }
@@ -259,6 +266,13 @@ export async function* streamGeminiChat(opts: {
maxOutputTokens: 8192, maxOutputTokens: 8192,
}; };
if (GEMINI_MODEL.includes("thinking") || GEMINI_MODEL.includes("pro")) {
config.thinkingConfig = {
thinkingBudget: 1024,
includeThoughts: true,
};
}
if (opts.systemPrompt) { if (opts.systemPrompt) {
config.systemInstruction = opts.systemPrompt; config.systemInstruction = opts.systemPrompt;
} }