Ver código fonte

fix: fix the method to detect vision model

Fred 1 ano atrás
pai
commit
43e5dc2292
2 arquivos alterados com 4 adições e 4 exclusões
  1. 1 1
      app/client/platforms/openai.ts
  2. 3 3
      app/utils.ts

+ 1 - 1
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,
       });
     }
 

+ 3 - 3
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")
   );
 }