fix: fix the method to detect vision model

This commit is contained in:
Fred 2024-02-28 11:33:43 +08:00
parent 08fa22749a
commit 43e5dc2292
No known key found for this signature in database
GPG Key ID: 4DABDA85EF70EC71
2 changed files with 4 additions and 4 deletions

View File

@ -116,7 +116,7 @@ export class ChatGPTApi implements LLMApi {
enumerable: true,
configurable: 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) {
return (
model.startsWith("gpt-4-vision") ||
model.startsWith("gemini-pro-vision") ||
!DEFAULT_MODELS.find((m) => m.name == model)
// model.startsWith("gpt-4-vision") ||
// model.startsWith("gemini-pro-vision") ||
model.includes("vision")
);
}