nemoguardrails.library.topic_safety.actions

nemoguardrails.library.topic_safety.actions.topic_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,
    events: typing.Optional[typing.List[dict]] = None,
    model_caches: typing.Optional[typing.Dict[str, nemoguardrails.llm.cache.CacheInterface]] = None,
    kwargs = {}
) -> dict

nemoguardrails.library.topic_safety.actions.TOPIC_SAFETY_MAX_TOKENS = 10

nemoguardrails.library.topic_safety.actions.TOPIC_SAFETY_OUTPUT_RESTRICTION = 'If any of the above conditions are violated, please respond with "off-topic". O...

nemoguardrails.library.topic_safety.actions.TOPIC_SAFETY_TEMPERATURE = 0.01

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