nemoguardrails.library.self_check.utils

Module Contents

Functions

Name	Description
`get_self_check_llm`	-
`get_self_check_prompt_task`	Resolve the prompt task name while supporting deprecated bare custom tasks.
`get_self_check_task_from_rail`	-
`parse_self_check_output`	-
`resolve_self_check_task`	-
`run_self_check_task`	-

Data

SELF_CHECK_INPUT_DEFAULT_TASK

SELF_CHECK_INPUT_FLOW

SELF_CHECK_INPUT_VARIANT_PARAM

SELF_CHECK_OUTPUT_DEFAULT_TASK

SELF_CHECK_OUTPUT_FLOW

SELF_CHECK_OUTPUT_VARIANT_PARAM

SELF_CHECK_VARIANT_PARAM

log

API

nemoguardrails.library.self_check.utils.get_self_check_llm(
    llms: typing.Dict[str, nemoguardrails.types.LLMModel],
    task: str,
    default_task: str,
    main_llm: typing.Optional[nemoguardrails.types.LLMModel]
) -> nemoguardrails.types.LLMModel

nemoguardrails.library.self_check.utils.get_self_check_prompt_task(
    task: str,
    default_task: str,
    available_prompt_tasks: typing.Optional[collections.abc.Collection[str]] = None
) -> str

Resolve the prompt task name while supporting deprecated bare custom tasks.

nemoguardrails.library.self_check.utils.get_self_check_task_from_rail(
    flow: typing.Any,
    flow_id: str,
    variant_param: str,
    default_task: str
) -> typing.Optional[str]

nemoguardrails.library.self_check.utils.parse_self_check_output(
    llm_task_manager: typing.Any,
    task: str,
    response: str
)

nemoguardrails.library.self_check.utils.resolve_self_check_task(
    variant: typing.Optional[str],
    context: typing.Optional[dict],
    events: typing.Optional[typing.List[dict]],
    triggered_rail_key: str,
    start_rail_event_type: str,
    flow_id: str,
    variant_param: str,
    default_task: str
) -> str

nemoguardrails.library.self_check.utils.run_self_check_task(
    task: str,
    prompt_context: typing.Dict[str, typing.Any],
    llms: typing.Dict[str, nemoguardrails.types.LLMModel],
    default_task: str,
    main_llm: typing.Optional[nemoguardrails.types.LLMModel],
    llm_task_manager: typing.Any,
    lowest_temperature: float,
    max_tokens: int = 1024
) -> tuple[bool, str]

async

nemoguardrails.library.self_check.utils.SELF_CHECK_INPUT_DEFAULT_TASK = 'self_check_input'

nemoguardrails.library.self_check.utils.SELF_CHECK_INPUT_FLOW = 'self check input'

nemoguardrails.library.self_check.utils.SELF_CHECK_INPUT_VARIANT_PARAM = SELF_CHECK_VARIANT_PARAM

nemoguardrails.library.self_check.utils.SELF_CHECK_OUTPUT_DEFAULT_TASK = 'self_check_output'

nemoguardrails.library.self_check.utils.SELF_CHECK_OUTPUT_FLOW = 'self check output'

nemoguardrails.library.self_check.utils.SELF_CHECK_OUTPUT_VARIANT_PARAM = SELF_CHECK_VARIANT_PARAM

nemoguardrails.library.self_check.utils.SELF_CHECK_VARIANT_PARAM = 'variant'

nemoguardrails.library.self_check.utils.log = logging.getLogger(__name__)