From d6903105879e020595938baac6dc6c6cdf542079 Mon Sep 17 00:00:00 2001
From: Markus Ylisiurunen <8409947+markusylisiurunen@users.noreply.github.com>
Date: Thu, 18 Dec 2025 14:03:28 +0200
Subject: [PATCH] Fix Gemini 3 Flash Preview thinking levels (#212)

* use the correct Gemini 3 Flash Preview thinking levels

* fix a build error

* add changelog entry

* regenerate models

* make less assumptions about future models
---
 package-lock.json                   | 10 +++++-----
 packages/ai/CHANGELOG.md            |  4 ++++
 packages/ai/package.json            |  2 +-
 packages/ai/src/providers/google.ts |  2 ++
 packages/ai/src/stream.ts           | 28 +++++++++++++++++++++++-----
 5 files changed, 35 insertions(+), 11 deletions(-)

diff --git a/package-lock.json b/package-lock.json
index 9bd8c93b..13b1bffa 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -675,9 +675,9 @@
 			}
 		},
 		"node_modules/@google/genai": {
-			"version": "1.31.0",
-			"resolved": "https://registry.npmjs.org/@google/genai/-/genai-1.31.0.tgz",
-			"integrity": "sha512-rK0RKXxNkbK35eDl+G651SxtxwHNEOogjyeZJUJe+Ed4yxu3xy5ufCiU0+QLT7xo4M9Spey8OAYfD8LPRlYBKw==",
+			"version": "1.34.0",
+			"resolved": "https://registry.npmjs.org/@google/genai/-/genai-1.34.0.tgz",
+			"integrity": "sha512-vu53UMPvjmb7PGzlYu6Tzxso8Dfhn+a7eQFaS2uNemVtDZKwzSpJ5+ikqBbXplF7RGB1STcVDqCkPvquiwb2sw==",
 			"license": "Apache-2.0",
 			"dependencies": {
 				"google-auth-library": "^10.3.0",
@@ -687,7 +687,7 @@
 				"node": ">=20.0.0"
 			},
 			"peerDependencies": {
-				"@modelcontextprotocol/sdk": "^1.20.1"
+				"@modelcontextprotocol/sdk": "^1.24.0"
 			},
 			"peerDependenciesMeta": {
 				"@modelcontextprotocol/sdk": {
@@ -6197,7 +6197,7 @@
 			"license": "MIT",
 			"dependencies": {
 				"@anthropic-ai/sdk": "0.71.2",
-				"@google/genai": "1.31.0",
+				"@google/genai": "1.34.0",
 				"@mistralai/mistralai": "1.10.0",
 				"@sinclair/typebox": "^0.34.41",
 				"ajv": "^8.17.1",
diff --git a/packages/ai/CHANGELOG.md b/packages/ai/CHANGELOG.md
index 815d3ceb..be5c003d 100644
--- a/packages/ai/CHANGELOG.md
+++ b/packages/ai/CHANGELOG.md
@@ -2,6 +2,10 @@
 
 ## [Unreleased]
 
+### Added
+
+- **Gemini 3 Flash thinking support**: Extended thinking level support for Gemini 3 Flash models (MINIMAL, LOW, MEDIUM, HIGH) to match Pro models' capabilities. ([#212](https://github.com/badlogic/pi-mono/pull/212) by [@markusylisiurunen](https://github.com/markusylisiurunen))
+
 ## [0.22.3] - 2025-12-16
 
 ### Added
diff --git a/packages/ai/package.json b/packages/ai/package.json
index 4a7559e5..2cc66760 100644
--- a/packages/ai/package.json
+++ b/packages/ai/package.json
@@ -21,7 +21,7 @@
 	},
 	"dependencies": {
 		"@anthropic-ai/sdk": "0.71.2",
-		"@google/genai": "1.31.0",
+		"@google/genai": "1.34.0",
 		"@mistralai/mistralai": "1.10.0",
 		"@sinclair/typebox": "^0.34.41",
 		"ajv": "^8.17.1",
diff --git a/packages/ai/src/providers/google.ts b/packages/ai/src/providers/google.ts
index bf9e8050..8fcd9ef3 100644
--- a/packages/ai/src/providers/google.ts
+++ b/packages/ai/src/providers/google.ts
@@ -468,6 +468,8 @@ function mapStopReason(reason: FinishReason): StopReason {
 		case FinishReason.SAFETY:
 		case FinishReason.IMAGE_SAFETY:
 		case FinishReason.IMAGE_PROHIBITED_CONTENT:
+		case FinishReason.IMAGE_RECITATION:
+		case FinishReason.IMAGE_OTHER:
 		case FinishReason.RECITATION:
 		case FinishReason.FINISH_REASON_UNSPECIFIED:
 		case FinishReason.OTHER:
diff --git a/packages/ai/src/stream.ts b/packages/ai/src/stream.ts
index 382abd14..ea52a387 100644
--- a/packages/ai/src/stream.ts
+++ b/packages/ai/src/stream.ts
@@ -174,14 +174,14 @@ function mapOptionsForApi<TApi extends Api>(
 			const googleModel = model as Model<"google-generative-ai">;
 			const effort = clampReasoning(options.reasoning)!;
 
-			// Gemini 3 Pro models use thinkingLevel exclusively instead of thinkingBudget.
+			// Gemini 3 models use thinkingLevel exclusively instead of thinkingBudget.
 			// https://ai.google.dev/gemini-api/docs/thinking#set-budget
-			if (isGemini3ProModel(googleModel)) {
+			if (isGemini3ProModel(googleModel) || isGemini3FlashModel(googleModel)) {
 				return {
 					...base,
 					thinking: {
 						enabled: true,
-						level: getGoogleThinkingLevel(effort),
+						level: getGemini3ThinkingLevel(effort, googleModel),
 					},
 				} satisfies GoogleOptions;
 			}
@@ -210,13 +210,31 @@ function isGemini3ProModel(model: Model<"google-generative-ai">): boolean {
 	return model.id.includes("3-pro");
 }
 
-function getGoogleThinkingLevel(effort: ClampedReasoningEffort): ThinkingLevel {
-	// Gemini 3 Pro only supports LOW/HIGH (for now)
+function isGemini3FlashModel(model: Model<"google-generative-ai">): boolean {
+	// Covers gemini-3-flash, gemini-3-flash-preview, and possible other prefixed ids in the future
+	return model.id.includes("3-flash");
+}
+
+function getGemini3ThinkingLevel(effort: ClampedReasoningEffort, model: Model<"google-generative-ai">): ThinkingLevel {
+	if (isGemini3ProModel(model)) {
+		// Gemini 3 Pro only supports LOW/HIGH (for now)
+		switch (effort) {
+			case "minimal":
+			case "low":
+				return ThinkingLevel.LOW;
+			case "medium":
+			case "high":
+				return ThinkingLevel.HIGH;
+		}
+	}
+	// Gemini 3 Flash supports all four levels
 	switch (effort) {
 		case "minimal":
+			return ThinkingLevel.MINIMAL;
 		case "low":
 			return ThinkingLevel.LOW;
 		case "medium":
+			return ThinkingLevel.MEDIUM;
 		case "high":
 			return ThinkingLevel.HIGH;
 	}