feat: support visual model file upload
This commit is contained in:
parent
fe9a17693c
commit
991dd0490f
|
@ -531,7 +531,7 @@ export function ChatActions(props: {
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
const show = isVisionModel(currentModel);
|
const show = isVisionModel(currentModel);
|
||||||
setShowUploadImage(show);
|
setShowUploadImage(show);
|
||||||
setShowUploadFile(isEnableRAG && !show && isSupportRAGModel(currentModel));
|
setShowUploadFile(isEnableRAG && isSupportRAGModel(currentModel));
|
||||||
if (!show) {
|
if (!show) {
|
||||||
props.setAttachImages([]);
|
props.setAttachImages([]);
|
||||||
props.setUploading(false);
|
props.setUploading(false);
|
||||||
|
|
|
@ -267,6 +267,14 @@ export function isVisionModel(model: string) {
|
||||||
}
|
}
|
||||||
|
|
||||||
export function isSupportRAGModel(modelName: 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(
|
return DEFAULT_MODELS.filter((model) => model.provider.id === "openai").some(
|
||||||
(model) => model.name === modelName,
|
(model) => model.name === modelName,
|
||||||
);
|
);
|
||||||
|
|
Loading…
Reference in New Issue