diff --git a/apps/models_provider/models/model_management.py b/apps/models_provider/models/model_management.py index 9f0a70e52..c1c36a426 100644 --- a/apps/models_provider/models/model_management.py +++ b/apps/models_provider/models/model_management.py @@ -46,4 +46,4 @@ class Model(AppModelMixin): class Meta: db_table = "model" - unique_together = ['name', 'user_id'] + unique_together = ['name', 'workspace_id'] diff --git a/apps/models_provider/views/model.py b/apps/models_provider/views/model.py index 14832a50e..dbfd95c69 100644 --- a/apps/models_provider/views/model.py +++ b/apps/models_provider/views/model.py @@ -73,8 +73,9 @@ class ModelSetting(APIView): ) def post(self, request: Request, workspace_id: str): return result.success( - ModelSerializer.Create(data={**request.data, 'user_id': request.user.id}).insert(workspace_id, - with_valid=True)) + ModelSerializer.Create( + data={**request.data, 'user_id': request.user.id, 'workspace_id': workspace_id}).insert(workspace_id, + with_valid=True)) # @extend_schema(methods=['PUT'], # summary=_('Update model'), @@ -266,4 +267,4 @@ class ModelList(APIView): return result.success( ModelSerializer.Query( data={**query_params_to_single_dict(request.query_params)}).model_list(workspace_id=workspace_id, - with_valid=True)) + with_valid=True))