7x31
/
ChatGPT-Next-Web
mirror of https://github.com/Yidadaa/ChatGPT-Next-Web.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat: swap name and displayName for bytedance in custom models

This commit is contained in:
Dogtiti 2024-07-06 22:59:20 +08:00
parent 2ec8b7a804
commit 1caa61f4c0
4 changed files with 23 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
app/client/api.ts
View File

@ -225,6 +225,8 @@ export function getClientApi(provider: ServiceProvider): ClientApi {
return new ClientApi(ModelProvider.GeminiPro); return new ClientApi(ModelProvider.GeminiPro);
case ServiceProvider.Anthropic: case ServiceProvider.Anthropic:
return new ClientApi(ModelProvider.Claude); return new ClientApi(ModelProvider.Claude);
case ServiceProvider.ByteDance:
return new ClientApi(ModelProvider.Doubao);
default: default:
return new ClientApi(ModelProvider.GPT); return new ClientApi(ModelProvider.GPT);
} }

6
app/config/server.ts
View File

@ -32,13 +32,13 @@ declare global {
GOOGLE_API_KEY?: string; GOOGLE_API_KEY?: string;
GOOGLE_URL?: string; GOOGLE_URL?: string;
// google tag manager
GTM_ID?: string;
// bytedance only // bytedance only
BYTEDANCE_URL?: string; BYTEDANCE_URL?: string;
BYTEDANCE_API_KEY?: string; BYTEDANCE_API_KEY?: string;
// google tag manager
GTM_ID?: string;
// custom template for preprocessing user input // custom template for preprocessing user input
DEFAULT_INPUT_TEMPLATE?: string; DEFAULT_INPUT_TEMPLATE?: string;
} }

9
app/constant.ts
View File

@ -183,7 +183,14 @@ const anthropicModels = [
"claude-3-5-sonnet-20240620", "claude-3-5-sonnet-20240620",
]; ];
const bytedanceModels = ["ep-20240520082937-424bw=Doubao-lite-4k"]; const bytedanceModels = [
"Doubao-lite-4k",
"Doubao-lite-32k",
"Doubao-lite-128k",
"Doubao-pro-4k",
"Doubao-pro-32k",
"Doubao-pro-128k",
];
export const DEFAULT_MODELS = [ export const DEFAULT_MODELS = [
...openaiModels.map((name) => ({ ...openaiModels.map((name) => ({

12
app/utils/model.ts
View File

@ -39,7 +39,7 @@ export function collectModelTable(
const available = !m.startsWith("-"); const available = !m.startsWith("-");
const nameConfig = const nameConfig =
m.startsWith("+") || m.startsWith("-") ? m.slice(1) : m; m.startsWith("+") || m.startsWith("-") ? m.slice(1) : m;
const [name, displayName] = nameConfig.split("="); let [name, displayName] = nameConfig.split("=");
// enable or disable all models // enable or disable all models
if (name === "all") { if (name === "all") {
@ -50,9 +50,17 @@ export function collectModelTable(
// 1. find model by name(), and set available value // 1. find model by name(), and set available value
let count = 0; let count = 0;
for (const fullName in modelTable) { for (const fullName in modelTable) {
if (fullName.split("@").shift() == name) { const [modelName, providerName] = fullName.split("@");
if (modelName === name) {
count += 1; count += 1;
modelTable[fullName]["available"] = available; modelTable[fullName]["available"] = available;
// swap name and displayName for bytedance
if (providerName === "bytedance") {
const tempName = name;
name = displayName;
displayName = tempName;
modelTable[fullName]["name"] = name;
}
if (displayName) { if (displayName) {
modelTable[fullName]["displayName"] = displayName; modelTable[fullName]["displayName"] = displayName;
} }