瀏覽代碼

Improve [Utils] Check Vision Model

- [+] refactor(utils.ts): improve isVisionModel function to use array.some instead of model.includes
H0llyW00dzZ 1 年之前
父節點
當前提交
a4c54cae60
共有 1 個文件被更改,包括 7 次插入5 次删除
  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));
 }