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="请给基础模型设置一个名称" /> + + + + 私有 + 公用 + +