diff --git a/apps/models_provider/serializers/model_serializer.py b/apps/models_provider/serializers/model_serializer.py index 84977af73de..0c60472da10 100644 --- a/apps/models_provider/serializers/model_serializer.py +++ b/apps/models_provider/serializers/model_serializer.py @@ -485,18 +485,19 @@ def is_valid(self, *, raise_exception=False): super().is_valid(raise_exception=True) validated_data = self.validated_data - model = ( - QuerySet(Model) - .filter( - id=validated_data.get("id"), - workspace_id=validated_data.get("workspace_id"), - ) - .first() - ) + model = QuerySet(Model).filter( + id=validated_data["id"], + ).first() if model is None: raise AppApiException(500, _("Model does not exist")) + if model.workspace_id == "None": + return model + + if model.workspace_id != validated_data["workspace_id"]: + raise AppApiException(500, _("Model does not exist")) + return model def get_model_params(self, with_valid=True):