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