fix: shared model edit error

This commit is contained in:
wxg0103 2025-06-25 10:49:10 +08:00
parent 656c035b53
commit 93530e0a9b

View File

@ -126,7 +126,7 @@ class ModelSerializer(serializers.Serializer):
if with_valid: if with_valid:
super().is_valid(raise_exception=True) super().is_valid(raise_exception=True)
model = QuerySet(Model).get( model = QuerySet(Model).get(
id=self.data.get('id'), workspace_id=self.data.get('workspace_id') id=self.data.get('id'), workspace_id=self.data.get('workspace_id', 'None')
) )
return ModelSerializer.model_to_dict(model) return ModelSerializer.model_to_dict(model)
@ -135,7 +135,7 @@ class ModelSerializer(serializers.Serializer):
if with_valid: if with_valid:
super().is_valid(raise_exception=True) super().is_valid(raise_exception=True)
model = QuerySet(Model).filter(id=self.data.get("id"), model = QuerySet(Model).filter(id=self.data.get("id"),
workspace_id=self.data.get('workspace_id')).first() workspace_id=self.data.get('workspace_id', 'None')).first()
if model is None: if model is None:
raise AppApiException(500, _('Model does not exist')) raise AppApiException(500, _('Model does not exist'))
return {'id': str(model.id), 'provider': model.provider, 'name': model.name, 'model_type': model.model_type, return {'id': str(model.id), 'provider': model.provider, 'name': model.name, 'model_type': model.model_type,
@ -269,7 +269,7 @@ class ModelSerializer(serializers.Serializer):
super().is_valid(raise_exception=True) super().is_valid(raise_exception=True)
if QuerySet(Model).filter( if QuerySet(Model).filter(
name=self.data.get('name'), name=self.data.get('name'),
workspace_id=self.data.get('workspace_id') workspace_id=self.data.get('workspace_id', 'None')
).exists(): ).exists():
raise AppApiException( raise AppApiException(
500, 500,