Explorar el Código

Merge pull request #4306 from H0llyW00dzZ/simplify-cherry-pick

[Cherry Pick] Improve [Utils] Check Vision Model
fred-bf hace 1 año
padre
commit
3ba984d09e
Se han modificado 1 ficheros con 7 adiciones y 5 borrados
  1. 7 5
      app/utils.ts

+ 7 - 5
app/utils.ts

@@ -292,9 +292,11 @@ export function getMessageImages(message: RequestMessage): string[] {
 }
 
 export function isVisionModel(model: string) {
-  return (
-    // model.startsWith("gpt-4-vision") ||
-    // model.startsWith("gemini-pro-vision") ||
-    model.includes("vision")
-  );
+  // 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",
+  ];
+
+  return visionKeywords.some(keyword => model.includes(keyword));
 }