From 4755cf6a25d358f850b1f387e6b4cac2280b9473 Mon Sep 17 00:00:00 2001
From: jialin <jialinkuang@126.com>
Date: Wed, 2 Apr 2025 17:54:26 +0800
Subject: [PATCH] fix: do not display modelscope source

---
 src/pages/llmodels/config/button-actions.ts | 4 +++-
 src/pages/llmodels/config/index.ts          | 2 ++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/pages/llmodels/config/button-actions.ts b/src/pages/llmodels/config/button-actions.ts
index 10299bb3..49bbdc70 100644
--- a/src/pages/llmodels/config/button-actions.ts
+++ b/src/pages/llmodels/config/button-actions.ts
@@ -2,6 +2,7 @@ import IconFont from '@/components/icon-font';
 import HotKeys from '@/config/hotkeys';
 import {
   DeleteOutlined,
+  DownloadOutlined,
   EditOutlined,
   ExperimentOutlined,
   RetweetOutlined,
@@ -17,6 +18,7 @@ const icons = {
   DeleteOutlined: React.createElement(DeleteOutlined),
   ThunderboltOutlined: React.createElement(ThunderboltOutlined),
   RetweetOutlined: React.createElement(RetweetOutlined),
+  DownloadOutlined: React.createElement(DownloadOutlined),
   Stop: React.createElement(IconFont, { type: 'icon-stop1' }),
   Play: React.createElement(IconFont, { type: 'icon-outline-play' }),
   Catalog: React.createElement(IconFont, { type: 'icon-catalog' }),
@@ -199,7 +201,7 @@ export const modelFileActions = [
   {
     label: 'resources.modelfiles.retry.download',
     key: 'retry',
-    icon: icons.RetweetOutlined
+    icon: icons.DownloadOutlined
   },
   {
     label: 'common.button.delete',
diff --git a/src/pages/llmodels/config/index.ts b/src/pages/llmodels/config/index.ts
index aeabc068..238610ec 100644
--- a/src/pages/llmodels/config/index.ts
+++ b/src/pages/llmodels/config/index.ts
@@ -186,6 +186,7 @@ interface ModelSource {
   modelscope_value: string;
   local_path: string;
   local_path_value: string;
+  model_scope: string;
 }
 
 export const modelSourceMap: ModelSource = {
@@ -195,6 +196,7 @@ export const modelSourceMap: ModelSource = {
   ollama_library_value: 'ollama_library',
   modelScope: 'ModelScope',
   modelscope_value: 'model_scope',
+  model_scope: 'ModelScope',
   local_path: 'Local Path',
   local_path_value: 'local_path'
 };