Merge pull request #4162 from fred-bf/fix/identify-vision-model

fix: fix the method to detect vision model
This commit is contained in:
fred-bf 2024-02-28 11:35:22 +08:00 committed by GitHub
commit 8b821ac0c9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 4 deletions

View File

@ -116,7 +116,7 @@ export class ChatGPTApi implements LLMApi {
enumerable: true, enumerable: true,
configurable: true, configurable: true,
writable: true, writable: true,
value: Math.max(modelConfig.max_tokens, 4096), value: modelConfig.max_tokens,
}); });
} }

View File

@ -292,8 +292,8 @@ export function getMessageImages(message: RequestMessage): string[] {
export function isVisionModel(model: string) { export function isVisionModel(model: string) {
return ( return (
model.startsWith("gpt-4-vision") || // model.startsWith("gpt-4-vision") ||
model.startsWith("gemini-pro-vision") || // model.startsWith("gemini-pro-vision") ||
!DEFAULT_MODELS.find((m) => m.name == model) model.includes("vision")
); );
} }