nemoguardrails.library.content_safety.actions

Module Contents

Functions

Name	Description
`_detect_language`	-
`_get_reasoning_enabled`	-
`_get_refusal_message`	-
`content_safety_check_input`	-
`content_safety_check_output`	-
`content_safety_check_output_mapping`	Mapping function for content_safety_check_output.
`detect_language`	-

Data

DEFAULT_REFUSAL_MESSAGES

SUPPORTED_LANGUAGES

log

API

nemoguardrails.library.content_safety.actions._detect_language(
    text: str
) -> typing.Optional[str]

nemoguardrails.library.content_safety.actions._get_reasoning_enabled(
    llm_task_manager: nemoguardrails.llm.taskmanager.LLMTaskManager
) -> bool

nemoguardrails.library.content_safety.actions._get_refusal_message(
    lang: str,
    custom_messages: typing.Optional[typing.Dict[str, str]]
) -> str

nemoguardrails.library.content_safety.actions.content_safety_check_input(
    llms: typing.Dict[str, nemoguardrails.types.LLMModel],
    llm_task_manager: nemoguardrails.llm.taskmanager.LLMTaskManager,
    model_name: typing.Optional[str] = None,
    context: typing.Optional[dict] = None,
    model_caches: typing.Optional[typing.Dict[str, nemoguardrails.llm.cache.CacheInterface]] = None,
    kwargs = {}
) -> dict

async

nemoguardrails.library.content_safety.actions.content_safety_check_output(
    llms: typing.Dict[str, nemoguardrails.types.LLMModel],
    llm_task_manager: nemoguardrails.llm.taskmanager.LLMTaskManager,
    model_name: typing.Optional[str] = None,
    context: typing.Optional[dict] = None,
    model_caches: typing.Optional[typing.Dict[str, nemoguardrails.llm.cache.CacheInterface]] = None,
    kwargs = {}
) -> dict

async

nemoguardrails.library.content_safety.actions.content_safety_check_output_mapping(
    result: dict
) -> bool

Mapping function for content_safety_check_output.

Returns: bool

True if the content should be blocked (i.e. allowed is False),

nemoguardrails.library.content_safety.actions.detect_language(
    context: typing.Optional[dict] = None,
    config: typing.Optional[dict] = None
) -> dict

async

nemoguardrails.library.content_safety.actions.DEFAULT_REFUSAL_MESSAGES: Dict[str, str] = {'en': "I'm sorry, I can't respond to that.", 'es': 'Lo siento, no puedo respond...

nemoguardrails.library.content_safety.actions.SUPPORTED_LANGUAGES: FrozenSet[str] = frozenset({'en', 'es', 'zh', 'de', 'fr', 'hi', 'ja', 'ar', 'th'})

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