Cohee, when I catch you Cohee

src/shared/api-schemas/google-ai.ts

@@ -76,6 +76,7 @@ export const GoogleAIV1GenerateContentSchema = z
         topK: z.number().min(1).max(40).optional(),
         stopSequences: z.array(z.string().max(500)).max(5).optional(),
         thinkingConfig: z.object({
+          enable_thinking: z.boolean().optional(),
           includeThoughts: z.boolean().optional(),
           thinkingBudget: z.union([
             z.literal("auto"),
@@ -83,7 +84,21 @@ export const GoogleAIV1GenerateContentSchema = z
           ]).optional()
         }).optional()
       })
-      .default({}),
+      .default({})
+      .transform((config) => {
+        // Auto-enable thinking when thinkingBudget is set
+        if (config.thinkingConfig?.thinkingBudget !== undefined && 
+            config.thinkingConfig.enable_thinking === undefined) {
+          return {
+            ...config,
+            thinkingConfig: {
+              ...config.thinkingConfig,
+              enable_thinking: true
+            }
+          };
+        }
+        return config;
+      }),
   })
   .strip();
 export type GoogleAIChatMessage = z.infer<
@@ -182,4 +197,4 @@ export function containsImageContent(contents: GoogleAIChatMessage[]): boolean {
     const parts = Array.isArray(content.parts) ? content.parts : [content.parts];
     return parts.some(part => 'inlineData' in part);
   });
-}
+}