image

models/cloudbrain_image.go

@@ -68,6 +68,7 @@ type SearchImageOptions struct {
 	IncludeCustom       bool
 	IncludeOwnerOnly    bool
 	Topics              string
+	CloudbrainType      int
 	ListOptions
 	SearchOrderBy
 }
@@ -411,6 +412,10 @@ func SearchImageCondition(opts *SearchImageOptions) builder.Cond {
 
 	}
 
+	if opts.CloudbrainType > 0 {
+		cond = cond.And(builder.Eq{"cloudbrain_type": opts.CloudbrainType})
+	}
+
 	return cond
 }
 

routers/repo/cloudbrain.go

@@ -1037,6 +1037,7 @@ func GetPublicImages(ctx *context.Context) {
 		IncludeOfficialOnly: ctx.QueryBool("recommend"),
 		SearchOrderBy:       "type desc, num_stars desc,id desc",
 		Status:              models.IMAGE_STATUS_SUCCESS,
+		CloudbrainType:      ctx.QueryInt("cloudbrainType"),
 	}
 
 	getImages(ctx, &opts)

web_src/js/components/images/Images.vue

@@ -672,7 +672,7 @@ export default {
   },
   filters:{
       transformType(val){
-        if(val==0){
+        if(val==0 || val==2){
             return "GPU"
         }
       },

web_src/js/components/images/selectGrampusImages.vue

@@ -84,7 +84,7 @@
                 currentPagePublic: 1,
                 pageSizePublic: 5,
                 totalNumPublic: 0,
-                paramsPublic: { page: 1, pageSize: 5, q: '', recommend: false },
+                paramsPublic: { page: 1, pageSize: 5, q: '', recommend: false,cloudbrainType: 2 },
                 tableDataPublic: [],
                 loadingPublic: false,
 

web_src/js/features/images.js

@@ -218,7 +218,7 @@ export default async function initImage(){
         
         new Vue({
             el:el,            
-            render: h => h(selectImages)
+            render: h => h(selectGrampusImages)
         });
     }
     initVueImages()