From 2d9e216e0af06fc32f4224eec0fcf602883da570 Mon Sep 17 00:00:00 2001
From: Sahaj Jain <sahajjain13@gmail.com>
Date: Sat, 19 Jul 2025 21:21:02 +0530
Subject: [PATCH] limit thinking budget to optimise pricing

---
 app/api/generate-theme/route.ts | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/app/api/generate-theme/route.ts b/app/api/generate-theme/route.ts
index bd8b4549..842c0209 100644
--- a/app/api/generate-theme/route.ts
+++ b/app/api/generate-theme/route.ts
@@ -4,7 +4,7 @@ import { getCurrentUserId, logError } from "@/lib/shared";
 import { validateSubscriptionAndUsage } from "@/lib/subscription";
 import { SubscriptionRequiredError } from "@/types/errors";
 import { requestSchema, responseSchema, SYSTEM_PROMPT } from "@/utils/ai/generate-theme";
-import { createGoogleGenerativeAI } from "@ai-sdk/google";
+import { createGoogleGenerativeAI, GoogleGenerativeAIProviderOptions } from "@ai-sdk/google";
 import { Ratelimit } from "@upstash/ratelimit";
 import { kv } from "@vercel/kv";
 import { generateText, Output } from "ai";
@@ -61,6 +61,13 @@ export async function POST(req: NextRequest) {
       system: SYSTEM_PROMPT,
       messages,
       abortSignal: req.signal,
+      providerOptions: {
+        google: {
+          thinkingConfig: {
+            thinkingBudget: 128,
+          },
+        } satisfies GoogleGenerativeAIProviderOptions,
+      },
     });
 
     if (usage) {