PyRIT icon indicating copy to clipboard operation
PyRIT copied to clipboard

[DRAFT] [BREAKING] FEAT Refactor Harm Category as StrEnum

Open eugeniavkim opened this issue 6 months ago • 0 comments

Description

In order to make harm_categories easier to search through and group by some alias names, this PR refactors our previous String harm categories into a class HarmCategory with categories that are not included in the list to be recognized in an OTHER category. This also does not include all harm categories, but others can continue to add harm areas that they would like to probe for and score with.

This PR is a breaking change as it does affect all datasets, SeedPrompt initialization, and tests.

The following items must be complete before changing from DRAFT Status:

  • [ ] Refactor each dataset to map its harm categories to the new class HarmCategory
  • [ ] Update DB queries

eugeniavkim avatar Jul 03 '25 00:07 eugeniavkim