From 991dd0490f83aae5681c11c8f3c8d083f0bf40f2 Mon Sep 17 00:00:00 2001
From: Hk-Gosuto <yominogosuto@gmail.com>
Date: Sun, 2 Jun 2024 20:05:56 +0800
Subject: [PATCH] feat: support visual model file upload

---
 app/components/chat.tsx | 2 +-
 app/utils.ts            | 8 ++++++++
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/app/components/chat.tsx b/app/components/chat.tsx
index 9db721600..c7e4af82a 100644
--- a/app/components/chat.tsx
+++ b/app/components/chat.tsx
@@ -531,7 +531,7 @@ export function ChatActions(props: {
   useEffect(() => {
     const show = isVisionModel(currentModel);
     setShowUploadImage(show);
-    setShowUploadFile(isEnableRAG && !show && isSupportRAGModel(currentModel));
+    setShowUploadFile(isEnableRAG && isSupportRAGModel(currentModel));
     if (!show) {
       props.setAttachImages([]);
       props.setUploading(false);
diff --git a/app/utils.ts b/app/utils.ts
index 0c3a92f62..f3a8edac4 100644
--- a/app/utils.ts
+++ b/app/utils.ts
@@ -267,6 +267,14 @@ export function isVisionModel(model: string) {
 }
 
 export function isSupportRAGModel(modelName: string) {
+  const specialModels = [
+    "gpt-4-turbo",
+    "gpt-4-turbo-2024-04-09",
+    "gpt-4o",
+    "gpt-4o-2024-05-13",
+  ];
+  if (specialModels.some((keyword) => modelName === keyword)) return true;
+  if (isVisionModel(modelName)) return false;
   return DEFAULT_MODELS.filter((model) => model.provider.id === "openai").some(
     (model) => model.name === modelName,
   );