Pipeline API

The pipeline module orchestrates the complete red teaming workflow.

Constants

MODEL_CLASSES

Dictionary mapping model names to their implementation classes. Allows dynamic instantiation of models based on configuration strings.

ATTACK_TYPES

Dictionary categorizing attack types and their corresponding attack classes. Used for organizing attacks by their strategy/approach.

ATTACK_CLASSES

Dictionary mapping attack names to their implementation classes and types. Allows dynamic instantiation of attacks based on configuration strings.

EVALUATOR_CLASSES

Dictionary mapping evaluator names to their implementation classes. Allows dynamic instantiation of evaluators based on configuration strings.

Pipeline Functions

Attack Setup

Attack Prompt Generation

Model Response Collection

Response Evaluation

Results Saving

See Also