feat: add gemini flash into vision model list

This commit is contained in:
Fred 2024-05-15 15:42:06 +08:00
parent 6612550c06
commit 4789a7f6a9
No known key found for this signature in database
GPG Key ID: 4DABDA85EF70EC71
1 changed files with 6 additions and 3 deletions

View File

@ -290,16 +290,19 @@ export function getMessageImages(message: RequestMessage): string[] {
} }
export function isVisionModel(model: string) { export function isVisionModel(model: string) {
// Note: This is a better way using the TypeScript feature instead of `&&` or `||` (ts v5.5.0-dev.20240314 I've been using) // 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 = [ const visionKeywords = [
"vision", "vision",
"claude-3", "claude-3",
"gemini-1.5-pro", "gemini-1.5-pro",
"gemini-1.5-flash",
]; ];
const isGpt4Turbo = model.includes("gpt-4-turbo") && !model.includes("preview"); const isGpt4Turbo =
model.includes("gpt-4-turbo") && !model.includes("preview");
return visionKeywords.some((keyword) => model.includes(keyword)) || isGpt4Turbo; return (
visionKeywords.some((keyword) => model.includes(keyword)) || isGpt4Turbo
);
} }