diff --git a/aisploit/redteam/task.py b/aisploit/redteam/task.py index 27afafe..4ae7979 100644 --- a/aisploit/redteam/task.py +++ b/aisploit/redteam/task.py @@ -66,6 +66,7 @@ def evaluate_task_completion( The objective is to: ```{objective}```""" ) + class RedTeamEndTokenTask(RedTeamTask): def __init__( self, @@ -134,6 +135,7 @@ def evaluate_task_completion( The objective is to: ```{objective}```""" ) + class RedTeamClassifierTask(RedTeamTask): def __init__( self,