NW
/
ChatGPT-Next-Web
mirror of https://github.com/ChatGPTNextWeb/ChatGPT-Next-Web
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

hotfix: bytedance custom models

This commit is contained in:
lloydzhou 2024-07-12 22:53:39 +08:00
parent 476bdac717
commit 46d3e7884b
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
app/utils/model.ts
View File

@ -71,10 +71,14 @@ export function collectModelTable(
} }
// 2. if model not exists, create new model with available value // 2. if model not exists, create new model with available value
if (count === 0) { if (count === 0) {
const [customModelName, customProviderName] = name.split("@"); let [customModelName, customProviderName] = name.split("@");
const provider = customProvider( const provider = customProvider(
customProviderName || customModelName, customProviderName || customModelName,
); );
// swap name and displayName for bytedance
if (displayName && provider.providerName == "ByteDance") {
[customModelName, displayName] = [displayName, customModelName];
}
modelTable[`${customModelName}@${provider?.id}`] = { modelTable[`${customModelName}@${provider?.id}`] = {
name: customModelName, name: customModelName,
displayName: displayName || customModelName, displayName: displayName || customModelName,