pyrit.prompt_converter.VariationConverter

class VariationConverter(*, converter_target: PromptChatTarget, prompt_template: SeedPrompt = None)

Bases: PromptConverter

__init__(*, converter_target: PromptChatTarget, prompt_template: SeedPrompt = None)

Methods

__init__(*, converter_target[, prompt_template])
convert_async(*, prompt[, input_type])	Generates variations of the input prompts using the converter target.
convert_tokens_async(*, prompt[, ...])	Converts substrings within a prompt that are enclosed by specified start and end tokens.
get_identifier()
input_supported(input_type)	Checks if the input type is supported by the converter
output_supported(output_type)	Checks if the output type is supported by the converter
send_variation_prompt_async(request)

Attributes

supported_input_types	Returns a list of supported input types for the converter.
supported_output_types	Returns a list of supported output types for the converter.

async convert_async(*, prompt: str, input_type: Literal['text', 'image_path', 'audio_path', 'video_path', 'url', 'error'] = 'text') → ConverterResult

Generates variations of the input prompts using the converter target. :param prompts: list of prompts to convert

Returns:

list of prompt variations generated by the converter target

Return type:

target_responses

input_supported(input_type: Literal['text', 'image_path', 'audio_path', 'video_path', 'url', 'error']) → bool

Checks if the input type is supported by the converter

Parameters:

input_type – The input type to check

Returns:

True if the input type is supported, False otherwise

Return type:

bool

output_supported(output_type: Literal['text', 'image_path', 'audio_path', 'video_path', 'url', 'error']) → bool

Checks if the output type is supported by the converter

Parameters:

output_type – The output type to check

Returns:

True if the output type is supported, False otherwise

Return type:

bool

async send_variation_prompt_async(request)