From 68f26d63d766b110c4e1027fa8b697cf667ca8b3 Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Wed, 25 Jun 2025 14:52:06 +0800
Subject: [PATCH] fix: Model data processing (#3383)

---
 ui/src/api/model/model.ts | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/ui/src/api/model/model.ts b/ui/src/api/model/model.ts
index 938557344..acd75c843 100644
--- a/ui/src/api/model/model.ts
+++ b/ui/src/api/model/model.ts
@@ -37,7 +37,19 @@ const getSelectModelList: (
   data?: ListModelRequest,
   loading?: Ref<boolean>,
 ) => Promise<Result<Array<Model>>> = (data, loading) => {
-  return get(`${prefix.value}/model_list`, data, loading)
+  return get(`${prefix.value}/model_list`, data, loading).then((ok) => {
+    return {
+      ...ok,
+      data: [
+        ...ok.data.shared_model.map((m: any) => {
+          return { ...m, type: 'share' }
+        }),
+        ...ok.data.model.map((m: any) => {
+          return { ...m, type: 'workspace' }
+        }),
+      ],
+    }
+  })
 }
 
 /**