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

---
 frontend/packages/core/src/const/ai-service/const.tsx     | 4 ++--
 frontend/packages/core/src/const/ai-service/type.ts       | 1 +
 .../pages/aiService/api/AiServiceInsideRouterCreate.tsx   | 8 +++++---
 .../src/pages/aiSetting/contexts/ModelsDetailTable.tsx    | 2 +-
 4 files changed, 9 insertions(+), 6 deletions(-)

diff --git a/frontend/packages/core/src/const/ai-service/const.tsx b/frontend/packages/core/src/const/ai-service/const.tsx
index ccf72b07..a058f636 100644
--- a/frontend/packages/core/src/const/ai-service/const.tsx
+++ b/frontend/packages/core/src/const/ai-service/const.tsx
@@ -19,11 +19,11 @@ export const AI_SERVICE_ROUTER_TABLE_COLUMNS: PageProColumns<AiServiceRouterTabl
   },
   {
     title: '模型',
-    dataIndex: ['model', 'logo'],
+    dataIndex: ['model', 'name'],
     ellipsis: true,
     render: (_: React.ReactNode, entity: AiServiceRouterTableListItem) => (
       <div className="flex items-center gap-[2px] ">
-        <span>{entity.model.id}</span>
+        <span>{entity.model.name}</span>
       </div>
     )
   },
diff --git a/frontend/packages/core/src/const/ai-service/type.ts b/frontend/packages/core/src/const/ai-service/type.ts
index d7086490..0c3e2841 100644
--- a/frontend/packages/core/src/const/ai-service/type.ts
+++ b/frontend/packages/core/src/const/ai-service/type.ts
@@ -104,6 +104,7 @@ export type AiServiceRouterTableListItem = {
     model:{
         id:string
         logo:string
+        name:string
     }
 };
 
diff --git a/frontend/packages/core/src/pages/aiService/api/AiServiceInsideRouterCreate.tsx b/frontend/packages/core/src/pages/aiService/api/AiServiceInsideRouterCreate.tsx
index 834110ad..bace0c54 100644
--- a/frontend/packages/core/src/pages/aiService/api/AiServiceInsideRouterCreate.tsx
+++ b/frontend/packages/core/src/pages/aiService/api/AiServiceInsideRouterCreate.tsx
@@ -80,7 +80,7 @@ const AiServiceInsideRouterCreate = () => {
             timeout,
             retry,
             aiPrompt: { variables: variables, prompt: prompt },
-            aiModel: { id: defaultLlm?.id, provider: defaultLlm?.provider, config: defaultLlm?.config, type: defaultLlm?.type },
+            aiModel: { id: defaultLlm?.id, provider: defaultLlm?.provider, config: defaultLlm?.config, type: defaultLlm?.type, name: defaultLlm?.name },
             disabled
           }
           return fetchData<BasicResponse<null>>('service/ai-router', {
@@ -230,7 +230,8 @@ const AiServiceInsideRouterCreate = () => {
                 return {
                   ...prev,
                   logo: llmSetting?.logo,
-                  scopes: llmSetting?.scopes
+                  scopes: llmSetting?.scopes,
+                  name: llmSetting?.name
                 } as AiProviderDefaultConfig & { config: string }
               })
             }
@@ -273,7 +274,8 @@ const AiServiceInsideRouterCreate = () => {
                 return {
                   ...prev,
                   logo: llmSetting?.logo,
-                  scopes: llmSetting?.scopes
+                  scopes: llmSetting?.scopes,
+                  name: llmSetting?.name
                 } as AiProviderDefaultConfig & { config: string }
               })
             }
diff --git a/frontend/packages/core/src/pages/aiSetting/contexts/ModelsDetailTable.tsx b/frontend/packages/core/src/pages/aiSetting/contexts/ModelsDetailTable.tsx
index 32318081..fd658a54 100644
--- a/frontend/packages/core/src/pages/aiSetting/contexts/ModelsDetailTable.tsx
+++ b/frontend/packages/core/src/pages/aiSetting/contexts/ModelsDetailTable.tsx
@@ -156,7 +156,7 @@ const ModelsDetailTable = (props: { providerID?: string }) => {
         const tableData = response.data.llms.map((item) => {
           return {
             ...item,
-            modelValue: `${response.data.provider.id}/${item.id}`
+            modelValue: `${response.data.provider.id}/${item.name}`
           }
         })
         return {