Add files via upload

easyjailbreak/selector/RandomSelector.py

@@ -0,0 +1,34 @@
+"""
+RandomSelectPolicy class
+============================
+"""
+import random
+from easyjailbreak.selector import SelectPolicy
+from easyjailbreak.datasets import JailbreakDataset
+
+__all__ = ["RandomSelectPolicy"]
+
+class RandomSelectPolicy(SelectPolicy):
+    """
+    A selection policy that randomly selects an instance from a JailbreakDataset.
+    It extends the SelectPolicy abstract base class, providing a concrete implementation
+    for the random selection strategy.
+    """
+
+    def __init__(self, Datasets: JailbreakDataset):
+        """
+        Initializes the RandomSelectPolicy with a given JailbreakDataset.
+
+        :param ~JailbreakDataset Datasets: The dataset from which instances will be randomly selected.
+        """
+        super().__init__(Datasets)
+
+    def select(self) -> JailbreakDataset:
+        """
+        Selects an instance randomly from the dataset and increments its visited count.
+
+        :return ~JailbreakDataset: The randomly selected instance from the dataset.
+        """
+        seed = random.choice(self.Datasets._dataset)
+        seed.visited_num += 1
+        return JailbreakDataset([seed])