nemoguardrails.library.hf_classifier.actions

HuggingFace classifier-based detection actions.

Module Contents

Functions

Name	Description
`_classify_and_check`	Classify text and check against blocked labels.
`_extract_text`	-
`_hf_classifier_output_mapping`	Map action result to streaming blocked semantic: True=blocked.
`hf_classifier_check_input`	-
`hf_classifier_check_output`	-
`hf_classifier_check_retrieval`	-

Data

log

API

nemoguardrails.library.hf_classifier.actions._classify_and_check(
    classifier_name: str,
    text: str,
    config: typing.Optional[nemoguardrails.RailsConfig]
) -> bool

async

Classify text and check against blocked labels.

Returns True if allowed, False if blocked.

nemoguardrails.library.hf_classifier.actions._extract_text(
    context: typing.Optional[dict],
    key: str
) -> str

nemoguardrails.library.hf_classifier.actions._hf_classifier_output_mapping(
    result: bool
) -> bool

Map action result to streaming blocked semantic: True=blocked.

nemoguardrails.library.hf_classifier.actions.hf_classifier_check_input(
    classifier: str,
    config: typing.Optional[nemoguardrails.RailsConfig] = None,
    context: typing.Optional[dict] = None,
    kwargs = {}
) -> bool

async

nemoguardrails.library.hf_classifier.actions.hf_classifier_check_output(
    classifier: str,
    config: typing.Optional[nemoguardrails.RailsConfig] = None,
    context: typing.Optional[dict] = None,
    model_name: typing.Optional[str] = None,
    kwargs = {}
) -> bool

async

nemoguardrails.library.hf_classifier.actions.hf_classifier_check_retrieval(
    classifier: str,
    config: typing.Optional[nemoguardrails.RailsConfig] = None,
    context: typing.Optional[dict] = None,
    kwargs = {}
) -> bool

async

nemoguardrails.library.hf_classifier.actions.log = logging.getLogger(__name__)