From a4c54cae60820a8a026481b91a11a9e7a842699c Mon Sep 17 00:00:00 2001
From: H0llyW00dzZ <priv8@btz.pm>
Date: Fri, 15 Mar 2024 09:33:21 +0700
Subject: [PATCH] Improve [Utils] Check Vision Model

- [+] refactor(utils.ts): improve isVisionModel function to use array.some instead of model.includes
---
 app/utils.ts | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/app/utils.ts b/app/utils.ts
index 8b755afeac1..b4fc1980ce3 100644
--- a/app/utils.ts
+++ b/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));
 }