feat: 支持向量模型

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:

ui/src/api/type/model.ts

@@ -53,6 +53,7 @@ interface Model {
    * 模型类型
    */
   model_type: string
+  permission_type: 'PUBLIC' | 'PRIVATE'
   /**
    * 基础模型
    */

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 = []

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 = []