Feat: add description to leaderboard selection

gpu-mode · Feb 15, 2025 · 9d0a74b · 9d0a74b
1 parent de3b4f2
commit 9d0a74b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/src/models/types.go b/src/models/types.go
@@ -1,12 +1,13 @@
 package models
 
 type LeaderboardItem struct {
-	TitleText string
+	TitleText     string
+	TaskDescription string
 }
 
 func (i LeaderboardItem) FilterValue() string { return i.TitleText }
 func (i LeaderboardItem) Title() string       { return i.TitleText }
-func (i LeaderboardItem) Description() string { return "" }
+func (i LeaderboardItem) Description() string { return i.TaskDescription }
 
 type GpuItem struct {
 	TitleText string

diff --git a/src/service/api.go b/src/service/api.go
@@ -41,9 +41,14 @@ func FetchLeaderboards() ([]models.LeaderboardItem, error) {
 		return nil, err
 	}
 
+
 	leaderboardNames := make([]models.LeaderboardItem, len(leaderboards))
 	for i, lb := range leaderboards {
-		leaderboardNames[i] = models.LeaderboardItem{TitleText: lb["name"].(string)}
+		task := lb["task"].(map[string]interface{})
+		leaderboardNames[i] = models.LeaderboardItem{
+			TitleText:         lb["name"].(string),
+			TaskDescription:   task["description"].(string),
+		}
 	}
 
 	return leaderboardNames, nil