From fa13d06f45adb0c1824e75aa89b88bb494e72c67 Mon Sep 17 00:00:00 2001
From: reanon <>
Date: Wed, 25 Jun 2025 08:12:15 +0200
Subject: [PATCH] imagegen ultra ISNT gemini-ultra

---
 src/shared/models.ts | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/shared/models.ts b/src/shared/models.ts
index ae68c1a..58b28e6 100644
--- a/src/shared/models.ts
+++ b/src/shared/models.ts
@@ -264,6 +264,11 @@ export function getClaudeModelFamily(model: string): AnthropicModelFamily {
 }
 
 export function getGoogleAIModelFamily(model: string): GoogleAIModelFamily {
+  // Ignore any models with "imagen" in the name
+  if (model.toLowerCase().includes("imagen")) {
+    return "gemini-pro"; // Default to pro for imagen models
+  }
+  
   return model.includes("ultra")
     ? "gemini-ultra"
     : model.includes("flash")