From 8e3963468e04a470a98d9805507a58ccb45633db Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <shaohu.zhang@fit2cloud.com>
Date: Thu, 18 Jul 2024 18:34:37 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=94=AF=E6=8C=81=E5=90=91=E9=87=8F?=
 =?UTF-8?q?=E6=A8=A1=E5=9E=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../setting/serializers/provider_serializers.py |  9 ++++++---
 ui/src/api/type/model.ts                        |  1 +
 .../template/component/CreateModelDialog.vue    | 17 ++++++++++++++---
 ui/src/views/template/component/EditModel.vue   | 16 +++++++++++++---
 4 files changed, 34 insertions(+), 9 deletions(-)

diff --git a/apps/setting/serializers/provider_serializers.py b/apps/setting/serializers/provider_serializers.py
index c155ca1cd..2a4a147f2 100644
--- a/apps/setting/serializers/provider_serializers.py
+++ b/apps/setting/serializers/provider_serializers.py
@@ -87,7 +87,8 @@ 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} for model in
+                 'model_name': model.model_name, 'status': model.status, 'meta': model.meta,
+                 'permission_type': model.permission_type} for model in
                 model_query_set.filter(**query_params).order_by("-create_time")]
 
     class Edit(serializers.Serializer):
@@ -198,7 +199,8 @@ class ModelSerializer(serializers.Serializer):
                 'meta': model.meta,
                 'credential': ModelProvideConstants[model.provider].value.get_model_credential(model.model_type,
                                                                                                model.model_name).encryption_dict(
-                    credential)}
+                    credential),
+                'permission_type': model.permission_type}
 
     class Operate(serializers.Serializer):
         id = serializers.UUIDField(required=True, error_messages=ErrMessage.uuid("模型id"))
@@ -224,7 +226,8 @@ 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, }
+                    'meta': model.meta
+                    }
 
         def delete(self, with_valid=True):
             if with_valid:
diff --git a/ui/src/api/type/model.ts b/ui/src/api/type/model.ts
index e07b36159..487cd439a 100644
--- a/ui/src/api/type/model.ts
+++ b/ui/src/api/type/model.ts
@@ -53,6 +53,7 @@ interface Model {
    * 模型类型
    */
   model_type: string
+  permission_type: 'PUBLIC' | 'PRIVATE'
   /**
    * 基础模型
    */
diff --git a/ui/src/views/template/component/CreateModelDialog.vue b/ui/src/views/template/component/CreateModelDialog.vue
index e01cc1efc..a4aed1c22 100644
--- a/ui/src/views/template/component/CreateModelDialog.vue
+++ b/ui/src/views/template/component/CreateModelDialog.vue
@@ -55,6 +55,15 @@
             placeholder="请给基础模型设置一个名称"
           />
         </el-form-item>
+        <el-form-item prop="model_type" :rules="base_form_data_rule.permission_type">
+          <template #label>
+            <span>权限</span>
+          </template>
+          <el-radio-group v-model="base_form_data.permission_type">
+            <el-radio value="PRIVATE" size="large" border>私有</el-radio>
+            <el-radio value="PUBLIC" size="large" border>公用</el-radio>
+          </el-radio-group>
+        </el-form-item>
         <el-form-item prop="model_type" :rules="base_form_data_rule.model_type">
           <template #label>
             <span>模型类型</span>
@@ -74,6 +83,7 @@
             ></el-option>
           </el-select>
         </el-form-item>
+
         <el-form-item prop="model_name" :rules="base_form_data_rule.model_name">
           <template #label>
             <div class="flex align-center" style="display: inline-flex">
@@ -150,17 +160,18 @@ const dialogVisible = ref<boolean>(false)
 
 const base_form_data_rule = ref<FormRules>({
   name: { required: true, trigger: 'blur', message: '模型名不能为空' },
+  permission_type: { required: true, trigger: 'change', message: '权限不能为空' },
   model_type: { required: true, trigger: 'change', message: '模型类型不能为空' },
   model_name: { required: true, trigger: 'change', message: '基础模型不能为空' }
 })
 
 const base_form_data = ref<{
   name: string
-
+  permission_type: string
   model_type: string
 
   model_name: string
-}>({ name: '', model_type: '', model_name: '' })
+}>({ name: '', model_type: '', model_name: '', permission_type: 'PRIVATE' })
 
 const credential_form_data = ref<Dict<any>>({})
 
@@ -212,7 +223,7 @@ const list_base_model = (model_type: any) => {
 }
 
 const close = () => {
-  base_form_data.value = { name: '', model_type: '', model_name: '' }
+  base_form_data.value = { name: '', model_type: '', model_name: '', permission_type: 'PRIVATE' }
   credential_form_data.value = {}
   model_form_field.value = []
   base_model_list.value = []
diff --git a/ui/src/views/template/component/EditModel.vue b/ui/src/views/template/component/EditModel.vue
index d506f041c..6a8451169 100644
--- a/ui/src/views/template/component/EditModel.vue
+++ b/ui/src/views/template/component/EditModel.vue
@@ -48,6 +48,15 @@
             placeholder="请给基础模型设置一个名称"
           />
         </el-form-item>
+        <el-form-item prop="model_type" :rules="base_form_data_rule.permission_type">
+          <template #label>
+            <span>权限</span>
+          </template>
+          <el-radio-group v-model="base_form_data.permission_type">
+            <el-radio value="PRIVATE" size="large" border>私有</el-radio>
+            <el-radio value="PUBLIC" size="large" border>公用</el-radio>
+          </el-radio-group>
+        </el-form-item>
         <el-form-item prop="model_type" :rules="base_form_data_rule.model_type">
           <template #label>
             <span>模型类型</span>
@@ -151,11 +160,11 @@ const base_form_data_rule = ref<FormRules>({
 
 const base_form_data = ref<{
   name: string
-
+  permission_type: string
   model_type: string
 
   model_name: string
-}>({ name: '', model_type: '', model_name: '' })
+}>({ name: '', model_type: '', model_name: '', permission_type: 'PRIVATE' })
 
 const credential_form_data = ref<Dict<any>>({})
 
@@ -204,6 +213,7 @@ const open = (provider: Provider, model: Model) => {
 
     base_form_data.value = {
       name: model.name,
+      permission_type: model.permission_type,
       model_type: model.model_type,
       model_name: model.model_name
     }
@@ -214,7 +224,7 @@ const open = (provider: Provider, model: Model) => {
 }
 
 const close = () => {
-  base_form_data.value = { name: '', model_type: '', model_name: '' }
+  base_form_data.value = { name: '', model_type: '', model_name: '', permission_type: '' }
   dynamicsFormRef.value?.ruleFormRef?.resetFields()
   credential_form_data.value = {}
   model_form_field.value = []