Update model.ts

This commit is contained in:
Mr. Z 2024-07-25 20:03:54 +08:00 committed by GitHub
parent 5c04d3c5ea
commit c4334d4e5f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

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