Skip to content

Commit de1ec6a

Browse files
committed
refactor: improve model data serialization and update user field names
1 parent 179b8b5 commit de1ec6a

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

apps/models_provider/serializers/model_serializer.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -383,8 +383,11 @@ def share_list(self, workspace_id, with_valid=True):
383383
self.is_valid(raise_exception=True)
384384
user_id = self.data.get("user_id")
385385
query_params = self._build_query_params(workspace_id, False, user_id)
386-
return [self._build_model_data(model) for model in
387-
query_params.get('model_query_set').order_by("-create_time")]
386+
return [
387+
self._build_model_data(
388+
model
389+
) for model in query_params.get('model_query_set').order_by("-create_time")
390+
]
388391

389392
def model_list(self, workspace_id, with_valid=True):
390393
if with_valid:
@@ -451,7 +454,8 @@ def _build_model_data(self, model):
451454
'status': model.status,
452455
'meta': model.meta,
453456
'user_id': model.user_id,
454-
'username': model.user.nick_name
457+
'username': model.user.username,
458+
'nick_name': model.user.nick_name,
455459
}
456460

457461
def page(self, current_page, page_size):

0 commit comments

Comments
 (0)