pyrit.prompt_converter.TranslationConverter

pyrit.prompt_converter.TranslationConverter#

class TranslationConverter(*, converter_target: PromptChatTarget, language: str, prompt_template: SeedPrompt | None = None, max_retries: int = 3, max_wait_time_in_seconds: int = 60)[source]#

Bases: PromptConverter

Translates prompts into different languages using an LLM.

__init__(*, converter_target: PromptChatTarget, language: str, prompt_template: SeedPrompt | None = None, max_retries: int = 3, max_wait_time_in_seconds: int = 60)[source]#

Initializes the converter with the target chat support, language, and optional prompt template.

Parameters:

  • converter_target (PromptChatTarget) – The target chat support for the conversion which will translate.

  • language (str) – The language for the conversion. E.g. Spanish, French, leetspeak, etc.

  • prompt_template (SeedPrompt, Optional) – The prompt template for the conversion.

  • max_retries (int) – Maximum number of retries for the conversion.

  • max_wait_time_in_seconds (int) – Maximum wait time in seconds between retries.

Raises:

ValueError – If the language is not provided.

Methods

__init__(*, converter_target, language[, ...])

Initializes the converter with the target chat support, language, and optional prompt template.

convert_async(*, prompt[, input_type])

Converts the given prompt by translating it using the converter target.

convert_tokens_async(*, prompt[, ...])

Converts substrings within a prompt that are enclosed by specified start and end tokens.

get_identifier()

Returns an identifier dictionary for the converter.

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.

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', 'reasoning', 'error'] = 'text') → ConverterResult[source]#

Converts the given prompt by translating it using the converter target.

Parameters:

prompt (str) – The prompt to be converted.

Returns:

The result containing the generated version of the prompt.

Return type:

ConverterResult

Raises:

ValueError – If the input type is not supported.

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

Checks if the input type is supported by the converter.

Parameters:

input_type (PromptDataType) – 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', 'reasoning', 'error']) → bool[source]#

Checks if the output type is supported by the converter.

Parameters:

output_type (PromptDataType) – The output type to check.

Returns:

True if the output type is supported, False otherwise.

Return type:

bool

Contents