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

2026-06-15 11:46:20 -07:00
parent e58972d594
commit dfb79f3acd
1 changed files with 14 additions and 0 deletions
--- a/vibn-frontend/lib/ai/gemini-chat.ts
+++ b/vibn-frontend/lib/ai/gemini-chat.ts
@@ -124,6 +124,13 @@ export async function callGeminiChat(opts: {
      maxOutputTokens: 8192,
    };

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

+    if (GEMINI_MODEL.includes("thinking") || GEMINI_MODEL.includes("pro")) {
+      config.thinkingConfig = {
+        thinkingBudget: 1024,
+        includeThoughts: true,
+      };
+    }
+
    if (opts.systemPrompt) {
      config.systemInstruction = opts.systemPrompt;
    }