nemoguardrails.library.crowdstrike_aidr.actions

View as Markdown

Module Contents

Classes

Functions

Data

log

API

class nemoguardrails.library.crowdstrike_aidr.actions.GuardChatCompletionsResponse()

Bases: BaseModel

result
GuardChatCompletionsResult
class nemoguardrails.library.crowdstrike_aidr.actions.GuardChatCompletionsResult()

Bases: BaseModel

blocked
Optional[bool] = None

Whether or not the prompt triggered a block detection.

bot_message
Optional[str] = None
guard_output
Optional[GuardOutput] = None

Updated structured prompt, if applicable.

transformed
Optional[bool] = None

Whether or not the original input was transformed.

user_message
Optional[str] = None
class nemoguardrails.library.crowdstrike_aidr.actions.GuardOutput

Bases: typing.TypedDict

messages
list[Message]
class nemoguardrails.library.crowdstrike_aidr.actions.Message()

Bases: BaseModel

content
str
role
str
nemoguardrails.library.crowdstrike_aidr.actions.crowdstrike_aidr_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.crowdstrike_aidr.actions.GuardChatCompletionsResult
async
nemoguardrails.library.crowdstrike_aidr.actions.get_crowdstrike_aidr_config(
config: nemoguardrails.rails.llm.config.RailsConfig
) -> nemoguardrails.rails.llm.config.CrowdStrikeAIDRRailConfig
nemoguardrails.library.crowdstrike_aidr.actions.log = logging.getLogger(__name__)