From 9733efab86dce7e83505c4c9c9f04e6a9ed0a373 Mon Sep 17 00:00:00 2001
From: liwei <liwei-8466@qq.com>
Date: Fri, 21 Nov 2025 09:29:47 +0800
Subject: [PATCH] feat: add Nano Banana 2 model to Fal and Google providers

---
 src/server/ai/provider/fal.ts    | 8 ++++++++
 src/server/ai/provider/google.ts | 7 +++++++
 2 files changed, 15 insertions(+)

diff --git a/src/server/ai/provider/fal.ts b/src/server/ai/provider/fal.ts
index 1b243dc..485c0c3 100644
--- a/src/server/ai/provider/fal.ts
+++ b/src/server/ai/provider/fal.ts
@@ -30,6 +30,13 @@ const Fal: AiProvider = {
 	enabledByDefault: true,
 	settings: falSettingsSchema,
 	models: [
+		{
+			id: "fal-ai/nano-banana-pro",
+			name: "Nano Banana 2",
+			ability: "i2i",
+			maxInputImages: 4,
+			enabledByDefault: true,
+		},
 		{
 			id: "fal-ai/gemini-25-flash-image",
 			name: "Nano Banana",
@@ -70,6 +77,7 @@ const Fal: AiProvider = {
 			const genType = chooseAblility(request, model.ability);
 			let endpoint = "";
 			switch (request.modelId) {
+				case "fal-ai/nano-banana-pro":
 				case "fal-ai/gemini-25-flash-image":
 					if (genType === "i2i") {
 						endpoint = "/edit";
diff --git a/src/server/ai/provider/google.ts b/src/server/ai/provider/google.ts
index 064e1ba..441fe7c 100644
--- a/src/server/ai/provider/google.ts
+++ b/src/server/ai/provider/google.ts
@@ -89,6 +89,13 @@ const Google: AiProvider = {
 	enabledByDefault: true,
 	settings: googleSettingsSchema,
 	models: [
+		{
+			id: "gemini-3-pro-image-preview",
+			name: "Nano Banana 2",
+			ability: "i2i",
+			maxInputImages: 4,
+			enabledByDefault: true,
+		},
 		{
 			id: "gemini-2.5-flash-image-preview",
 			name: "Nano Banana",