Merge pull request ChatGPTNextWeb#5071 from ZTH7/main

Fix defaultModel undefined error
cosmic-latte · Jul 25, 2024 · f5499ff · f5499ff
2 parents 941a03e + c4334d4
commit f5499ff
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/app/utils/model.ts b/app/utils/model.ts
@@ -99,12 +99,18 @@ export function collectModelTableWithDefaultModel(
 ) {
   let modelTable = collectModelTable(models, customModels);
   if (defaultModel && defaultModel !== "") {
-    modelTable[defaultModel] = {
-      ...modelTable[defaultModel],
-      name: defaultModel,
-      available: true,
-      isDefault: true,
-    };
+    if (defaultModel.includes('@')) {
+      if (defaultModel in modelTable) {
+        modelTable[defaultModel].isDefault = true;
+      }
+    } else {
+      for (const key of Object.keys(modelTable)) {
+        if (modelTable[key].available && key.split('@').shift() == defaultModel) {
+          modelTable[key].isDefault = true;
+          break;
+        }
+      }
+    }
   }
   return modelTable;
 }