diff --git a/src/shared/api-schemas/google-ai.ts b/src/shared/api-schemas/google-ai.ts
index 2ad9504..e100244 100644
--- a/src/shared/api-schemas/google-ai.ts
+++ b/src/shared/api-schemas/google-ai.ts
@@ -75,7 +75,12 @@ export const GoogleAIV1GenerateContentSchema = z
         topP: z.number().min(0).max(1).optional(),
         topK: z.number().min(1).max(40).optional(),
         stopSequences: z.array(z.string().max(500)).max(5).optional(),
-        thinkingConfig: z.object({includeThoughts: z.boolean().optional()}).optional()
+        thinkingConfig: z.object({includeThoughts: z.boolean().optional()}).optional(),
+        // Support for new Gemini 2.5 thinking budget
+        thinking_budget: z.union([
+          z.literal("auto"),
+          z.number().int().min(0).max(24576)
+        ]).optional()
       })
       .default({}),
   })