Skip to content

Commit

Permalink
support gpt-4o
Browse files Browse the repository at this point in the history
  • Loading branch information
leo4life2 committed May 13, 2024
1 parent 6b30e16 commit ef5f910
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 8 deletions.
6 changes: 5 additions & 1 deletion app/constant.ts
Original file line number Diff line number Diff line change
Expand Up @@ -127,6 +127,8 @@ export const KnowledgeCutOffDate: Record<string, string> = {
"gpt-4-turbo": "2023-12",
"gpt-4-turbo-2024-04-09": "2023-12",
"gpt-4-turbo-preview": "2023-12",
"gpt-4o": "2023-10",
"gpt-4o-2024-05-13": "2023-10",
"gpt-4-vision-preview": "2023-04",
// After improvements,
// it's now easier to add "KnowledgeCutOffDate" instead of stupid hardcoding it, as was done previously.
Expand All @@ -144,8 +146,10 @@ const openaiModels = [
"gpt-4-32k-0613",
"gpt-4-turbo",
"gpt-4-turbo-preview",
"gpt-4o",
"gpt-4o-2024-05-13",
"gpt-4-vision-preview",
"gpt-4-turbo-2024-04-09",
"gpt-4-turbo-2024-04-09"
];

const googleModels = [
Expand Down
11 changes: 4 additions & 7 deletions app/utils.ts
Original file line number Diff line number Diff line change
Expand Up @@ -290,16 +290,13 @@ export function getMessageImages(message: RequestMessage): 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",
"gemini-1.5-pro",
"gpt-4-turbo",
"gpt-4o",
];

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

0 comments on commit ef5f910

Please sign in to comment.