From 98938b8ccf77b8f5308acc62a8c0828f11cd6880 Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Mon, 22 Jul 2024 12:03:05 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E6=A8=A1=E5=9E=8B?=
 =?UTF-8?q?=E7=AE=A1=E7=90=86=E4=B8=AD,=E5=85=AC=E5=85=B1=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B=E4=B8=8D=E8=83=BD=E4=BF=AE=E6=94=B9=20(#826)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../serializers/provider_serializers.py       |  2 +-
 ui/src/api/type/model.ts                      |  1 +
 ui/src/views/template/component/ModelCard.vue | 20 ++++++++++++++-----
 3 files changed, 17 insertions(+), 6 deletions(-)

diff --git a/apps/setting/serializers/provider_serializers.py b/apps/setting/serializers/provider_serializers.py
index a0f28e326..a7b23fe2f 100644
--- a/apps/setting/serializers/provider_serializers.py
+++ b/apps/setting/serializers/provider_serializers.py
@@ -91,7 +91,7 @@ class ModelSerializer(serializers.Serializer):
             return [
                 {'id': str(model.id), 'provider': model.provider, 'name': model.name, 'model_type': model.model_type,
                  'model_name': model.model_name, 'status': model.status, 'meta': model.meta,
-                 'permission_type': model.permission_type} for model in
+                 'permission_type': model.permission_type, 'user_id': model.user_id} for model in
                 model_query_set.filter(**query_params).order_by("-create_time")]
 
     class Edit(serializers.Serializer):
diff --git a/ui/src/api/type/model.ts b/ui/src/api/type/model.ts
index fcca438d5..8a41d68bf 100644
--- a/ui/src/api/type/model.ts
+++ b/ui/src/api/type/model.ts
@@ -53,6 +53,7 @@ interface Model {
    * 模型类型
    */
   model_type: string
+  user_id: string
   permission_type: 'PUBLIC' | 'PRIVATE'
   /**
    * 基础模型
diff --git a/ui/src/views/template/component/ModelCard.vue b/ui/src/views/template/component/ModelCard.vue
index 3d91533f9..9984d8690 100644
--- a/ui/src/views/template/component/ModelCard.vue
+++ b/ui/src/views/template/component/ModelCard.vue
@@ -47,7 +47,12 @@
 
       <div class="percentage-label flex-center">
         正在下载中 <span class="dotting"></span>
-        <el-button link type="primary" class="ml-16" @click.stop="cancelDownload"
+        <el-button
+          link
+          type="primary"
+          class="ml-16"
+          :disabled="!is_permisstion"
+          @click.stop="cancelDownload"
           >取消下载</el-button
         >
       </div>
@@ -56,7 +61,7 @@
     <template #mouseEnter>
       <div class="operation-button">
         <el-tooltip effect="dark" content="修改" placement="top">
-          <el-button text @click.stop="openEditModel">
+          <el-button text :disabled="!is_permisstion" @click.stop="openEditModel">
             <el-icon>
               <component
                 :is="
@@ -68,9 +73,8 @@
             </el-icon>
           </el-button>
         </el-tooltip>
-
         <el-tooltip effect="dark" content="删除" placement="top">
-          <el-button text @click.stop="deleteModel">
+          <el-button :disabled="!is_permisstion" text @click.stop="deleteModel">
             <el-icon><Delete /></el-icon>
           </el-button>
         </el-tooltip>
@@ -86,14 +90,20 @@ import { computed, ref, onMounted, onBeforeUnmount } from 'vue'
 import EditModel from '@/views/template/component/EditModel.vue'
 import DownloadLoading from '@/components/loading/DownloadLoading.vue'
 import { MsgConfirm } from '@/utils/message'
-
+import useStore from '@/stores'
 const props = defineProps<{
   model: Model
   provider_list: Array<Provider>
   updateModelById: (model_id: string, model: Model) => void
 }>()
+
+const { user } = useStore()
 const downModel = ref<Model>()
 
+const is_permisstion = computed(() => {
+  console.log(user.userInfo?.id, props.model.user_id)
+  return user.userInfo?.id == props.model.user_id
+})
 const currentModel = computed(() => {
   if (downModel.value) {
     return downModel.value