feat: o1 model

This commit is contained in:
Hk-Gosuto 2024-09-17 16:53:45 +08:00
parent ab7c13c59b
commit 128038312f
2 changed files with 10 additions and 3 deletions

View File

@ -198,9 +198,12 @@ export const KnowledgeCutOffDate: Record<string, string> = {
"gpt-4o": "2023-10",
"gpt-4o-2024-05-13": "2023-10",
"gpt-4o-2024-08-06": "2023-10",
"chatgpt-4o-latest": "2023-10",
"gpt-4o-mini": "2023-10",
"gpt-4o-mini-2024-07-18": "2023-10",
"gpt-4-vision-preview": "2023-04",
"o1-mini": "2023-10",
"o1-preview": "2023-10",
// After improvements,
// it's now easier to add "KnowledgeCutOffDate" instead of stupid hardcoding it, as was done previously.
"gemini-pro": "2023-12",
@ -237,11 +240,15 @@ const openaiModels = [
"gpt-4o",
"gpt-4o-2024-05-13",
"gpt-4o-2024-08-06",
"chatgpt-4o-latest",
"gpt-4o-mini",
"gpt-4o-mini-2024-07-18",
"gpt-4-vision-preview",
"gpt-4-turbo-2024-04-09",
"gpt-4-1106-preview",
"dall-e-3",
"o1-mini",
"o1-preview",
];
const googleModels = [

View File

@ -306,7 +306,7 @@ export function isFunctionCallModel(modelName: string) {
"claude-3-5-sonnet-20240620",
];
if (specialModels.some((keyword) => modelName === keyword)) return true;
return DEFAULT_MODELS.filter((model) => model.provider.id === "openai").some(
(model) => model.name === modelName,
);
return DEFAULT_MODELS.filter(
(model) => model.provider.id === "openai" && !model.name.includes("o1"),
).some((model) => model.name === modelName);
}