textattack.prompt_augmentation package

Prompt Augmentation

This package includes functions used to augment a prompt for a LLM

class textattack.prompt_augmentation.prompt_augmentation_pipeline.PromptAugmentationPipeline(augmenter, llm)[source]

Bases: object

A prompt augmentation pipeline to augment a prompt and obtain the responses from a LLM on the augmented prompts.

Parameters:

augmenter (textattack.Augmenter) – the augmenter to use to augment the prompt
llm (textattack.ModelWrapper) – the LLM to generate responses to the augmented data