nemoguardrails.library.pangea.actions

View as Markdown

Module Contents

Classes

Functions

Data

log

API

class nemoguardrails.library.pangea.actions.Message()

Bases: BaseModel

content
str
role
str
class nemoguardrails.library.pangea.actions.TextGuardResponse()

Bases: BaseModel

result
TextGuardResult
class nemoguardrails.library.pangea.actions.TextGuardResult()

Bases: BaseModel

blocked
Optional[bool] = None

Whether or not the prompt triggered a block detection.

bot_message
Optional[str] = None
prompt_messages
Optional[list[Message]] = None

Updated structured prompt, if applicable.

transformed
Optional[bool] = None

Whether or not the original input was transformed.

user_message
Optional[str] = None
nemoguardrails.library.pangea.actions.get_pangea_config(
config: nemoguardrails.rails.llm.config.RailsConfig
) -> nemoguardrails.rails.llm.config.PangeaRailConfig
nemoguardrails.library.pangea.actions.pangea_ai_guard(
mode: typing_extensions.Literal['input', 'output'],
config: nemoguardrails.rails.llm.config.RailsConfig,
context: collections.abc.Mapping[str, typing.Any] = {},
user_message: typing.Optional[str] = None,
bot_message: typing.Optional[str] = None
) -> nemoguardrails.library.pangea.actions.TextGuardResult
async
nemoguardrails.library.pangea.actions.log = logging.getLogger(__name__)