From a9168c7b91a89b6e71db480b8281275aaec7c95d Mon Sep 17 00:00:00 2001
From: ningyv <1793599591@qq.com>
Date: Tue, 11 Mar 2025 10:23:26 +0800
Subject: [PATCH] feat: feature/1.6-Integrate custom model

---
 .../pages/aiService/api/AiServiceInsideRouterCreate.tsx  | 9 +++++----
 frontend/packages/core/src/pages/system/SystemConfig.tsx | 2 +-
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/frontend/packages/core/src/pages/aiService/api/AiServiceInsideRouterCreate.tsx b/frontend/packages/core/src/pages/aiService/api/AiServiceInsideRouterCreate.tsx
index 72eebb2d..834110ad 100644
--- a/frontend/packages/core/src/pages/aiService/api/AiServiceInsideRouterCreate.tsx
+++ b/frontend/packages/core/src/pages/aiService/api/AiServiceInsideRouterCreate.tsx
@@ -162,7 +162,8 @@ const AiServiceInsideRouterCreate = () => {
                 provider: aiModel?.provider,
                 id: aiModel?.id,
                 config: aiModel.config,
-                type: aiModel?.type
+                type: aiModel?.type,
+                name: aiModel?.name
               }) as AiProviderDefaultConfig & { config: string }
           )
           getDefaultModelConfig({
@@ -216,8 +217,8 @@ const AiServiceInsideRouterCreate = () => {
                   provider: localId,
                   name: aiServiceInfo?.name,
                   config: llmSetting?.defaultConfig || '',
+                  ...(llmSetting ?? {}),
                   type: 'local',
-                  ...(llmSetting ?? {})
                 } as AiProviderDefaultConfig & { config: string }
               })
             }
@@ -259,8 +260,8 @@ const AiServiceInsideRouterCreate = () => {
                   provider: data.provider.id,
                   name: data.provider.name,
                   config: llmSetting?.config || '',
+                  ...(llmSetting ?? {}),
                   type: 'online',
-                  ...(llmSetting ?? {})
                 } as AiProviderDefaultConfig & { config: string }
               })
             }
@@ -385,7 +386,7 @@ const AiServiceInsideRouterCreate = () => {
                 className="flex items-center  h-[24px] ai-setting-svg-container "
                 dangerouslySetInnerHTML={{ __html: defaultLlm?.logo || '' }}
               ></span>
-              <span>{defaultLlm?.id || defaultLlm?.defaultLlm}</span>
+              <span>{defaultLlm?.name || defaultLlm?.id || defaultLlm?.defaultLlm}</span>
               {defaultLlm?.scopes?.map((x) => <Tag>{x?.toLocaleUpperCase()}</Tag>)}
             </div>
           </Button>
diff --git a/frontend/packages/core/src/pages/system/SystemConfig.tsx b/frontend/packages/core/src/pages/system/SystemConfig.tsx
index 7062c530..f3eacd6e 100644
--- a/frontend/packages/core/src/pages/system/SystemConfig.tsx
+++ b/frontend/packages/core/src/pages/system/SystemConfig.tsx
@@ -138,7 +138,7 @@ const SystemConfig = forwardRef<SystemConfigHandle>((_, ref) => {
       const { code, data, msg } = response
       if (code === STATUS_CODE.SUCCESS) {
         const localModelList = data.llms?.map((x: any) => {
-          return { ...x, label: x.id, value: x.id }
+          return { ...x, label: x.name || x.id, value: x.id }
         })
         setModelList(localModelList)
         if (setDefaultLlm && localModelList.length > 0) {