pyrit/orchestrator/__init__.py

# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.

from pyrit.orchestrator.orchestrator_class import Orchestrator
from pyrit.orchestrator.models.orchestrator_result import OrchestratorResult, OrchestratorResultStatus
from pyrit.orchestrator.multi_turn.multi_turn_orchestrator import MultiTurnOrchestrator
from pyrit.orchestrator.multi_turn.tree_of_attacks_with_pruning_orchestrator import TreeOfAttacksWithPruningOrchestrator
from pyrit.orchestrator.scoring_orchestrator import ScoringOrchestrator
from pyrit.orchestrator.single_turn.prompt_sending_orchestrator import PromptSendingOrchestrator
from pyrit.orchestrator.single_turn.question_answer_benchmark_orchestrator import QuestionAnsweringBenchmarkOrchestrator
from pyrit.orchestrator.single_turn.role_play_orchestrator import RolePlayOrchestrator, RolePlayPaths
from pyrit.orchestrator.single_turn.context_compliance_orchestrator import (
    ContextComplianceOrchestrator,
    ContextDescriptionPaths,
)

from pyrit.orchestrator.fuzzer_orchestrator import FuzzerOrchestrator
from pyrit.orchestrator.multi_turn.crescendo_orchestrator import CrescendoOrchestrator
from pyrit.orchestrator.multi_turn.pair_orchestrator import PAIROrchestrator
from pyrit.orchestrator.multi_turn.red_teaming_orchestrator import RedTeamingOrchestrator
from pyrit.orchestrator.single_turn.flip_attack_orchestrator import FlipAttackOrchestrator
from pyrit.orchestrator.skeleton_key_orchestrator import SkeletonKeyOrchestrator
from pyrit.orchestrator.single_turn.many_shot_jailbreak_orchestrator import ManyShotJailbreakOrchestrator
from pyrit.orchestrator.xpia_orchestrator import (
    XPIAManualProcessingOrchestrator,
    XPIAOrchestrator,
    XPIATestOrchestrator,
)

__all__ = [
    "ContextComplianceOrchestrator",
    "ContextDescriptionPaths",
    "CrescendoOrchestrator",
    "FlipAttackOrchestrator",
    "FuzzerOrchestrator",
    "OrchestratorResult",
    "OrchestratorResultStatus",
    "MultiTurnOrchestrator",
    "Orchestrator",
    "PAIROrchestrator",
    "PromptSendingOrchestrator",
    "QuestionAnsweringBenchmarkOrchestrator",
    "RedTeamingOrchestrator",
    "RolePlayOrchestrator",
    "RolePlayPaths",
    "ScoringOrchestrator",
    "SkeletonKeyOrchestrator",
    "ManyShotJailbreakOrchestrator",
    "TreeOfAttacksWithPruningOrchestrator",
    "XPIAManualProcessingOrchestrator",
    "XPIAOrchestrator",
    "XPIATestOrchestrator",
]