Source code for pyrit.prompt_converter.human_in_the_loop_converter
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import logging
from typing import Optional
from pyrit.models import PromptDataType
from pyrit.prompt_converter import PromptConverter, ConverterResult
logger = logging.getLogger(__name__)
[docs]
class HumanInTheLoopConverter(PromptConverter):
"""
Allows review of each prompt sent to a target before sending it. User can choose to send the prompt as is,
modify the prompt, or run the prompt through one of the passed-in converters before sending it.
Args:
converters: (List[PromptConverter], Optional): List of possible converters to run input through.
"""
[docs]
def __init__(
self,
converters: Optional[list[PromptConverter]] = None,
):
self._converters = converters or []
[docs]
async def convert_async(self, *, prompt: str, input_type: PromptDataType = "text") -> ConverterResult:
"""
Before sending a prompt to a target, user is given three options to choose from:
(1) Proceed with sending the prompt as is.
(2) Manually modify the prompt.
(3) Run the prompt through a converter before sending it.
Args:
prompt (str): The prompt to be added to the image.
input_type (PromptDataType): Type of data
Returns:
ConverterResult: The filename of the converted image as a ConverterResult Object
"""
user_input = ""
if self._converters:
while user_input not in ["1", "2", "3"]:
user_input = input(
f"This is the next {input_type} prompt to be sent: [{prompt}] \
Select an option: (1) Proceed with sending the prompt as is. \
(2) Manually modify the prompt. (3) Run the prompt through a \
converter before sending it. Please enter the number of the \
choice you'd like."
).strip()
else:
while user_input not in ["1", "2"]:
user_input = input(
f"This is the next {input_type} prompt to be sent: [{prompt}] \
Select an option: (1) Proceed with sending the prompt as is. \
(2) Manually modify the prompt. \
Please enter the number of the choice you'd like."
).strip()
if user_input == "1":
return ConverterResult(output_text=prompt, output_type=input_type)
elif user_input == "2":
new_input = input("Enter new prompt to send: ")
return await self.convert_async(prompt=new_input, input_type=input_type)
elif user_input == "3":
if self._converters:
converters_str = str([converter.__class__.__name__ for converter in self._converters])
converter_index = -1
while not 0 <= converter_index < len(self._converters):
converter_index = int(
input(
f"The available converters are {converters_str}. \
Enter the index of the converter you would like to run on \
the input (0 to {len(self._converters)-1})"
).strip()
)
converter = self._converters[converter_index]
new_result = await converter.convert_async(prompt=prompt, input_type=input_type)
return await self.convert_async(prompt=new_result.output_text, input_type=new_result.output_type)
else:
raise ValueError("No converters were passed into the HumanInTheLoopConverter")
return ConverterResult(output_text=prompt, output_type=input_type)
