Change method

This commit is contained in:
Mr. Z 2024-07-25 17:59:15 +08:00 committed by GitHub
parent 46a47db2d8
commit 5c04d3c5ea
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 19 additions and 7 deletions

View File

@ -99,13 +99,25 @@ export function collectModelTableWithDefaultModel(
) {
let modelTable = collectModelTable(models, customModels);
if (defaultModel && defaultModel !== "") {
modelTable[defaultModel] = {
...modelTable[defaultModel],
name: modelTable[defaultModel]?.name ?? defaultModel,
displayName: modelTable[defaultModel]?.displayName ?? defaultModel,
available: true,
isDefault: true,
};
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,
};
} else {
for (const key of Object.keys(modelTable)) {
if (modelTable[key].available && key.startsWith(modelName)) {
modelTable[key].isDefault = true;
break;
}
}
}
}
return modelTable;
}