shuffle titles for broader topic

2025-07-22 13:41:39 +02:00
parent 4e205a7b1a
commit 19ef90f8e4
1 changed files with 8 additions and 2 deletions
@@ -9,7 +9,13 @@ import 'package:flutter_mozilla_components/src/pigeons/gecko.g.dart';
 final _apiInstance = GeckoMlApi();

 class GeckoMlService {
-  Future<String> getContainerTopic(Set<String> titles, {int maxCount = 10}) {
-    return _apiInstance.getContainerTopic(titles.take(maxCount).toList());
+  Future<String> getContainerTopic(Set<String> titles, {int maxCount = 8}) {
+    var selectedTitles = titles.toList();
+    if (selectedTitles.length > maxCount) {
+      //TODO: Randomize for now, maybe use clusters later
+      selectedTitles = (selectedTitles..shuffle()).take(maxCount).toList();
+    }
+
+    return _apiInstance.getContainerTopic(selectedTitles);
  }
 }