Parcourir la source

feat: add gemini flash into vision model list

Fred il y a 1 an
Parent
commit
4789a7f6a9
1 fichiers modifiés avec 6 ajouts et 3 suppressions
  1. 6 3
      app/utils.ts

+ 6 - 3
app/utils.ts

@@ -290,16 +290,19 @@ export function getMessageImages(message: RequestMessage): string[] {
 }
 
 export function isVisionModel(model: string) {
-  
   // Note: This is a better way using the TypeScript feature instead of `&&` or `||` (ts v5.5.0-dev.20240314 I've been using)
 
   const visionKeywords = [
     "vision",
     "claude-3",
     "gemini-1.5-pro",
+    "gemini-1.5-flash",
   ];
 
-  const isGpt4Turbo = model.includes("gpt-4-turbo") && !model.includes("preview");
+  const isGpt4Turbo =
+    model.includes("gpt-4-turbo") && !model.includes("preview");
 
-  return visionKeywords.some((keyword) => model.includes(keyword)) || isGpt4Turbo;
+  return (
+    visionKeywords.some((keyword) => model.includes(keyword)) || isGpt4Turbo
+  );
 }