`stages.text.classifiers.aegis_utils`#

Module Contents#

Functions#

Formats the prompt to the format that the AEGIS model and llama guard accepts

Data#

`ACCESS_ERROR_MESSAGE`
`AEGIS_LABELS`
`unsafe_categories`

API#

stages.text.classifiers.aegis_utils.ACCESS_ERROR_MESSAGE = <Multiline-String>#

stages.text.classifiers.aegis_utils.AEGIS_LABELS#: [‘unknown’, ‘safe’, ‘O1’, ‘O2’, ‘O3’, ‘O4’, ‘O5’, ‘O6’, ‘O7’, ‘O8’, ‘O9’, ‘O10’, ‘O11’, ‘O12’, ‘O13’…

stages.text.classifiers.aegis_utils.format_aegis(message: str) → str#: Formats the prompt to the format that the AEGIS model and llama guard accepts

stages.text.classifiers.aegis_utils.unsafe_categories = <Multiline-String>#