PyRIT/pyrit/executor/attack/multi_turn/__init__.py at 933f4a523468403f2fa7168947d4e698a32918e5 · ValbuenaVC/PyRIT · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.

from pyrit.executor.attack.multi_turn.multi_branch_attack import (
    MultiBranchAttack,
    MultiBranchAttackContext,
    MultiBranchAttackResult,
    MultiBranchCommandEnum
)

from pyrit.executor.attack.multi_turn.multi_turn_attack_strategy import (
    ConversationSession,
    MultiTurnAttackContext,
    MultiTurnAttackStrategy,
)

from pyrit.executor.attack.multi_turn.crescendo import CrescendoAttack, CrescendoAttackContext, CrescendoAttackResult
from pyrit.executor.attack.multi_turn.multi_prompt_sending import (
    MultiPromptSendingAttack,
    MultiPromptSendingAttackContext,
)
from pyrit.executor.attack.multi_turn.red_teaming import RedTeamingAttack, RTASystemPromptPaths
from pyrit.executor.attack.multi_turn.tree_of_attacks import (
    TreeOfAttacksWithPruningAttack,
    TAPAttack,
    TAPAttackResult,
    TAPAttackContext,
)

__all__ = [
    "ConversationSession",

    "MultiBranchAttack",
    "MultiBranchAttackContext",
    "MultiBranchAttackResult",
    "MultiBranchCommandEnum",

    "MultiTurnAttackContext",
    "MultiTurnAttackStrategy",
    "MultiPromptSendingAttack",
    "MultiPromptSendingAttackContext",
    "CrescendoAttack",
    "CrescendoAttackContext",
    "CrescendoAttackResult",
    "RedTeamingAttack",
    "RTASystemPromptPaths",
    "TreeOfAttacksWithPruningAttack",
    "TAPAttack",
    "TAPAttackResult",
    "TAPAttackContext",
]