imagegen ultra ISNT gemini-ultra

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")