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¶
Running the Pipeline - Usage guide and examples