use regex to split modelName and providerName to process multi '@' symbols

This commit is contained in:
Q.A.zh
2024-08-09 16:05:28 +00:00
parent cf1c8e8f2a
commit b3c38a1ee4
4 changed files with 6 additions and 6 deletions

View File

@@ -79,10 +79,10 @@ export function collectModelTable(
);
} else {
// 1. find model by name, and set available value
const [customModelName, customProviderName] = name.split("@");
const [customModelName, customProviderName] = name.split(/@(?=[^@]*$)/);
let count = 0;
for (const fullName in modelTable) {
const [modelName, providerName] = fullName.split("@");
const [modelName, providerName] = fullName.split(/@(?=[^@]*$)/);
if (
customModelName == modelName &&
(customProviderName === undefined ||
@@ -102,7 +102,7 @@ export function collectModelTable(
}
// 2. if model not exists, create new model with available value
if (count === 0) {
let [customModelName, customProviderName] = name.split("@");
let [customModelName, customProviderName] = name.split(/@(?=[^@]*$)/);
const provider = customProvider(
customProviderName || customModelName,
);