⚡️ again, improve image support autodetection

continuedev · Apr 18, 2024 · b50c327 · b50c327
1 parent 91b4ce2
commit b50c327
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/core/llm/autodetect.ts b/core/llm/autodetect.ts
@@ -57,14 +57,17 @@ const MODEL_SUPPORTS_IMAGES: string[] = [
   "claude-3",
   "gemini-ultra",
   "gemini-1.5-pro",
+  "sonnet",
+  "opus",
+  "haiku",
 ];
 
 function modelSupportsImages(provider: ModelProvider, model: string): boolean {
   if (!PROVIDER_SUPPORTS_IMAGES.includes(provider)) {
     return false;
   }
 
-  if (MODEL_SUPPORTS_IMAGES.includes(model)) {
+  if (MODEL_SUPPORTS_IMAGES.includes(model.toLowerCase())) {
     return true;
   }