From dfb79f3acdb6437d47d18864978090d7d2c6eb54 Mon Sep 17 00:00:00 2001
From: mawkone <mark@getacquired.com>
Date: Mon, 15 Jun 2026 11:46:20 -0700
Subject: [PATCH] Enable thinking stream in frontend lib/ai/gemini-chat.ts

---
 vibn-frontend/lib/ai/gemini-chat.ts | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/vibn-frontend/lib/ai/gemini-chat.ts b/vibn-frontend/lib/ai/gemini-chat.ts
index f0f75e6..ac2355f 100644
--- 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;
     }