PyRIT/pyrit/executor/attack/__init__.py at 46aaeaabbe5eb4800320b95ee1458b9532531ea2 · microsoft/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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.

"""Attack executor module."""

from pyrit.executor.attack.component import (
    ConversationManager,
    ConversationState,
    PrependedConversationConfig,
)
from pyrit.executor.attack.compound import (
    SequenceCompletionPolicy,
    SequentialAttack,
    SequentialAttackResult,
    SequentialChildAttack,
)
from pyrit.executor.attack.core import (
    AttackAdversarialConfig,
    AttackContext,
    AttackConverterConfig,
    AttackExecutor,
    AttackExecutorResult,
    AttackParameters,
    AttackScoringConfig,
    AttackStrategy,
)
from pyrit.executor.attack.multi_turn import (
    ChunkedRequestAttack,
    ChunkedRequestAttackContext,
    ConversationSession,
    CrescendoAttack,
    CrescendoAttackContext,
    CrescendoAttackResult,
    MultiPromptSendingAttack,
    MultiPromptSendingAttackParameters,
    MultiTurnAttackContext,
    MultiTurnAttackStrategy,
    PAIRAttack,
    RedTeamingAttack,
    RTASystemPromptPaths,
    TAPAttack,
    TAPAttackContext,
    TAPAttackResult,
    TAPSystemPromptPaths,
    TreeOfAttacksWithPruningAttack,
    generate_simulated_conversation_async,
)
from pyrit.executor.attack.single_turn import (
    BijectionAttack,
    ContextComplianceAttack,
    FlipAttack,
    ManyShotJailbreakAttack,
    PromptSendingAttack,
    RolePlayAttack,
    RolePlayPaths,
    SingleTurnAttackContext,
    SingleTurnAttackStrategy,
    SkeletonKeyAttack,
)
from pyrit.executor.attack.streaming import BargeInAttack, BargeInAttackContext

# Backward-compatibility aliases — import from pyrit.output.attack_result directly.
# TODO: Remove these re-exports in two releases (target removal: 0.16.0).
from pyrit.output.attack_result.base import AttackResultPrinterBase as AttackResultPrinter
from pyrit.output.attack_result.markdown import MarkdownAttackResultMemoryPrinter as MarkdownAttackResultPrinter
from pyrit.output.attack_result.pretty import PrettyAttackResultMemoryPrinter as ConsoleAttackResultPrinter

__all__ = [
    "AttackAdversarialConfig",
    "AttackContext",
    "AttackConverterConfig",
    "AttackExecutor",
    "AttackExecutorResult",
    "AttackParameters",
    "AttackResultPrinter",
    "AttackScoringConfig",
    "AttackStrategy",
    "BargeInAttack",
    "BargeInAttackContext",
    "ChunkedRequestAttack",
    "ChunkedRequestAttackContext",
    "ConsoleAttackResultPrinter",
    "ContextComplianceAttack",
    "ConversationManager",
    "ConversationSession",
    "ConversationState",
    "CrescendoAttack",
    "CrescendoAttackContext",
    "CrescendoAttackResult",
    "BijectionAttack",
    "FlipAttack",
    "ManyShotJailbreakAttack",
    "MarkdownAttackResultPrinter",
    "MultiPromptSendingAttack",
    "MultiPromptSendingAttackParameters",
    "MultiTurnAttackContext",
    "MultiTurnAttackStrategy",
    "PAIRAttack",
    "PrependedConversationConfig",
    "PromptSendingAttack",
    "RTASystemPromptPaths",
    "RedTeamingAttack",
    "RolePlayAttack",
    "RolePlayPaths",
    "SequenceCompletionPolicy",
    "SequentialAttack",
    "SequentialAttackResult",
    "SequentialChildAttack",
    "SingleTurnAttackContext",
    "SingleTurnAttackStrategy",
    "SkeletonKeyAttack",
    "TAPAttack",
    "TAPAttackContext",
    "TAPAttackResult",
    "TAPSystemPromptPaths",
    "TreeOfAttacksWithPruningAttack",
    "generate_simulated_conversation_async",
]