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, );