diff --git a/server/biz/model/dto/training.go b/server/biz/model/dto/training.go index 83f50d3..851a098 100644 --- a/server/biz/model/dto/training.go +++ b/server/biz/model/dto/training.go @@ -457,6 +457,7 @@ type ModelCatalogItem struct { Notes string `json:"notes"` PublishedAt *gtime.Time `json:"publishedAt"` DownloadUrl string `json:"downloadUrl"` // /download/trainings/<文件名前缀>.tflite(前缀空回退数据集名) + CoverUrl string `json:"coverUrl"` // /api/v1/app/cover?namePrefix=<前缀>(App 模型清单缩略图) } type ModelCatalogRes struct { diff --git a/server/biz/service/model_version.go b/server/biz/service/model_version.go index 3bd078e..54d0469 100644 --- a/server/biz/service/model_version.go +++ b/server/biz/service/model_version.go @@ -3,6 +3,7 @@ package service import ( "context" "encoding/json" + "net/url" "observer-server/biz/dao" "observer-server/biz/model/dto" @@ -39,6 +40,7 @@ func (s *modelVersionService) ClientCatalog(ctx context.Context) (*dto.ModelCata Notes: v.Notes, PublishedAt: v.CreatedAt, DownloadUrl: "/download/trainings/" + modelNames[v.DatasetId] + ".tflite", + CoverUrl: "/api/v1/app/cover?namePrefix=" + url.PathEscape(modelNames[v.DatasetId]), }) } return &dto.ModelCatalogRes{List: items}, nil