Update model.ts

mingyue0094 · Jul 25, 2024 · c4334d4 · c4334d4
1 parent 5c04d3c
commit c4334d4
Showing 1 changed file with 5 additions and 12 deletions.
diff --git a/app/utils/model.ts b/app/utils/model.ts
@@ -99,20 +99,13 @@ export function collectModelTableWithDefaultModel(
 ) {
   let modelTable = collectModelTable(models, customModels);
   if (defaultModel && defaultModel !== "") {
-    const [modelName, providerName] = defaultModel.split("@");
-    if (providerName && providerName != "") {
-      modelTable[defaultModel] = {
-        ...modelTable[defaultModel],
-        name: modelTable[defaultModel]?.name ?? modelName,
-        displayName:
-          modelTable[defaultModel]?.displayName ??
-          modelName + "(" + providerName + ")",
-        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.startsWith(modelName)) {
+        if (modelTable[key].available && key.split('@').shift() == defaultModel) {
           modelTable[key].isDefault = true;
           break;
         }