pyrit.datasets.fetch_harmbench_examples

pyrit.datasets.fetch_harmbench_examples#

fetch_harmbench_examples(source: str = 'https://raw.githubusercontent.com/centerforaisafety/HarmBench/c0423b9/data/behavior_datasets/harmbench_behaviors_text_all.csv', source_type: Literal['public_url'] = 'public_url', cache: bool = True, data_home: Path | None = None) → SeedPromptDataset[source]#

Fetch HarmBench examples and create a SeedPromptDataset.

Parameters:

  • source (str) – The source from which to fetch examples. Defaults to the HarmBench repository.

  • source_type (Literal["public_url"]) – The type of source (‘public_url’).

  • cache (bool) – Whether to cache the fetched examples. Defaults to True.

  • data_home (Optional[Path]) – Directory to store cached data. Defaults to None.

Returns:

A SeedPromptDataset containing the examples.

Return type:

SeedPromptDataset

Note

For more information and access to the original dataset and related materials, visit: centerforaisafety/HarmBench