support new vision models

This commit is contained in:
Leo Li 2024-04-10 05:33:54 -04:00
parent 6319f41b2c
commit f101ee3c4f
No known key found for this signature in database
GPG Key ID: DC99441AF9F08B7B

View File

@ -290,8 +290,8 @@ 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)
const visionKeywords = ["vision", "claude-3"]; const visionKeywords = ["vision", "claude-3"];
const isGpt4Turbo = model.includes("gpt-4-turbo") && !model.includes("preview");
return visionKeywords.some((keyword) => model.includes(keyword)); return visionKeywords.some((keyword) => model.includes(keyword)) || isGpt4Turbo;
} }