update chat model filters for openai (#2803)

Mintplex-Labs · Dec 11, 2024 · a69997a · a69997a
1 parent a0c5d89
commit a69997a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/server/utils/AiProviders/openAi/index.js b/server/utils/AiProviders/openAi/index.js
@@ -62,7 +62,9 @@ class OpenAiLLM {
   // we don't want to hit the OpenAI api every chat because it will get spammed
   // and introduce latency for no reason.
   async isValidChatCompletionModel(modelName = "") {
-    const isPreset = modelName.toLowerCase().includes("gpt");
+    const isPreset =
+      modelName.toLowerCase().includes("gpt") ||
+      modelName.toLowerCase().includes("o1");
     if (isPreset) return true;
 
     const model = await this.openai.models

diff --git a/server/utils/helpers/customModels.js b/server/utils/helpers/customModels.js
@@ -141,9 +141,17 @@ async function openAiModels(apiKey = null) {
     });
 
   const gpts = allModels
-    .filter((model) => model.id.startsWith("gpt") || model.id.startsWith("o1"))
     .filter(
-      (model) => !model.id.includes("vision") && !model.id.includes("instruct")
+      (model) =>
+        (model.id.includes("gpt") && !model.id.startsWith("ft:")) ||
+        model.id.includes("o1")
+    )
+    .filter(
+      (model) =>
+        !model.id.includes("vision") &&
+        !model.id.includes("instruct") &&
+        !model.id.includes("audio") &&
+        !model.id.includes("realtime")
     )
     .map((model) => {
       return {