Merge pull request ChatGPTNextWeb#4162 from fred-bf/fix/identify-visi…

…on-model fix: fix the method to detect vision model
sasukebinbin · Feb 28, 2024 · 8b821ac · 8b821ac
2 parents 08fa227 + 43e5dc2
commit 8b821ac
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/app/client/platforms/openai.ts b/app/client/platforms/openai.ts
@@ -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,
       });
     }
 

diff --git a/app/utils.ts b/app/utils.ts
@@ -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")
   );
 }