From 46d3e7884b82b591d6f0204ee4083384864d95e6 Mon Sep 17 00:00:00 2001
From: lloydzhou <lloydzhou@qq.com>
Date: Fri, 12 Jul 2024 22:53:39 +0800
Subject: [PATCH] hotfix: bytedance custom models

---
 app/utils/model.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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