From f020ab4b28eb96a4d832109a699c58a529932b92 Mon Sep 17 00:00:00 2001
From: CaptainB <bin@fit2cloud.com>
Date: Wed, 30 Jul 2025 10:27:08 +0800
Subject: [PATCH] feat: refactor getSelectModelList to use workspace_id from
 data and update model retrieval logic

---
 ui/src/api/system-resource-management/model.ts | 5 ++---
 ui/src/api/type/model.ts                       | 2 ++
 ui/src/views/knowledge/component/BaseForm.vue  | 9 ++++++++-
 3 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/ui/src/api/system-resource-management/model.ts b/ui/src/api/system-resource-management/model.ts
index 2a82a1da0..8c847a742 100644
--- a/ui/src/api/system-resource-management/model.ts
+++ b/ui/src/api/system-resource-management/model.ts
@@ -30,10 +30,9 @@ const getModelListPage: (
  */
 const getSelectModelList: (
   data?: ListModelRequest,
-  workspace_id?: string,
   loading?: Ref<boolean>,
-) => Promise<Result<Array<Model>>> = (data, workspace_id, loading) => {
-  return get(`${prefix}/workspace/${workspace_id}/model/model_list`, data, loading).then((ok) => {
+) => Promise<Result<Array<Model>>> = (data, loading) => {
+  return get(`${prefix}/model/model_list`, data, loading).then((ok) => {
     return {
       ...ok,
       data: [
diff --git a/ui/src/api/type/model.ts b/ui/src/api/type/model.ts
index ff72105b6..09a0092e3 100644
--- a/ui/src/api/type/model.ts
+++ b/ui/src/api/type/model.ts
@@ -37,6 +37,8 @@ interface ListModelRequest {
    * 供应商
    */
   provider?: string
+
+  workspace_id?: string
 }
 
 interface Model {
diff --git a/ui/src/views/knowledge/component/BaseForm.vue b/ui/src/views/knowledge/component/BaseForm.vue
index c0917668e..25b400d63 100644
--- a/ui/src/views/knowledge/component/BaseForm.vue
+++ b/ui/src/views/knowledge/component/BaseForm.vue
@@ -60,6 +60,7 @@ const form = ref<knowledgeData>({
   desc: '',
   embedding_model_id: '',
 })
+const workspace_id = ref('')
 
 const rules = reactive({
   name: [
@@ -96,6 +97,10 @@ watch(
       form.value.name = value.name
       form.value.desc = value.desc
       form.value.embedding_model_id = value.embedding_model_id
+      workspace_id.value = value.workspace_id || ''
+
+      // 重新刷新模型列表
+      getSelectModel()
     }
   },
   {
@@ -115,8 +120,9 @@ function validate() {
 
 function getSelectModel() {
   loading.value = true
+  console.log(workspace_id.value)
   loadSharedApi({ type: 'model', systemType: props.apiType })
-    .getSelectModelList({ model_type: 'EMBEDDING' })
+    .getSelectModelList({ model_type: 'EMBEDDING', workspace_id: workspace_id.value })
     .then((res: any) => {
       modelOptions.value = groupBy(res?.data, 'provider')
       loading.value = false
@@ -129,6 +135,7 @@ function getSelectModel() {
 onMounted(() => {
   getSelectModel()
 })
+
 onUnmounted(() => {
   form.value = {
     name: '',