merge: collectModelTableWithDefaultModel function

This commit is contained in:
Hk-Gosuto 2024-07-28 18:45:46 +08:00 committed by GitHub
parent 6014ab2e44
commit 52d5e963dd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -99,12 +99,18 @@ export function collectModelTableWithDefaultModel(
) { ) {
let modelTable = collectModelTable(models, customModels); let modelTable = collectModelTable(models, customModels);
if (defaultModel && defaultModel !== "") { if (defaultModel && defaultModel !== "") {
modelTable[defaultModel] = { if (defaultModel.includes('@')) {
...modelTable[defaultModel], if (defaultModel in modelTable) {
name: defaultModel, modelTable[defaultModel].isDefault = true;
available: true, }
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; return modelTable;
} }